Refresh the conversation from the db in the websocket flow

yxiaolu · May 31, 2024 · 2667ef4 · 2667ef4
1 parent fd07abb
commit 2667ef4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/khoj/routers/api_chat.py b/src/khoj/routers/api_chat.py
@@ -487,6 +487,10 @@ async def send_rate_limit_message(message: str):
             if conversation:
                 await sync_to_async(conversation.refresh_from_db)(fields=["conversation_log"])
             q = await websocket.receive_text()
+
+            # Refresh these because the connection to the database might have been closed
+            await conversation.arefresh_from_db()
+
         except WebSocketDisconnect:
             logger.debug(f"User {user} disconnected web socket")
             break