Update app.py

app.py

@@ -306,37 +306,6 @@ def rag_workflow(query):
     print("METADATA PREDICTION -------------------------:", library_usage_prediction)
     print(library_usage_prediction)
 
-
-    rewrite_prompt = (
-        f"""You are an intelligent assistant that helps users rewrite their queries.
-            The vectorstore consists of the source code and documentation of a Python library. 
-            The library's functions and methods have descriptive docstrings.           
-            Your task is to rewrite the query in a way that aligns with the language and structure of the 
-            library's methods and documentation, ensuring optimal retrieval of relevant information.
-
-            Guidelines for rewriting the query:
-                1. Identify the main action the user wants to perform (e.g., "Upload a file to a record," "Get users of a group").
-                2. Remove conversational elements like greetings or pleasantries (e.g., "Hello Chatbot", "I need you to help me with").
-                3. Exclude specific variable values (e.g., "ID of my record is '31'") unless essential to the intent.
-                4. Rephrase the query to match the format and keywords used in the docstrings, focusing on verbs and objects relevant to the action (e.g., "Add a record to a collection").
-                5. Given the query the user might need more than one action to achieve his goal. In this case the rewritten query has more than one action. 
-
-                Examples:
-                    - User query: "Create a Python script with a method that facilitates the creation of records. This method should accept an array of identifiers as a parameter and allow metadata to be added to each record."
-                    - Rewritten query: "create records, add metadata to record"
-                    - User query: "Hi, can you help me write Python code to add a record to a collection? The record ID is '45', and the collection ID is '12'."
-                      Rewritten query: "add a record to a collection"
-                    - User query: I need a python script with which i create a new record with the title: "Hello World"  and then link the record to a given collection.
-                      Rewritten query: "create a new record with title" , "link a record to a collection"
-
-            Based on these examples and guidelines, rewrite the following user query to align more effectively with the keywords used in the docstrings. 
-            Do not include any addition comments, explanations, or text.
-            
-            Original query:
-            {query}
-        """
-    )
-
     rewritten_query_response = llm.invoke(rewrite_prompt)
     rewritten_query = rewritten_query_response.content.strip()
     
@@ -349,6 +318,8 @@ def rag_workflow(query):
     doc_context = format_kadi_api_doc_context(kadi_apy_docs)
     code_context = format_kadi_apy_library_context(kadi_apy_sourcecode)
 
+
+
     
     print("H")
     print("::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::")
@@ -356,13 +327,17 @@ def rag_workflow(query):
         print(doc.metadata.get("source", "Unknown Type"))
         print("\n")
     print("::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::")
+    print("CODE_CONTEST\n:", code_context)
 
+    
     print("::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::")
     for doc in kadi_apy_docs:
         print(doc.metadata.get("source", "Unknown Type"))
         print("\n")
     print("::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::")
 
+    print(doc_context)
+    print("DOC_CONTEXT\n:", code_context)
 
     
     prompt = f"""You are an expert python developer. You are assisting in generating code for users who wants to make use of "kadi-apy", an API library. 
@@ -493,8 +468,6 @@ def main():
 
             user_txt.submit(check_input_text, user_txt, None).success(add_text, [chatbot, user_txt], [chatbot, user_txt]).then(bot_kadi, [chatbot], [chatbot])
             submit_btn.click(check_input_text, user_txt, None).success(add_text, [chatbot, user_txt], [chatbot, user_txt]).then(bot_kadi, [chatbot], [chatbot])
-            #user_txt.submit(check_input_text, user_txt, None).success(add_text, [chatbot, user_txt], [chatbot, user_txt]).then(bot_kadi, [chatbot], [chatbot, doc_citation])
-            #submit_btn.click(check_input_text, user_txt, None).success(add_text, [chatbot, user_txt], [chatbot, user_txt]).then(bot_kadi, [chatbot], [chatbot, doc_citation])
             clear_btn.click(lambda: None, None, chatbot, queue=False)
 
     demo.launch()