diff options
Diffstat (limited to 'continuedev/src')
| -rw-r--r-- | continuedev/src/continuedev/server/ide.py | 9 |
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) |