Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/open-webui.git" did not exist on "f2285f29b799c8e9d668f475651b961136a1ef56"
Commit 16fa4545 authored by Michael Poluektov's avatar Michael Poluektov
Browse files

fix banners env

parent e4c85921
Hide whitespace changes
Inline Side-by-side
Showing with 9 additions and 8 deletions
backend/config.py
View file @ 16fa4545
...@@ -5,10 +5,10 @@ import importlib.metadata ...@@ -5,10 +5,10 @@ import importlib.metadata
import pkgutil import pkgutil
import chromadb import chromadb
from chromadb import Settings from chromadb import Settings
from base64 import b64encode
from bs4 import BeautifulSoup from bs4 import BeautifulSoup
from typing import TypeVar, Generic, Union from typing import TypeVar, Generic
from pydantic import BaseModel from pydantic import BaseModel
from pydantic.error_wrappers import ValidationError
from typing import Optional from typing import Optional
from pathlib import Path from pathlib import Path
...@@ -19,7 +19,6 @@ import markdown ...@@ -19,7 +19,6 @@ import markdown
import requests import requests
import shutil import shutil
from secrets import token_bytes
from constants import ERROR_MESSAGES from constants import ERROR_MESSAGES
#################################### ####################################
...@@ -768,12 +767,14 @@ class BannerModel(BaseModel): ...@@ -768,12 +767,14 @@ class BannerModel(BaseModel):
dismissible: bool dismissible: bool
timestamp: int timestamp: int
try:
banners = json.loads(os.environ.get("WEBUI_BANNERS", "[]"))
banners = [BannerModel(**banner) for banner in banners]
except ValidationError as e:
print(f"Error loading WEBUI_BANNERS: {e}")
banners = []
WEBUI_BANNERS = PersistentConfig( WEBUI_BANNERS = PersistentConfig("WEBUI_BANNERS", "ui.banners", banners)
"WEBUI_BANNERS",
"ui.banners",
[BannerModel(**banner) for banner in json.loads("[]")],
)
SHOW_ADMIN_DETAILS = PersistentConfig( SHOW_ADMIN_DETAILS = PersistentConfig(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment