Harrison/open search kwargs (#2040)

Signed-off-by: Marcel Coetzee <marcelcoetzee@tutanota.com>
Co-authored-by: Marcel <34739235+Pipboyguy@users.noreply.github.com>
searx
Harrison Chase 1 year ago committed by GitHub
parent b26fa1935d
commit 0bbcc7815b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -35,11 +35,11 @@ def _import_bulk() -> Any:
return bulk return bulk
def _get_opensearch_client(opensearch_url: str) -> Any: def _get_opensearch_client(opensearch_url: str, **kwargs: Any) -> Any:
"""Get OpenSearch client from the opensearch_url, otherwise raise error.""" """Get OpenSearch client from the opensearch_url, otherwise raise error."""
try: try:
opensearch = _import_opensearch() opensearch = _import_opensearch()
client = opensearch(opensearch_url) client = opensearch(opensearch_url, **kwargs)
except ValueError as e: except ValueError as e:
raise ValueError( raise ValueError(
f"OpenSearch client string provided is not in proper format. " f"OpenSearch client string provided is not in proper format. "
@ -218,12 +218,16 @@ class OpenSearchVectorSearch(VectorStore):
""" """
def __init__( def __init__(
self, opensearch_url: str, index_name: str, embedding_function: Embeddings self,
opensearch_url: str,
index_name: str,
embedding_function: Embeddings,
**kwargs: Any,
): ):
"""Initialize with necessary components.""" """Initialize with necessary components."""
self.embedding_function = embedding_function self.embedding_function = embedding_function
self.index_name = index_name self.index_name = index_name
self.client = _get_opensearch_client(opensearch_url) self.client = _get_opensearch_client(opensearch_url, **kwargs)
def add_texts( def add_texts(
self, self,

Loading…
Cancel
Save