klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-09 05:39:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Refresh the conversation from the db in the websocket flow

Browse Source
This commit is contained in:
sabaimran
2024-05-31 16:15:56 +05:30
parent fd07abbfc8
commit 2667ef4544
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/khoj/routers/api_chat.py
View File

@@ -487,6 +487,10 @@ async def websocket_endpoint(
if conversation: if conversation:
await sync_to_async(conversation.refresh_from_db)(fields=["conversation_log"]) await sync_to_async(conversation.refresh_from_db)(fields=["conversation_log"])
q = await websocket.receive_text() q = await websocket.receive_text()
# Refresh these because the connection to the database might have been closed
await conversation.arefresh_from_db()
except WebSocketDisconnect: except WebSocketDisconnect:
logger.debug(f"User {user} disconnected web socket") logger.debug(f"User {user} disconnected web socket")
break break