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

Users can provide an Elasticsearch connection with custom headers. This
PR makes sure these headers are preserved when adding the langchain user
agent header.
pull/17479/head
Max Jakob 8 months ago committed by GitHub
parent 112e10e933
commit ab3d944667
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -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,

Loading…
Cancel
Save