cua-computer 0.4.2__tar.gz → 0.4.4__tar.gz

{cua_computer-0.4.2 → cua_computer-0.4.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cua-computer
-Version: 0.4.2
+Version: 0.4.4
 Summary: Computer-Use Interface (CUI) framework powering Cua
 Author-Email: TryCua <gh@trycua.com>
 Requires-Python: >=3.11

{cua_computer-0.4.2 → cua_computer-0.4.4}/computer/__init__.py

@@ -12,7 +12,6 @@ logger = logging.getLogger("computer")
 try:
     # Import from core telemetry
     from core.telemetry import (
-        flush,
         is_telemetry_enabled,
         record_event,
     )
@@ -30,9 +29,6 @@ try:
                 "python_version": sys.version,
             },
         )
-
-        # Flush events to ensure they're sent
-        flush()
     else:
         logger.info("Telemetry is disabled")
 except ImportError as e:

{cua_computer-0.4.2 → cua_computer-0.4.4}/computer/computer.py

@@ -9,10 +9,18 @@ import re
 from .logger import Logger, LogLevel
 import json
 import logging
-from .telemetry import record_computer_initialization
+from core.telemetry import is_telemetry_enabled, record_event
 import os
 from . import helpers
 
+import platform
+
+SYSTEM_INFO = {
+    "os": platform.system().lower(),
+    "os_version": platform.release(),
+    "python_version": platform.python_version(),
+}
+
 # Import provider related modules
 from .providers.base import VMProviderType
 from .providers.factory import VMProviderFactory
@@ -88,10 +96,11 @@ class Computer:
         self.logger = Logger("computer", verbosity)
         self.logger.info("Initializing Computer...")
 
-        if os_type == "macos":
-            image = "macos-sequoia-cua:latest"
-        elif os_type == "linux":
-            image = "trycua/cua-ubuntu:latest"
+        if not image:
+            if os_type == "macos":
+                image = "macos-sequoia-cua:latest"
+            elif os_type == "linux":
+                image = "trycua/cua-ubuntu:latest"
         image = str(image)
 
         # Store original parameters
@@ -151,6 +160,8 @@ class Computer:
             if not name:
                 # Normalize the name to be used for the VM
                 name = image.replace(":", "_")
+                # Remove any forward slashes
+                name = name.replace("/", "_")
 
             # Convert display parameter to Display object
             if isinstance(display, str):
@@ -189,8 +200,8 @@ class Computer:
         self.use_host_computer_server = use_host_computer_server
 
         # Record initialization in telemetry (if enabled)
-        if telemetry_enabled:
-            record_computer_initialization()
+        if telemetry_enabled and is_telemetry_enabled():
+            record_event("computer_initialized", SYSTEM_INFO)
         else:
             self.logger.debug("Telemetry disabled - skipping initialization tracking")
 

{cua_computer-0.4.2 → cua_computer-0.4.4}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "pdm.backend"
 
 [project]
 name = "cua-computer"
-version = "0.4.2"
+version = "0.4.4"
 description = "Computer-Use Interface (CUI) framework powering Cua"
 readme = "README.md"
 authors = [
@@ -57,7 +57,7 @@ target-version = [
 
 [tool.ruff]
 line-length = 100
-target-version = "0.4.2"
+target-version = "0.4.4"
 select = [
     "E",
     "F",
@@ -71,7 +71,7 @@ docstring-code-format = true
 
 [tool.mypy]
 strict = true
-python_version = "0.4.2"
+python_version = "0.4.4"
 ignore_missing_imports = true
 disallow_untyped_defs = true
 check_untyped_defs = true

cua_computer-0.4.2/computer/telemetry.py

@@ -1,116 +0,0 @@
-"""Computer telemetry for tracking anonymous usage and feature usage."""
-
-import logging
-import platform
-from typing import Any
-
-# Import the core telemetry module
-TELEMETRY_AVAILABLE = False
-
-try:
-    from core.telemetry import (
-        increment,
-        is_telemetry_enabled,
-        is_telemetry_globally_disabled,
-        record_event,
-    )
-
-    def increment_counter(counter_name: str, value: int = 1) -> None:
-        """Wrapper for increment to maintain backward compatibility."""
-        if is_telemetry_enabled():
-            increment(counter_name, value)
-
-    def set_dimension(name: str, value: Any) -> None:
-        """Set a dimension that will be attached to all events."""
-        logger = logging.getLogger("computer.telemetry")
-        logger.debug(f"Setting dimension {name}={value}")
-
-    TELEMETRY_AVAILABLE = True
-    logger = logging.getLogger("computer.telemetry")
-    logger.info("Successfully imported telemetry")
-except ImportError as e:
-    logger = logging.getLogger("computer.telemetry")
-    logger.warning(f"Could not import telemetry: {e}")
-    TELEMETRY_AVAILABLE = False
-
-
-# Local fallbacks in case core telemetry isn't available
-def _noop(*args: Any, **kwargs: Any) -> None:
-    """No-op function for when telemetry is not available."""
-    pass
-
-
-logger = logging.getLogger("computer.telemetry")
-
-# If telemetry isn't available, use no-op functions
-if not TELEMETRY_AVAILABLE:
-    logger.debug("Telemetry not available, using no-op functions")
-    record_event = _noop  # type: ignore
-    increment_counter = _noop  # type: ignore
-    set_dimension = _noop  # type: ignore
-    get_telemetry_client = lambda: None  # type: ignore
-    flush = _noop  # type: ignore
-    is_telemetry_enabled = lambda: False  # type: ignore
-    is_telemetry_globally_disabled = lambda: True  # type: ignore
-
-# Get system info once to use in telemetry
-SYSTEM_INFO = {
-    "os": platform.system().lower(),
-    "os_version": platform.release(),
-    "python_version": platform.python_version(),
-}
-
-
-def enable_telemetry() -> bool:
-    """Enable telemetry if available.
-
-    Returns:
-        bool: True if telemetry was successfully enabled, False otherwise
-    """
-    global TELEMETRY_AVAILABLE
-
-    # Check if globally disabled using core function
-    if TELEMETRY_AVAILABLE and is_telemetry_globally_disabled():
-        logger.info("Telemetry is globally disabled via environment variable - cannot enable")
-        return False
-
-    # Already enabled
-    if TELEMETRY_AVAILABLE:
-        return True
-
-    # Try to import and enable
-    try:
-        # Verify we can import core telemetry
-        from core.telemetry import record_event  # type: ignore
-
-        TELEMETRY_AVAILABLE = True
-        logger.info("Telemetry successfully enabled")
-        return True
-    except ImportError as e:
-        logger.warning(f"Could not enable telemetry: {e}")
-        return False
-
-
-def is_telemetry_enabled() -> bool:
-    """Check if telemetry is enabled.
-
-    Returns:
-        bool: True if telemetry is enabled, False otherwise
-    """
-    # Use the core function if available, otherwise use our local flag
-    if TELEMETRY_AVAILABLE:
-        from core.telemetry import is_telemetry_enabled as core_is_enabled
-
-        return core_is_enabled()
-    return False
-
-
-def record_computer_initialization() -> None:
-    """Record when a computer instance is initialized."""
-    if TELEMETRY_AVAILABLE and is_telemetry_enabled():
-        record_event("computer_initialized", SYSTEM_INFO)
-
-        # Set dimensions that will be attached to all events
-        set_dimension("os", SYSTEM_INFO["os"])
-        set_dimension("os_version", SYSTEM_INFO["os_version"])
-        set_dimension("python_version", SYSTEM_INFO["python_version"])