Change background task handling method

Author: deeptiongit

chatbot-core/api/routes/chatbot.py

@@ -47,15 +47,14 @@
 from api.services.memory import (
     delete_session,
     session_exists,
-    get_session,
+    persist_session,
     init_session,
 )
 from api.services.file_service import (
     process_uploaded_file,
     get_supported_extensions,
     FileProcessingError,
 )
-from api.services.sessionmanager import append_message
 
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.DEBUG)
@@ -212,9 +211,9 @@ def chatbot_reply(session_id: str, request: ChatRequest, _background_tasks: Back
         )
     reply =  get_chatbot_reply(session_id, request.message)
     _background_tasks.add_task(
-        append_message,
+        persist_session,
         session_id,
-        get_session(session_id).chat_memory.messages if get_session(session_id) else None,)
+        )
 
     return reply
 

chatbot-core/api/services/memory.py

@@ -8,7 +8,12 @@
 from threading import Lock
 from langchain.memory import ConversationBufferMemory
 from api.config.loader import CONFIG
-from api.services.sessionmanager import delete_session_file, load_session, session_exists_in_json
+from api.services.sessionmanager import(
+    delete_session_file,
+    load_session,
+    session_exists_in_json,
+    append_message
+)
 # sessionId --> {"memory": ConversationBufferMemory, "last_accessed": datetime}
 
 
@@ -73,6 +78,17 @@ def get_session(session_id: str) -> ConversationBufferMemory | None:
         return memory
 
 
+def persist_session(session_id: str)-> None:
+    """
+    Persist the current session messages to disk.
+
+    Args:
+        session_id (str): The session identifier.
+    """
+    session_data = get_session(session_id)
+    if session_data:
+        messages = list(session_data.chat_memory.messages)
+        append_message(session_id, messages)
 
 
 

chatbot-core/api/services/sessionmanager.py

@@ -3,7 +3,6 @@
 import json
 import uuid
 from threading import Lock
-from langchain.memory import ConversationBufferMemory
 
 
 
@@ -39,7 +38,7 @@ def _load_session_from_json(session_id: str) -> list:
         return json.load(f)
 
 
-def _append_message_to_json(session_id: str, messages:ConversationBufferMemory) -> None:
+def _append_message_to_json(session_id: str, messages:list) -> None:
     """
     Persist the current session messages as a full snapshot using atomic write.
     """