Update app.py

app.py

@@ -332,17 +332,33 @@ 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)
+    #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)
     
-    doc_context = format_kadi_api_doc_context(kadi_apy_docs)
     #print(doc_context)
-    code_context = format_kadi_apy_library_context(kadi_apy_library_docs)
     #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.")
+    metadata_prediction = llm.predict(prompt)
+    print(metadata_prediction)
+    
+    kadi_apy_docs = retrieve_within_kadiApy_docs (vector_store, query, k = 5)
+    kadi_apy_library_docs = vectorstore.similarity_search(query, filter={"usage": metadata_prediction})
+        
+    doc_context = format_kadi_api_doc_context(kadi_apy_docs)
+    code_context = format_kadi_apy_library_context(kadi_apy_library_docs)
 
 
-    
+    # Assuming kadi_apy_library_docs is a list of Document objects
+    for idx, doc in enumerate(kadi_apy_library_docs, start=1):
+        # Access the "source" metadata
+        source_value = doc.metadata.get("source", "No source found")
+        # Print the formatted output
+        print(f"=== Document {idx} ===")
+        print(f"Source: {source_value}")
+        print("======================\n")
+
     
     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. 
                     "Doc-context:" provides you with information how to use this API library by givnig code examples and code documentation.