Update kadi_apy_bot.py

kadi_apy_bot.py

@@ -18,7 +18,7 @@ class KadiAPYBot:
         #self.add_to_conversation(user_query=query)
         
         # Rewrite query
-        rewritten_query = self.rewrite_query(query)
+    #    rewritten_query = self.rewrite_query(query)
 
         # Predict library usage
     #    code_library_usage_prediction = self.predict_library_usage(query)
@@ -27,8 +27,8 @@ class KadiAPYBot:
     #    doc_contexts = self.retrieve_contexts(query, k=3, filter={"usage": "doc"})
     #    code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"usage": code_library_usage_prediction})
 
-        doc_contexts = self.retrieve_contexts(query, k=3, filter={"usage": "doc/"})
-        code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"usage": code_library_usage_prediction})
+        doc_contexts = self.retrieve_contexts(query, k=3, filter={"folder": "doc/"})
+        code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"folder": "kadi_apy/"})
         
         # Format contexts
     #   formatted_doc_contexts = self.format_documents(doc_contexts)
@@ -153,23 +153,14 @@ class KadiAPYBot:
         """
         Generate a response using the retrieved contexts and the LLM.
         """
-        prompt = f"""You are an expert python developer. You are assisting in generating code for users who want to programmatically 
-                    make use of a python library named 'kadiAPY' to interact with the API of a software. It provides an object-oriented 
-                    approach for interfacing with the API.
-
-                    You are given "Documentation Snippets" and "Code Snippets"
-                        "Documentation Snippets:" Contains a collection of potentially useful snippets, including code examples and documentation excerpts of "kadiAPY"
-                        "Code Snippets:" Contains potentially useful snippets from the source code of "kadiAPY"
-                        
-                    Based on the retrieved snippets (Documentation Snippets and Code Snippets) and the guidelines answer the "query".  
+        prompt = f"""You are an expert python developer. You are assisting in generating code for users.
                         
-                        General Guidelines:
-                        - If no related information is found from the snippets to answer the query, reply that you do not know.
+                    Based on the retrieved document and code snippets and the guidelines answer the "query".  
                         
                         Guidelines when generating code:
                         - Display the complete code first, followed by a concise explanation in no more than 5 sentences.
                         
-                Documentation Snippets:
+                Documentation context:
                 {doc_context}
                 Code Snippets:
                 {code_context}