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

feat(google_ai_studio_endpoints.py): support pass-through endpoint for all google ai studio requests

Browse source 
New Feature
This commit is contained in:
Krrish Dholakia 2024-08-17 10:46:59 -07:00
parent 668ea6cbc7
commit 29bedae79f
6 changed files with 186 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/proxy_server.py
View file

@ -227,6 +227,9 @@ from litellm.proxy.utils import (
send_email,
update_spend,
)
from litellm.proxy.vertex_ai_endpoints.google_ai_studio_endpoints import (
router as gemini_router,
)
from litellm.proxy.vertex_ai_endpoints.vertex_endpoints import router as vertex_router
from litellm.proxy.vertex_ai_endpoints.vertex_endpoints import set_default_vertex_config
from litellm.router import (
@ -9704,6 +9707,7 @@ def cleanup_router_config_variables():
app.include_router(router)
app.include_router(fine_tuning_router)
app.include_router(vertex_router)
app.include_router(gemini_router)
app.include_router(pass_through_router)
app.include_router(health_router)
app.include_router(key_management_router)