tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Projects 1 Releases 2 Packages 1 Activity Actions

fix: error handling

Browse source
This commit is contained in:
Timothy J. Baek 2024-02-23 17:36:38 -08:00
parent decba05c7a
commit d1f170dc77
1 changed files with 19 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

6
backend/config.py
View file

@ -106,9 +106,10 @@ CHANGELOG = changelog_json
CUSTOM_NAME = os.environ.get("CUSTOM_NAME", "") CUSTOM_NAME = os.environ.get("CUSTOM_NAME", "")
if CUSTOM_NAME: if CUSTOM_NAME:
try:
r = requests.get(f"https://api.openwebui.com/api/v1/custom/{CUSTOM_NAME}") r = requests.get(f"https://api.openwebui.com/api/v1/custom/{CUSTOM_NAME}")
data = r.json() data = r.json()
if r.ok:
if "logo" in data: if "logo" in data:
url = ( url = (
f"https://api.openwebui.com{data['logo']}" f"https://api.openwebui.com{data['logo']}"
@ -123,6 +124,9 @@ if CUSTOM_NAME:
shutil.copyfileobj(r.raw, f) shutil.copyfileobj(r.raw, f)
WEBUI_NAME = data["name"] WEBUI_NAME = data["name"]
except Exception as e:
print(e)
pass
#################################### ####################################