escobar 0.1.84__py3-none-any.whl → 0.1.86__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.84'
+__version__ = VERSION = '0.1.86'

escobar/handlers.py

@@ -343,7 +343,8 @@ class WebSocketProxyHandler(tornado.websocket.WebSocketHandler):
                 additional_headers=headers,
                 ssl=ssl_context,
                 ping_interval=20,
-                ping_timeout=10
+                ping_timeout=10,
+                max_size=100 * 1024 * 1024 
             )
             
             end_time = time.time()
@@ -424,16 +425,36 @@ class WebSocketProxyHandler(tornado.websocket.WebSocketHandler):
                     break
                 print(f"[ESCOBAR-WS] === END TARGET MESSAGE #{message_count} ===")
                 
-        except websockets.exceptions.ConnectionClosed:
-            print(f"[ESCOBAR-WS] Target websocket connection closed normally")
-            logging.info("Target websocket connection closed")
+        except websockets.exceptions.ConnectionClosed as e:
+            print(f"[ESCOBAR-WS] === TARGET CONNECTION CLOSED DETAILS ===")
+            print(f"[ESCOBAR-WS] Target websocket connection closed")
+            print(f"[ESCOBAR-WS] Close code: {getattr(e, 'code', 'NO_CODE')}")
+            print(f"[ESCOBAR-WS] Close reason: {getattr(e, 'reason', 'NO_REASON')}")
+            print(f"[ESCOBAR-WS] Exception type: {type(e).__name__}")
+            print(f"[ESCOBAR-WS] Exception str: {str(e)}")
+            print(f"[ESCOBAR-WS] Target WS final state: {getattr(self.target_ws, 'state', 'NO_STATE') if self.target_ws else 'NO_TARGET_WS'}")
+            print(f"[ESCOBAR-WS] Messages processed before disconnect: {message_count}")
+            print(f"[ESCOBAR-WS] === END TARGET CONNECTION CLOSED DETAILS ===")
+            
+            logging.info(f"Target websocket connection closed - code: {getattr(e, 'code', 'NO_CODE')}, reason: {getattr(e, 'reason', 'NO_REASON')}")
+            
             if not self.is_closing:
                 print(f"[ESCOBAR-WS] Closing client connection due to target disconnect")
                 self.close(code=1011, reason="Target server disconnected")
         except Exception as e:
-            print(f"[ESCOBAR-WS] ERROR in target message forwarding:")
-            print(f"[ESCOBAR-WS]   Error: {str(e)}")
-            print(f"[ESCOBAR-WS]   Error type: {type(e).__name__}")
+            print(f"[ESCOBAR-WS] === TARGET MESSAGE FORWARDING ERROR ===")
+            print(f"[ESCOBAR-WS] Error: {str(e)}")
+            print(f"[ESCOBAR-WS] Error type: {type(e).__name__}")
+            print(f"[ESCOBAR-WS] Messages processed before error: {message_count}")
+            print(f"[ESCOBAR-WS] Target WS state: {getattr(self.target_ws, 'state', 'NO_STATE') if self.target_ws else 'NO_TARGET_WS'}")
+            if hasattr(e, '__dict__'):
+                print(f"[ESCOBAR-WS] Error attributes: {e.__dict__}")
+            if hasattr(e, 'errno'):
+                print(f"[ESCOBAR-WS] Errno: {e.errno}")
+            if hasattr(e, 'strerror'):
+                print(f"[ESCOBAR-WS] Strerror: {e.strerror}")
+            print(f"[ESCOBAR-WS] === END TARGET MESSAGE FORWARDING ERROR ===")
+            
             logging.error(f"Error receiving from target websocket: {str(e)}")
             if not self.is_closing:
                 print(f"[ESCOBAR-WS] Closing client connection due to target error")
@@ -463,17 +484,36 @@ class WebSocketProxyHandler(tornado.websocket.WebSocketHandler):
         """Safely close the target websocket connection"""
         print(f"[ESCOBAR-WS] === CLOSING TARGET CONNECTION ===")
         try:
-            if self.target_ws and not self.target_ws.closed:
-                print(f"[ESCOBAR-WS] Target WS is open, closing it")
-                await self.target_ws.close()
-                print(f"[ESCOBAR-WS] Target connection closed successfully")
-                logging.info("Target websocket connection closed")
+            if self.target_ws:
+                print(f"[ESCOBAR-WS] Target WS exists, checking state")
+                print(f"[ESCOBAR-WS] Target WS type: {type(self.target_ws)}")
+                print(f"[ESCOBAR-WS] Target WS state: {getattr(self.target_ws, 'state', 'NO_STATE_ATTR')}")
+                
+                # Check if connection is already closed using the correct websockets library API
+                try:
+                    is_closed = self.target_ws.state.name == 'CLOSED'
+                    print(f"[ESCOBAR-WS] Target WS is closed: {is_closed}")
+                except AttributeError:
+                    # Fallback: just try to close it regardless of state
+                    print(f"[ESCOBAR-WS] Cannot check state, will attempt to close anyway")
+                    is_closed = False
+                
+                if not is_closed:
+                    print(f"[ESCOBAR-WS] Target WS is open, closing it")
+                    await self.target_ws.close()
+                    print(f"[ESCOBAR-WS] Target connection closed successfully")
+                    logging.info("Target websocket connection closed")
+                else:
+                    print(f"[ESCOBAR-WS] Target WS already closed")
             else:
-                print(f"[ESCOBAR-WS] Target WS already closed or None")
+                print(f"[ESCOBAR-WS] No target WS to close")
         except Exception as e:
             print(f"[ESCOBAR-WS] ERROR closing target connection:")
             print(f"[ESCOBAR-WS]   Error: {str(e)}")
             print(f"[ESCOBAR-WS]   Error type: {type(e).__name__}")
+            print(f"[ESCOBAR-WS]   Target WS type: {type(self.target_ws) if self.target_ws else 'None'}")
+            if hasattr(e, '__dict__'):
+                print(f"[ESCOBAR-WS]   Error attributes: {e.__dict__}")
             logging.error(f"Error closing target websocket: {str(e)}")
         
         print(f"[ESCOBAR-WS] === END CLOSING TARGET CONNECTION ===")

escobar/labextension/package.json

@@ -1,6 +1,6 @@
 {
   "name": "escobar",
-  "version": "0.1.84",
+  "version": "0.1.86",
   "description": "AI CHAT EXTENSION",
   "keywords": [
     "jupyter",
@@ -123,7 +123,7 @@
     "outputDir": "escobar/labextension",
     "schemaDir": "schema",
     "_build": {
-      "load": "static/remoteEntry.31ea89463b75c6ad65cb.js",
+      "load": "static/remoteEntry.47499e159c447c9de143.js",
       "extension": "./extension",
       "style": "./style"
     }

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

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