From 7091a1a798b6600e693429206e96fba48cf4aaaa Mon Sep 17 00:00:00 2001 From: ccurme Date: Thu, 26 Sep 2024 10:31:33 -0400 Subject: [PATCH] openai[patch]: increase token limit in azure integration tests (#26901) `test_json_mode` occasionally runs into this --- .../openai/tests/integration_tests/chat_models/test_azure.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 b62e9dfdd1a..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 @@ -39,7 +39,7 @@ def _get_llm(**kwargs: Any) -> AzureChatOpenAI: @pytest.mark.scheduled @pytest.fixture def llm() -> AzureChatOpenAI: - return _get_llm(max_tokens=10) + return _get_llm(max_tokens=50) def test_chat_openai(llm: AzureChatOpenAI) -> None: