Pārlūkot izejas kodu

fix: error handling

Timothy J. Baek 1 gadu atpakaļ
vecāks
revīzija
d1f170dc77
1 mainītis faili ar 21 papildinājumiem un 17 dzēšanām
  1. 21 17
      backend/config.py

+ 21 - 17
backend/config.py

@@ -106,23 +106,27 @@ CHANGELOG = changelog_json
 
 CUSTOM_NAME = os.environ.get("CUSTOM_NAME", "")
 if CUSTOM_NAME:
-    r = requests.get(f"https://api.openwebui.com/api/v1/custom/{CUSTOM_NAME}")
-    data = r.json()
-
-    if "logo" in data:
-        url = (
-            f"https://api.openwebui.com{data['logo']}"
-            if data["logo"][0] == "/"
-            else data["logo"]
-        )
-
-        r = requests.get(url, stream=True)
-        if r.status_code == 200:
-            with open("./static/favicon.png", "wb") as f:
-                r.raw.decode_content = True
-                shutil.copyfileobj(r.raw, f)
-
-    WEBUI_NAME = data["name"]
+    try:
+        r = requests.get(f"https://api.openwebui.com/api/v1/custom/{CUSTOM_NAME}")
+        data = r.json()
+        if r.ok:
+            if "logo" in data:
+                url = (
+                    f"https://api.openwebui.com{data['logo']}"
+                    if data["logo"][0] == "/"
+                    else data["logo"]
+                )
+
+                r = requests.get(url, stream=True)
+                if r.status_code == 200:
+                    with open("./static/favicon.png", "wb") as f:
+                        r.raw.decode_content = True
+                        shutil.copyfileobj(r.raw, f)
+
+            WEBUI_NAME = data["name"]
+    except Exception as e:
+        print(e)
+        pass
 
 
 ####################################