hopx-ai 0.1.15__py3-none-any.whl

hopx_ai/_generated/models.py

@@ -0,0 +1,502 @@
+# generated by datamodel-codegen:
+#   filename:  openapi_specs_3.1.md
+#   timestamp: 2025-10-22T08:21:26+00:00
+
+from __future__ import annotations
+
+from enum import Enum
+from typing import Annotated, Any, Dict, List, Optional, Union
+
+from pydantic import AnyUrl, AwareDatetime, BaseModel, Field
+
+
+class Features(BaseModel):
+    code_execution: Optional[bool] = None
+    file_operations: Optional[bool] = None
+    terminal_access: Optional[bool] = None
+    websocket_streaming: Optional[bool] = None
+    rich_output: Optional[bool] = None
+    background_jobs: Optional[bool] = None
+    ipython_kernel: Optional[bool] = None
+    system_metrics: Optional[bool] = None
+    languages: Annotated[
+        Optional[List[str]], Field(examples=[['python', 'javascript', 'bash', 'go']])
+    ] = None
+
+
+class HealthResponse(BaseModel):
+    status: Annotated[Optional[str], Field(examples=['healthy'])] = None
+    agent: Annotated[Optional[str], Field(examples=['hopx-vm-agent-desktop'])] = None
+    version: Annotated[Optional[str], Field(examples=['3.1.1'])] = None
+    uptime: Annotated[Optional[str], Field(examples=['2h34m12s'])] = None
+    go_version: Annotated[Optional[str], Field(examples=['go1.22.2'])] = None
+    vm_id: Annotated[Optional[str], Field(examples=['1760954509layu9lw0'])] = None
+    features: Optional[Features] = None
+    active_streams: Annotated[Optional[int], Field(examples=[0])] = None
+
+
+class InfoResponse(BaseModel):
+    vm_id: Optional[str] = None
+    agent: Optional[str] = None
+    agent_version: Optional[str] = None
+    os: Optional[str] = None
+    arch: Optional[str] = None
+    go_version: Optional[str] = None
+    vm_ip: Optional[str] = None
+    vm_port: Optional[str] = None
+    start_time: Optional[AwareDatetime] = None
+    uptime: Annotated[Optional[float], Field(description='Uptime in seconds')] = None
+    endpoints: Annotated[
+        Optional[Dict[str, str]], Field(description='Map of endpoint names to HTTP methods + paths')
+    ] = None
+    features: Annotated[Optional[Dict[str, Any]], Field(description='Available features')] = None
+
+
+class Cpu(BaseModel):
+    usage_percent: Optional[float] = None
+    cores: Optional[int] = None
+
+
+class Memory(BaseModel):
+    total: Optional[int] = None
+    used: Optional[int] = None
+    free: Optional[int] = None
+    usage_percent: Optional[float] = None
+
+
+class Disk(BaseModel):
+    total: Optional[int] = None
+    used: Optional[int] = None
+    free: Optional[int] = None
+    usage_percent: Optional[float] = None
+
+
+class SystemMetrics(BaseModel):
+    cpu: Optional[Cpu] = None
+    memory: Optional[Memory] = None
+    disk: Optional[Disk] = None
+    uptime: Annotated[Optional[float], Field(description='System uptime in seconds')] = None
+
+
+class Language(Enum):
+    PYTHON = 'python'
+    PYTHON3 = 'python3'
+    NODE = 'node'
+    NODEJS = 'nodejs'
+    JAVASCRIPT = 'javascript'
+    JS = 'js'
+    BASH = 'bash'
+    SH = 'sh'
+    SHELL = 'shell'
+    GO = 'go'
+
+
+class ExecuteRequest(BaseModel):
+    code: Annotated[str, Field(description='Code to execute', examples=['print("Hello, World!")'])]
+    language: Annotated[Language, Field(description='Programming language', examples=['python'])]
+    env: Annotated[
+        Optional[Dict[str, str]],
+        Field(
+            description='Optional environment variables for this execution only.\n\n**Priority**: Request env > Global env > Agent env\n\n**Example**: `{"DATABASE_URL": "postgres://localhost/app", "DEBUG": "true"}`\n',
+            examples=[{'DATABASE_URL': 'postgres://localhost/testdb', 'DEBUG': 'true'}],
+        ),
+    ] = None
+    timeout: Annotated[Optional[int], Field(description='Timeout in seconds', ge=1, le=300)] = 30
+
+
+class ExecuteResponse(BaseModel):
+    stdout: Annotated[Optional[str], Field(description='Standard output')] = None
+    stderr: Annotated[Optional[str], Field(description='Standard error')] = None
+    exit_code: Annotated[Optional[int], Field(description='Exit code (0 = success)')] = None
+    execution_time: Annotated[Optional[float], Field(description='Execution time in seconds')] = (
+        None
+    )
+    timestamp: Optional[AwareDatetime] = None
+    language: Optional[str] = None
+    success: Annotated[
+        Optional[bool], Field(description='Whether execution succeeded (exit_code == 0)')
+    ] = None
+
+
+class BackgroundExecuteRequest(ExecuteRequest):
+    name: Annotated[
+        Optional[str], Field(description='Optional process name for identification')
+    ] = None
+
+
+class Status(Enum):
+    running = 'running'
+    completed = 'completed'
+    failed = 'failed'
+    killed = 'killed'
+
+
+class BackgroundExecuteResponse(BaseModel):
+    process_id: Annotated[Optional[str], Field(description='Unique process identifier')] = None
+    execution_id: Annotated[Optional[str], Field(description='Execution identifier')] = None
+    status: Optional[Status] = None
+    start_time: Optional[AwareDatetime] = None
+    message: Optional[str] = None
+    name: Annotated[Optional[str], Field(description='Process name (if provided)')] = None
+
+
+class Status1(Enum):
+    queued = 'queued'
+
+
+class AsyncExecuteResponse(BaseModel):
+    execution_id: Annotated[
+        Optional[str], Field(description='Unique execution identifier', examples=['abc123-def456'])
+    ] = None
+    status: Annotated[
+        Optional[Status1], Field(description='Execution status (always "queued" initially)')
+    ] = None
+    callback_url: Annotated[
+        Optional[AnyUrl],
+        Field(
+            description='URL that will receive execution results',
+            examples=['https://client.com/webhooks/execution'],
+        ),
+    ] = None
+    message: Annotated[
+        Optional[str],
+        Field(
+            description='Human-readable message',
+            examples=['Execution queued. Will POST to callback_url when complete.'],
+        ),
+    ] = None
+    timestamp: Annotated[Optional[AwareDatetime], Field(description='Queue timestamp')] = None
+
+
+class Status2(Enum):
+    COMPLETED = 'completed'
+    FAILED = 'failed'
+    TIMEOUT = 'timeout'
+
+
+class WebhookExecutionComplete(BaseModel):
+    execution_id: Annotated[
+        Optional[str],
+        Field(
+            description='Execution identifier (matches AsyncExecuteResponse)',
+            examples=['abc123-def456'],
+        ),
+    ] = None
+    status: Annotated[Optional[Status2], Field(description='Final execution status')] = None
+    stdout: Annotated[Optional[str], Field(description='Standard output')] = None
+    stderr: Annotated[Optional[str], Field(description='Standard error')] = None
+    exit_code: Annotated[Optional[int], Field(description='Exit code (0 = success)')] = None
+    execution_time: Annotated[
+        Optional[float], Field(description='Execution time in seconds', examples=[600.123])
+    ] = None
+    timestamp: Annotated[Optional[AwareDatetime], Field(description='Completion timestamp')] = None
+    error: Annotated[
+        Optional[str], Field(description='Error message (if status is "failed" or "timeout")')
+    ] = None
+
+
+class Status3(Enum):
+    running = 'running'
+    completed = 'completed'
+    failed = 'failed'
+    killed = 'killed'
+
+
+class ProcessInfo(BaseModel):
+    process_id: Optional[str] = None
+    execution_id: Optional[str] = None
+    name: Optional[str] = None
+    status: Optional[Status3] = None
+    language: Optional[str] = None
+    start_time: Optional[AwareDatetime] = None
+    end_time: Optional[AwareDatetime] = None
+    exit_code: Optional[int] = None
+    duration: Annotated[Optional[float], Field(description='Duration in seconds')] = None
+    pid: Annotated[Optional[int], Field(description='System process ID')] = None
+
+
+class ProcessListResponse(BaseModel):
+    processes: Optional[List[ProcessInfo]] = None
+    count: Optional[int] = None
+    timestamp: Optional[AwareDatetime] = None
+
+
+class Type(Enum):
+    image_png = 'image/png'
+    text_html = 'text/html'
+    application_json = 'application/json'
+    application_vnd_dataframe_json = 'application/vnd.dataframe+json'  # Pandas dataframe
+
+
+class Format(Enum):
+    base64 = 'base64'
+    html = 'html'
+    json = 'json'
+
+
+class Source(Enum):
+    matplotlib = 'matplotlib'
+    pandas = 'pandas'
+    plotly = 'plotly'
+    pandas_stdout = 'pandas_stdout'  # Pandas to stdout
+    other = 'other'
+
+
+class Metadata(BaseModel):
+    source: Optional[Source] = None
+    filename: Optional[str] = None
+    size: Optional[int] = None
+
+
+class RichOutput(BaseModel):
+    type: Annotated[Optional[Type], Field(description='MIME type')] = None
+    format: Optional[Format] = None
+    data: Annotated[
+        Optional[Union[str, Dict[str, Any]]], Field(description='Output data (base64 for images, HTML for tables, dict for dataframes)')
+    ] = None
+    metadata: Optional[Metadata] = None
+
+
+class CommandResponse(BaseModel):
+    stdout: Optional[str] = None
+    stderr: Optional[str] = None
+    exit_code: Optional[int] = None
+    execution_time: Optional[float] = None
+    command: Optional[str] = None
+    timestamp: Optional[AwareDatetime] = None
+
+
+class EnvVarsSetRequest(BaseModel):
+    env_vars: Annotated[
+        Dict[str, str],
+        Field(
+            description='Environment variables as key-value pairs',
+            examples=[
+                {
+                    'DATABASE_URL': 'postgres://prod-db/app',
+                    'API_KEY': 'sk-1234567890abcdef',
+                    'ENVIRONMENT': 'production',
+                }
+            ],
+        ),
+    ]
+    timestamp: Annotated[
+        Optional[AwareDatetime],
+        Field(
+            description='Optional timestamp for deduplication (prevents old requests from overwriting new ones)'
+        ),
+    ] = None
+    merge: Annotated[
+        Optional[bool],
+        Field(
+            description='If true, merge with existing env vars. If false (default), replace all.\n\n**Note**: PATCH /env always merges, PUT /env always replaces (this field is ignored for those endpoints)\n'
+        ),
+    ] = False
+
+
+class EnvVarsResponse(BaseModel):
+    env_vars: Annotated[
+        Optional[Dict[str, str]],
+        Field(
+            description='Environment variables as key-value pairs.\n\n**Security**: Sensitive values (containing KEY, SECRET, PASSWORD, TOKEN, etc.) are masked.\n',
+            examples=[
+                {
+                    'DATABASE_URL': 'postgres://localhost/app',
+                    'API_KEY': '***MASKED***',
+                    'DEBUG': 'true',
+                }
+            ],
+        ),
+    ] = None
+    count: Annotated[
+        Optional[int], Field(description='Number of environment variables', examples=[3])
+    ] = None
+    timestamp: Annotated[Optional[AwareDatetime], Field(description='Response timestamp')] = None
+
+
+class FileInfo(BaseModel):
+    name: Annotated[Optional[str], Field(description='File or directory name')] = None
+    path: Annotated[Optional[str], Field(description='Full path')] = None
+    size: Annotated[Optional[int], Field(description='Size in bytes')] = None
+    is_directory: Optional[bool] = None
+    modified_time: Optional[AwareDatetime] = None
+    permissions: Annotated[
+        Optional[str], Field(description='Unix permissions (e.g., drwxr-xr-x)')
+    ] = None
+
+
+class FileListResponse(BaseModel):
+    files: Optional[List[FileInfo]] = None
+    path: Annotated[Optional[str], Field(description='Directory path')] = None
+    count: Annotated[Optional[int], Field(description='Number of files')] = None
+
+
+class FileContentResponse(BaseModel):
+    content: Annotated[Optional[str], Field(description='File contents')] = None
+    path: Optional[str] = None
+    size: Annotated[Optional[int], Field(description='Size in bytes')] = None
+
+
+class FileWriteRequest(BaseModel):
+    path: Annotated[str, Field(description='File path to write', examples=['/workspace/script.py'])]
+    content: Annotated[str, Field(description='File contents')]
+
+
+class FileResponse(BaseModel):
+    message: Optional[str] = None
+    path: Optional[str] = None
+    success: Optional[bool] = None
+    size: Annotated[Optional[int], Field(description='File size (for write operations)')] = None
+    timestamp: Optional[AwareDatetime] = None
+
+
+class VNCInfo(BaseModel):
+    url: Annotated[
+        Optional[str], Field(description='VNC connection URL', examples=['vnc://vm.hopx.dev:5901'])
+    ] = None
+    password: Annotated[Optional[str], Field(description='VNC password')] = None
+    display: Annotated[Optional[int], Field(description='X11 display number', examples=[1])] = None
+    port: Annotated[Optional[int], Field(description='VNC port', examples=[5901])] = None
+    websocket_url: Annotated[
+        Optional[str], Field(description='noVNC WebSocket URL (if available)')
+    ] = None
+
+
+class WindowInfo(BaseModel):
+    id: Annotated[Optional[int], Field(description='Window ID (X11 window identifier)')] = None
+    title: Annotated[
+        Optional[str], Field(description='Window title', examples=['Firefox - Mozilla Firefox'])
+    ] = None
+    x: Annotated[Optional[int], Field(description='X coordinate')] = None
+    y: Annotated[Optional[int], Field(description='Y coordinate')] = None
+    width: Annotated[Optional[int], Field(description='Window width')] = None
+    height: Annotated[Optional[int], Field(description='Window height')] = None
+    is_active: Annotated[
+        Optional[bool], Field(description='Whether this window is currently active')
+    ] = None
+    is_minimized: Annotated[
+        Optional[bool], Field(description='Whether this window is minimized')
+    ] = None
+    pid: Annotated[Optional[int], Field(description='Process ID owning this window')] = None
+
+
+class Status4(Enum):
+    RECORDING = 'recording'
+    STOPPED = 'stopped'
+    FAILED = 'failed'
+
+
+class Format1(Enum):
+    MP4 = 'mp4'
+    WEBM = 'webm'
+
+
+class RecordingInfo(BaseModel):
+    recording_id: Annotated[Optional[str], Field(description='Unique recording identifier')] = None
+    status: Annotated[Optional[Status4], Field(description='Recording status')] = None
+    start_time: Annotated[Optional[AwareDatetime], Field(description='Recording start time')] = None
+    end_time: Annotated[
+        Optional[AwareDatetime], Field(description='Recording end time (if stopped)')
+    ] = None
+    duration: Annotated[Optional[float], Field(description='Recording duration in seconds')] = None
+    file_path: Annotated[Optional[str], Field(description='Path to recorded video file')] = None
+    file_size: Annotated[Optional[int], Field(description='Video file size in bytes')] = None
+    format: Annotated[Optional[Format1], Field(description='Video format')] = None
+
+
+class Display(BaseModel):
+    id: Optional[int] = None
+    name: Optional[str] = None
+    width: Optional[int] = None
+    height: Optional[int] = None
+    primary: Optional[bool] = None
+
+
+class DisplayInfo(BaseModel):
+    width: Annotated[
+        Optional[int], Field(description='Display width in pixels', examples=[1920])
+    ] = None
+    height: Annotated[
+        Optional[int], Field(description='Display height in pixels', examples=[1080])
+    ] = None
+    depth: Annotated[
+        Optional[int], Field(description='Color depth (bits per pixel)', examples=[24])
+    ] = None
+    refresh_rate: Annotated[
+        Optional[int], Field(description='Refresh rate in Hz', examples=[60])
+    ] = None
+    displays: Annotated[
+        Optional[List[Display]],
+        Field(description='List of available displays (multi-monitor support)'),
+    ] = None
+
+
+class Format2(Enum):
+    png = 'png'
+    jpeg = 'jpeg'
+
+
+class ScreenshotResponse(BaseModel):
+    image: Annotated[Optional[str], Field(description='Base64-encoded image data')] = None
+    format: Annotated[Optional[Format2], Field(description='Image format')] = None
+    width: Annotated[Optional[int], Field(description='Image width in pixels')] = None
+    height: Annotated[Optional[int], Field(description='Image height in pixels')] = None
+    size: Annotated[Optional[int], Field(description='Image size in bytes')] = None
+    timestamp: Annotated[Optional[AwareDatetime], Field(description='Screenshot timestamp')] = None
+
+
+class MetricsSnapshot(BaseModel):
+    uptime_seconds: Annotated[Optional[float], Field(description='Agent uptime in seconds')] = None
+    total_requests: Annotated[Optional[int], Field(description='Total HTTP requests handled')] = (
+        None
+    )
+    total_errors: Annotated[Optional[int], Field(description='Total errors encountered')] = None
+    active_executions: Annotated[
+        Optional[int], Field(description='Current active code executions')
+    ] = None
+    total_executions: Annotated[
+        Optional[int], Field(description='Total code executions completed')
+    ] = None
+
+
+class Code(Enum):
+    METHOD_NOT_ALLOWED = 'METHOD_NOT_ALLOWED'
+    INVALID_JSON = 'INVALID_JSON'
+    MISSING_PARAMETER = 'MISSING_PARAMETER'
+    PATH_NOT_ALLOWED = 'PATH_NOT_ALLOWED'
+    FILE_NOT_FOUND = 'FILE_NOT_FOUND'
+    PERMISSION_DENIED = 'PERMISSION_DENIED'
+    COMMAND_FAILED = 'COMMAND_FAILED'
+    EXECUTION_TIMEOUT = 'EXECUTION_TIMEOUT'
+    EXECUTION_FAILED = 'EXECUTION_FAILED'
+    INTERNAL_ERROR = 'INTERNAL_ERROR'
+    INVALID_PATH = 'INVALID_PATH'
+    FILE_ALREADY_EXISTS = 'FILE_ALREADY_EXISTS'
+    DIRECTORY_NOT_FOUND = 'DIRECTORY_NOT_FOUND'
+    INVALID_REQUEST = 'INVALID_REQUEST'
+    PROCESS_NOT_FOUND = 'PROCESS_NOT_FOUND'
+    DESKTOP_NOT_AVAILABLE = 'DESKTOP_NOT_AVAILABLE'
+
+
+class ErrorResponse(BaseModel):
+    error: Annotated[
+        str, Field(description='Human-readable error message', examples=['File not found'])
+    ]
+    code: Annotated[
+        Optional[Code],
+        Field(description='Machine-readable error code', examples=['FILE_NOT_FOUND']),
+    ] = None
+    request_id: Annotated[
+        Optional[str],
+        Field(
+            description='Request ID for tracing (from X-Request-ID header)',
+            examples=['550e8400-e29b-41d4-a716-446655440000'],
+        ),
+    ] = None
+    timestamp: AwareDatetime
+    path: Annotated[
+        Optional[str], Field(description='Related file path (for file operation errors)')
+    ] = None
+    details: Annotated[Optional[Dict[str, Any]], Field(description='Additional error context')] = (
+        None
+    )

