llama-index-vector-stores-opensearch 0.1.11__tar.gz → 0.1.12__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of llama-index-vector-stores-opensearch might be problematic. Click here for more details.
- {llama_index_vector_stores_opensearch-0.1.11 → llama_index_vector_stores_opensearch-0.1.12}/PKG-INFO +1 -1
- {llama_index_vector_stores_opensearch-0.1.11 → llama_index_vector_stores_opensearch-0.1.12}/llama_index/vector_stores/opensearch/base.py +7 -2
- {llama_index_vector_stores_opensearch-0.1.11 → llama_index_vector_stores_opensearch-0.1.12}/pyproject.toml +1 -1
- {llama_index_vector_stores_opensearch-0.1.11 → llama_index_vector_stores_opensearch-0.1.12}/README.md +0 -0
- {llama_index_vector_stores_opensearch-0.1.11 → llama_index_vector_stores_opensearch-0.1.12}/llama_index/vector_stores/opensearch/__init__.py +0 -0
|
@@ -20,6 +20,7 @@ from llama_index.core.vector_stores.utils import (
|
|
|
20
20
|
node_to_metadata_dict,
|
|
21
21
|
)
|
|
22
22
|
from opensearchpy import AsyncOpenSearch
|
|
23
|
+
from opensearchpy.client import Client as OSClient
|
|
23
24
|
from opensearchpy.exceptions import NotFoundError
|
|
24
25
|
from opensearchpy.helpers import async_bulk
|
|
25
26
|
|
|
@@ -64,8 +65,10 @@ class OpensearchVectorClient:
|
|
|
64
65
|
embedding_field: str = "embedding",
|
|
65
66
|
text_field: str = "content",
|
|
66
67
|
method: Optional[dict] = None,
|
|
68
|
+
engine: Optional[str] = "nmslib",
|
|
67
69
|
max_chunk_bytes: int = 1 * 1024 * 1024,
|
|
68
70
|
search_pipeline: Optional[str] = None,
|
|
71
|
+
os_client: Optional[OSClient] = None,
|
|
69
72
|
**kwargs: Any,
|
|
70
73
|
):
|
|
71
74
|
"""Init params."""
|
|
@@ -73,7 +76,7 @@ class OpensearchVectorClient:
|
|
|
73
76
|
method = {
|
|
74
77
|
"name": "hnsw",
|
|
75
78
|
"space_type": "l2",
|
|
76
|
-
"engine":
|
|
79
|
+
"engine": engine,
|
|
77
80
|
"parameters": {"ef_construction": 256, "m": 48},
|
|
78
81
|
}
|
|
79
82
|
if embedding_field is None:
|
|
@@ -102,7 +105,9 @@ class OpensearchVectorClient:
|
|
|
102
105
|
}
|
|
103
106
|
},
|
|
104
107
|
}
|
|
105
|
-
self._os_client = self._get_async_opensearch_client(
|
|
108
|
+
self._os_client = os_client or self._get_async_opensearch_client(
|
|
109
|
+
self._endpoint, **kwargs
|
|
110
|
+
)
|
|
106
111
|
not_found_error = self._import_not_found_error()
|
|
107
112
|
|
|
108
113
|
event_loop = asyncio.get_event_loop()
|
|
File without changes
|