Update app.py

app.py

@@ -325,12 +325,6 @@ def format_kadi_api_doc_context(docs):
 
                    
 def rag_workflow(query):
-    
-
-
-    
-    rewritten_query_response = llm.invoke("Hello")
-    rewritten_query = rewritten_query_response.content.strip()
 
 
     prompt = (
@@ -348,6 +342,40 @@ def rag_workflow(query):
     )
     metadata_prediction = llm.predict(prompt)
 
+
+
+    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, which enables users to 
+            programmatically interact with a REST-like API of a software system. The library 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 method to achieve his goal. 
+
+                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"
+
+            Based on these examples and guidelines, rewrite the following user query to align more effectively with the keywords used in the docstrings.
+            Original query:
+            {query}
+        """
+    )
+
+    rewritten_query_response = llm.invoke("Hello")
+    rewritten_query = rewritten_query_response.content.strip()
+    print("AAAAArewritten_query -------------------------:", metadata_prediction)
+    print(rewritten_query)
+
+
     print("METADATA PREDICTION -------------------------:", metadata_prediction)
     print(metadata_prediction)