Skip to content

Commit 5249655

Browse files
ccurmeshinxi
authored andcommitted
openai[patch]: fix Azure LLM test (#29302)
The tokens I get are: ``` ['', '\n\n', 'The', ' sun', ' was', ' setting', ' over', ' the', ' horizon', ',', ' casting', ''] ``` so possibly an extra empty token is included in the output. lmk @efriis if we should look into this further.
1 parent c42af7d commit 5249655

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
@@ -148,7 +148,7 @@ def test_openai_streaming_callback() -> None:
148148
verbose=True,
149149
)
150150
llm.invoke("Write me a sentence with 100 words.")
151-
assert callback_handler.llm_streams == 11
151+
assert callback_handler.llm_streams == 12
152152

153153

154154
@pytest.mark.scheduled
@@ -171,5 +171,5 @@ async def test_openai_async_streaming_callback() -> None:
171171
verbose=True,
172172
)
173173
result = await llm.agenerate(["Write me a sentence with 100 words."])
174-
assert callback_handler.llm_streams == 11
174+
assert callback_handler.llm_streams == 12
175175
assert isinstance(result, LLMResult)

0 commit comments

Comments
 (0)