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

Merge pull request #883 from open-webui/custom-name

Browse source 
fix: error handling
This commit is contained in:
Timothy Jaeryang Baek 2024-02-23 20:37:22 -05:00 committed by GitHub
commit f69c0d2e85
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 19 additions and 15 deletions
Show stats 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
#################################### ####################################