bupa1018 commited on
Commit
acbd799
·
1 Parent(s): 008e2b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -344,7 +344,7 @@ def rag_workflow(query):
344
  print(rewritten_query)
345
 
346
  kadi_apy_docs = vector_store.similarity_search(query, k=5, filter={"usage": "doc"})
347
- kadi_apy_docs = vector_store.similarity_search(query, k=5, filter={"usage": library_usage_prediction})
348
 
349
  doc_context = format_kadi_api_doc_context(kadi_apy_docs)
350
  code_context = format_kadi_apy_library_context(kadi_apy_sourcecode)
 
344
  print(rewritten_query)
345
 
346
  kadi_apy_docs = vector_store.similarity_search(query, k=5, filter={"usage": "doc"})
347
+ kadi_apy_sourcecode = vector_store.similarity_search(query, k=5, filter={"usage": library_usage_prediction})
348
 
349
  doc_context = format_kadi_api_doc_context(kadi_apy_docs)
350
  code_context = format_kadi_apy_library_context(kadi_apy_sourcecode)