fix: :bug: catch error when workspace uri isn't defined

1 files changed, 5 insertions, 4 deletions

diff --git a/continuedev/src/continuedev/server/ide.py b/continuedev/src/continuedev/server/ide.py
index cf8b32a1..87374928 100644
--- a/continuedev/src/continuedev/server/ide.py
+++ b/continuedev/src/continuedev/server/ide.py
@@ -154,6 +154,7 @@ class IdeProtocolServer(AbstractIdeProtocolServer):
 
     async def _send_json(self, message_type: str, data: Any):
         if self.websocket.application_state == WebSocketState.DISCONNECTED:
+            print("Tried to send message, but websocket is disconnected", message_type)
             return
         print("Sending IDE message: ", message_type)
         await self.websocket.send_json({
@@ -435,6 +436,10 @@ class IdeProtocolServer(AbstractIdeProtocolServer):
 @router.websocket("/ws")
 async def websocket_endpoint(websocket: WebSocket, session_id: str = None):
     try:
+        await websocket.accept()
+        print("Accepted websocket connection from, ", websocket.client)
+        await websocket.send_json({"messageType": "connected", "data": {}})
+
         # Start meilisearch
         try:
             await start_meilisearch()
@@ -442,10 +447,6 @@ async def websocket_endpoint(websocket: WebSocket, session_id: str = None):
             print("Failed to start MeiliSearch")
             print(e)
 
-        await websocket.accept()
-        print("Accepted websocket connection from, ", websocket.client)
-        await websocket.send_json({"messageType": "connected", "data": {}})
-
         def handle_msg(msg):
             message = json.loads(msg)