test fix

TransformerOptimus · Aug 23, 2023 · f17d035 · f17d035
1 parent bcdc9b4
commit f17d035
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/superagi/agent/tool_builder.py b/superagi/agent/tool_builder.py
@@ -81,7 +81,7 @@ def build_tool(self, tool: Tool):
         return new_object
 
     def set_default_params_tool(self, tool, agent_config, agent_execution_config, model_api_key: str,
-                                resource_summary: str = ""):
+                                resource_summary: str = "",memory=None):
         """
         Set the default parameters for the tools.
 
@@ -113,9 +113,9 @@ def set_default_params_tool(self, tool, agent_config, agent_execution_config, mo
                                                 agent_execution_id=self.agent_execution_id)
         if hasattr(tool, 'tool_response_manager'):
             tool.tool_response_manager = ToolResponseQueryManager(session=self.session,
-                                                                  agent_execution_id=self.agent_execution_id)
+                                                                  agent_execution_id=self.agent_execution_id,memory=memory)
 
         if tool.name == "QueryResourceTool":
             tool.description = tool.description.replace("{summary}", resource_summary)
 
-        return tool
+        return tool