cua-agent 0.4.0b1__tar.gz → 0.4.0b3__tar.gz

{cua_agent-0.4.0b1 → cua_agent-0.4.0b3}/PKG-INFO

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

{cua_agent-0.4.0b1 → cua_agent-0.4.0b3}/agent/__init__.py

@@ -16,4 +16,4 @@ __all__ = [
     "AgentResponse"
 ]
 
-__version__ = "0.4.0b1"
+__version__ = "0.4.0b3"

{cua_agent-0.4.0b1 → cua_agent-0.4.0b3}/agent/adapters/huggingfacelocal_adapter.py

@@ -1,10 +1,17 @@
 import asyncio
 import warnings
 from typing import Iterator, AsyncIterator, Dict, List, Any, Optional
-import torch
-from transformers import Qwen2_5_VLForConditionalGeneration, AutoProcessor
 from litellm.types.utils import GenericStreamingChunk, ModelResponse
-from litellm import CustomLLM, completion, acompletion
+from litellm.llms.custom_llm import CustomLLM
+from litellm import completion, acompletion
+
+# Try to import HuggingFace dependencies
+try:
+    import torch
+    from transformers import Qwen2_5_VLForConditionalGeneration, AutoProcessor
+    HF_AVAILABLE = True
+except ImportError:
+    HF_AVAILABLE = False
 
 
 class HuggingFaceLocalAdapter(CustomLLM):
@@ -102,6 +109,12 @@ class HuggingFaceLocalAdapter(CustomLLM):
         Returns:
             Generated text response
         """
+        if not HF_AVAILABLE:
+            raise ImportError(
+                "HuggingFace transformers dependencies not found. "
+                "Please install with: pip install \"cua-agent[uitars-hf]\""
+            )
+        
         # Extract messages and model from kwargs
         messages = kwargs.get('messages', [])
         model_name = kwargs.get('model', 'ByteDance-Seed/UI-TARS-1.5-7B')

{cua_agent-0.4.0b1 → cua_agent-0.4.0b3}/agent/loops/omniparser.py

@@ -84,7 +84,7 @@ except ImportError:
     pass
 OMNIPARSER_SINGLETON = None
 
-def get_parser() -> OmniParser:
+def get_parser():
     global OMNIPARSER_SINGLETON
     if OMNIPARSER_SINGLETON is None:
         OMNIPARSER_SINGLETON = OmniParser()

{cua_agent-0.4.0b1 → cua_agent-0.4.0b3}/pyproject.toml

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