hopx_ai/_temp_async_token.py

@@ -0,0 +1,14 @@
+# Token management pentru AsyncSandbox (copiat din sandbox.py)
+
+from datetime import datetime, timedelta
+from dataclasses import dataclass
+from typing import Dict, Optional
+
+@dataclass
+class TokenData:
+    """JWT token data."""
+    token: str
+    expires_at: datetime
+
+# Global token cache (shared between Sandbox instances)
+_token_cache: Dict[str, TokenData] = {}

hopx_ai/_test_env_fix.py

@@ -0,0 +1,30 @@
+# Quick test pentru env vars fix
+import sys
+sys.path.insert(0, "/var/www/sdks/python")
+
+from hopx_ai import Sandbox
+
+sandbox = Sandbox.create(
+    template="code-interpreter",
+    api_key="hopx_live_Lap0VJrWLii8.KSN6iLWELs13jHt960gSK9Eq63trgPApqMf7yLGVTNo"
+)
+
+print("1️⃣  Set env var...")
+sandbox.env.update({"TEST_VAR": "hello_world"})
+all_vars = sandbox.env.get_all()
+print(f"✅ TEST_VAR = {all_vars.get('TEST_VAR')}")
+
+print("\n2️⃣  Update env var...")
+sandbox.env.update({"TEST_VAR": "updated_value", "ANOTHER_VAR": "test123"})
+all_vars = sandbox.env.get_all()
+print(f"✅ TEST_VAR = {all_vars.get('TEST_VAR')}")
+print(f"✅ ANOTHER_VAR = {all_vars.get('ANOTHER_VAR')}")
+
+print("\n3️⃣  Delete env var...")
+sandbox.env.delete("TEST_VAR")
+all_vars = sandbox.env.get_all()
+print(f"✅ TEST_VAR after delete = {all_vars.get('TEST_VAR')}")
+print(f"✅ ANOTHER_VAR still = {all_vars.get('ANOTHER_VAR')}")
+
+sandbox.kill()
+print("\n✅ Test PASSED!")

