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

ensure screenshot / output call exists

Browse Source
This commit is contained in:
Dillon DuPont
2025-08-08 18:50:12 -04:00
parent fa88836132
commit 6d42c5d939
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
libs/python/agent/agent/integrations/hud/agent.py
View File

@@ -175,7 +175,8 @@ class ComputerAgent(Agent[BaseComputerAgent, dict[str, Any]]):
# Stop at the last user message
break
if last_computer_calls and observation.screenshot:
if last_computer_calls:
screenshot_b64 = await self.hud_computer.screenshot()
# Add computer_call_output for each unresponded computer_call
for call_id in reversed(last_computer_calls): # Maintain order
self.conversation_history.append({
@@ -183,7 +184,7 @@ class ComputerAgent(Agent[BaseComputerAgent, dict[str, Any]]):
"call_id": call_id,
"output": {
"type": "input_image",
"image_url": f"data:image/png;base64,{observation.screenshot}"
"image_url": f"data:image/png;base64,{screenshot_b64}"
}
})
else: