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: test_router.py

Browse source 
fix router testing

n
This commit is contained in:
Krrish Dholakia 2023-12-07 20:10:36 -08:00
parent a350b80d28
commit dfba305508
2 changed files with 7 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/tests/test_router.py
View file

@ -626,13 +626,14 @@ def test_openai_completion_on_router():
messages=[
{
"role": "user",
"content": "hello from litellm test",
"content": f"hello from litellm test {time.time()}",
}
],
stream=True
)
complete_response = ""
print(response)
# if you want to see all the attributes and methods
async for chunk in response:
print(chunk)
complete_response += chunk.choices[0].delta.content or ""
@ -675,7 +676,7 @@ def test_openai_completion_on_router():
except Exception as e:
traceback.print_exc()
pytest.fail(f"Error occurred: {e}")
# test_openai_completion_on_router()
test_openai_completion_on_router()
def test_reading_keys_os_environ():
@ -742,7 +743,7 @@ def test_reading_keys_os_environ():
traceback.print_exc()
pytest.fail(f"Error occurred: {e}")
test_reading_keys_os_environ()
# test_reading_keys_os_environ()
def test_reading_openai_keys_os_environ():