openai[patch]: increase token limit in azure integration tests (#26901)

`test_json_mode` occasionally runs into this
This commit is contained in:
ccurme 2024-09-26 10:31:33 -04:00 committed by GitHub
parent 2ea5f60cc5
commit 7091a1a798
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -39,7 +39,7 @@ def _get_llm(**kwargs: Any) -> AzureChatOpenAI:
@pytest.mark.scheduled @pytest.mark.scheduled
@pytest.fixture @pytest.fixture
def llm() -> AzureChatOpenAI: def llm() -> AzureChatOpenAI:
return _get_llm(max_tokens=10) return _get_llm(max_tokens=50)
def test_chat_openai(llm: AzureChatOpenAI) -> None: def test_chat_openai(llm: AzureChatOpenAI) -> None: