Update app.py

app.py

@@ -332,10 +332,29 @@ def rag_workflow(query):
     rewritten_query_response = llm.invoke("Hello")
     rewritten_query = rewritten_query_response.content.strip()
 
+
+    prompt = (
+        f"""The query is: '{query}'.
+            Based on the user's query, assist them by determining which technical document they should read to interact with the software named 'Kadi4Mat'. 
+            There are three different technical documents to choose from:
+                - Document 1: Provides information on how to use a Python library to interact with the HTTP API of 'Kadi4Mat'.
+                - Document 2: Provides information on how to use a Python library to implement custom CLI commands to interact with 'Kadi4Mat'.
+                - Document 3: Provides prewritten CLI command lines and guides on how to use the command-line interface to interact with 'Kadi4Mat'.
+    
+            Your task is to select the single most likely option. 
+                If Document 1 is the best choice, respond with 'kadi-apy python library'. 
+                If Document 2 is the best choice, respond with 'kadi-apy python cli library'. 
+                If Document 3 is the best choice, respond with 'kadi-apy command line interface'. 
+            Respond with only the exact corresponding option and do not include any additional comments, explanations, or text."
+        """
+    )
+    metadata_prediction = llm.predict(prompt)
+
+
     prompt = (
         f"The query is: '{query}'. "
         "Based on this query, choose the most appropriate option from the following: "
-        "'kadiAPY library"', 'kadiAPY cli library', 'kadiAPY command line'. "
+        "'kadiAPY library"', 'kadiAPY cli library', 'kadiAPY command line'"
         "Respond only with one of these options—do not add any additional text or commentary."
     )
     metadata_prediction = llm.predict(prompt)