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

(fix) use dynamic redirect urls

Browse source
This commit is contained in:
ishaan-jaff 2024-01-27 15:13:43 -08:00
parent 30d4ec9221
commit 378ab212c1
1 changed files with 11 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

15
litellm/proxy/proxy_server.py
View file

@ -2854,8 +2854,11 @@ async def user_auth(request: Request):
@app.get("/google-login", tags=["experimental"])
async def google_login():
GOOGLE_REDIRECT_URI = "http://localhost:4000/google-callback"
async def google_login(request: Request):
scheme = request.url.scheme
host = request.url.hostname
port = request.url.port or 4000
GOOGLE_REDIRECT_URI = f"{scheme}://{host}:{port}/google-callback"
GOOGLE_CLIENT_ID = (
"246483686424-clje5sggkjma26ilktj6qssakqhoon0m.apps.googleusercontent.com"
)
@ -2864,10 +2867,14 @@ async def google_login():
@app.get("/google-callback", tags=["experimental"], response_model=GenerateKeyResponse)
async def google_callback(code: str):
async def google_callback(code: str, request: Request):
import httpx
GOOGLE_REDIRECT_URI = "http://localhost:4000/google-callback"
scheme = request.url.scheme
host = request.url.hostname
port = request.url.port or 4000
GOOGLE_REDIRECT_URI = f"{scheme}://{host}:{port}/google-callback"
GOOGLE_CLIENT_ID = (
"246483686424-clje5sggkjma26ilktj6qssakqhoon0m.apps.googleusercontent.com"
)