llama-index-vector-stores-chroma 0.1.7__py3-none-any.whl → 0.1.9__py3-none-any.whl

llama_index/vector_stores/chroma/base.py

@@ -48,6 +48,10 @@ def _transform_chroma_filter_operator(operator: str) -> str:
         return "$gte"
     elif operator == "<=":
         return "$lte"
+    elif operator == "in":
+        return "$in"
+    elif operator == "nin":
+        return "$nin"
     else:
         raise ValueError(f"Filter operator {operator} not supported")
 
@@ -144,7 +148,7 @@ class ChromaVectorStore(BasePydanticVectorStore):
     persist_dir: Optional[str]
     collection_kwargs: Dict[str, Any] = Field(default_factory=dict)
 
-    _collection: Any = PrivateAttr()
+    _collection: Collection = PrivateAttr()
 
     def __init__(
         self,
@@ -231,6 +235,32 @@ class ChromaVectorStore(BasePydanticVectorStore):
     def class_name(cls) -> str:
         return "ChromaVectorStore"
 
+    def get_nodes(
+        self,
+        node_ids: Optional[List[str]],
+        filters: Optional[List[MetadataFilters]] = None,
+    ) -> List[BaseNode]:
+        """Get nodes from index.
+
+        Args:
+            node_ids (List[str]): list of node ids
+            filters (List[MetadataFilters]): list of metadata filters
+
+        """
+        if not self._collection:
+            raise ValueError("Collection not initialized")
+
+        node_ids = node_ids or []
+
+        if filters:
+            where = _to_chroma_filter(filters)
+        else:
+            where = {}
+
+        result = self._get(None, where=where, ids=node_ids)
+
+        return result.nodes
+
     def add(self, nodes: List[BaseNode], **add_kwargs: Any) -> List[str]:
         """Add nodes to index.
 
@@ -282,6 +312,35 @@ class ChromaVectorStore(BasePydanticVectorStore):
         """
         self._collection.delete(where={"document_id": ref_doc_id})
 
+    def delete_nodes(
+        self,
+        node_ids: Optional[List[str]] = None,
+        filters: Optional[List[MetadataFilters]] = None,
+    ) -> None:
+        """Delete nodes from index.
+
+        Args:
+            node_ids (List[str]): list of node ids
+            filters (List[MetadataFilters]): list of metadata filters
+
+        """
+        if not self._collection:
+            raise ValueError("Collection not initialized")
+
+        node_ids = node_ids or []
+
+        if filters:
+            where = _to_chroma_filter(filters)
+        else:
+            where = {}
+
+        self._collection.delete(ids=node_ids, where=where)
+
+    def clear(self) -> None:
+        """Clear the collection."""
+        ids = self._collection.get()["ids"]
+        self._collection.delete(ids=ids)
+
     @property
     def client(self) -> Any:
         """Return client."""
@@ -326,7 +385,7 @@ class ChromaVectorStore(BasePydanticVectorStore):
             **kwargs,
         )
 
-        logger.debug(f"> Top {len(results['documents'])} nodes:")
+        logger.debug(f"> Top {len(results['documents'][0])} nodes:")
         nodes = []
         similarities = []
         ids = []
@@ -367,7 +426,9 @@ class ChromaVectorStore(BasePydanticVectorStore):
 
         return VectorStoreQueryResult(nodes=nodes, similarities=similarities, ids=ids)
 
-    def _get(self, limit: int, where: dict, **kwargs) -> VectorStoreQueryResult:
+    def _get(
+        self, limit: Optional[int], where: dict, **kwargs
+    ) -> VectorStoreQueryResult:
         results = self._collection.get(
             limit=limit,
             where=where,

{llama_index_vector_stores_chroma-0.1.7.dist-info → llama_index_vector_stores_chroma-0.1.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llama-index-vector-stores-chroma
-Version: 0.1.7
+Version: 0.1.9
 Summary: llama-index vector_stores chroma integration
 License: MIT
 Author: Your Name

llama_index_vector_stores_chroma-0.1.9.dist-info/RECORD

@@ -0,0 +1,5 @@
+llama_index/vector_stores/chroma/__init__.py,sha256=QNMK-nHKEt-wmks5mhWfdOKDybpmsqrL4neV-HCA6N4,101
+llama_index/vector_stores/chroma/base.py,sha256=_PxgEShceRgIq_f4l-T19GlUQSQm_4akn7BYlUluPk4,14842
+llama_index_vector_stores_chroma-0.1.9.dist-info/METADATA,sha256=Yr549GoPezUy0Hfb-BwH5sgfCVKeTpiHWKXVi2qR3Us,653
+llama_index_vector_stores_chroma-0.1.9.dist-info/WHEEL,sha256=d2fvjOD7sXsVzChCqf0Ty0JbHKBaLYwDbGQDwQTnJ50,88
+llama_index_vector_stores_chroma-0.1.9.dist-info/RECORD,,

llama_index_vector_stores_chroma-0.1.7.dist-info/RECORD

@@ -1,5 +0,0 @@
-llama_index/vector_stores/chroma/__init__.py,sha256=QNMK-nHKEt-wmks5mhWfdOKDybpmsqrL4neV-HCA6N4,101
-llama_index/vector_stores/chroma/base.py,sha256=OlB4EU_CZZTdlIxyPWEyliydz1__nvLbXpY7H8TNi34,13268
-llama_index_vector_stores_chroma-0.1.7.dist-info/METADATA,sha256=NfLG6nmPL5KxbgQjxTnbg4rlYiSzDEKNiiat9ZYcqiM,653
-llama_index_vector_stores_chroma-0.1.7.dist-info/WHEEL,sha256=d2fvjOD7sXsVzChCqf0Ty0JbHKBaLYwDbGQDwQTnJ50,88
-llama_index_vector_stores_chroma-0.1.7.dist-info/RECORD,,