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

(test) async_fn with stream

Browse source
This commit is contained in:
ishaan-jaff 2023-11-10 17:47:15 -08:00
parent c2c186eb28
commit 9d3a28e391
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/test_async_fn.py
View file

@ -42,7 +42,7 @@ def test_async_response():
def test_get_response_streaming():
import asyncio
async def test_async_call():
user_message = "Hello, how are you?"
user_message = "write a short poem in one sentence"
messages = [{"content": user_message, "role": "user"}]
try:
response = await acompletion(model="azure/chatgpt-v-2", messages=messages, stream=True)
@ -62,7 +62,7 @@ def test_get_response_streaming():
print(f"output: {output}")
assert output is not None, "output cannot be None."
assert isinstance(output, str), "output needs to be of type str"
assert len(output) > 0, "Length of output needs to be greater than 0."
assert len(output) > 0, f"Length of output needs to be greater than 0. {output}"
except Exception as e:
pytest.fail(f"An exception occurred: {e}")