Adding a new container to an existing pod doesnt break everything!

websocket_server/events/logs.py

@@ -173,8 +173,8 @@ def register_socketio_handlers(socketio):
 
         context = json.loads(context_data)
         user_sid = context.get("user_sid")
-        with base.connected_sids_lock:
-            if user_sid in base.connected_sids:
+        with connected_sids_lock:
+            if user_sid in connected_sids:
                 socketio.emit("user_log_stream_update", {
                     "logs": logs,
                     "request_id": request_id

websocket_server/events/terminal.py

@@ -8,7 +8,7 @@ from flask import request
 
 from websocket_server.config import get_redis_client
 from websocket_server.events import base
-from websocket_server.shared_state import connected_workers, connected_sids_lock, connected_sids
+from websocket_server.shared_state import connected_workers
 
 logger = logging.getLogger("websocket_server")
 socketio = base.socketio

websocket_server/events/vnc.py

@@ -6,7 +6,7 @@ import requests
 from flask import request
 
 from websocket_server.events import base
-from websocket_server.shared_state import connected_workers, connected_sids_lock, connected_sids
+from websocket_server.shared_state import connected_workers
 
 
 logger = logging.getLogger("websocket_server")

websocket_server/worker_manager.py

@@ -36,7 +36,6 @@ def notify_worker_online(worker_id):
 
     try:
         response = requests.put(f"{api_server_url}/workload_hosts/{worker_id}", json=payload, headers=headers)
-        logger.debug(response.text)
         logger.info(f"[{worker_id}] API server acknowledged online state")
     except Exception as e:
         logger.error(f"[{worker_id}] Failed to notify API server of online status: {e}")