From f966cce26ab660293c673293561d91e40f1c04fc Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Tue, 30 Jan 2024 16:47:12 -0800 Subject: [PATCH] (feat) add litellm login to proxy --- litellm/proxy/proxy_server.py | 78 +++++++++++++++++++++++++++++++++++ 1 file changed, 78 insertions(+) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index f1ec2744c..3ed91684a 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -3103,6 +3103,84 @@ async def user_info( ) +html_form = """ + + + + LiteLLM Login + + + +
+

LiteLLM Login

+ + + + + +
+ + +""" +from fastapi import FastAPI, Form +from fastapi.responses import HTMLResponse + + +@router.get("/login/page") +async def login_page(): + return HTMLResponse(content=html_form, status_code=200) + + +@router.get("/login") +async def login(username: str = Form(...), password: str = Form(...)): + # Here you can perform authentication logic + # For simplicity, let's just print the received credentials + # print(f"Received username: {username}, password: {password}") + return {"message": "Login successful"} + + @router.post( "/user/update", tags=["user management"], dependencies=[Depends(user_api_key_auth)] )