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

(testing) Router add testing coverage (#6253)

Browse source 
* test: add more router code coverage

* test: additional router testing coverage

* fix: fix linting error

* test: fix tests for ci/cd

* test: fix test

* test: handle flaky tests

---------

Co-authored-by: Krrish Dholakia <krrishdholakia@gmail.com>
This commit is contained in:
Ishaan Jaff 2024-10-16 20:02:27 +05:30 • committed by GitHub
parent b72a47d092
commit dee6de0105
7 changed files with 706 additions and 106 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/router_utils/router_callbacks/track_deployment_metrics.py
View file

@ -24,7 +24,7 @@ else:
def increment_deployment_successes_for_current_minute(
litellm_router_instance: LitellmRouter,
deployment_id: str,
):
) -> str:
"""
In-Memory: Increments the number of successes for the current minute for a deployment_id
"""
@ -35,6 +35,7 @@ def increment_deployment_successes_for_current_minute(
value=1,
ttl=60,
)
return key
def increment_deployment_failures_for_current_minute(