phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(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(): def test_get_response_streaming():
import asyncio import asyncio
async def test_async_call(): 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"}] messages = [{"content": user_message, "role": "user"}]
try: try:
response = await acompletion(model="azure/chatgpt-v-2", messages=messages, stream=True) 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}") print(f"output: {output}")
assert output is not None, "output cannot be None." assert output is not None, "output cannot be None."
assert isinstance(output, str), "output needs to be of type str" 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: except Exception as e:
pytest.fail(f"An exception occurred: {e}") pytest.fail(f"An exception occurred: {e}")