fix knowledge history

NLP_model/chatbot.py

@@ -18,7 +18,7 @@ import functools
 import hashlib
 import logging
 import random
-from mongodb import get_chat_history
+from mongodb import get_chat_history, get_request_history
 
 # Configure logging
 logging.basicConfig(
@@ -110,6 +110,7 @@ You have to use core knowledge and conversation history to chat with users, who
 
 Return Format:
 Respond in friendly, natural, concise and use only English like a real tour guide. 
+Return in format that use for telegram
 
 Warning:
 Let's support users like a real tour guide, not a bot. The information in core knowledge is your own knowledge.
@@ -135,6 +136,10 @@ def get_history(user_id):
     """Get conversation history for a specific user from MongoDB"""
     return get_chat_history(user_id)
 
+def get_request(user_id):
+    """Get conversation history for a specific user from MongoDB"""
+    return get_request_history(user_id)
+
 
 def get_chain():
     """Get the retrieval chain with Pinecone vector store (singleton pattern)"""
@@ -151,7 +156,7 @@ def get_chain():
             text_key="text"
         )
         
-        _retriever_instance = vectorstore.as_retriever(search_kwargs={"k": 5})
+        _retriever_instance = vectorstore.as_retriever(search_kwargs={"k": 7})
         logger.info(f"Pinecone retriever initialized in {time.time() - start_time:.2f} seconds")
         return _retriever_instance
     except Exception as e:
@@ -218,7 +223,7 @@ def chat(request, user_id="default_user"):
         current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
         
         # Sử dụng invoke thay vì get_relevant_documents
-        retrieved_docs = retriever.invoke(request)
+        retrieved_docs = retriever.invoke(get_request_history(user_id))
         context = "\n".join([doc.page_content for doc in retrieved_docs])
         
         # Sử dụng generate_content thay vì invoke cho model Gemini

mongodb.py

@@ -22,20 +22,6 @@ collection = db[MONGODB_COLLECTION]
 
 
 def get_chat_history(user_id: int) -> str:
-    """
-    Lấy lịch sử chat cho user_id cho trước từ MongoDB và ghép thành chuỗi theo định dạng:
-    
-    Bot: ...
-    User: ...
-    Bot: ...
-    ...
-    
-    Giả sử:
-      - Các document chứa trường "user_id" để lọc theo user_id.
-      - Trường "factor" xác định nguồn tin (nếu factor == "user" thì là tin của User,
-        còn lại coi là tin của Bot/RAG).
-      - Trường "timestamp" dùng để sắp xếp theo thời gian (nếu có).
-    """
     try:
         # Truy vấn tất cả các document có user_id, sắp xếp theo timestamp tăng dần
         # Nếu không có trường timestamp, có thể sort theo _id
@@ -64,6 +50,36 @@ def get_chat_history(user_id: int) -> str:
         logger.error(f"Lỗi khi lấy lịch sử chat cho user_id {user_id}: {e}")
         return ""
 
+def get_request_history(user_id: int) -> str:
+    try:
+        # Truy vấn tất cả các document có user_id, sắp xếp theo timestamp tăng dần
+        # Nếu không có trường timestamp, có thể sort theo _id
+        docs = list(collection.find({"user_id": user_id}).sort("timestamp", -1).limit(15))
+        docs.reverse()
+
+        if not docs:
+            logger.info(f"Không tìm thấy dữ liệu cho user_id: {user_id}")
+            return ""
+        
+        conversation_lines = []
+        for doc in docs:
+            factor = doc.get("factor", "").lower()
+            action = doc.get("action", "").lower()
+            message = doc.get("message", "")
+            
+            if action == "freely asking":
+                # Only keep the most recent 3 messages
+                if len(conversation_lines) >= 3:
+                    conversation_lines.pop(0)  # Remove oldest message
+                conversation_lines.append(message)
+        
+        # Ghép các dòng thành chuỗi, mỗi dòng cách nhau bằng xuống dòng
+        logger.info("User ID: " + str(user_id))
+        return "\n".join(conversation_lines)
+    except Exception as e:
+        logger.error(f"Lỗi khi lấy lịch sử request cho user_id {user_id}: {e}")
+        return ""
+
 # if __name__ == '__main__':
 #     user_id = int(input("Nhập user_id cần lấy lịch sử chat: ").strip())
 #     history = get_chat_history(user_id)