From ddb38d388ca9e1b1fe94e28c3b7317dc8d35b52e Mon Sep 17 00:00:00 2001 From: "open-swe[bot]" Date: Mon, 15 Sep 2025 14:19:37 +0000 Subject: [PATCH] Apply patch [skip ci] --- .../langchain_openai/chat_models/azure.py | 1 - .../unit_tests/chat_models/test_azure.py | 19 ++++++++----------- 2 files changed, 8 insertions(+), 12 deletions(-) diff --git a/libs/partners/openai/langchain_openai/chat_models/azure.py b/libs/partners/openai/langchain_openai/chat_models/azure.py index 5aa42e5c1d7..128de9954a6 100644 --- a/libs/partners/openai/langchain_openai/chat_models/azure.py +++ b/libs/partners/openai/langchain_openai/chat_models/azure.py @@ -1168,4 +1168,3 @@ class AzureChatOpenAI(BaseChatOpenAI): return super().with_structured_output( schema, method=method, include_raw=include_raw, strict=strict, **kwargs ) - diff --git a/libs/partners/openai/tests/unit_tests/chat_models/test_azure.py b/libs/partners/openai/tests/unit_tests/chat_models/test_azure.py index 10c044d8d9b..97134831e11 100644 --- a/libs/partners/openai/tests/unit_tests/chat_models/test_azure.py +++ b/libs/partners/openai/tests/unit_tests/chat_models/test_azure.py @@ -113,7 +113,7 @@ def test_max_tokens_and_max_completion_tokens_parameters() -> None: api_key="test-key", max_tokens=100, ) - + # Test max_completion_tokens parameter (using alias) llm_max_completion_tokens = AzureChatOpenAI( azure_deployment="gpt-35-turbo", @@ -122,16 +122,18 @@ def test_max_tokens_and_max_completion_tokens_parameters() -> None: api_key="test-key", max_completion_tokens=100, ) - + # Both should have the same max_tokens value assert llm_max_tokens.max_tokens == 100 assert llm_max_completion_tokens.max_tokens == 100 - + # Test that both produce the same payload messages = [HumanMessage("Hello")] payload_max_tokens = llm_max_tokens._get_request_payload(messages) - payload_max_completion_tokens = llm_max_completion_tokens._get_request_payload(messages) - + payload_max_completion_tokens = llm_max_completion_tokens._get_request_payload( + messages + ) + # Both payloads should be identical and contain max_tokens # (AzureChatOpenAI doesn't have the conversion logic that ChatOpenAI has) expected_payload = { @@ -140,7 +142,7 @@ def test_max_tokens_and_max_completion_tokens_parameters() -> None: "stream": False, "max_tokens": 100, } - + assert payload_max_tokens == expected_payload assert payload_max_completion_tokens == expected_payload assert payload_max_tokens == payload_max_completion_tokens @@ -183,8 +185,3 @@ def test_chat_completions_api_uses_model_name() -> None: assert payload["model"] == "gpt-5" assert "messages" in payload # Chat Completions API uses 'messages' assert "input" not in payload - - - - -