github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-22 14:49:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[patch]: ElasticsearchStore: preserve user headers (#16830)

Browse Source 
Users can provide an Elasticsearch connection with custom headers. This
PR makes sure these headers are preserved when adding the langchain user
agent header.
This commit is contained in:
Max Jakob 2024-02-13 21:37:35 +01:00 committed by GitHub
parent 112e10e933
commit ab3d944667
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libs/community/langchain_community/vectorstores/elasticsearch.py
View File

@ -528,9 +528,9 @@ class ElasticsearchStore(VectorStore):
self.strategy = strategy self.strategy = strategy
if es_connection is not None: if es_connection is not None:
self.client = es_connection.options( headers = dict(es_connection._headers)
headers={"user-agent": self.get_user_agent()} headers.update({"user-agent": self.get_user_agent()})
) self.client = es_connection.options(headers=headers)
elif es_url is not None or es_cloud_id is not None: elif es_url is not None or es_cloud_id is not None:
self.client = ElasticsearchStore.connect_to_elasticsearch( self.client = ElasticsearchStore.connect_to_elasticsearch(
es_url=es_url, es_url=es_url,