cua-agent 0.4.33__tar.gz → 0.4.34__tar.gz

{cua_agent-0.4.33 → cua_agent-0.4.34}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cua-agent
-Version: 0.4.33
+Version: 0.4.34
 Summary: CUA (Computer Use) Agent for AI-driven computer interaction
 Author-Email: TryCua <gh@trycua.com>
 Requires-Python: >=3.12
@@ -43,6 +43,10 @@ Requires-Dist: torch; extra == "internvl-hf"
 Requires-Dist: transformers>=4.55.0; extra == "internvl-hf"
 Requires-Dist: einops; extra == "internvl-hf"
 Requires-Dist: timm; extra == "internvl-hf"
+Provides-Extra: moondream3
+Requires-Dist: accelerate; extra == "moondream3"
+Requires-Dist: torch; extra == "moondream3"
+Requires-Dist: transformers>=4.55.0; extra == "moondream3"
 Provides-Extra: ui
 Requires-Dist: gradio>=5.23.3; extra == "ui"
 Requires-Dist: python-dotenv>=1.0.1; extra == "ui"

{cua_agent-0.4.33 → cua_agent-0.4.34}/agent/loops/moondream3.py

@@ -20,8 +20,6 @@ import io
 from typing import Dict, List, Any, Optional, Tuple, Any
 
 from PIL import Image, ImageDraw, ImageFont
-import torch
-from transformers import AutoModelForCausalLM
 import litellm
 
 from ..decorators import register_agent
@@ -41,12 +39,19 @@ def get_moondream_model() -> Any:
     """Get a singleton instance of the Moondream3 preview model."""
     global _MOONDREAM_SINGLETON
     if _MOONDREAM_SINGLETON is None:
-        _MOONDREAM_SINGLETON = AutoModelForCausalLM.from_pretrained(
-            "moondream/moondream3-preview",
-            trust_remote_code=True,
-            torch_dtype=torch.bfloat16,
-            device_map="cuda",
-        )
+        try:
+            import torch
+            from transformers import AutoModelForCausalLM
+            _MOONDREAM_SINGLETON = AutoModelForCausalLM.from_pretrained(
+                "moondream/moondream3-preview",
+                trust_remote_code=True,
+                torch_dtype=torch.bfloat16,
+                device_map="cuda",
+            )
+        except ImportError as e:
+            raise RuntimeError(
+                "moondream3 requires torch and transformers. Install with: pip install cua-agent[moondream3]"
+            ) from e
     return _MOONDREAM_SINGLETON
 
 

{cua_agent-0.4.33 → cua_agent-0.4.34}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "pdm.backend"
 
 [project]
 name = "cua-agent"
-version = "0.4.33"
+version = "0.4.34"
 description = "CUA (Computer Use) Agent for AI-driven computer interaction"
 readme = "README.md"
 authors = [
@@ -62,6 +62,11 @@ internvl-hf = [
     "einops",
     "timm",
 ]
+moondream3 = [
+    "accelerate",
+    "torch",
+    "transformers>=4.55.0",
+]
 ui = [
     "gradio>=5.23.3",
     "python-dotenv>=1.0.1",