From 0bbcc7815bf7a5b64efd39951ed2355d682790c7 Mon Sep 17 00:00:00 2001 From: Harrison Chase Date: Mon, 27 Mar 2023 07:56:09 -0700 Subject: [PATCH] Harrison/open search kwargs (#2040) Signed-off-by: Marcel Coetzee Co-authored-by: Marcel <34739235+Pipboyguy@users.noreply.github.com> --- langchain/vectorstores/opensearch_vector_search.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/langchain/vectorstores/opensearch_vector_search.py b/langchain/vectorstores/opensearch_vector_search.py index 4e1d7357..b387bc71 100644 --- a/langchain/vectorstores/opensearch_vector_search.py +++ b/langchain/vectorstores/opensearch_vector_search.py @@ -35,11 +35,11 @@ def _import_bulk() -> Any: 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.""" try: opensearch = _import_opensearch() - client = opensearch(opensearch_url) + client = opensearch(opensearch_url, **kwargs) except ValueError as e: raise ValueError( f"OpenSearch client string provided is not in proper format. " @@ -218,12 +218,16 @@ class OpenSearchVectorSearch(VectorStore): """ 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.""" self.embedding_function = embedding_function self.index_name = index_name - self.client = _get_opensearch_client(opensearch_url) + self.client = _get_opensearch_client(opensearch_url, **kwargs) def add_texts( self,