cua-agent 0.4.29__tar.gz → 0.4.31__tar.gz

{cua_agent-0.4.29 → cua_agent-0.4.31}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cua-agent
-Version: 0.4.29
+Version: 0.4.31
 Summary: CUA (Computer Use) Agent for AI-driven computer interaction
 Author-Email: TryCua <gh@trycua.com>
 Requires-Python: >=3.12

{cua_agent-0.4.29 → cua_agent-0.4.31}/agent/integrations/hud/__init__.py

@@ -11,6 +11,7 @@ Exports:
 import time
 from typing import Any, Optional
 
+from agent.computers import is_agent_computer
 from datasets import load_dataset, Dataset
 from hud.datasets import Task, run_dataset
 from hud import trace
@@ -55,6 +56,15 @@ async def run_single_task(
     sample_task = dataset[task_id]  # type: ignore[index]
     task_prompt = sample_task.get("prompt", f"Task {sample_task.get('id', 0)}")  # type: ignore[attr-defined]
 
+    # Filter any existing Computer tools
+    # The eval framework will add its own Computer tool per task
+    if tools:
+        tools = [
+            tool 
+            for tool in tools 
+            if not is_agent_computer(tool)
+        ]
+    
     with trace(name=task_prompt):
         task = Task(**sample_task)  # type: ignore[arg-type]
 
@@ -118,6 +128,15 @@ async def run_full_dataset(
         dataset_name = "custom"
         job_name = job_name or f"Evaluation {time.strftime('%H:%M %Y-%m-%d')}"
 
+    # Filter any existing Computer tools
+    # The eval framework will add its own Computer tool per task
+    if tools:
+        tools = [
+            tool 
+            for tool in tools 
+            if not is_agent_computer(tool)
+        ]
+    
     # Execute evaluation
     return await run_dataset(
         name=job_name,

{cua_agent-0.4.29 → cua_agent-0.4.31}/agent/integrations/hud/agent.py

@@ -128,10 +128,11 @@ class MCPComputerAgent(MCPAgent):
         }
         agent_tools: list[Any] = [computer_shim]
         if tools:
-            for tool in tools:
-                if is_agent_computer(tool):
-                    raise ValueError(f"Too many Computer tools: MCPComputerAgent already includes a Computer interface. Received a Computer tool in tools= (e.g., {tool!r}). Remove it and retry.")
-            agent_tools.extend(tools)
+            agent_tools.extend([
+                tool 
+                for tool in tools 
+                if not is_agent_computer(tool)
+            ])
         
         agent_kwargs = {
             "model": self.model,

{cua_agent-0.4.29 → cua_agent-0.4.31}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "pdm.backend"
 
 [project]
 name = "cua-agent"
-version = "0.4.29"
+version = "0.4.31"
 description = "CUA (Computer Use) Agent for AI-driven computer interaction"
 readme = "README.md"
 authors = [