fix: hybrid search

This commit is contained in:
Timothy J. Baek
2024-09-13 01:21:47 -04:00
parent 939bfd153e
commit 823093eea6
3 changed files with 3 additions and 6 deletions

View File

@@ -70,13 +70,10 @@ class ChromaClient:
# Get all the items in the collection.
collection = self.client.get_collection(name=collection_name)
if collection:
result = collection.get()
return GetResult(
**{
"ids": [result["ids"]],
"distances": [result["distances"]],
"documents": [result["documents"]],
"metadatas": [result["metadatas"]],
}

View File

@@ -113,7 +113,7 @@ class MilvusClient:
collection_name=f"{self.collection_prefix}_{collection_name}",
filter='id != ""',
)
return self._result_to_query_result(result)
return GetResult(**self._result_to_query_result(result))
def insert(self, collection_name: str, items: list[VectorItem]):
# Insert the items into the collection, if the collection does not exist, it will be created.