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'],
|