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

elasticsearch[patch]: fix integration tests for release (#18980)

Browse Source
This commit is contained in:
Erick Friis 2024-03-12 10:22:07 -07:00 committed by GitHub
parent 0d1f6c417c
commit a29e8d8594
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
libs/partners/elasticsearch/tests/integration_tests/test_retrievers.py
View File

@ -1,5 +1,6 @@
"""Test ElasticsearchRetriever functionality.""" """Test ElasticsearchRetriever functionality."""
import os
import re import re
import uuid import uuid
from typing import Any, Dict from typing import Any, Dict
@ -77,11 +78,19 @@ class TestElasticsearchRetriever:
def body_func(query: str) -> Dict: def body_func(query: str) -> Dict:
return {"query": {"match": {text_field: {"query": query}}}} return {"query": {"match": {text_field: {"query": query}}}}
es_url = os.environ.get("ES_URL", "http://localhost:9200")
cloud_id = os.environ.get("ES_CLOUD_ID")
api_key = os.environ.get("ES_API_KEY")
config = (
{"cloud_id": cloud_id, "api_key": api_key} if cloud_id else {"url": es_url}
)
retriever = ElasticsearchRetriever.from_es_params( retriever = ElasticsearchRetriever.from_es_params(
url="http://localhost:9200",
index_name=index_name, index_name=index_name,
body_func=body_func, body_func=body_func,
content_field=text_field, content_field=text_field,
**config, # type: ignore[arg-type]
) )
index_test_data(retriever.es_client, index_name, text_field) index_test_data(retriever.es_client, index_name, text_field)