tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-03 12:24:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

test_async_custom_handler_stream

Browse Source
This commit is contained in:
Ishaan Jaffer
2025-09-27 12:37:56 -07:00
parent 02cc9133a5
commit 9a35d760cf
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/local_testing/test_custom_logger.py
+3 -3
View File
@@ -160,7 +160,7 @@ def test_completion_azure_stream_moderation_failure():
]
try:
response = completion(
model="azure/chatgpt-v-3",
model="azure/gpt-4.1-nano",
messages=messages,
mock_response="Exception: content_filter_policy",
stream=True,
@@ -195,7 +195,7 @@ def test_async_custom_handler_stream():
async def test_1():
nonlocal complete_streaming_response
response = await litellm.acompletion(
model="azure/chatgpt-v-3", messages=messages, stream=True
model="azure/gpt-4.1-nano", messages=messages, stream=True
)
async for chunk in response:
complete_streaming_response += (
@@ -239,7 +239,7 @@ def test_azure_completion_stream():
complete_streaming_response = ""
response = litellm.completion(
model="azure/chatgpt-v-3", messages=messages, stream=True
model="azure/gpt-4.1-nano", messages=messages, stream=True
)
for chunk in response:
complete_streaming_response += chunk["choices"][0]["delta"]["content"] or ""