Add timeout possibility

BlueBrain · Nov 19, 2024 · ca6760b · ca6760b
1 parent b3c6d2d
commit ca6760b
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/swarm_copy/run.py b/swarm_copy/run.py
@@ -186,7 +186,12 @@ async def handle_tool_call(
                 )
 
                 if not approval_json:
-                    continue
+                    return {
+                        "role": "tool",
+                        "tool_call_id": tool_call.id,
+                        "tool_name": name,
+                        "content": "Tool execution was cancelled due to timeout.",
+                    }, None
 
                 approval = json.loads(approval_json)
                 if approval["status"] == "approved":