phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 03:34:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(utils.py): return last streaming chunk

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-29 12:11:08 -08:00
parent ff865ab2c5
commit 94de89c812
2 changed files with 8 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

14
litellm/tests/test_custom_logger.py
View file

@ -37,15 +37,7 @@ def test_chat_openai():
"role": "user",
"content": "Hi 👋 - i'm openai"
}],
stream=True,
complete_response = True)
response2 = completion(model="gpt-3.5-turbo",
messages=[{
"role": "user",
"content": "Hi 👋 - i'm not openai"
}],
stream=True,
complete_response = True)
stream=True)
time.sleep(1)
assert customHandler.success == True
except Exception as e:
@ -53,7 +45,7 @@ def test_chat_openai():
pass
# test_chat_openai()
test_chat_openai()
def test_completion_azure_stream_moderation_failure():
try:
@ -80,7 +72,7 @@ def test_completion_azure_stream_moderation_failure():
except Exception as e:
pytest.fail(f"Error occurred: {e}")
test_completion_azure_stream_moderation_failure()
# test_completion_azure_stream_moderation_failure()
# def custom_callback(