github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-05-21 15:07:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

openai[patch]: fix Azure LLM test (#29302)

Browse Source 
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.
This commit is contained in:
ccurme 2025-01-19 12:25:42 -05:00 committed by GitHub
parent 6b249a0dc2
commit c20f7418c7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
libs/partners/openai/tests/integration_tests/llms/test_azure.py
View File

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