scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 89

Merge 33f80f3f77 into 67c7b97fd2

Browse source
This commit is contained in:
ryan-crabbe-berri 2026-08-27 16:20:30 -05:00 • committed by GitHub
commit beac3d7622
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
tests/local_testing/test_custom_logger.py
View file

@ -211,7 +211,9 @@ def test_async_custom_handler_stream():
async def test_1():
nonlocal complete_streaming_response
response = await litellm.acompletion(
model="azure/gpt-4.1-mini", messages=messages, stream=True
model=f"azure/{os.environ.get('CI_CD_DEFAULT_AZURE_MODEL', 'gpt-4.1-mini')}",
messages=messages,
stream=True,
)
async for chunk in response:
complete_streaming_response += (
@ -255,7 +257,9 @@ def test_azure_completion_stream():
complete_streaming_response = ""
response = litellm.completion(
model="azure/gpt-4.1-mini", messages=messages, stream=True
model=f"azure/{os.environ.get('CI_CD_DEFAULT_AZURE_MODEL', 'gpt-4.1-mini')}",
messages=messages,
stream=True,
)
for chunk in response:
complete_streaming_response += chunk["choices"][0]["delta"]["content"] or ""