llama-index-vector-stores-chroma 0.1.9__tar.gz → 0.2.0__tar.gz

{llama_index_vector_stores_chroma-0.1.9 → llama_index_vector_stores_chroma-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llama-index-vector-stores-chroma
-Version: 0.1.9
+Version: 0.2.0
 Summary: llama-index vector_stores chroma integration
 License: MIT
 Author: Your Name
@@ -11,8 +11,9 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: chromadb (>=0.4.0,<0.6.0)
-Requires-Dist: llama-index-core (>=0.10.1,<0.11.0)
+Requires-Dist: llama-index-core (>=0.11.0,<0.12.0)
 Description-Content-Type: text/markdown
 
 # LlamaIndex Vector_Stores Integration: Chroma

{llama_index_vector_stores_chroma-0.1.9 → llama_index_vector_stores_chroma-0.2.0}/llama_index/vector_stores/chroma/base.py

@@ -164,13 +164,6 @@ class ChromaVectorStore(BasePydanticVectorStore):
     ) -> None:
         """Init params."""
         collection_kwargs = collection_kwargs or {}
-        if chroma_collection is None:
-            client = chromadb.HttpClient(host=host, port=port, ssl=ssl, headers=headers)
-            self._collection = client.get_or_create_collection(
-                name=collection_name, **collection_kwargs
-            )
-        else:
-            self._collection = cast(Collection, chroma_collection)
 
         super().__init__(
             host=host,
@@ -181,6 +174,13 @@ class ChromaVectorStore(BasePydanticVectorStore):
             persist_dir=persist_dir,
             collection_kwargs=collection_kwargs or {},
         )
+        if chroma_collection is None:
+            client = chromadb.HttpClient(host=host, port=port, ssl=ssl, headers=headers)
+            self._collection = client.get_or_create_collection(
+                name=collection_name, **collection_kwargs
+            )
+        else:
+            self._collection = cast(Collection, chroma_collection)
 
     @classmethod
     def from_collection(cls, collection: Any) -> "ChromaVectorStore":
@@ -443,7 +443,7 @@ class ChromaVectorStore(BasePydanticVectorStore):
             results["ids"] = [[]]
 
         for node_id, text, metadata in zip(
-            results["ids"][0], results["documents"], results["metadatas"]
+            results["ids"], results["documents"], results["metadatas"]
         ):
             try:
                 node = metadata_dict_to_node(metadata)

{llama_index_vector_stores_chroma-0.1.9 → llama_index_vector_stores_chroma-0.2.0}/pyproject.toml

@@ -27,12 +27,12 @@ exclude = ["**/BUILD"]
 license = "MIT"
 name = "llama-index-vector-stores-chroma"
 readme = "README.md"
-version = "0.1.9"
+version = "0.2.0"
 
 [tool.poetry.dependencies]
 python = ">=3.8.1,<4.0"
-llama-index-core = "^0.10.1"
 chromadb = ">=0.4.0,<0.6.0"
+llama-index-core = "^0.11.0"
 
 [tool.poetry.group.dev.dependencies]
 ipython = "8.10.0"