From e53e91bcb2be6ee39fb033ec424e6d307b02aba3 Mon Sep 17 00:00:00 2001 From: Chester Curme Date: Wed, 29 Oct 2025 10:50:25 -0400 Subject: [PATCH] update openai --- libs/partners/openai/langchain_openai/chat_models/base.py | 6 ++---- .../chat_models/__snapshots__/test_azure_standard.ambr | 1 + .../chat_models/__snapshots__/test_base_standard.ambr | 1 + .../chat_models/__snapshots__/test_responses_standard.ambr | 1 + 4 files changed, 5 insertions(+), 4 deletions(-) diff --git a/libs/partners/openai/langchain_openai/chat_models/base.py b/libs/partners/openai/langchain_openai/chat_models/base.py index 35beb08ec5f..873bd9ae8c3 100644 --- a/libs/partners/openai/langchain_openai/chat_models/base.py +++ b/libs/partners/openai/langchain_openai/chat_models/base.py @@ -550,6 +550,8 @@ class BaseChatOpenAI(BaseChatModel): !!! warning "Behavior changed in 0.3.35" Enabled for default base URL and client. """ + model_provider: str | None = "openai" + """THe model provider name (openai).""" max_retries: int | None = None """Maximum number of retries to make when generating.""" presence_penalty: float | None = None @@ -1034,7 +1036,6 @@ class BaseChatOpenAI(BaseChatModel): if usage_metadata and isinstance(message_chunk, AIMessageChunk): message_chunk.usage_metadata = usage_metadata - message_chunk.response_metadata["model_provider"] = "openai" return ChatGenerationChunk( message=message_chunk, generation_info=generation_info or None ) @@ -1398,7 +1399,6 @@ class BaseChatOpenAI(BaseChatModel): generations.append(gen) llm_output = { "token_usage": token_usage, - "model_provider": "openai", "model_name": response_dict.get("model", self.model_name), "system_fingerprint": response_dict.get("system_fingerprint", ""), } @@ -4081,7 +4081,6 @@ def _construct_lc_result_from_responses_api( if metadata: response_metadata.update(metadata) # for compatibility with chat completion calls. - response_metadata["model_provider"] = "openai" response_metadata["model_name"] = response_metadata.get("model") if response.usage: usage_metadata = _create_usage_metadata_responses( @@ -4277,7 +4276,6 @@ def _convert_responses_chunk_to_generation_chunk( tool_call_chunks: list = [] additional_kwargs: dict = {} response_metadata = metadata or {} - response_metadata["model_provider"] = "openai" usage_metadata = None chunk_position: Literal["last"] | None = None id = None diff --git a/libs/partners/openai/tests/unit_tests/chat_models/__snapshots__/test_azure_standard.ambr b/libs/partners/openai/tests/unit_tests/chat_models/__snapshots__/test_azure_standard.ambr index 3faf95d352f..4bdbb12debd 100644 --- a/libs/partners/openai/tests/unit_tests/chat_models/__snapshots__/test_azure_standard.ambr +++ b/libs/partners/openai/tests/unit_tests/chat_models/__snapshots__/test_azure_standard.ambr @@ -15,6 +15,7 @@ }), 'max_retries': 2, 'max_tokens': 100, + 'model_provider': 'openai', 'openai_api_key': dict({ 'id': list([ 'AZURE_OPENAI_API_KEY', 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 223d5418075..1b4de63ab42 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', + 'model_provider': 'openai', '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 88ea956d7fb..12eab029f55 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', + 'model_provider': 'openai', 'openai_api_key': dict({ 'id': list([ 'OPENAI_API_KEY',