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

test(test_openai_endpoints.py): make test stricter

Browse source
This commit is contained in:
Krrish Dholakia 2024-04-20 12:11:54 -07:00
parent 47e9d5f2ec
commit 1507b23e30
2 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
proxy_server_config.yaml
View file

@ -96,9 +96,9 @@ litellm_settings:
router_settings: router_settings:
routing_strategy: usage-based-routing-v2 routing_strategy: usage-based-routing-v2
redis_host: os.environ/REDIS_HOST # redis_host: os.environ/REDIS_HOST
redis_password: os.environ/REDIS_PASSWORD # redis_password: os.environ/REDIS_PASSWORD
redis_port: os.environ/REDIS_PORT # redis_port: os.environ/REDIS_PORT
enable_pre_call_checks: true enable_pre_call_checks: true
general_settings: general_settings:

5
tests/test_openai_endpoints.py
View file

@ -260,7 +260,10 @@ async def test_chat_completion_ratelimit():
await asyncio.gather(*tasks) await asyncio.gather(*tasks)
pytest.fail("Expected at least 1 call to fail") pytest.fail("Expected at least 1 call to fail")
except Exception as e: except Exception as e:
pass if "Request did not return a 200 status code: 429" in str(e):
pass
else:
pytest.fail(f"Wrong error received - {str(e)}")
@pytest.mark.asyncio @pytest.mark.asyncio