hopx_ai/_utils.py

@@ -0,0 +1,9 @@
+"""Utility functions."""
+
+from typing import Any, Dict
+
+
+def remove_none_values(data: Dict[str, Any]) -> Dict[str, Any]:
+    """Remove None values from dictionary."""
+    return {k: v for k, v in data.items() if v is not None}
+

hopx_ai/_ws_client.py

@@ -0,0 +1,141 @@
+"""WebSocket client for real-time streaming."""
+
+import json
+import asyncio
+import logging
+from typing import Optional, Dict, Any, AsyncIterator, Callable
+from urllib.parse import urlparse
+
+try:
+    import websockets
+    from websockets.client import WebSocketClientProtocol
+    WEBSOCKETS_AVAILABLE = True
+except ImportError:
+    WEBSOCKETS_AVAILABLE = False
+    WebSocketClientProtocol = Any  # type: ignore
+
+logger = logging.getLogger(__name__)
+
+
+class WebSocketClient:
+    """
+    WebSocket client for Agent API streaming.
+    
+    Handles WebSocket connections with automatic reconnection,
+    message protocol, and async iteration.
+    """
+    
+    def __init__(self, agent_url: str):
+        """
+        Initialize WebSocket client.
+        
+        Args:
+            agent_url: Agent base URL (https://...)
+        """
+        if not WEBSOCKETS_AVAILABLE:
+            raise ImportError(
+                "websockets library is required for WebSocket features. "
+                "Install with: pip install websockets"
+            )
+        
+        self.agent_url = agent_url.rstrip('/')
+        # Convert https:// to wss:// for WebSocket
+        parsed = urlparse(self.agent_url)
+        ws_scheme = 'wss' if parsed.scheme == 'https' else 'ws'
+        self.ws_base_url = f"{ws_scheme}://{parsed.netloc}"
+        
+        logger.debug(f"WebSocket client initialized: {self.ws_base_url}")
+    
+    async def connect(
+        self,
+        endpoint: str,
+        *,
+        timeout: Optional[int] = None
+    ) -> WebSocketClientProtocol:
+        """
+        Connect to WebSocket endpoint.
+        
+        Args:
+            endpoint: WebSocket endpoint path (e.g., "/terminal")
+            timeout: Connection timeout in seconds
+        
+        Returns:
+            WebSocket connection
+        """
+        url = f"{self.ws_base_url}{endpoint}"
+        logger.debug(f"Connecting to WebSocket: {url}")
+        
+        try:
+            ws = await asyncio.wait_for(
+                websockets.connect(url),
+                timeout=timeout
+            )
+            logger.debug(f"WebSocket connected: {endpoint}")
+            return ws
+        except asyncio.TimeoutError:
+            raise TimeoutError(f"WebSocket connection timeout: {endpoint}")
+        except Exception as e:
+            logger.error(f"WebSocket connection failed: {e}")
+            raise
+    
+    async def send_message(
+        self,
+        ws: WebSocketClientProtocol,
+        message: Dict[str, Any]
+    ) -> None:
+        """
+        Send JSON message over WebSocket.
+        
+        Args:
+            ws: WebSocket connection
+            message: Message dictionary
+        """
+        await ws.send(json.dumps(message))
+        logger.debug(f"Sent WS message: {message.get('type', 'unknown')}")
+    
+    async def receive_message(
+        self,
+        ws: WebSocketClientProtocol
+    ) -> Dict[str, Any]:
+        """
+        Receive and parse JSON message from WebSocket.
+        
+        Args:
+            ws: WebSocket connection
+        
+        Returns:
+            Parsed message dictionary
+        """
+        data = await ws.recv()
+        if isinstance(data, bytes):
+            data = data.decode('utf-8')
+        message = json.loads(data)
+        logger.debug(f"Received WS message: {message.get('type', 'unknown')}")
+        return message
+    
+    async def iter_messages(
+        self,
+        ws: WebSocketClientProtocol
+    ) -> AsyncIterator[Dict[str, Any]]:
+        """
+        Iterate over incoming messages.
+        
+        Args:
+            ws: WebSocket connection
+        
+        Yields:
+            Parsed message dictionaries
+        """
+        try:
+            async for data in ws:
+                if isinstance(data, bytes):
+                    data = data.decode('utf-8')
+                message = json.loads(data)
+                logger.debug(f"Yielding WS message: {message.get('type', 'unknown')}")
+                yield message
+        except websockets.exceptions.ConnectionClosed:
+            logger.debug("WebSocket connection closed")
+    
+    def __repr__(self) -> str:
+        return f"<WebSocketClient url={self.ws_base_url}>"
+