llama-index-vector-stores-chroma 0.4.0__py3-none-any.whl → 0.4.2__py3-none-any.whl

llama_index/vector_stores/chroma/base.py

@@ -2,7 +2,7 @@
 
 import logging
 import math
-from typing import Any, Dict, Generator, List, Optional, cast
+from typing import Any, Dict, Generator, List, Optional, Union, cast
 
 import chromadb
 from chromadb.api.models.Collection import Collection
@@ -95,7 +95,8 @@ MAX_CHUNK_SIZE = 41665  # One less than the max chunk size for ChromaDB
 def chunk_list(
     lst: List[BaseNode], max_chunk_size: int
 ) -> Generator[List[BaseNode], None, None]:
-    """Yield successive max_chunk_size-sized chunks from lst.
+    """
+    Yield successive max_chunk_size-sized chunks from lst.
 
     Args:
         lst (List[BaseNode]): list of nodes with embeddings
@@ -103,13 +104,15 @@ def chunk_list(
 
     Yields:
         Generator[List[BaseNode], None, None]: list of nodes with embeddings
+
     """
     for i in range(0, len(lst), max_chunk_size):
         yield lst[i : i + max_chunk_size]
 
 
 class ChromaVectorStore(BasePydanticVectorStore):
-    """Chroma vector store.
+    """
+    Chroma vector store.
 
     In this vector store, embeddings are stored within a ChromaDB collection.
 
@@ -142,7 +145,7 @@ class ChromaVectorStore(BasePydanticVectorStore):
 
     collection_name: Optional[str]
     host: Optional[str]
-    port: Optional[str]
+    port: Optional[Union[str, int]]
     ssl: bool
     headers: Optional[Dict[str, str]]
     persist_dir: Optional[str]
@@ -155,7 +158,7 @@ class ChromaVectorStore(BasePydanticVectorStore):
         chroma_collection: Optional[Any] = None,
         collection_name: Optional[str] = None,
         host: Optional[str] = None,
-        port: Optional[str] = None,
+        port: Optional[Union[str, int]] = None,
         ssl: bool = False,
         headers: Optional[Dict[str, str]] = None,
         persist_dir: Optional[str] = None,
@@ -199,7 +202,7 @@ class ChromaVectorStore(BasePydanticVectorStore):
         cls,
         collection_name: str,
         host: Optional[str] = None,
-        port: Optional[str] = None,
+        port: Optional[Union[str, int]] = None,
         ssl: bool = False,
         headers: Optional[Dict[str, str]] = None,
         persist_dir: Optional[str] = None,
@@ -240,7 +243,8 @@ class ChromaVectorStore(BasePydanticVectorStore):
         node_ids: Optional[List[str]],
         filters: Optional[List[MetadataFilters]] = None,
     ) -> List[BaseNode]:
-        """Get nodes from index.
+        """
+        Get nodes from index.
 
         Args:
             node_ids (List[str]): list of node ids
@@ -262,7 +266,8 @@ class ChromaVectorStore(BasePydanticVectorStore):
         return result.nodes
 
     def add(self, nodes: List[BaseNode], **add_kwargs: Any) -> List[str]:
-        """Add nodes to index.
+        """
+        Add nodes to index.
 
         Args:
             nodes: List[BaseNode]: list of nodes with embeddings
@@ -317,7 +322,8 @@ class ChromaVectorStore(BasePydanticVectorStore):
         node_ids: Optional[List[str]] = None,
         filters: Optional[List[MetadataFilters]] = None,
     ) -> None:
-        """Delete nodes from index.
+        """
+        Delete nodes from index.
 
         Args:
             node_ids (List[str]): list of node ids
@@ -331,10 +337,10 @@ class ChromaVectorStore(BasePydanticVectorStore):
 
         if filters:
             where = _to_chroma_filter(filters)
-        else:
-            where = None
+            self._collection.delete(ids=node_ids, where=where)
 
-        self._collection.delete(ids=node_ids, where=where)
+        else:
+            self._collection.delete(ids=node_ids)
 
     def clear(self) -> None:
         """Clear the collection."""
@@ -347,7 +353,8 @@ class ChromaVectorStore(BasePydanticVectorStore):
         return self._collection
 
     def query(self, query: VectorStoreQuery, **kwargs: Any) -> VectorStoreQueryResult:
-        """Query index for top k most similar nodes.
+        """
+        Query index for top k most similar nodes.
 
         Args:
             query_embedding (List[float]): query embedding
@@ -378,12 +385,19 @@ class ChromaVectorStore(BasePydanticVectorStore):
     def _query(
         self, query_embeddings: List["float"], n_results: int, where: dict, **kwargs
     ) -> VectorStoreQueryResult:
-        results = self._collection.query(
-            query_embeddings=query_embeddings,
-            n_results=n_results,
-            where=where,
-            **kwargs,
-        )
+        if where:
+            results = self._collection.query(
+                query_embeddings=query_embeddings,
+                n_results=n_results,
+                where=where,
+                **kwargs,
+            )
+        else:
+            results = self._collection.query(
+                query_embeddings=query_embeddings,
+                n_results=n_results,
+                **kwargs,
+            )
 
         logger.debug(f"> Top {len(results['documents'][0])} nodes:")
         nodes = []
@@ -429,11 +443,17 @@ class ChromaVectorStore(BasePydanticVectorStore):
     def _get(
         self, limit: Optional[int], where: dict, **kwargs
     ) -> VectorStoreQueryResult:
-        results = self._collection.get(
-            limit=limit,
-            where=where,
-            **kwargs,
-        )
+        if where:
+            results = self._collection.get(
+                limit=limit,
+                where=where,
+                **kwargs,
+            )
+        else:
+            results = self._collection.get(
+                limit=limit,
+                **kwargs,
+            )
 
         logger.debug(f"> Top {len(results['documents'])} nodes:")
         nodes = []

llama_index_vector_stores_chroma-0.4.2.dist-info/METADATA

@@ -0,0 +1,13 @@
+Metadata-Version: 2.4
+Name: llama-index-vector-stores-chroma
+Version: 0.4.2
+Summary: llama-index vector_stores chroma integration
+Author-email: Your Name <you@example.com>
+License-Expression: MIT
+License-File: LICENSE
+Requires-Python: <4.0,>=3.10
+Requires-Dist: chromadb>=0.5.17
+Requires-Dist: llama-index-core<0.13,>=0.12.0
+Description-Content-Type: text/markdown
+
+# LlamaIndex Vector_Stores Integration: Chroma

llama_index_vector_stores_chroma-0.4.2.dist-info/RECORD

@@ -0,0 +1,7 @@
+llama_index/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+llama_index/vector_stores/chroma/__init__.py,sha256=QNMK-nHKEt-wmks5mhWfdOKDybpmsqrL4neV-HCA6N4,101
+llama_index/vector_stores/chroma/base.py,sha256=Sv1ZGpa_e9ZCRfhfMi8xLmcd_A586Z0TpoRESCcxglE,15359
+llama_index_vector_stores_chroma-0.4.2.dist-info/METADATA,sha256=4xusRlNXdisiDzItXkZtXo-aL7Lx3iD2FMlBjBr885g,413
+llama_index_vector_stores_chroma-0.4.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+llama_index_vector_stores_chroma-0.4.2.dist-info/licenses/LICENSE,sha256=JPQLUZD9rKvCTdu192Nk0V5PAwklIg6jANii3UmTyMs,1065
+llama_index_vector_stores_chroma-0.4.2.dist-info/RECORD,,

{llama_index_vector_stores_chroma-0.4.0.dist-info → llama_index_vector_stores_chroma-0.4.2.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.9.0
+Generator: hatchling 1.27.0
 Root-Is-Purelib: true
 Tag: py3-none-any

llama_index_vector_stores_chroma-0.4.2.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+The MIT License
+
+Copyright (c) Jerry Liu
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

llama_index_vector_stores_chroma-0.4.0.dist-info/METADATA

@@ -1,20 +0,0 @@
-Metadata-Version: 2.1
-Name: llama-index-vector-stores-chroma
-Version: 0.4.0
-Summary: llama-index vector_stores chroma integration
-License: MIT
-Author: Your Name
-Author-email: you@example.com
-Requires-Python: >=3.9,<4.0
-Classifier: License :: OSI Approved :: MIT License
-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.5.17)
-Requires-Dist: llama-index-core (>=0.12.0,<0.13.0)
-Description-Content-Type: text/markdown
-
-# LlamaIndex Vector_Stores Integration: Chroma
-

llama_index_vector_stores_chroma-0.4.0.dist-info/RECORD

@@ -1,6 +0,0 @@
-llama_index/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-llama_index/vector_stores/chroma/__init__.py,sha256=QNMK-nHKEt-wmks5mhWfdOKDybpmsqrL4neV-HCA6N4,101
-llama_index/vector_stores/chroma/base.py,sha256=bMnmCNGwNSPbuCUi_J_DUfvpucNibq8l2U9C4AZOwXY,14845
-llama_index_vector_stores_chroma-0.4.0.dist-info/METADATA,sha256=jicQJ7vcOH6oPwjdE5ty21WYwzU9jRrb4xuE1xffnq4,696
-llama_index_vector_stores_chroma-0.4.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-llama_index_vector_stores_chroma-0.4.0.dist-info/RECORD,,