Update kadi_apy_bot.py

kadi_apy_bot.py

@@ -10,7 +10,7 @@ class KadiAPYBot:
 
 
 
-    def process_query(self, query):
+    def process_query(self, query, chat_history):
         """
         Process a user query, handle history, retrieve contexts, and generate a response.
         """
@@ -132,30 +132,7 @@ class KadiAPYBot:
         context = self.vector_store.similarity_search(query = query, k=k, filter=filter)       
         return context
 
-    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")
-            all_metadata = doc.metadata
-            
-            metadata_str = ", ".join(f"{key}: {value}" for key, value in all_metadata.items())
-            print("\n")
-            print("------------------------------Beneath is retrieved doc------------------------------------------------")
-            print(metadata_str)
-            formatted_docs.append(metadata_str)
-            print("\n")
-            formatted_docs.append("\n")
-            formatted_docs.append(doc.page_content)
-            print(doc.page_content)
-            print("\n\n")
-            print("------------------------------End  of retrived doc------------------------------------------------")
-            formatted_docs.append("\n\n")
-            
-        return formatted_docs
-
-    def generate_response(self, query, doc_context, code_context):
+    def generate_response(self, query, history, doc_context, code_context):
         """
         Generate a response using the retrieved contexts and the LLM.
         """
@@ -185,4 +162,39 @@ class KadiAPYBot:
                     Query:
                     {query}
                 """
-        return self.llm.invoke(prompt).content
+        return self.llm.invoke(prompt).content
+
+        
+    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")
+            all_metadata = doc.metadata
+            
+            metadata_str = ", ".join(f"{key}: {value}" for key, value in all_metadata.items())
+            print("\n")
+            print("------------------------------Beneath is retrieved doc------------------------------------------------")
+            print(metadata_str)
+            formatted_docs.append(metadata_str)
+            print("\n")
+            formatted_docs.append("\n")
+            formatted_docs.append(doc.page_content)
+            print(doc.page_content)
+            print("\n\n")
+            print("------------------------------End  of retrived doc------------------------------------------------")
+            formatted_docs.append("\n\n")
+            
+        return formatted_docs
+
+    def format_history(conversation_history):
+    formatted_history = []
+    for i, entry in enumerate(conversation_history, start=1):
+        user_query = entry.get("query", "No query provided")
+        assistant_response = entry.get("response", "No response yet")  # Updated label
+        formatted_history.append(f"Turn {i}:")
+        formatted_history.append(f"User Query: {user_query}")
+        formatted_history.append(f"Assistant Response: {assistant_response}")  # Using "Assistant Response"
+        formatted_history.append("\n")
+    return "\n".join(formatted_history)