Update kadi_apy_bot.py

kadi_apy_bot.py

@@ -25,7 +25,7 @@ class KadiAPYBot:
         
         # Retrieve contexts
         doc_contexts = self.retrieve_contexts(query, k=5, filter={"usage": "doc"})
-        code_contexts = self.retrieve_contexts(rewritten_query, k=3, filter={"usage": code_library_usage_prediction})
+        code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"usage": code_library_usage_prediction})
         
         # Format contexts
         formatted_doc_contexts = self.format_documents(doc_contexts)
@@ -40,6 +40,7 @@ class KadiAPYBot:
         return response
 
 
+    #not supported yet, need session handling in app.py
     def add_to_conversation(self, user_query=None, llm_response=None):
         """
         Add either the user's query, the LLM's response, or both to the conversation history.
@@ -124,6 +125,7 @@ class KadiAPYBot:
 
     def format_documents(self, documents):
         formatted_docs = []
+        print("################################# start of doc #######################################")
         for i, doc in enumerate(documents, start=1):
             formatted_docs.append(f"Snippet {i}: \n")            
             formatted_docs.append("\n")
@@ -140,7 +142,7 @@ class KadiAPYBot:
             print(doc.page_content)
             print("\n\n")
             print("------------------------------End  of retrived doc------------------------------------------------")
-            formatted_docs.append("\n\n\n")
+            formatted_docs.append("\n\n")
             
         return formatted_docs