From 86f028fce0a8891abe3a83802a320ea9250182ea Mon Sep 17 00:00:00 2001 From: Chester Curme Date: Thu, 12 Sep 2024 11:23:39 -0400 Subject: [PATCH] fix integration test --- .../tests/integration_tests/test_retrievers.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/libs/elasticsearch/tests/integration_tests/test_retrievers.py b/libs/elasticsearch/tests/integration_tests/test_retrievers.py index 4775f88..15d0307 100644 --- a/libs/elasticsearch/tests/integration_tests/test_retrievers.py +++ b/libs/elasticsearch/tests/integration_tests/test_retrievers.py @@ -58,7 +58,9 @@ def test_user_agent_header(self, es_client: Elasticsearch, index_name: str) -> N assert retriever.es_client user_agent = retriever.es_client._headers["User-Agent"] assert ( - re.match(r"^langchain-py-r/\d+\.\d+\.\d+(?:rc\d+)?$", user_agent) + re.match( + r"^langchain-py-r/\d+\.\d+\.\d+(?:rc\d+)?(?:\.dev\d+)?$", user_agent + ) is not None ), f"The string '{user_agent}' does not match the expected pattern." @@ -68,7 +70,9 @@ def test_user_agent_header(self, es_client: Elasticsearch, index_name: str) -> N search_request = es_client.transport.requests[-1] # type: ignore[attr-defined] user_agent = search_request["headers"]["User-Agent"] assert ( - re.match(r"^langchain-py-r/\d+\.\d+\.\d+(?:rc\d+)?$", user_agent) + re.match( + r"^langchain-py-r/\d+\.\d+\.\d+(?:rc\d+)?(?:\.dev\d+)?$", user_agent + ) is not None ), f"The string '{user_agent}' does not match the expected pattern."