unrealon 2.0.5__py3-none-any.whl → 2.0.7__py3-none-any.whl

{unrealon-2.0.5.dist-info → unrealon-2.0.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unrealon
-Version: 2.0.5
+Version: 2.0.7
 Summary: Enterprise-grade web scraping platform with AI-powered automation and real-time orchestration capabilities
 Author-email: UnrealOn Team <team@unrealon.com>
 License: MIT

{unrealon-2.0.5.dist-info → unrealon-2.0.7.dist-info}/RECORD

@@ -28,14 +28,14 @@ unrealon_browser/stealth/nodriver_stealth.pyc,sha256=SnDMdeG_W2LDK_3UfrjDeolMpqx
 unrealon_browser/stealth/playwright_stealth.pyc,sha256=Y-IiRTu136R6LIstGdKbns1j-CODX3sTGVT0IGx_MF4,6529
 unrealon_browser/stealth/scanner_tester.pyc,sha256=ki_Lp7zBp12iWgj3LGPnXTwemwgaGG1UrWeM1IXB6Gs,21711
 unrealon_browser/stealth/undetected_chrome.pyc,sha256=HNp2_84zwp-JG-Y3jqQwCNldnG6P5bBnbf9dBV_pOuw,12765
-unrealon_core/__init__.py,sha256=ZA2VNptlNYmeX9cyfGklKDnEMR_peu2lG24nBvAPbLU,5317
+unrealon_core/__init__.py,sha256=ZAinyQDsAS63xqnl5uqUKTna9M-xFDs6fLwy8-Hh-X0,5599
 unrealon_core/version.py,sha256=qZOlKA_Hsz7_KXsCLO_0l9Mf0u_iNTxvHZPV21DwCqs,5803
 unrealon_core/config/__init__.py,sha256=57-KZaTDya0oyfOCfZ3pU1xLLbnZPBgqwfP9VrfhcKE,395
 unrealon_core/config/environment.py,sha256=TACbyjr3lxrA9R_Ve0LLqyLnRk3FShm2imQjhiP9i98,4589
 unrealon_core/config/urls.py,sha256=7ScpWEhSu1kewhMaKhuqgiMHIv_fyI1TyRiS-f2wrLo,2821
 unrealon_core/enums/__init__.py,sha256=C-fqNVSWofnaq2Ed6GbGtck8WkN8gpHYA_NiIlaEbTI,520
 unrealon_core/enums/status.py,sha256=DT6rth7Ve7chsYhJtccY6plsgQQ8jEMhfhcZTpK308c,6861
-unrealon_core/enums/types.py,sha256=hf1NkA2sBIfgtNHqdQPr7Uow_E2SeDGh1Jpxls0RFYg,8704
+unrealon_core/enums/types.py,sha256=o4df_zPVE5ByF4P5t_dnwmr1GVQytl5ppKNQBCu0i2A,8784
 unrealon_core/error_handling/__init__.py,sha256=s7VXUJG7nvD6Od76-ZfFmn5e6r29mxNusP8YPba73GM,1248
 unrealon_core/error_handling/circuit_breaker.py,sha256=IPT7I8nrDm4eDojrjMgM3-vLIaz-0raS-9W3LUpFnhY,10066
 unrealon_core/error_handling/error_context.py,sha256=jfluYy1FVyyU_8dwjBN1w2vPq9oLRYLg2fgndlc_8As,9798
@@ -61,8 +61,9 @@ unrealon_core/models/logging.py,sha256=MKGjMmhJhediGNVKmX0wfm1GO2kfvEqOp2cCfq42V
 unrealon_core/models/task.py,sha256=QSOQMXNbvD7vfP6ddYUBPwD7jj0N9_zQxjwDMrznXo4,435
 unrealon_core/models/typed_responses.py,sha256=shEvV4A7mYUKIzG3zD4zoRO2PmVLL3iLtGjzRGSW8Ks,7034
 unrealon_core/models/websocket_session.py,sha256=q3RXQP2CwRTgUTYMRnb4xzzdGWUOeL8maYxVFPmF92w,5119
-unrealon_core/models/websocket/__init__.py,sha256=F-KDDZAMJvCVQMo2Z-xLPQmG0B8JEQ2auxxI89ZCsro,2551
+unrealon_core/models/websocket/__init__.py,sha256=wBgv1ZI_0_SiC2AX-UOA5fivi4rf3GEFhLDG159-x_s,2845
 unrealon_core/models/websocket/base.py,sha256=gmkJ-vLpsajc4aGlPBeMURIFNg3AgX4yYSnJQAqqkf0,1388
+unrealon_core/models/websocket/broadcast.py,sha256=XoefNORneicJUQxutFkzuPJDEAEyQlZEWtmBqL97yVg,4126
 unrealon_core/models/websocket/config.py,sha256=AZz9wK8j0UIkRkDE17Nsxdocimh-41MJQkS-_VzTnM0,4665
 unrealon_core/models/websocket/driver.py,sha256=Bi1B7nSSWkLFCwfIOLJSkPUxGS5r9x4Mw9XWvHuSxa0,5206
 unrealon_core/models/websocket/errors.py,sha256=l5w1rCx_RrzdISPwLOoKtoPCuSNjHL5XExyFaHgayS4,3095
@@ -92,16 +93,16 @@ unrealon_driver/decorators/task.py,sha256=nnG-S99BSdtKYXZ97XBkQLDc-jpc4a6SKv-6b7
 unrealon_driver/decorators/timing.py,sha256=Bmtsp2tWxY4EOq3onF1OTSk0rqf9lQNlX4RL-hyr4ZM,3927
 unrealon_driver/driver/__init__.py,sha256=P_NJXcsigIjROwErqUXgS0VneUfGggFL4bWDbdB73iw,419
 unrealon_driver/driver/communication/__init__.py,sha256=SalrzX_55WXlkbS9P6nm6PhddoI8pquURqx3103Q708,205
-unrealon_driver/driver/communication/session.py,sha256=vRarQAUunmqltD1p2Z7Jms5xBK543XarfCUWAC_G8XQ,6936
-unrealon_driver/driver/communication/websocket_client.py,sha256=h23aS5v0Tt6_BCA2fSlij9Vab7b4OH2UM7C6Z42YY2E,7412
+unrealon_driver/driver/communication/session.py,sha256=DYN_Q3Qm3XuOi-dM8aNihJfQJl7n8cgB1GHP621ZosQ,7552
+unrealon_driver/driver/communication/websocket_client.py,sha256=VPsICBvGHunuCGZvorvPCF01Qdvp7QWyTDX0hkYXRwo,7910
 unrealon_driver/driver/core/__init__.py,sha256=ZvJQp1zO7pj6tBNYTJk2fj-0ZMiQTQEk-I9hXalNsfg,235
 unrealon_driver/driver/core/config.py,sha256=jWJjRll19VlL4iM5Q-J3o9qwYeH89Iuj1_3KayM6fCk,5914
 unrealon_driver/driver/core/driver.py,sha256=NI-pdhnduRyHLsfFr8HmP2gp7pR1pWB4vBIJkMJ2cls,7886
 unrealon_driver/driver/factory/__init__.py,sha256=XrjBhOaLvC3MIG5PAFIYS_xYXFDz5JizpFvmQcwA7mU,189
 unrealon_driver/driver/factory/manager_factory.py,sha256=zJt63N8oWNJS0aNbLy7WjY4GCeUwLjvtWGdqdpucufU,4792
 unrealon_driver/driver/lifecycle/__init__.py,sha256=KnkXklezAOIbXcCzEU_XSOt32z7tz1zIGclXYXTkO8k,286
-unrealon_driver/driver/lifecycle/daemon.py,sha256=6cglmGsMLmMgqkXm0cquzcQiUNLRJ2abrj37bJysIUA,2508
-unrealon_driver/driver/lifecycle/initialization.py,sha256=jwamdzG3IDKwYiTbMnGkBOJeWeTlPxIlQkOrsc22qCo,3649
+unrealon_driver/driver/lifecycle/daemon.py,sha256=KHAzpiWFu3HRElRtzSEStmI74bMivFjfCAFlXha87KU,2609
+unrealon_driver/driver/lifecycle/initialization.py,sha256=R4MgfkSNnfAdMO0Kp1Cx42cfNqq8VIxj_mGX7ECXad4,4406
 unrealon_driver/driver/lifecycle/shutdown.py,sha256=h-6hKjDP30OjKiSgLe3UIfpS5wDbQFRaYWnEPosyTe0,1426
 unrealon_driver/driver/monitoring/__init__.py,sha256=MJjLm4b5pzSpXGOq50e-MKUkyToiUbARzkGqNqN-iJg,162
 unrealon_driver/driver/monitoring/health.py,sha256=rOH0wChDBTnDbV7MstB21JsJJUMGMeq3crb47RLL2ek,2164
@@ -113,16 +114,16 @@ unrealon_driver/managers/base.py,sha256=GkuXillg9uqqnx6RL682fmKgK-7JyqYlH6DFUgyN
 unrealon_driver/managers/browser.py,sha256=9kwdmWPxWFpNQ1KejtpzdN615V8PBJu8Y2nUOewQaeQ,3489
 unrealon_driver/managers/cache.py,sha256=c0tPKQ5KFd_Un1U8mw3j1WPuycxg863MMWNMveVF_2I,3506
 unrealon_driver/managers/http.py,sha256=EjlpoTRuhpsgzzrEARxRlbGczzua7hnKFVq06bvCgTM,3624
-unrealon_driver/managers/logger.py,sha256=w-JiouSPN8YZKTgDshANv6EpiKEJ9r6R1vxQexQp32M,10243
+unrealon_driver/managers/logger.py,sha256=PL3rA9ZQl12jJU0EiPAkLwJ6eDHQfIzr8-nc8bVivKQ,10526
 unrealon_driver/managers/proxy.py,sha256=Y_3ok2iGefeYHcTtBQIEK3Rnr8qRnPsP90hOOYPUMz0,3449
 unrealon_driver/managers/registry.py,sha256=--oNPU-65e8J21ubJufyEOc1TirnzJIvpvuY_j7rH7Q,2666
 unrealon_driver/managers/threading.py,sha256=djw5cSC99dfBKmep3IJ_8IgxQceMXtNvCp5fIxHM0TY,1702
 unrealon_driver/managers/update.py,sha256=-hohVxGXpj5bZ6ZTQN6NH1RK9Pd6GVzCMtu3GS2SdcQ,3582
 unrealon_driver/utils/__init__.py,sha256=qxXVoQJVdLJhaLBXk_LZV_062AhrvBrMPXWAKfEc3C4,104
 unrealon_driver/utils/time.py,sha256=Oxk1eicKeZl8ZWbf7gu1Ll716k6CpXmVj67FHSnPIsA,184
-unrealon-2.0.5.dist-info/LICENSE,sha256=eEH8mWZW49YMpl4Sh5MtKqkZ8aVTzKQXiNPEnvL14ns,1070
-unrealon-2.0.5.dist-info/METADATA,sha256=LvQG4LPi81KyP5XKqPdEg1ZspdFc2inV9XP0SM6qqW8,15688
-unrealon-2.0.5.dist-info/WHEEL,sha256=pL8R0wFFS65tNSRnaOVrsw9EOkOqxLrlUPenUYnJKNo,91
-unrealon-2.0.5.dist-info/entry_points.txt,sha256=k0qM-eotpajkKUq-almJmxj9afhXprZ6IkvQkSdcKhI,104
-unrealon-2.0.5.dist-info/top_level.txt,sha256=Gu8IeIfIVfUxdi-h-F0nKMQxo15pjhHZ0aTadXTpRE8,47
-unrealon-2.0.5.dist-info/RECORD,,
+unrealon-2.0.7.dist-info/LICENSE,sha256=eEH8mWZW49YMpl4Sh5MtKqkZ8aVTzKQXiNPEnvL14ns,1070
+unrealon-2.0.7.dist-info/METADATA,sha256=6emT2rPYUexfKJNIKu4CuuhuSCYptvwMahqhmiXqzRs,15688
+unrealon-2.0.7.dist-info/WHEEL,sha256=pL8R0wFFS65tNSRnaOVrsw9EOkOqxLrlUPenUYnJKNo,91
+unrealon-2.0.7.dist-info/entry_points.txt,sha256=k0qM-eotpajkKUq-almJmxj9afhXprZ6IkvQkSdcKhI,104
+unrealon-2.0.7.dist-info/top_level.txt,sha256=Gu8IeIfIVfUxdi-h-F0nKMQxo15pjhHZ0aTadXTpRE8,47
+unrealon-2.0.7.dist-info/RECORD,,

unrealon_core/__init__.py

@@ -42,7 +42,10 @@ from .models.communication import (
     create_error_message, create_ack_message,
     # Data models for strict typing
     TaskAssignmentData, TaskResultData,
-    DriverRegistrationData, HeartbeatData, LogEntryData, LogContext
+    DriverRegistrationData, HeartbeatData, LogEntryData, LogContext,
+    # Broadcast models
+    DriverBroadcastData, DriverRegisterBroadcast,
+    DriverHeartbeatBroadcast, DriverDisconnectBroadcast
 )
 from .enums.status import DriverStatus, TaskStatus, ProxyStatus, LogLevel
 from .enums.types import MessageType, ProxyType, TaskPriority
@@ -127,6 +130,12 @@ __all__ = [
     "LogEntryData",
     "LogContext",
     
+    # Broadcast models
+    "DriverBroadcastData",
+    "DriverRegisterBroadcast",
+    "DriverHeartbeatBroadcast", 
+    "DriverDisconnectBroadcast",
+    
     # Status enums
     "DriverStatus",
     "TaskStatus",

unrealon_core/enums/types.py

@@ -53,6 +53,7 @@ class MessageType(str, Enum):
     LOG_BATCH = "log_batch"                       # Batch of log messages
     
     # System messages
+    WELCOME = "welcome"                           # Welcome message from server
     PING = "ping"                                 # Ping message
     PONG = "pong"                                 # Pong response
     ERROR = "error"                               # Error message

unrealon_core/models/websocket/__init__.py

@@ -38,6 +38,12 @@ from .proxy import (
 )
 
 from .utils import create_error_message, create_ack_message
+from .broadcast import (
+    DriverBroadcastData,
+    DriverRegisterBroadcast,
+    DriverHeartbeatBroadcast,
+    DriverDisconnectBroadcast
+)
 
 __all__ = [
     # Base
@@ -87,5 +93,11 @@ __all__ = [
     
     # Utilities
     'create_error_message',
-    'create_ack_message'
+    'create_ack_message',
+    
+    # Broadcast messages
+    'DriverBroadcastData',
+    'DriverRegisterBroadcast',
+    'DriverHeartbeatBroadcast',
+    'DriverDisconnectBroadcast'
 ]

unrealon_core/models/websocket/broadcast.py

@@ -0,0 +1,168 @@
+"""
+Broadcast WebSocket Models.
+
+Strictly typed models for broadcasting driver events to monitoring clients.
+These are Server → Client messages for real-time monitoring.
+
+Phase 2: Core Systems - WebSocket Bridge
+"""
+
+from typing import List, Optional
+from pydantic import Field, ConfigDict
+
+from ..base import UnrealOnBaseModel
+from .base import WebSocketMessage, MessageType
+
+
+class DriverBroadcastData(UnrealOnBaseModel):
+    """Driver broadcast data payload - strictly typed."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    driver_id: str = Field(
+        description="Unique driver identifier",
+        min_length=1
+    )
+    
+    driver_type: str = Field(
+        description="Type of driver (e.g., 'universal', 'ecommerce')",
+        min_length=1
+    )
+    
+    status: str = Field(
+        default="active",
+        pattern=r"^(active|idle|busy|offline|error)$",
+        description="Current driver status"
+    )
+    
+    capabilities: List[str] = Field(
+        default_factory=list,
+        description="List of supported task types"
+    )
+    
+    active_tasks: int = Field(
+        default=0,
+        ge=0,
+        description="Number of currently active tasks"
+    )
+    
+    completed_tasks: int = Field(
+        default=0,
+        ge=0,
+        description="Total completed tasks since startup"
+    )
+    
+    failed_tasks: int = Field(
+        default=0,
+        ge=0,
+        description="Total failed tasks since startup"
+    )
+    
+    uptime_seconds: float = Field(
+        default=0.0,
+        ge=0.0,
+        description="Driver uptime in seconds"
+    )
+    
+    last_seen: Optional[str] = Field(
+        default=None,
+        description="ISO timestamp of last heartbeat"
+    )
+    
+    connected_at: Optional[str] = Field(
+        default=None,
+        description="ISO timestamp when driver connected"
+    )
+    
+    disconnected_at: Optional[str] = Field(
+        default=None,
+        description="ISO timestamp when driver disconnected"
+    )
+    
+    # Additional fields for complete driver information
+    version: Optional[str] = Field(
+        default=None,
+        description="Driver version"
+    )
+    
+    driver_name: Optional[str] = Field(
+        default=None,
+        description="Human-readable driver name"
+    )
+    
+    environment: Optional[str] = Field(
+        default=None,
+        description="Driver environment (development, production, etc.)"
+    )
+    
+    region: Optional[str] = Field(
+        default=None,
+        description="Driver region"
+    )
+    
+    max_concurrent_tasks: Optional[int] = Field(
+        default=None,
+        ge=1,
+        description="Maximum concurrent tasks"
+    )
+    
+    tags: List[str] = Field(
+        default_factory=list,
+        description="Driver tags"
+    )
+
+
+class DriverRegisterBroadcast(WebSocketMessage):
+    """Driver registration broadcast message (Server → Monitoring Clients)."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    type: MessageType = Field(
+        default=MessageType.DRIVER_REGISTER
+    )
+    
+    data: DriverBroadcastData = Field(
+        description="Driver registration broadcast data"
+    )
+
+
+class DriverHeartbeatBroadcast(WebSocketMessage):
+    """Driver heartbeat broadcast message (Server → Monitoring Clients)."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    type: MessageType = Field(
+        default=MessageType.DRIVER_HEARTBEAT,
+        frozen=True
+    )
+    
+    data: DriverBroadcastData = Field(
+        description="Driver heartbeat broadcast data"
+    )
+
+
+class DriverDisconnectBroadcast(WebSocketMessage):
+    """Driver disconnect broadcast message (Server → Monitoring Clients)."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    type: MessageType = Field(
+        default=MessageType.DRIVER_DISCONNECT,
+        frozen=True
+    )
+    
+    data: DriverBroadcastData = Field(
+        description="Driver disconnect broadcast data"
+    )

unrealon_driver/driver/communication/session.py

@@ -12,6 +12,7 @@ from pydantic import BaseModel, Field
 
 from unrealon_core.models.websocket import (
     TaskAssignmentData,
+    TaskAssignmentMessage,
     TaskResultData,
     WebSocketMessage
 )
@@ -69,21 +70,26 @@ class DriverSession:
         
         try:
             # Connect WebSocket
+            logger.info(f"🔌 Connecting WebSocket for driver: {self.driver_id}")
             self.status = SessionStatus.CONNECTING
             if not await self.websocket_client.connect():
+                logger.error(f"❌ WebSocket connection failed for driver: {self.driver_id}")
                 self.status = SessionStatus.ERROR
                 return False
             
+            logger.info(f"✅ WebSocket connected for driver: {self.driver_id}")
             self.status = SessionStatus.CONNECTED
             self.stats.connected_at = utc_now()
             
             # Register driver
+            logger.info(f"📝 Registering driver: {self.driver_id} with capabilities: {capabilities}")
             if await self.register(capabilities or []):
                 self.status = SessionStatus.REGISTERED
                 self.stats.registered_at = utc_now()
-                logger.info(f"Session started for driver: {self.driver_id}")
+                logger.info(f"✅ Session started for driver: {self.driver_id}")
                 return True
             else:
+                logger.error(f"❌ Driver registration failed for: {self.driver_id}")
                 self.status = SessionStatus.ERROR
                 return False
                 
@@ -139,8 +145,11 @@ class DriverSession:
     async def _handle_task_assignment(self, message: WebSocketMessage):
         """Handle task assignment."""
         try:
-            # Validate task data
-            task_data = TaskAssignmentData.model_validate(message.data)
+            # Parse as TaskAssignmentMessage
+            
+            # Convert to proper task assignment message
+            task_message = TaskAssignmentMessage.model_validate(message.model_dump())
+            task_data = task_message.data
             
             # Find handler
             handler = self.task_handlers.get(task_data.task_type)

unrealon_driver/driver/communication/websocket_client.py

@@ -40,9 +40,11 @@ class WebSocketClient:
     - Clean error handling
     """
     
-    def __init__(self, websocket_url: str, driver_id: str):
+    def __init__(self, websocket_url: str, driver_id: str, custom_logger=None):
         self.websocket_url = websocket_url
         self.driver_id = driver_id
+        # Use custom logger if provided, otherwise use default
+        self._logger = custom_logger if custom_logger else logger
         self.websocket: Optional[ClientConnection] = None
         self.connected = False
         self.running = False
@@ -94,20 +96,20 @@ class WebSocketClient:
             self.websocket = None
         
         self.connected = False
-        logger.info("WebSocket client stopped")
+        self._logger.info("WebSocket client stopped")
     
     async def _establish_connection(self) -> bool:
         """Establish WebSocket connection."""
         try:
-            logger.info(f"Connecting to WebSocket: {self.websocket_url}")
+            self._logger.info(f"Connecting to WebSocket: {self.websocket_url}")
             self.websocket = await websockets.connect(self.websocket_url)
             self.connected = True
             self.reconnect_attempts = 0
             self.reconnect_delay = 1.0
-            logger.info("WebSocket connected successfully")
+            self._logger.info("WebSocket connected successfully")
             return True
         except Exception as e:
-            logger.error(f"WebSocket connection failed: {e}")
+            self._logger.error(f"WebSocket connection failed: {e}")
             self.connected = False
             return False
     
@@ -115,10 +117,10 @@ class WebSocketClient:
         """Monitor connection and handle reconnection."""
         while self.running:
             if not self.connected and self.running:
-                logger.info(f"Attempting reconnection (attempt {self.reconnect_attempts + 1})")
+                self._logger.info(f"Attempting reconnection (attempt {self.reconnect_attempts + 1})")
                 
                 if await self._establish_connection():
-                    logger.info("Reconnection successful")
+                    self._logger.info("Reconnection successful")
                 else:
                     self.reconnect_attempts += 1
                     # Exponential backoff
@@ -136,13 +138,16 @@ class WebSocketClient:
             if self.connected and self.websocket and self.message_queue:
                 try:
                     message = self.message_queue.popleft()
+                    self._logger.info(f"📤 Sending WebSocket message: {message[:200]}...")  # Log first 200 chars
                     await self.websocket.send(message)
+                    self._logger.info(f"✅ Message sent successfully")
                 except (websockets.exceptions.ConnectionClosed, ConnectionResetError):
                     self.connected = False
                     # Put message back in queue
                     self.message_queue.appendleft(message)
+                    self._logger.warning("🔌 Connection lost, message queued for retry")
                 except Exception as e:
-                    logger.error(f"Error sending message: {e}")
+                    self._logger.error(f"❌ Error sending message: {e}")
             else:
                 await asyncio.sleep(0.1)
     
@@ -163,9 +168,9 @@ class WebSocketClient:
                         
                 except (websockets.exceptions.ConnectionClosed, ConnectionResetError):
                     self.connected = False
-                    logger.warning("WebSocket connection lost")
+                    self._logger.warning("WebSocket connection lost")
                 except Exception as e:
-                    logger.error(f"Error receiving message: {e}")
+                    self._logger.error(f"Error receiving message: {e}")
             else:
                 await asyncio.sleep(0.1)
     
@@ -197,9 +202,9 @@ class WebSocketClient:
             # Queue for sending
             self.send(registration_message)
             
-            logger.info(f"Driver registration queued: {self.driver_id}")
+            self._logger.info(f"Driver registration queued: {self.driver_id}")
             return True
             
         except Exception as e:
-            logger.error(f"Driver registration failed: {e}")
+            self._logger.error(f"Driver registration failed: {e}")
             return False

unrealon_driver/driver/lifecycle/daemon.py

@@ -47,6 +47,7 @@ class DaemonManager:
         
         try:
             # Initialize driver with capabilities
+            logger.info(f"🎯 Daemon initializing driver with capabilities: {driver.capabilities}")
             success = await DriverInitializer.initialize_driver(driver, driver.capabilities)
             if not success:
                 raise RuntimeError("Driver initialization failed")

unrealon_driver/driver/lifecycle/initialization.py

@@ -10,6 +10,8 @@ from typing import List, TYPE_CHECKING
 from ..core.config import DriverMode
 from ..communication.websocket_client import WebSocketClient
 from ..communication.session import DriverSession
+from unrealon_core.config.environment import get_environment_config
+
 
 if TYPE_CHECKING:
     from ..core.driver import UniversalDriver
@@ -37,6 +39,14 @@ class DriverInitializer:
         
         try:
             logger.info(f"Initializing UniversalDriver: {driver.driver_id}")
+            logger.info(f"🔧 Driver mode: {driver.config.mode}")
+            logger.info(f"📡 WebSocket URL: {driver.config.effective_websocket_url}")
+            logger.info(f"🎯 Capabilities: {capabilities}")
+            
+            # Log environment info if available
+            env_config = get_environment_config()
+            logger.info(f"🌐 Environment: {env_config.environment.value}")
+            print(f"🌐 Environment: {env_config.environment}")
             
             # Initialize manager system
             if not await driver.manager_registry.initialize_all():
@@ -65,9 +75,12 @@ class DriverInitializer:
         
         # Setup WebSocket client if URL available
         if websocket_url:
+            # Pass driver's logger to WebSocket client for unified logging
+            custom_logger = driver.logger_manager.local_logger if driver.logger_manager else None
             driver.websocket_client = WebSocketClient(
                 websocket_url,
-                driver.driver_id
+                driver.driver_id,
+                custom_logger=custom_logger
             )
             
             # Set WebSocket client for logger (always for logging)

unrealon_driver/managers/logger.py

@@ -89,6 +89,11 @@ class LoggerManager(BaseManager):
             self.local_logger.addHandler(handler)
             self.local_logger.setLevel(getattr(logging, self.config.log_level))
             
+            # Also configure root unrealon_driver logger to use same handler
+            unrealon_logger = logging.getLogger('unrealon_driver')
+            unrealon_logger.addHandler(handler)
+            unrealon_logger.setLevel(getattr(logging, self.config.log_level))
+            
         except Exception as e:
             logger.error(f"Failed to setup local logging: {e}")