Merge branch 'BerriAI:main' into main

This commit is contained in:
Utkash Dubey 2025-03-10 14:42:30 -07:00 committed by GitHub
commit 1f17daf52c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
194 changed files with 6724 additions and 2181 deletions

View file

@ -392,3 +392,25 @@ def test_router_redis_cache():
router._update_redis_cache(cache=redis_cache)
assert router.cache.redis_cache == redis_cache
def test_router_handle_clientside_credential():
deployment = {
"model_name": "gemini/*",
"litellm_params": {"model": "gemini/*"},
"model_info": {
"id": "1",
},
}
router = Router(model_list=[deployment])
new_deployment = router._handle_clientside_credential(
deployment=deployment,
kwargs={
"api_key": "123",
"metadata": {"model_group": "gemini/gemini-1.5-flash"},
},
)
assert new_deployment.litellm_params.api_key == "123"
assert len(router.get_model_list()) == 2