From fd1f6fb574d313c335723868a9b49bac934db892 Mon Sep 17 00:00:00 2001 From: Bagatur Date: Tue, 26 Nov 2024 15:57:05 -0800 Subject: [PATCH] fmt --- .../tests/integration_tests/chat_models/test_azure.py | 10 ---------- 1 file changed, 10 deletions(-) diff --git a/libs/partners/openai/tests/integration_tests/chat_models/test_azure.py b/libs/partners/openai/tests/integration_tests/chat_models/test_azure.py index 153152b144f..4ed531ad119 100644 --- a/libs/partners/openai/tests/integration_tests/chat_models/test_azure.py +++ b/libs/partners/openai/tests/integration_tests/chat_models/test_azure.py @@ -262,13 +262,3 @@ async def test_json_mode_async(llm: AzureChatOpenAI) -> None: assert isinstance(full, AIMessageChunk) assert isinstance(full.content, str) assert json.loads(full.content) == {"a": 1} - - -def test_o1_max_tokens() -> None: - response = ChatOpenAI(model="o1-mini", max_tokens=10).invoke("how are you") # type: ignore[call-arg] - assert isinstance(response, AIMessage) - - response = ChatOpenAI(model="gpt-4o", max_completion_tokens=10).invoke( - "how are you" - ) - assert isinstance(response, AIMessage)