fix
Browse files- controller.py +1 -1
controller.py
CHANGED
@@ -24,7 +24,7 @@ class Controller:
|
|
24 |
log_response("document \n {}".format(self.agent_config.document))
|
25 |
log_response("image \n {}".format(self.agent_config.image))
|
26 |
log_response("context \n {}".format(self.agent_config.context))
|
27 |
-
selected_tools = [self.agent_config.tool_loader.tools[idx] for idx, checkbox in enumerate(self.agent_config.
|
28 |
agent = CustomHfAgent(
|
29 |
url_endpoint=self.agent_config.url_endpoint,
|
30 |
token=os.environ['HF_token'],
|
|
|
24 |
log_response("document \n {}".format(self.agent_config.document))
|
25 |
log_response("image \n {}".format(self.agent_config.image))
|
26 |
log_response("context \n {}".format(self.agent_config.context))
|
27 |
+
selected_tools = [self.agent_config.tool_loader.tools[idx] for idx, checkbox in enumerate(self.agent_config.s_tool_checkboxes) if checkbox]
|
28 |
agent = CustomHfAgent(
|
29 |
url_endpoint=self.agent_config.url_endpoint,
|
30 |
token=os.environ['HF_token'],
|