mirror/computer
1
0
Fork 0
mirror of https://github.com/trycua/computer.git synced 2026-01-04 04:19:57 -06:00
Code Issues Packages Projects Releases Wiki Activity

Loop agent proxy

Browse Source
This commit is contained in:
Dillon DuPont
2025-08-28 18:35:44 -04:00
parent db652a5998
commit 3cb45f1e9b
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
libs/python/computer-server/computer_server/main.py
View File

@@ -646,7 +646,7 @@ async def agent_response_endpoint(
total_output: List[Any] = []
total_usage: Dict[str, Any] = {}
turns = 0
pending_computer_call_ids = set()
try:
async for result in agent.run(messages):
total_output += result["output"]
@@ -658,8 +658,13 @@ async def agent_response_endpoint(
total_usage[k] = total_usage.get(k, 0) + v
else:
total_usage[k] = v
turns += 1
if turns > 2:
for msg in result.get("output", []):
if msg.get("type") == "computer_call":
pending_computer_call_ids.add(msg["call_id"])
elif msg.get("type") == "computer_call_output":
pending_computer_call_ids.discard(msg["call_id"])
# exit if no pending computer calls
if not pending_computer_call_ids:
break
except Exception as e:
logger.error(f"Error running agent: {str(e)}")