diff --git a/libs/partners/openai/tests/integration_tests/chat_models/test_azure_standard.py b/libs/partners/openai/tests/integration_tests/chat_models/test_azure_standard.py index 8cddaac726e30..41c20a942ac52 100644 --- a/libs/partners/openai/tests/integration_tests/chat_models/test_azure_standard.py +++ b/libs/partners/openai/tests/integration_tests/chat_models/test_azure_standard.py @@ -3,6 +3,7 @@ import os from typing import Type +import pytest from langchain_core.language_models import BaseChatModel from langchain_standard_tests.integration_tests import ChatModelIntegrationTests @@ -30,3 +31,7 @@ def chat_model_params(self) -> dict: "azure_endpoint": OPENAI_API_BASE, "api_key": OPENAI_API_KEY, } + + @pytest.mark.xfail(reason="Not yet supported.") + def test_usage_metadata_streaming(self, model: BaseChatModel) -> None: + super().test_usage_metadata_streaming(model) diff --git a/libs/partners/openai/tests/integration_tests/chat_models/test_base_standard.py b/libs/partners/openai/tests/integration_tests/chat_models/test_base_standard.py index 0227a07f78739..888998b237191 100644 --- a/libs/partners/openai/tests/integration_tests/chat_models/test_base_standard.py +++ b/libs/partners/openai/tests/integration_tests/chat_models/test_base_standard.py @@ -18,7 +18,7 @@ def chat_model_class(self) -> Type[BaseChatModel]: @property def chat_model_params(self) -> dict: - return {"model": "gpt-4o"} + return {"model": "gpt-4o", "stream_usage": True} @property def supports_image_inputs(self) -> bool: