minor fix

marifl · Jul 8, 2023 · 4499083 · 4499083
1 parent a7e49f6
commit 4499083
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/superagi/agent/super_agi.py b/superagi/agent/super_agi.py
@@ -90,7 +90,7 @@ def from_llm_and_tools(
             tools=tools
         )
 
-    def fetch_agent_feeds(self, session, agent_execution_id, agent_id):
+    def fetch_agent_feeds(self, session, agent_execution_id):
         agent_feeds = session.query(AgentExecutionFeed.role, AgentExecutionFeed.feed) \
             .filter(AgentExecutionFeed.agent_execution_id == agent_execution_id) \
             .order_by(asc(AgentExecutionFeed.created_at)) \
@@ -116,8 +116,7 @@ def execute(self, workflow_step: AgentWorkflowStep):
         task_queue = TaskQueue(str(agent_execution_id))
 
         token_limit = TokenCounter.token_limit()
-        agent_feeds = self.fetch_agent_feeds(session, self.agent_config["agent_execution_id"],
-                                             self.agent_config["agent_id"])
+        agent_feeds = self.fetch_agent_feeds(session, self.agent_config["agent_execution_id"])
         current_calls = 0
         if len(agent_feeds) <= 0:
             task_queue.clear_tasks()