cua-agent 0.4.33__py3-none-any.whl → 0.4.34__py3-none-any.whl

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.dist-info → cua_agent-0.4.34.dist-info}/METADATA

@@ -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.dist-info → cua_agent-0.4.34.dist-info}/RECORD

@@ -43,7 +43,7 @@ agent/loops/gta1.py,sha256=uGIcUH5ChzO75eGvoQxuKMBWjX-1J9-xmC7vPetobjU,5831
 agent/loops/holo.py,sha256=peQ0xx4XQDBQ3g2XKRLCgyrU_2PkXe3RaysNBqFyS90,7481
 agent/loops/internvl.py,sha256=iQs6DSoP9JOyUxRAz_HPuv4Hi2Sbv-Jc3022W-oPX5Y,6596
 agent/loops/model_types.csv,sha256=GmFn4x80yoUpQZuQ-GXtJkPVlOLYWZ5u_5A73HRyeNE,112
-agent/loops/moondream3.py,sha256=_h4k6Z7VyBU8bw-Av0RiOt-GanCJqU46ZLF03eyRgug,17828
+agent/loops/moondream3.py,sha256=ZmTcJxLxHzrxQqbXHyTJxv5bCSVjotjo-cIWH8u9olk,18085
 agent/loops/omniparser.py,sha256=-db8JUL2Orn47ERIaLbuNShAXn4LeIgYzRWphn_9Dg4,15071
 agent/loops/openai.py,sha256=2typWRS7j2sVm52AzwwigPniCrdw9IVvllypjXN2mKI,8694
 agent/loops/opencua.py,sha256=Chb4UASHDrdcX_fO__Gw2e9ay4Hl6Vq38K5x-IoHyuo,4432
@@ -57,7 +57,7 @@ agent/ui/__main__.py,sha256=vudWXYvGM0aNT5aZ94HPtGW8YXOZ4cLXepHyhUM_k1g,73
 agent/ui/gradio/__init__.py,sha256=yv4Mrfo-Sj2U5sVn_UJHAuwYCezo-5O4ItR2C9jzNko,145
 agent/ui/gradio/app.py,sha256=Ol97YEbwREZZQ9_PMjVHlfOcu9BGsawxgAGAm79hT80,9117
 agent/ui/gradio/ui_components.py,sha256=dJUvKDmc1oSejtoR_gU_oWWYwxaOOQyPloSYRGMrUCQ,36068
-cua_agent-0.4.33.dist-info/METADATA,sha256=LIl2V_RBTn8mrq69gBc_7fpWfQnxBHEnkInmnToJ8Qw,6470
-cua_agent-0.4.33.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
-cua_agent-0.4.33.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
-cua_agent-0.4.33.dist-info/RECORD,,
+cua_agent-0.4.34.dist-info/METADATA,sha256=bfV4ikbYKrLbB3IQV6naEBEeSJK8nS6SzdgWmcBR_60,6649
+cua_agent-0.4.34.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
+cua_agent-0.4.34.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+cua_agent-0.4.34.dist-info/RECORD,,