Przeglądaj źródła

refac: auth required for litellm in dev

Timothy J. Baek 1 rok temu
rodzic
commit
8e52ba8b91
1 zmienionych plików z 6 dodań i 7 usunięć
  1. 6 7
      backend/apps/litellm/main.py

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

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