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

test: refactor testing order

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-24 12:47:28 -08:00
parent 2a033fd8a2
commit 16e1070dbe
4 changed files with 45 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/router.py
View file

@ -761,4 +761,11 @@ class Router:
raise ValueError("No models available.")
def flush_cache(self):
self.cache.flush_cache()
self.cache.flush_cache()
def reset(self):
## clean up on close
litellm.success_callback = []
litellm.failure_callback = []
self.flush_cache()