bupa1018 commited on
Commit
582f629
·
1 Parent(s): bcf5ba4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -270,19 +270,18 @@ def retrieve_within_kadiApy_library(vectorstore, query, k):
270
  return retrieved_docs
271
 
272
  def retrieve_within_kadiApy_cli_library(vectorstore, query, k):
273
- filter_criteria = {"usage": "cli_library", "visibility" : "public"}
274
  retrieved_docs = vectorstore.similarity_search(query=query, k=k, filter=filter_criteria)
275
  return retrieved_docs
276
 
277
  def retrieve_within_kadiApy_cli_library_excluding_cli_commands(vectorstore, query, k):
278
- filter_criteria = {"usage": "cli_library", "visibility" : "public"}
279
- retrieved_docs = vectorstore.similarity_search(query=query, k=k, filter=filter_criteria)
280
  filtered_docs = [doc for doc in documents if "command" not in doc.metadata]
281
  return filtered_docs
282
 
283
 
284
  def retrieve_kadiApy_cli_commands(vectorstore, query, k):
285
- filter_criteria = {"usage": "cli_library", "type": "command"}
286
  results = vectorstore.similarity_search(query=query, k=k, filter=filter_criteria)
287
  return results
288
 
 
270
  return retrieved_docs
271
 
272
  def retrieve_within_kadiApy_cli_library(vectorstore, query, k):
273
+ filter_criteria = {"usage": "cli_library"}
274
  retrieved_docs = vectorstore.similarity_search(query=query, k=k, filter=filter_criteria)
275
  return retrieved_docs
276
 
277
  def retrieve_within_kadiApy_cli_library_excluding_cli_commands(vectorstore, query, k):
278
+ filter_criteria = {"usage": "cli_library"}
 
279
  filtered_docs = [doc for doc in documents if "command" not in doc.metadata]
280
  return filtered_docs
281
 
282
 
283
  def retrieve_kadiApy_cli_commands(vectorstore, query, k):
284
+ filter_criteria = {"type": "command"}
285
  results = vectorstore.similarity_search(query=query, k=k, filter=filter_criteria)
286
  return results
287