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

cleanup async func test

Browse source
This commit is contained in:
ishaan-jaff 2023-09-07 13:56:46 -07:00
parent a611409e0f
commit 301eb36095
1 changed files with 2 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

16
litellm/tests/test_async_fn.py
View file

@ -44,8 +44,8 @@ async def test_get_response_streaming():
output += token output += token
print(output) print(output)
assert output is not None, "Agent output cannot be None." assert output is not None, "output cannot be None."
assert isinstance(output, str), "Agent 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, "Length of output needs to be greater than 0."
except Exception as e: except Exception as e:
@ -56,15 +56,3 @@ async def test_get_response_streaming():
# print(response) # print(response)
# async def test_get_response():
# user_message = "Hello, how are you?"
# messages = [{"content": user_message, "role": "user"}]
# try:
# response = await acreate(model="gpt-3.5-turbo", messages=messages)
# except Exception as e:
# pytest.fail(f"error occurred: {e}")
# return response
# response = asyncio.run(test_get_response())
# print(response)