llama-index-vector-stores-opensearch 0.1.12__py3-none-any.whl → 0.1.13__py3-none-any.whl

llama_index/vector_stores/opensearch/base.py

@@ -214,7 +214,10 @@ class OpensearchVectorClient:
         pre_filter = []
         if filters is not None:
             for f in filters.legacy_filters():
-                pre_filter.append({f.key: json.loads(str(f.value))})
+                if isinstance(f.value, str):
+                    pre_filter.append({f.key: f.value})
+                else:
+                    pre_filter.append({f.key: json.loads(str(f.value))})
 
         return pre_filter
 
@@ -389,6 +392,41 @@ class OpensearchVectorClient:
         }
         await self._os_client.delete_by_query(index=self._index, body=search_query)
 
+    async def delete_nodes(
+        self,
+        node_ids: Optional[List[str]] = None,
+        filters: Optional[MetadataFilters] = None,
+        **delete_kwargs: Any,
+    ) -> None:
+        """Deletes nodes.
+
+        Args:
+            node_ids (Optional[List[str]], optional): IDs of nodes to delete. Defaults to None.
+            filters (Optional[MetadataFilters], optional): Metadata filters. Defaults to None.
+        """
+        if not node_ids and not filters:
+            return
+
+        query = {"query": {"bool": {"filter": []}}}
+        if node_ids:
+            query["query"]["bool"]["filter"].append({"terms": {"_id": node_ids or []}})
+
+        if filters:
+            for filter in self._parse_filters(filters):
+                newfilter = {}
+
+                for key in filter:
+                    newfilter[f"metadata.{key}.keyword"] = filter[key]
+
+                query["query"]["bool"]["filter"].append({"term": newfilter})
+
+        await self._os_client.delete_by_query(index=self._index, body=query)
+
+    async def clear(self) -> None:
+        """Clears index."""
+        query = {"query": {"bool": {"filter": []}}}
+        await self._os_client.delete_by_query(index=self._index, body=query)
+
     async def aquery(
         self,
         query_mode: VectorStoreQueryMode,
@@ -574,6 +612,44 @@ class OpensearchVectorStore(BasePydanticVectorStore):
         """
         await self._client.delete_by_doc_id(ref_doc_id)
 
+    async def adelete_nodes(
+        self,
+        node_ids: Optional[List[str]] = None,
+        filters: Optional[MetadataFilters] = None,
+        **delete_kwargs: Any,
+    ) -> None:
+        """Deletes nodes async.
+
+        Args:
+            node_ids (Optional[List[str]], optional): IDs of nodes to delete. Defaults to None.
+            filters (Optional[MetadataFilters], optional): Metadata filters. Defaults to None.
+        """
+        await self._client.delete_nodes(node_ids, filters, **delete_kwargs)
+
+    def delete_nodes(
+        self,
+        node_ids: Optional[List[str]] = None,
+        filters: Optional[MetadataFilters] = None,
+        **delete_kwargs: Any,
+    ) -> None:
+        """Deletes nodes.
+
+        Args:
+            node_ids (Optional[List[str]], optional): IDs of nodes to delete. Defaults to None.
+            filters (Optional[MetadataFilters], optional): Metadata filters. Defaults to None.
+        """
+        asyncio.get_event_loop().run_until_complete(
+            self.adelete_nodes(node_ids, filters, **delete_kwargs)
+        )
+
+    async def aclear(self) -> None:
+        """Clears index."""
+        await self._client.clear()
+
+    def clear(self) -> None:
+        """Clears index."""
+        asyncio.get_event_loop().run_until_complete(self.aclear())
+
     def query(self, query: VectorStoreQuery, **kwargs: Any) -> VectorStoreQueryResult:
         """
         Query index for top k most similar nodes.

{llama_index_vector_stores_opensearch-0.1.12.dist-info → llama_index_vector_stores_opensearch-0.1.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llama-index-vector-stores-opensearch
-Version: 0.1.12
+Version: 0.1.13
 Summary: llama-index vector_stores opensearch integration
 License: MIT
 Author: Your Name

llama_index_vector_stores_opensearch-0.1.13.dist-info/RECORD

@@ -0,0 +1,5 @@
+llama_index/vector_stores/opensearch/__init__.py,sha256=U1_XAkZb6zcskOk4s10NB8Tjs9AZRGdRQLzOGpbWdBA,176
+llama_index/vector_stores/opensearch/base.py,sha256=YShOE7dzy3L5G8N9iobA6aenZLF_dJACs1aCBrQIPyg,22584
+llama_index_vector_stores_opensearch-0.1.13.dist-info/METADATA,sha256=X2CxNuyUSEoChYA5Rss4QThW0HarjyC3UQpWE24reOU,729
+llama_index_vector_stores_opensearch-0.1.13.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+llama_index_vector_stores_opensearch-0.1.13.dist-info/RECORD,,

llama_index_vector_stores_opensearch-0.1.12.dist-info/RECORD

@@ -1,5 +0,0 @@
-llama_index/vector_stores/opensearch/__init__.py,sha256=U1_XAkZb6zcskOk4s10NB8Tjs9AZRGdRQLzOGpbWdBA,176
-llama_index/vector_stores/opensearch/base.py,sha256=sSR4o3c14oTM9s8LCWJRh_7LJxViVk7vOb98Xk4E-jA,19952
-llama_index_vector_stores_opensearch-0.1.12.dist-info/METADATA,sha256=uXlsSYc_ruWaDwmaB3I29LYEYXtOoRDs-Fie_7b1aWA,729
-llama_index_vector_stores_opensearch-0.1.12.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-llama_index_vector_stores_opensearch-0.1.12.dist-info/RECORD,,