Update app.py

app.py

@@ -643,7 +643,7 @@ class PersonalizedKazakhAssistant:
             self.user_memories[user_id] = ConversationBufferMemory(
                 memory_key='chat_history',
                 return_messages=True,
-                max_token_limit=10000 
+                max_token_limit=5000 
             )
         return self.user_memories[user_id]
     
@@ -697,7 +697,7 @@ class PersonalizedKazakhAssistant:
         elif message.lower().startswith('/help'):
             return self.get_help_message()
 
-        retrieved_docs = self.vectorstore.similarity_search(message, k=15)
+        retrieved_docs = self.vectorstore.similarity_search(message, k=8)
         context = "\n".join([doc.page_content for doc in retrieved_docs])
         
         memory = self.get_user_memory(user_id)
@@ -886,7 +886,7 @@ class PersonalizedKazakhAssistant:
         for term in sorted(self.known_terms):  
             normalized_term = self.normalize_term(term)
             if normalized_term not in shown_words and len(term.split()) == 1:
-                retrieved_docs = self.vectorstore.similarity_search(term, k=15)
+                retrieved_docs = self.vectorstore.similarity_search(term, k=1)
                 for doc in retrieved_docs:
                     lines = doc.page_content.replace('\r\n', '\n').replace('\r', '\n').split('\n')
                     for line in lines:
@@ -916,7 +916,7 @@ class PersonalizedKazakhAssistant:
         for term in sorted(self.known_terms):  
             normalized_term = self.normalize_term(term)
             if normalized_term not in shown_idioms and len(term.split()) > 1:
-                retrieved_docs = self.vectorstore.similarity_search(term, k=15)
+                retrieved_docs = self.vectorstore.similarity_search(term, k=1)
                 for doc in retrieved_docs:
                     lines = doc.page_content.replace('\r\n', '\n').replace('\r', '\n').split('\n')
                     for line in lines: