github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-05 21:12:48 +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
1 changed files with 3 additions and 3 deletions
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
if es_connection is not None:
self.client = es_connection.options(
headers={"user-agent": self.get_user_agent()}
)
headers = dict(es_connection._headers)
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:
self.client = ElasticsearchStore.connect_to_elasticsearch(
es_url=es_url,