github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-18 18:53:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Patch enable to use Amazon OpenSearch Serverless(aoss) for Semantic Cache store (#25833)

Browse Source 
- [x] **PR title**: "community: Patch enable to use Amazon OpenSearch
Serverless for Semantic Cache store"

- [x] **PR message**: 
- **Description:** OpenSearchSemanticCache class support Amazon
OpenSearch Serverless for Semantic Cache store, it's only required to
pass auth(http_auth) parameter to initializer
    - **Dependencies:** none

If no one reviews your PR within a few days, please @-mention one of
baskaryan, efriis, eyurtsev, ccurme, vbarda, hwchase17.

---------

Co-authored-by: Jinoos Lee <jinoos@amazon.com>
This commit is contained in:
Jinoos Lee 2024-08-29 22:28:22 +09:00 committed by GitHub
parent 1ad621120d
commit 703af9ffe3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
libs/community/langchain_community/cache.py
View File

@ -2279,7 +2279,11 @@ class OpenSearchSemanticCache(BaseCache):
"""Cache that uses OpenSearch vector store backend""" """Cache that uses OpenSearch vector store backend"""
def __init__( def __init__(
self, opensearch_url: str, embedding: Embeddings, score_threshold: float = 0.2 self,
opensearch_url: str,
embedding: Embeddings,
score_threshold: float = 0.2,
**kwargs: Any,
): ):
""" """
Args: Args:
@ -2300,6 +2304,7 @@ class OpenSearchSemanticCache(BaseCache):
self.opensearch_url = opensearch_url self.opensearch_url = opensearch_url
self.embedding = embedding self.embedding = embedding
self.score_threshold = score_threshold self.score_threshold = score_threshold
self.connection_kwargs = kwargs
def _index_name(self, llm_string: str) -> str: def _index_name(self, llm_string: str) -> str:
hashed_index = _hash(llm_string) hashed_index = _hash(llm_string)
@ -2317,6 +2322,7 @@ class OpenSearchSemanticCache(BaseCache):
opensearch_url=self.opensearch_url, opensearch_url=self.opensearch_url,
index_name=index_name, index_name=index_name,
embedding_function=self.embedding, embedding_function=self.embedding,
**self.connection_kwargs,
) )
# create index for the vectorstore # create index for the vectorstore