cua-agent 0.2.11__tar.gz → 0.2.13__tar.gz

{cua_agent-0.2.11 → cua_agent-0.2.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cua-agent
-Version: 0.2.11
+Version: 0.2.13
 Summary: CUA (Computer Use) Agent for AI-driven computer interaction
 Author-Email: TryCua <gh@trycua.com>
 Requires-Python: >=3.11
@@ -24,6 +24,7 @@ Requires-Dist: httpx<0.29.0,>=0.27.0; extra == "openai"
 Provides-Extra: uitars
 Requires-Dist: httpx<0.29.0,>=0.27.0; extra == "uitars"
 Provides-Extra: uitars-mlx
+Requires-Dist: mlx-vlm>=0.1.27; sys_platform == "darwin" and extra == "uitars-mlx"
 Provides-Extra: ui
 Requires-Dist: gradio<6.0.0,>=5.23.3; extra == "ui"
 Requires-Dist: python-dotenv<2.0.0,>=1.0.1; extra == "ui"
@@ -67,6 +68,7 @@ Requires-Dist: requests<3.0.0,>=2.31.0; extra == "all"
 Requires-Dist: ollama<0.5.0,>=0.4.7; extra == "all"
 Requires-Dist: gradio<6.0.0,>=5.23.3; extra == "all"
 Requires-Dist: python-dotenv<2.0.0,>=1.0.1; extra == "all"
+Requires-Dist: mlx-vlm>=0.1.27; sys_platform == "darwin" and extra == "all"
 Description-Content-Type: text/markdown
 
 <div align="center">
@@ -105,10 +107,7 @@ pip install "cua-agent[anthropic]" # Anthropic Cua Loop
 pip install "cua-agent[uitars]"    # UI-Tars support
 pip install "cua-agent[omni]" # Cua Loop based on OmniParser (includes Ollama for local models)
 pip install "cua-agent[ui]" # Gradio UI for the agent
-
-# For local UI-TARS with MLX support, you need to manually install mlx-vlm:
-pip install "cua-agent[uitars-mlx]"
-pip install git+https://github.com/ddupont808/mlx-vlm.git@stable/fix/qwen2-position-id # PR: https://github.com/Blaizzy/mlx-vlm/pull/349
+pip install "cua-agent[uitars-mlx]" # MLX UI-Tars support
 ```
 
 ## Run

{cua_agent-0.2.11 → cua_agent-0.2.13}/README.md

@@ -34,10 +34,7 @@ pip install "cua-agent[anthropic]" # Anthropic Cua Loop
 pip install "cua-agent[uitars]"    # UI-Tars support
 pip install "cua-agent[omni]" # Cua Loop based on OmniParser (includes Ollama for local models)
 pip install "cua-agent[ui]" # Gradio UI for the agent
-
-# For local UI-TARS with MLX support, you need to manually install mlx-vlm:
-pip install "cua-agent[uitars-mlx]"
-pip install git+https://github.com/ddupont808/mlx-vlm.git@stable/fix/qwen2-position-id # PR: https://github.com/Blaizzy/mlx-vlm/pull/349
+pip install "cua-agent[uitars-mlx]" # MLX UI-Tars support
 ```
 
 ## Run

{cua_agent-0.2.11 → cua_agent-0.2.13}/agent/__init__.py

@@ -6,7 +6,7 @@ import logging
 __version__ = "0.1.0"
 
 # Initialize logging
-logger = logging.getLogger("cua.agent")
+logger = logging.getLogger("agent")
 
 # Initialize telemetry when the package is imported
 try:

{cua_agent-0.2.11 → cua_agent-0.2.13}/agent/core/agent.py

@@ -11,10 +11,8 @@ from .types import AgentResponse
 from .factory import LoopFactory
 from .provider_config import DEFAULT_MODELS, ENV_VARS
 
-logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-
 class ComputerAgent:
     """A computer agent that can perform automated tasks using natural language instructions."""
 

{cua_agent-0.2.11 → cua_agent-0.2.13}/agent/core/telemetry.py

@@ -34,7 +34,7 @@ flush = _default_flush
 is_telemetry_enabled = _default_is_telemetry_enabled
 is_telemetry_globally_disabled = _default_is_telemetry_globally_disabled
 
-logger = logging.getLogger("cua.agent.telemetry")
+logger = logging.getLogger("agent.telemetry")
 
 try:
     # Import from core telemetry

{cua_agent-0.2.11 → cua_agent-0.2.13}/agent/providers/omni/loop.py

@@ -26,10 +26,8 @@ from .api_handler import OmniAPIHandler
 from .tools.manager import ToolManager
 from .tools import ToolResult
 
-logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-
 def extract_data(input_string: str, data_type: str) -> str:
     """Extract content from code blocks."""
     pattern = f"```{data_type}" + r"(.*?)(```|$)"

{cua_agent-0.2.11 → cua_agent-0.2.13}/agent/providers/uitars/loop.py

@@ -25,10 +25,8 @@ from .prompts import COMPUTER_USE, SYSTEM_PROMPT, MAC_SPECIFIC_NOTES
 from .clients.oaicompat import OAICompatClient
 from .clients.mlxvlm import MLXVLMUITarsClient
 
-logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-
 class UITARSLoop(BaseLoop):
     """UI-TARS-specific implementation of the agent loop.
 

{cua_agent-0.2.11 → cua_agent-0.2.13}/agent/ui/gradio/app.py

@@ -132,6 +132,13 @@ class GradioChatScreenshotHandler(DefaultCallbackHandler):
 # Detect if current device is MacOS
 is_mac = platform.system().lower() == "darwin"
 
+# Detect if lume is available (host device is macOS)
+is_lume_available = is_mac or (os.environ.get("PYLUME_HOST", "localhost") != "localhost")
+
+print("PYLUME_HOST: ", os.environ.get("PYLUME_HOST", "localhost"))
+print("is_mac: ", is_mac)
+print("Lume available: ", is_lume_available)
+
 # Map model names to specific provider model names
 MODEL_MAPPINGS = {
     "openai": {
@@ -733,9 +740,9 @@ if __name__ == "__main__":
                     is_mac = platform.system().lower() == "darwin"
                     
                     providers = ["cloud"]
-                    if is_mac:
+                    if is_lume_available:
                         providers += ["lume"]
-                    elif is_windows:
+                    if is_windows:
                         providers += ["winsandbox"]
 
                     computer_provider = gr.Radio(

{cua_agent-0.2.11 → cua_agent-0.2.13}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "pdm.backend"
 
 [project]
 name = "cua-agent"
-version = "0.2.11"
+version = "0.2.13"
 description = "CUA (Computer Use) Agent for AI-driven computer interaction"
 readme = "README.md"
 authors = [
@@ -39,7 +39,9 @@ openai = [
 uitars = [
     "httpx>=0.27.0,<0.29.0",
 ]
-uitars-mlx = []
+uitars-mlx = [
+    "mlx-vlm>=0.1.27; sys_platform == 'darwin'",
+]
 ui = [
     "gradio>=5.23.3,<6.0.0",
     "python-dotenv>=1.0.1,<2.0.0",
@@ -86,6 +88,7 @@ all = [
     "ollama>=0.4.7,<0.5.0",
     "gradio>=5.23.3,<6.0.0",
     "python-dotenv>=1.0.1,<2.0.0",
+    "mlx-vlm>=0.1.27; sys_platform == 'darwin'",
 ]
 
 [tool.pdm]
@@ -109,7 +112,7 @@ target-version = [
 
 [tool.ruff]
 line-length = 100
-target-version = "0.2.11"
+target-version = "0.2.13"
 select = [
     "E",
     "F",
@@ -123,7 +126,7 @@ docstring-code-format = true
 
 [tool.mypy]
 strict = true
-python_version = "0.2.11"
+python_version = "0.2.13"
 ignore_missing_imports = true
 disallow_untyped_defs = true
 check_untyped_defs = true