Update app.py

app.py

@@ -332,15 +332,14 @@ def rag_workflow(query):
     rewritten_query_response = llm.invoke("Hello")
     rewritten_query = rewritten_query_response.content.strip()
 
-    #kadi_apy_docs = retrieve_within_kadiApy_docs (vector_store, query, k = 5)       
-    #kadi_apy_library_docs = retrieve_within_kadiApy_library (vector_store, query, k = 10)
-    
-    #print(doc_context)
-    #print("AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA:",code_context)
-
-    
-    prompt = f"The query is: '{query}'. What type of metadata applies to this query? Options: python library, python cli library, command line interface."
+    prompt = (
+        f"The query is: '{query}'. "
+        "Based on this query, choose the most appropriate option from the following: "
+        "'python library', 'python cli library', 'command line interface'. "
+        "Respond only with one of these options—do not add any additional text or commentary."
+    )
     metadata_prediction = llm.predict(prompt)
+
     print("METADATA PREDICTION -------------------------:", metadata_prediction)
     print(metadata_prediction)