Improve logging

bonk1t · Dec 19, 2023 · 3cc7fd6 · 3cc7fd6
1 parent c274dc8
commit 3cc7fd6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/nalgonda/routers/v1/api/agency.py b/nalgonda/routers/v1/api/agency.py
@@ -42,6 +42,7 @@ async def post_agency_message(request: AgencyMessagePostRequest) -> dict:
 
         new_thread_id = await agency_manager.refresh_thread_id(agency, agency_id, thread_id)
         if new_thread_id is not None:
+            logger.info(f"Thread ID changed from {thread_id} to {new_thread_id}")
             return {"response": response, "thread_id": new_thread_id}
 
         return {"response": response}

diff --git a/nalgonda/routers/v1/websocket.py b/nalgonda/routers/v1/websocket.py
@@ -73,6 +73,7 @@ async def websocket_receive_and_process_messages(
 
             new_thread_id = await agency_manager.refresh_thread_id(agency, agency_id, thread_id)
             if new_thread_id is not None:
+                logger.info(f"Thread ID changed from {thread_id} to {new_thread_id}")
                 await connection_manager.send_message(json.dumps({"thread_id": new_thread_id}), websocket)
                 thread_id = new_thread_id