Handling exception on agent run

TransformerOptimus · Jul 14, 2023 · dc35026 · dc35026
1 parent 4d5c292
commit dc35026
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/superagi/jobs/agent_executor.py b/superagi/jobs/agent_executor.py
@@ -247,7 +247,14 @@ def execute_next_action(self, agent_execution_id):
         agent_workflow_step = session.query(AgentWorkflowStep).filter(
             AgentWorkflowStep.id == agent_execution.current_step_id).first()
 
-        response = spawned_agent.execute(agent_workflow_step)
+        try:
+            response = spawned_agent.execute(agent_workflow_step)
+        except RuntimeError as e:
+            # If our execution encounters an error we return and attempt to retry
+            logger.error("Error executing the agent:", e)
+            superagi.worker.execute_agent.apply_async((agent_execution_id, datetime.now()), countdown=15)
+            session.close()
+            return
 
         if "retry" in response and response["retry"]:
             superagi.worker.execute_agent.apply_async((agent_execution_id, datetime.now()), countdown=10)