Skip to content

Commit 851fd43

Browse files
authored
openai[patch]: relax Azure llm streaming callback test (#31319)
Effectively reverts #29302, but check that counts are "less than" instead of equal to an expected count.
1 parent 092697d commit 851fd43

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

libs/partners/openai/tests/integration_tests/llms/test_azure.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ def test_openai_streaming_callback() -> None:
149149
verbose=True,
150150
)
151151
llm.invoke("Write me a sentence with 100 words.")
152-
assert callback_handler.llm_streams == 12
152+
assert callback_handler.llm_streams < 15
153153

154154

155155
@pytest.mark.scheduled
@@ -172,5 +172,5 @@ async def test_openai_async_streaming_callback() -> None:
172172
verbose=True,
173173
)
174174
result = await llm.agenerate(["Write me a sentence with 100 words."])
175-
assert callback_handler.llm_streams == 12
175+
assert callback_handler.llm_streams < 15
176176
assert isinstance(result, LLMResult)

0 commit comments

Comments
 (0)