refac: auth required for litellm in dev

8e52ba8b · Timothy J. Baek · ed0d4b5a · 8e52ba8b
Commit 8e52ba8b authored Mar 20, 2024 by Timothy J. Baek
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 7 deletions

backend/apps/litellm/main.py backend/apps/litellm/main.py +6 -7

No files found.
--- a/backend/apps/litellm/main.py
+++ b/backend/apps/litellm/main.py
@@ -47,13 +47,12 @@ async def auth_middleware(request: Request, call_next):
    auth_header = request.headers.get("Authorization", "")
    request.state.user = None
-    if ENV != "dev":
+    try:
-        try:
+        user = get_current_user(get_http_authorization_cred(auth_header))
-            user = get_current_user(get_http_authorization_cred(auth_header))
+        print(user)
-            print(user)
+        request.state.user = user
-            request.state.user = user
+    except Exception as e:
-        except Exception as e:
+        return JSONResponse(status_code=400, content={"detail": str(e)})
-            return JSONResponse(status_code=400, content={"detail": str(e)})
    response = await call_next(request)
    return response