Prechádzať zdrojové kódy

Merge pull request #17175 from garylab/bugfix/api-config

Bugfix: Add verify token from headers also for /api/config endpoint
Tim Jaeryang Baek 1 mesiac pred
rodič
commit
e5829572ff
1 zmenil súbory, kde vykonal 11 pridanie a 1 odobranie
  1. 11 1
      backend/open_webui/main.py

+ 11 - 1
backend/open_webui/main.py

@@ -1649,8 +1649,18 @@ async def list_tasks_by_chat_id_endpoint(
 @app.get("/api/config")
 async def get_app_config(request: Request):
     user = None
-    if "token" in request.cookies:
+    token = None
+
+    auth_header = request.headers.get("Authorization")
+    if auth_header:
+        cred = get_http_authorization_cred(auth_header)
+        if cred:
+            token = cred.credentials
+
+    if not token and "token" in request.cookies:
         token = request.cookies.get("token")
+
+    if token:
         try:
             data = decode_token(token)
         except Exception as e: