Skip to content

Commit d92ca82

Browse files
authored
Remove text field from milvus query for list/delete API (NVIDIA#242)
1 parent 19f5722 commit d92ca82

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

RAG/src/chain_server/utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -517,7 +517,7 @@ def get_docs_vectorstore_langchain(vectorstore: VectorStore) -> List[str]:
517517
elif settings.vector_store.name == "milvus":
518518
# Getting all the ID's > 0
519519
if vectorstore.col:
520-
milvus_data = vectorstore.col.query(expr="pk >= 0", output_fields=["pk", "source", "text"])
520+
milvus_data = vectorstore.col.query(expr="pk >= 0", output_fields=["pk", "source"])
521521
filenames = set([extract_filename(metadata) for metadata in milvus_data])
522522
return filenames
523523
except Exception as e:
@@ -582,7 +582,7 @@ def del_docs_vectorstore_langchain(vectorstore: VectorStore, filenames: List[str
582582
logger.info(f"Deleted documents with filenames {filename}")
583583
elif settings.vector_store.name == "milvus":
584584
# Getting all the ID's > 0
585-
milvus_data = vectorstore.col.query(expr="pk >= 0", output_fields=["pk", "source", "text"])
585+
milvus_data = vectorstore.col.query(expr="pk >= 0", output_fields=["pk", "source"])
586586
for filename in filenames:
587587
# get ids with filename in metadata
588588
ids_list = [metadata["pk"] for metadata in milvus_data if extract_filename(metadata) == filename]

0 commit comments

Comments
 (0)