From fc9eb3051fd5a7c9cad57b5d6cd93374bd8210fb Mon Sep 17 00:00:00 2001 From: Nate Sesti Date: Thu, 27 Jul 2023 19:16:17 -0700 Subject: fix: :bug: catch error when workspace uri isn't defined --- continuedev/src/continuedev/server/ide.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) (limited to 'continuedev/src') 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) -- cgit v1.2.3-70-g09d2