phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

Merge pull request #5260 from BerriAI/google_ai_studio_pass_through

Browse source 
Pass-through endpoints for Gemini - Google AI Studio
This commit is contained in:
Krish Dholakia 2024-08-17 13:51:51 -07:00 • committed by GitHub
commit ff6ff133ee
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
9 changed files with 479 additions and 31 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/proxy_server.py
View file

@ -230,6 +230,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 (
@ -9734,6 +9737,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)