|
@@ -109,7 +109,10 @@ class ChromaClient:
|
|
|
|
|
|
def insert(self, collection_name: str, items: list[VectorItem]):
|
|
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.
|
|
# Insert the items into the collection, if the collection does not exist, it will be created.
|
|
- collection = self.client.get_or_create_collection(name=collection_name)
|
|
|
|
|
|
+ collection = self.client.get_or_create_collection(
|
|
|
|
+ name=collection_name,
|
|
|
|
+ metadata={"hnsw:space": "cosine"}
|
|
|
|
+ )
|
|
|
|
|
|
ids = [item["id"] for item in items]
|
|
ids = [item["id"] for item in items]
|
|
documents = [item["text"] for item in items]
|
|
documents = [item["text"] for item in items]
|