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

test(test_rules.py): fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-05-11 21:22:37 -07:00
parent a456f6bf2b
commit e8437e52fa
1 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/tests/test_rules.py
View file

@ -132,12 +132,15 @@ def test_post_call_rule_streaming():
) )
def test_post_call_processing_error_async_response(): @pytest.mark.asyncio
response = asyncio.run( async def test_post_call_processing_error_async_response():
acompletion( try:
response = await acompletion(
model="command-nightly", # Just used as an example model="command-nightly", # Just used as an example
messages=[{"content": "Hello, how are you?", "role": "user"}], messages=[{"content": "Hello, how are you?", "role": "user"}],
api_base="https://openai-proxy.berriai.repl.co", # Just used as an example api_base="https://openai-proxy.berriai.repl.co", # Just used as an example
custom_llm_provider="openai", custom_llm_provider="openai",
) )
) pytest.fail("This call should have failed")
except Exception as e:
pass