From 6e2d4f4273d28e85c643ecdae73d855fa314fd32 Mon Sep 17 00:00:00 2001 From: ccurme Date: Mon, 17 Aug 2026 13:23:00 -0400 Subject: [PATCH] chore(openai): update snapshots (#39657) --- .../chat_models/__snapshots__/test_base_standard.ambr | 1 + .../chat_models/__snapshots__/test_responses_standard.ambr | 1 + .../tests/unit_tests/chat_models/test_base_standard.py | 7 +++++++ .../unit_tests/chat_models/test_responses_standard.py | 6 +++++- 4 files changed, 14 insertions(+), 1 deletion(-) diff --git a/libs/partners/openai/tests/unit_tests/chat_models/__snapshots__/test_base_standard.ambr b/libs/partners/openai/tests/unit_tests/chat_models/__snapshots__/test_base_standard.ambr index 223d541807..5bbc93b27d 100644 --- a/libs/partners/openai/tests/unit_tests/chat_models/__snapshots__/test_base_standard.ambr +++ b/libs/partners/openai/tests/unit_tests/chat_models/__snapshots__/test_base_standard.ambr @@ -11,6 +11,7 @@ 'max_retries': 2, 'max_tokens': 100, 'model_name': 'gpt-3.5-turbo', + 'openai_api_base': 'https://api.openai.com/v1', 'openai_api_key': dict({ 'id': list([ 'OPENAI_API_KEY', diff --git a/libs/partners/openai/tests/unit_tests/chat_models/__snapshots__/test_responses_standard.ambr b/libs/partners/openai/tests/unit_tests/chat_models/__snapshots__/test_responses_standard.ambr index 88ea956d7f..b0f1da1a37 100644 --- a/libs/partners/openai/tests/unit_tests/chat_models/__snapshots__/test_responses_standard.ambr +++ b/libs/partners/openai/tests/unit_tests/chat_models/__snapshots__/test_responses_standard.ambr @@ -11,6 +11,7 @@ 'max_retries': 2, 'max_tokens': 100, 'model_name': 'gpt-3.5-turbo', + 'openai_api_base': 'https://api.openai.com/v1', 'openai_api_key': dict({ 'id': list([ 'OPENAI_API_KEY', diff --git a/libs/partners/openai/tests/unit_tests/chat_models/test_base_standard.py b/libs/partners/openai/tests/unit_tests/chat_models/test_base_standard.py index cec64d9f59..e7041aac67 100644 --- a/libs/partners/openai/tests/unit_tests/chat_models/test_base_standard.py +++ b/libs/partners/openai/tests/unit_tests/chat_models/test_base_standard.py @@ -11,6 +11,13 @@ class TestOpenAIStandard(ChatModelUnitTests): def chat_model_class(self) -> type[BaseChatModel]: return ChatOpenAI + @property + def chat_model_params(self) -> dict: + return { + "base_url": "https://api.openai.com/v1", + "stream_usage": True, + } + @property def init_from_env_params(self) -> tuple[dict, dict, dict]: return ( diff --git a/libs/partners/openai/tests/unit_tests/chat_models/test_responses_standard.py b/libs/partners/openai/tests/unit_tests/chat_models/test_responses_standard.py index 0f460a4d24..23ae442ca3 100644 --- a/libs/partners/openai/tests/unit_tests/chat_models/test_responses_standard.py +++ b/libs/partners/openai/tests/unit_tests/chat_models/test_responses_standard.py @@ -13,7 +13,11 @@ class TestOpenAIResponses(ChatModelUnitTests): @property def chat_model_params(self) -> dict: - return {"use_responses_api": True} + return { + "use_responses_api": True, + "base_url": "https://api.openai.com/v1", + "stream_usage": True, + } @property def init_from_env_params(self) -> tuple[dict, dict, dict]: