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

llama_index/vector_stores/opensearch/base.py

@@ -20,6 +20,7 @@ from llama_index.core.vector_stores.utils import (
     node_to_metadata_dict,
 )
 from opensearchpy import AsyncOpenSearch
+from opensearchpy.client import Client as OSClient
 from opensearchpy.exceptions import NotFoundError
 from opensearchpy.helpers import async_bulk
 
@@ -64,8 +65,10 @@ class OpensearchVectorClient:
         embedding_field: str = "embedding",
         text_field: str = "content",
         method: Optional[dict] = None,
+        engine: Optional[str] = "nmslib",
         max_chunk_bytes: int = 1 * 1024 * 1024,
         search_pipeline: Optional[str] = None,
+        os_client: Optional[OSClient] = None,
         **kwargs: Any,
     ):
         """Init params."""
@@ -73,7 +76,7 @@ class OpensearchVectorClient:
             method = {
                 "name": "hnsw",
                 "space_type": "l2",
-                "engine": "nmslib",
+                "engine": engine,
                 "parameters": {"ef_construction": 256, "m": 48},
             }
         if embedding_field is None:
@@ -102,7 +105,9 @@ class OpensearchVectorClient:
                 }
             },
         }
-        self._os_client = self._get_async_opensearch_client(self._endpoint, **kwargs)
+        self._os_client = os_client or self._get_async_opensearch_client(
+            self._endpoint, **kwargs
+        )
         not_found_error = self._import_not_found_error()
 
         event_loop = asyncio.get_event_loop()

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

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

llama_index_vector_stores_opensearch-0.1.12.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=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,,

llama_index_vector_stores_opensearch-0.1.11.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=bHE7FV-CLU-x_ChG9CdoE4WzT1YkkZyIg23PYB7Eg30,19780
-llama_index_vector_stores_opensearch-0.1.11.dist-info/METADATA,sha256=JrqjHpIkrncELU0t2pNIKbbYaAahVnFP8r9Tz0LeYrE,729
-llama_index_vector_stores_opensearch-0.1.11.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-llama_index_vector_stores_opensearch-0.1.11.dist-info/RECORD,,