Update app.py
Browse files
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"
|
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 |
-
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 = {"
|
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 |
|