schedule agent fix (#1105)

* schedule agent fix --------- Co-authored-by: Rounak Bhatia <rounak@contlo.com>
TransformerOptimus · Aug 23, 2023 · f558f09 · f558f09
1 parent c990e16
commit f558f09
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 13 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
diff --git a/superagi/jobs/scheduling_executor.py b/superagi/jobs/scheduling_executor.py
@@ -49,16 +49,11 @@ def execute_scheduled_agent(self, agent_id: int, name: str):
         session.add(db_agent_execution)
         session.commit()
 
-        goal_value = session.query(AgentConfiguration.value).filter(AgentConfiguration.agent_id == agent_id).filter(AgentConfiguration.key == 'goal').first()[0]
-        instruction_value = session.query(AgentConfiguration.value).filter(AgentConfiguration.agent_id == agent_id).filter(AgentConfiguration.key == 'instruction').first()[0]
-
-        agent_execution_configs = {
-            "goal": goal_value,
-            "instruction": instruction_value
-        }
-
-
-        AgentExecutionConfiguration.add_or_update_agent_execution_config(session= session, execution=db_agent_execution,agent_execution_configs=agent_execution_configs)
+        agent_execution_id = db_agent_execution.id
+        agent_configurations = session.query(AgentConfiguration).filter(AgentConfiguration.agent_id == agent_id).all()
+        for agent_config in agent_configurations:
+            agent_execution_config = AgentExecutionConfiguration(agent_execution_id=agent_execution_id, key=agent_config.key, value=agent_config.value)
+            session.add(agent_execution_config)
 
 
         organisation = agent.get_agent_organisation(session)