Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -345,7 +345,7 @@ def rag_workflow(query):
|
|
345 |
print(metadata_prediction)
|
346 |
|
347 |
kadi_apy_docs = retrieve_within_kadiApy_docs (vector_store, query, k = 5)
|
348 |
-
kadi_apy_library_docs =
|
349 |
|
350 |
doc_context = format_kadi_api_doc_context(kadi_apy_docs)
|
351 |
code_context = format_kadi_apy_library_context(kadi_apy_library_docs)
|
|
|
345 |
print(metadata_prediction)
|
346 |
|
347 |
kadi_apy_docs = retrieve_within_kadiApy_docs (vector_store, query, k = 5)
|
348 |
+
kadi_apy_library_docs = vector_store.similarity_search(query, filter={"usage": metadata_prediction})
|
349 |
|
350 |
doc_context = format_kadi_api_doc_context(kadi_apy_docs)
|
351 |
code_context = format_kadi_apy_library_context(kadi_apy_library_docs)
|