From 9e068fad4fa94ad10830973d1ebedc61b14f8c0e Mon Sep 17 00:00:00 2001 From: Debanjum Singh Solanky Date: Tue, 30 Apr 2024 14:17:52 +0530 Subject: [PATCH] Handle null ref, when refresh conversation from db in websocket chat --- src/khoj/routers/api_chat.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/khoj/routers/api_chat.py b/src/khoj/routers/api_chat.py index c854d9de..2a1bbc5e 100644 --- a/src/khoj/routers/api_chat.py +++ b/src/khoj/routers/api_chat.py @@ -338,7 +338,8 @@ async def websocket_endpoint( await websocket.accept() while connection_alive: try: - await sync_to_async(conversation.refresh_from_db)(fields=["conversation_log"]) + if conversation: + await sync_to_async(conversation.refresh_from_db)(fields=["conversation_log"]) q = await websocket.receive_text() except WebSocketDisconnect: logger.debug(f"User {user} disconnected web socket")