escobar 0.1.89__py3-none-any.whl → 0.1.90__py3-none-any.whl

escobar/_version.py

@@ -1,4 +1,4 @@
 # This file is auto-generated by Hatchling. As such, do not:
 #   - modify
 #   - track in version control e.g. be sure to add to .gitignore
-__version__ = VERSION = '0.1.89'
+__version__ = VERSION = '0.1.90'

escobar/handlers.py

@@ -6,6 +6,8 @@ import websockets
 import ssl
 import logging
 import time
+import threading
+import psutil
 from urllib.parse import urlparse, urlunparse
 from jupyter_server.base.handlers import JupyterHandler
 from jupyter_server.utils import url_path_join
@@ -152,8 +154,91 @@ class ProxyHandler(JupyterHandler):
 class WebSocketProxyHandler(tornado.websocket.WebSocketHandler):
     """
     WebSocket proxy handler that forwards connections from /ws to target server
+    Enhanced with comprehensive logging for debugging intermittent connection issues
     """
     
+    # Class-level connection tracking
+    _connection_count = 0
+    _active_connections = {}
+    
+    def _log_system_state(self, context=""):
+        """Log current system state for debugging"""
+        try:
+            # System resource information
+            cpu_percent = psutil.cpu_percent(interval=0.1)
+            memory = psutil.virtual_memory()
+            
+            print(f"[ESCOBAR-WS] === SYSTEM STATE {context} ===")
+            print(f"[ESCOBAR-WS] CPU Usage: {cpu_percent}%")
+            print(f"[ESCOBAR-WS] Memory Usage: {memory.percent}% ({memory.used / 1024 / 1024:.1f}MB used)")
+            print(f"[ESCOBAR-WS] Active Connections: {len(self._active_connections)}")
+            print(f"[ESCOBAR-WS] Total Connections Created: {self._connection_count}")
+            print(f"[ESCOBAR-WS] Current Thread: {threading.current_thread().name}")
+            print(f"[ESCOBAR-WS] === END SYSTEM STATE ===")
+        except Exception as e:
+            print(f"[ESCOBAR-WS] Error logging system state: {e}")
+    
+    def _log_connection_lifecycle(self, event, details=None):
+        """Log connection lifecycle events with timing"""
+        timestamp = time.time()
+        connection_id = getattr(self, 'connection_id', 'UNKNOWN')
+        
+        print(f"[ESCOBAR-WS] === CONNECTION LIFECYCLE EVENT ===")
+        print(f"[ESCOBAR-WS] Event: {event}")
+        print(f"[ESCOBAR-WS] Connection ID: {connection_id}")
+        print(f"[ESCOBAR-WS] Timestamp: {timestamp}")
+        print(f"[ESCOBAR-WS] Human Time: {time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(timestamp))}")
+        if details:
+            print(f"[ESCOBAR-WS] Details: {details}")
+        print(f"[ESCOBAR-WS] === END LIFECYCLE EVENT ===")
+    
+    def _log_message_timing(self, direction, message_size, start_time=None):
+        """Log message timing and throughput"""
+        current_time = time.time()
+        
+        print(f"[ESCOBAR-WS] === MESSAGE TIMING ===")
+        print(f"[ESCOBAR-WS] Direction: {direction}")
+        print(f"[ESCOBAR-WS] Message Size: {message_size} bytes")
+        print(f"[ESCOBAR-WS] Timestamp: {current_time}")
+        
+        if start_time:
+            duration = current_time - start_time
+            throughput = message_size / duration if duration > 0 else 0
+            print(f"[ESCOBAR-WS] Processing Duration: {duration:.4f}s")
+            print(f"[ESCOBAR-WS] Throughput: {throughput:.2f} bytes/sec")
+        
+        print(f"[ESCOBAR-WS] === END MESSAGE TIMING ===")
+    
+    def _log_target_connection_health(self):
+        """Log target connection health metrics"""
+        if not self.target_ws:
+            print(f"[ESCOBAR-WS] === TARGET HEALTH: NO CONNECTION ===")
+            return
+        
+        print(f"[ESCOBAR-WS] === TARGET CONNECTION HEALTH ===")
+        try:
+            print(f"[ESCOBAR-WS] Target State: {self.target_ws.state}")
+            print(f"[ESCOBAR-WS] Target State Name: {self.target_ws.state.name}")
+            print(f"[ESCOBAR-WS] Target State Value: {self.target_ws.state.value}")
+            
+            # Check if we can access additional properties
+            if hasattr(self.target_ws, 'ping_interval'):
+                print(f"[ESCOBAR-WS] Ping Interval: {self.target_ws.ping_interval}")
+            if hasattr(self.target_ws, 'ping_timeout'):
+                print(f"[ESCOBAR-WS] Ping Timeout: {self.target_ws.ping_timeout}")
+            if hasattr(self.target_ws, 'close_timeout'):
+                print(f"[ESCOBAR-WS] Close Timeout: {self.target_ws.close_timeout}")
+                
+        except Exception as e:
+            print(f"[ESCOBAR-WS] Error checking target health: {e}")
+        
+        print(f"[ESCOBAR-WS] === END TARGET HEALTH ===")
+    
+    def _generate_connection_id(self):
+        """Generate unique connection ID for tracking"""
+        WebSocketProxyHandler._connection_count += 1
+        return f"CONN-{WebSocketProxyHandler._connection_count}-{int(time.time())}"
+    
     def _resolve_target_url_for_docker(self, url):
         """
         Resolve target URL for Docker environment.
@@ -214,6 +299,25 @@ class WebSocketProxyHandler(tornado.websocket.WebSocketHandler):
         super().__init__(*args, **kwargs)
         self.target_ws = None
         
+        # Generate unique connection ID for tracking
+        self.connection_id = self._generate_connection_id()
+        self.connection_start_time = time.time()
+        
+        # Add to active connections tracking
+        WebSocketProxyHandler._active_connections[self.connection_id] = {
+            'start_time': self.connection_start_time,
+            'handler': self
+        }
+        
+        # Log connection creation
+        self._log_connection_lifecycle("HANDLER_CREATED", {
+            'connection_id': self.connection_id,
+            'total_connections': len(WebSocketProxyHandler._active_connections)
+        })
+        
+        # Log system state at connection creation
+        self._log_system_state("CONNECTION_INIT")
+        
         # Debug environment information
         print(f"[ESCOBAR-WS] Environment WEBSOCKET_PROXY_TARGET: {os.getenv('WEBSOCKET_PROXY_TARGET', 'NOT_SET')}")
         print(f"[ESCOBAR-WS] Running in container: {os.path.exists('/.dockerenv')}")
@@ -232,6 +336,9 @@ class WebSocketProxyHandler(tornado.websocket.WebSocketHandler):
         print(f"[ESCOBAR-WS] All WEBSOCKET environment vars: {websocket_env_vars}")
         
         self.is_closing = False
+        self.message_count_sent = 0
+        self.message_count_received = 0
+        self.last_activity_time = time.time()
     
     def _get_user_bonnie_url(self):
         """

escobar/labextension/package.json

@@ -1,6 +1,6 @@
 {
   "name": "escobar",
-  "version": "0.1.89",
+  "version": "0.1.90",
   "description": "AI CHAT EXTENSION",
   "keywords": [
     "jupyter",
@@ -123,7 +123,7 @@
     "outputDir": "escobar/labextension",
     "schemaDir": "schema",
     "_build": {
-      "load": "static/remoteEntry.a8ae0b901c4a30b269e4.js",
+      "load": "static/remoteEntry.c9a1accf3c70050fa880.js",
       "extension": "./extension",
       "style": "./style"
     }

escobar/labextension/schemas/escobar/package.json.orig

@@ -1,6 +1,6 @@
 {
   "name": "escobar",
-  "version": "0.1.89",
+  "version": "0.1.90",
   "description": "AI CHAT EXTENSION",
   "keywords": [
     "jupyter",