env.py 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719
  1. import importlib.metadata
  2. import json
  3. import logging
  4. import os
  5. import pkgutil
  6. import sys
  7. import shutil
  8. from uuid import uuid4
  9. from pathlib import Path
  10. from cryptography.hazmat.primitives import serialization
  11. import markdown
  12. from bs4 import BeautifulSoup
  13. from open_webui.constants import ERROR_MESSAGES
  14. ####################################
  15. # Load .env file
  16. ####################################
  17. OPEN_WEBUI_DIR = Path(__file__).parent # the path containing this file
  18. print(OPEN_WEBUI_DIR)
  19. BACKEND_DIR = OPEN_WEBUI_DIR.parent # the path containing this file
  20. BASE_DIR = BACKEND_DIR.parent # the path containing the backend/
  21. print(BACKEND_DIR)
  22. print(BASE_DIR)
  23. try:
  24. from dotenv import find_dotenv, load_dotenv
  25. load_dotenv(find_dotenv(str(BASE_DIR / ".env")))
  26. except ImportError:
  27. print("dotenv not installed, skipping...")
  28. DOCKER = os.environ.get("DOCKER", "False").lower() == "true"
  29. # device type embedding models - "cpu" (default), "cuda" (nvidia gpu required) or "mps" (apple silicon) - choosing this right can lead to better performance
  30. USE_CUDA = os.environ.get("USE_CUDA_DOCKER", "false")
  31. if USE_CUDA.lower() == "true":
  32. try:
  33. import torch
  34. assert torch.cuda.is_available(), "CUDA not available"
  35. DEVICE_TYPE = "cuda"
  36. except Exception as e:
  37. cuda_error = (
  38. "Error when testing CUDA but USE_CUDA_DOCKER is true. "
  39. f"Resetting USE_CUDA_DOCKER to false: {e}"
  40. )
  41. os.environ["USE_CUDA_DOCKER"] = "false"
  42. USE_CUDA = "false"
  43. DEVICE_TYPE = "cpu"
  44. else:
  45. DEVICE_TYPE = "cpu"
  46. try:
  47. import torch
  48. if torch.backends.mps.is_available() and torch.backends.mps.is_built():
  49. DEVICE_TYPE = "mps"
  50. except Exception:
  51. pass
  52. ####################################
  53. # LOGGING
  54. ####################################
  55. GLOBAL_LOG_LEVEL = os.environ.get("GLOBAL_LOG_LEVEL", "").upper()
  56. if GLOBAL_LOG_LEVEL in logging.getLevelNamesMapping():
  57. logging.basicConfig(stream=sys.stdout, level=GLOBAL_LOG_LEVEL, force=True)
  58. else:
  59. GLOBAL_LOG_LEVEL = "INFO"
  60. log = logging.getLogger(__name__)
  61. log.info(f"GLOBAL_LOG_LEVEL: {GLOBAL_LOG_LEVEL}")
  62. if "cuda_error" in locals():
  63. log.exception(cuda_error)
  64. del cuda_error
  65. log_sources = [
  66. "AUDIO",
  67. "COMFYUI",
  68. "CONFIG",
  69. "DB",
  70. "IMAGES",
  71. "MAIN",
  72. "MODELS",
  73. "OLLAMA",
  74. "OPENAI",
  75. "RAG",
  76. "WEBHOOK",
  77. "SOCKET",
  78. "OAUTH",
  79. ]
  80. SRC_LOG_LEVELS = {}
  81. for source in log_sources:
  82. log_env_var = source + "_LOG_LEVEL"
  83. SRC_LOG_LEVELS[source] = os.environ.get(log_env_var, "").upper()
  84. if SRC_LOG_LEVELS[source] not in logging.getLevelNamesMapping():
  85. SRC_LOG_LEVELS[source] = GLOBAL_LOG_LEVEL
  86. log.info(f"{log_env_var}: {SRC_LOG_LEVELS[source]}")
  87. log.setLevel(SRC_LOG_LEVELS["CONFIG"])
  88. WEBUI_NAME = os.environ.get("WEBUI_NAME", "Open WebUI")
  89. if WEBUI_NAME != "Open WebUI":
  90. WEBUI_NAME += " (Open WebUI)"
  91. WEBUI_FAVICON_URL = "https://openwebui.com/favicon.png"
  92. TRUSTED_SIGNATURE_KEY = os.environ.get("TRUSTED_SIGNATURE_KEY", "")
  93. ####################################
  94. # ENV (dev,test,prod)
  95. ####################################
  96. ENV = os.environ.get("ENV", "dev")
  97. FROM_INIT_PY = os.environ.get("FROM_INIT_PY", "False").lower() == "true"
  98. if FROM_INIT_PY:
  99. PACKAGE_DATA = {"version": importlib.metadata.version("open-webui")}
  100. else:
  101. try:
  102. PACKAGE_DATA = json.loads((BASE_DIR / "package.json").read_text())
  103. except Exception:
  104. PACKAGE_DATA = {"version": "0.0.0"}
  105. VERSION = PACKAGE_DATA["version"]
  106. INSTANCE_ID = os.environ.get("INSTANCE_ID", str(uuid4()))
  107. # Function to parse each section
  108. def parse_section(section):
  109. items = []
  110. for li in section.find_all("li"):
  111. # Extract raw HTML string
  112. raw_html = str(li)
  113. # Extract text without HTML tags
  114. text = li.get_text(separator=" ", strip=True)
  115. # Split into title and content
  116. parts = text.split(": ", 1)
  117. title = parts[0].strip() if len(parts) > 1 else ""
  118. content = parts[1].strip() if len(parts) > 1 else text
  119. items.append({"title": title, "content": content, "raw": raw_html})
  120. return items
  121. try:
  122. changelog_path = BASE_DIR / "CHANGELOG.md"
  123. with open(str(changelog_path.absolute()), "r", encoding="utf8") as file:
  124. changelog_content = file.read()
  125. except Exception:
  126. changelog_content = (pkgutil.get_data("open_webui", "CHANGELOG.md") or b"").decode()
  127. # Convert markdown content to HTML
  128. html_content = markdown.markdown(changelog_content)
  129. # Parse the HTML content
  130. soup = BeautifulSoup(html_content, "html.parser")
  131. # Initialize JSON structure
  132. changelog_json = {}
  133. # Iterate over each version
  134. for version in soup.find_all("h2"):
  135. version_number = version.get_text().strip().split(" - ")[0][1:-1] # Remove brackets
  136. date = version.get_text().strip().split(" - ")[1]
  137. version_data = {"date": date}
  138. # Find the next sibling that is a h3 tag (section title)
  139. current = version.find_next_sibling()
  140. while current and current.name != "h2":
  141. if current.name == "h3":
  142. section_title = current.get_text().lower() # e.g., "added", "fixed"
  143. section_items = parse_section(current.find_next_sibling("ul"))
  144. version_data[section_title] = section_items
  145. # Move to the next element
  146. current = current.find_next_sibling()
  147. changelog_json[version_number] = version_data
  148. CHANGELOG = changelog_json
  149. ####################################
  150. # SAFE_MODE
  151. ####################################
  152. SAFE_MODE = os.environ.get("SAFE_MODE", "false").lower() == "true"
  153. ####################################
  154. # ENABLE_FORWARD_USER_INFO_HEADERS
  155. ####################################
  156. ENABLE_FORWARD_USER_INFO_HEADERS = (
  157. os.environ.get("ENABLE_FORWARD_USER_INFO_HEADERS", "False").lower() == "true"
  158. )
  159. ####################################
  160. # WEBUI_BUILD_HASH
  161. ####################################
  162. WEBUI_BUILD_HASH = os.environ.get("WEBUI_BUILD_HASH", "dev-build")
  163. ####################################
  164. # DATA/FRONTEND BUILD DIR
  165. ####################################
  166. DATA_DIR = Path(os.getenv("DATA_DIR", BACKEND_DIR / "data")).resolve()
  167. if FROM_INIT_PY:
  168. NEW_DATA_DIR = Path(os.getenv("DATA_DIR", OPEN_WEBUI_DIR / "data")).resolve()
  169. NEW_DATA_DIR.mkdir(parents=True, exist_ok=True)
  170. # Check if the data directory exists in the package directory
  171. if DATA_DIR.exists() and DATA_DIR != NEW_DATA_DIR:
  172. log.info(f"Moving {DATA_DIR} to {NEW_DATA_DIR}")
  173. for item in DATA_DIR.iterdir():
  174. dest = NEW_DATA_DIR / item.name
  175. if item.is_dir():
  176. shutil.copytree(item, dest, dirs_exist_ok=True)
  177. else:
  178. shutil.copy2(item, dest)
  179. # Zip the data directory
  180. shutil.make_archive(DATA_DIR.parent / "open_webui_data", "zip", DATA_DIR)
  181. # Remove the old data directory
  182. shutil.rmtree(DATA_DIR)
  183. DATA_DIR = Path(os.getenv("DATA_DIR", OPEN_WEBUI_DIR / "data"))
  184. STATIC_DIR = Path(os.getenv("STATIC_DIR", OPEN_WEBUI_DIR / "static"))
  185. FONTS_DIR = Path(os.getenv("FONTS_DIR", OPEN_WEBUI_DIR / "static" / "fonts"))
  186. FRONTEND_BUILD_DIR = Path(os.getenv("FRONTEND_BUILD_DIR", BASE_DIR / "build")).resolve()
  187. if FROM_INIT_PY:
  188. FRONTEND_BUILD_DIR = Path(
  189. os.getenv("FRONTEND_BUILD_DIR", OPEN_WEBUI_DIR / "frontend")
  190. ).resolve()
  191. ####################################
  192. # Database
  193. ####################################
  194. # Check if the file exists
  195. if os.path.exists(f"{DATA_DIR}/ollama.db"):
  196. # Rename the file
  197. os.rename(f"{DATA_DIR}/ollama.db", f"{DATA_DIR}/webui.db")
  198. log.info("Database migrated from Ollama-WebUI successfully.")
  199. else:
  200. pass
  201. DATABASE_URL = os.environ.get("DATABASE_URL", f"sqlite:///{DATA_DIR}/webui.db")
  202. DATABASE_TYPE = os.environ.get("DATABASE_TYPE")
  203. DATABASE_USER = os.environ.get("DATABASE_USER")
  204. DATABASE_PASSWORD = os.environ.get("DATABASE_PASSWORD")
  205. DATABASE_CRED = ""
  206. if DATABASE_USER:
  207. DATABASE_CRED += f"{DATABASE_USER}"
  208. if DATABASE_PASSWORD:
  209. DATABASE_CRED += f":{DATABASE_PASSWORD}"
  210. DB_VARS = {
  211. "db_type": DATABASE_TYPE,
  212. "db_cred": DATABASE_CRED,
  213. "db_host": os.environ.get("DATABASE_HOST"),
  214. "db_port": os.environ.get("DATABASE_PORT"),
  215. "db_name": os.environ.get("DATABASE_NAME"),
  216. }
  217. if all(DB_VARS.values()):
  218. DATABASE_URL = f"{DB_VARS['db_type']}://{DB_VARS['db_cred']}@{DB_VARS['db_host']}:{DB_VARS['db_port']}/{DB_VARS['db_name']}"
  219. # Replace the postgres:// with postgresql://
  220. if "postgres://" in DATABASE_URL:
  221. DATABASE_URL = DATABASE_URL.replace("postgres://", "postgresql://")
  222. DATABASE_SCHEMA = os.environ.get("DATABASE_SCHEMA", None)
  223. DATABASE_POOL_SIZE = os.environ.get("DATABASE_POOL_SIZE", None)
  224. if DATABASE_POOL_SIZE != None:
  225. try:
  226. DATABASE_POOL_SIZE = int(DATABASE_POOL_SIZE)
  227. except Exception:
  228. DATABASE_POOL_SIZE = None
  229. DATABASE_POOL_MAX_OVERFLOW = os.environ.get("DATABASE_POOL_MAX_OVERFLOW", 0)
  230. if DATABASE_POOL_MAX_OVERFLOW == "":
  231. DATABASE_POOL_MAX_OVERFLOW = 0
  232. else:
  233. try:
  234. DATABASE_POOL_MAX_OVERFLOW = int(DATABASE_POOL_MAX_OVERFLOW)
  235. except Exception:
  236. DATABASE_POOL_MAX_OVERFLOW = 0
  237. DATABASE_POOL_TIMEOUT = os.environ.get("DATABASE_POOL_TIMEOUT", 30)
  238. if DATABASE_POOL_TIMEOUT == "":
  239. DATABASE_POOL_TIMEOUT = 30
  240. else:
  241. try:
  242. DATABASE_POOL_TIMEOUT = int(DATABASE_POOL_TIMEOUT)
  243. except Exception:
  244. DATABASE_POOL_TIMEOUT = 30
  245. DATABASE_POOL_RECYCLE = os.environ.get("DATABASE_POOL_RECYCLE", 3600)
  246. if DATABASE_POOL_RECYCLE == "":
  247. DATABASE_POOL_RECYCLE = 3600
  248. else:
  249. try:
  250. DATABASE_POOL_RECYCLE = int(DATABASE_POOL_RECYCLE)
  251. except Exception:
  252. DATABASE_POOL_RECYCLE = 3600
  253. RESET_CONFIG_ON_START = (
  254. os.environ.get("RESET_CONFIG_ON_START", "False").lower() == "true"
  255. )
  256. ENABLE_REALTIME_CHAT_SAVE = (
  257. os.environ.get("ENABLE_REALTIME_CHAT_SAVE", "False").lower() == "true"
  258. )
  259. ####################################
  260. # REDIS
  261. ####################################
  262. REDIS_URL = os.environ.get("REDIS_URL", "")
  263. REDIS_CLUSTER = os.environ.get("REDIS_CLUSTER", "False").lower() == "true"
  264. REDIS_KEY_PREFIX = os.environ.get("REDIS_KEY_PREFIX", "open-webui")
  265. REDIS_SENTINEL_HOSTS = os.environ.get("REDIS_SENTINEL_HOSTS", "")
  266. REDIS_SENTINEL_PORT = os.environ.get("REDIS_SENTINEL_PORT", "26379")
  267. # Maximum number of retries for Redis operations when using Sentinel fail-over
  268. REDIS_SENTINEL_MAX_RETRY_COUNT = os.environ.get("REDIS_SENTINEL_MAX_RETRY_COUNT", "2")
  269. try:
  270. REDIS_SENTINEL_MAX_RETRY_COUNT = int(REDIS_SENTINEL_MAX_RETRY_COUNT)
  271. if REDIS_SENTINEL_MAX_RETRY_COUNT < 1:
  272. REDIS_SENTINEL_MAX_RETRY_COUNT = 2
  273. except ValueError:
  274. REDIS_SENTINEL_MAX_RETRY_COUNT = 2
  275. ####################################
  276. # UVICORN WORKERS
  277. ####################################
  278. # Number of uvicorn worker processes for handling requests
  279. UVICORN_WORKERS = os.environ.get("UVICORN_WORKERS", "1")
  280. try:
  281. UVICORN_WORKERS = int(UVICORN_WORKERS)
  282. if UVICORN_WORKERS < 1:
  283. UVICORN_WORKERS = 1
  284. except ValueError:
  285. UVICORN_WORKERS = 1
  286. log.info(f"Invalid UVICORN_WORKERS value, defaulting to {UVICORN_WORKERS}")
  287. ####################################
  288. # WEBUI_AUTH (Required for security)
  289. ####################################
  290. WEBUI_AUTH = os.environ.get("WEBUI_AUTH", "True").lower() == "true"
  291. ENABLE_SIGNUP_PASSWORD_CONFIRMATION = (
  292. os.environ.get("ENABLE_SIGNUP_PASSWORD_CONFIRMATION", "False").lower() == "true"
  293. )
  294. WEBUI_AUTH_TRUSTED_EMAIL_HEADER = os.environ.get(
  295. "WEBUI_AUTH_TRUSTED_EMAIL_HEADER", None
  296. )
  297. WEBUI_AUTH_TRUSTED_NAME_HEADER = os.environ.get("WEBUI_AUTH_TRUSTED_NAME_HEADER", None)
  298. WEBUI_AUTH_TRUSTED_GROUPS_HEADER = os.environ.get(
  299. "WEBUI_AUTH_TRUSTED_GROUPS_HEADER", None
  300. )
  301. BYPASS_MODEL_ACCESS_CONTROL = (
  302. os.environ.get("BYPASS_MODEL_ACCESS_CONTROL", "False").lower() == "true"
  303. )
  304. WEBUI_AUTH_SIGNOUT_REDIRECT_URL = os.environ.get(
  305. "WEBUI_AUTH_SIGNOUT_REDIRECT_URL", None
  306. )
  307. ####################################
  308. # WEBUI_SECRET_KEY
  309. ####################################
  310. WEBUI_SECRET_KEY = os.environ.get(
  311. "WEBUI_SECRET_KEY",
  312. os.environ.get(
  313. "WEBUI_JWT_SECRET_KEY", "t0p-s3cr3t"
  314. ), # DEPRECATED: remove at next major version
  315. )
  316. WEBUI_SESSION_COOKIE_SAME_SITE = os.environ.get("WEBUI_SESSION_COOKIE_SAME_SITE", "lax")
  317. WEBUI_SESSION_COOKIE_SECURE = (
  318. os.environ.get("WEBUI_SESSION_COOKIE_SECURE", "false").lower() == "true"
  319. )
  320. WEBUI_AUTH_COOKIE_SAME_SITE = os.environ.get(
  321. "WEBUI_AUTH_COOKIE_SAME_SITE", WEBUI_SESSION_COOKIE_SAME_SITE
  322. )
  323. WEBUI_AUTH_COOKIE_SECURE = (
  324. os.environ.get(
  325. "WEBUI_AUTH_COOKIE_SECURE",
  326. os.environ.get("WEBUI_SESSION_COOKIE_SECURE", "false"),
  327. ).lower()
  328. == "true"
  329. )
  330. if WEBUI_AUTH and WEBUI_SECRET_KEY == "":
  331. raise ValueError(ERROR_MESSAGES.ENV_VAR_NOT_FOUND)
  332. ENABLE_COMPRESSION_MIDDLEWARE = (
  333. os.environ.get("ENABLE_COMPRESSION_MIDDLEWARE", "True").lower() == "true"
  334. )
  335. ####################################
  336. # LICENSE_KEY
  337. ####################################
  338. LICENSE_KEY = os.environ.get("LICENSE_KEY", "")
  339. LICENSE_BLOB = None
  340. LICENSE_BLOB_PATH = os.environ.get("LICENSE_BLOB_PATH", DATA_DIR / "l.data")
  341. if LICENSE_BLOB_PATH and os.path.exists(LICENSE_BLOB_PATH):
  342. with open(LICENSE_BLOB_PATH, "rb") as f:
  343. LICENSE_BLOB = f.read()
  344. LICENSE_PUBLIC_KEY = os.environ.get("LICENSE_PUBLIC_KEY", "")
  345. pk = None
  346. if LICENSE_PUBLIC_KEY:
  347. pk = serialization.load_pem_public_key(
  348. f"""
  349. -----BEGIN PUBLIC KEY-----
  350. {LICENSE_PUBLIC_KEY}
  351. -----END PUBLIC KEY-----
  352. """.encode(
  353. "utf-8"
  354. )
  355. )
  356. ####################################
  357. # MODELS
  358. ####################################
  359. MODELS_CACHE_TTL = os.environ.get("MODELS_CACHE_TTL", "1")
  360. if MODELS_CACHE_TTL == "":
  361. MODELS_CACHE_TTL = None
  362. else:
  363. try:
  364. MODELS_CACHE_TTL = int(MODELS_CACHE_TTL)
  365. except Exception:
  366. MODELS_CACHE_TTL = 1
  367. ####################################
  368. # WEBSOCKET SUPPORT
  369. ####################################
  370. ENABLE_WEBSOCKET_SUPPORT = (
  371. os.environ.get("ENABLE_WEBSOCKET_SUPPORT", "True").lower() == "true"
  372. )
  373. WEBSOCKET_MANAGER = os.environ.get("WEBSOCKET_MANAGER", "")
  374. WEBSOCKET_REDIS_URL = os.environ.get("WEBSOCKET_REDIS_URL", REDIS_URL)
  375. WEBSOCKET_REDIS_CLUSTER = (
  376. os.environ.get("WEBSOCKET_REDIS_CLUSTER", str(REDIS_CLUSTER)).lower() == "true"
  377. )
  378. websocket_redis_lock_timeout = os.environ.get("WEBSOCKET_REDIS_LOCK_TIMEOUT", "60")
  379. try:
  380. WEBSOCKET_REDIS_LOCK_TIMEOUT = int(websocket_redis_lock_timeout)
  381. except ValueError:
  382. WEBSOCKET_REDIS_LOCK_TIMEOUT = 60
  383. WEBSOCKET_SENTINEL_HOSTS = os.environ.get("WEBSOCKET_SENTINEL_HOSTS", "")
  384. WEBSOCKET_SENTINEL_PORT = os.environ.get("WEBSOCKET_SENTINEL_PORT", "26379")
  385. AIOHTTP_CLIENT_TIMEOUT = os.environ.get("AIOHTTP_CLIENT_TIMEOUT", "")
  386. if AIOHTTP_CLIENT_TIMEOUT == "":
  387. AIOHTTP_CLIENT_TIMEOUT = None
  388. else:
  389. try:
  390. AIOHTTP_CLIENT_TIMEOUT = int(AIOHTTP_CLIENT_TIMEOUT)
  391. except Exception:
  392. AIOHTTP_CLIENT_TIMEOUT = 300
  393. AIOHTTP_CLIENT_SESSION_SSL = (
  394. os.environ.get("AIOHTTP_CLIENT_SESSION_SSL", "True").lower() == "true"
  395. )
  396. AIOHTTP_CLIENT_TIMEOUT_MODEL_LIST = os.environ.get(
  397. "AIOHTTP_CLIENT_TIMEOUT_MODEL_LIST",
  398. os.environ.get("AIOHTTP_CLIENT_TIMEOUT_OPENAI_MODEL_LIST", "10"),
  399. )
  400. if AIOHTTP_CLIENT_TIMEOUT_MODEL_LIST == "":
  401. AIOHTTP_CLIENT_TIMEOUT_MODEL_LIST = None
  402. else:
  403. try:
  404. AIOHTTP_CLIENT_TIMEOUT_MODEL_LIST = int(AIOHTTP_CLIENT_TIMEOUT_MODEL_LIST)
  405. except Exception:
  406. AIOHTTP_CLIENT_TIMEOUT_MODEL_LIST = 10
  407. AIOHTTP_CLIENT_TIMEOUT_TOOL_SERVER_DATA = os.environ.get(
  408. "AIOHTTP_CLIENT_TIMEOUT_TOOL_SERVER_DATA", "10"
  409. )
  410. if AIOHTTP_CLIENT_TIMEOUT_TOOL_SERVER_DATA == "":
  411. AIOHTTP_CLIENT_TIMEOUT_TOOL_SERVER_DATA = None
  412. else:
  413. try:
  414. AIOHTTP_CLIENT_TIMEOUT_TOOL_SERVER_DATA = int(
  415. AIOHTTP_CLIENT_TIMEOUT_TOOL_SERVER_DATA
  416. )
  417. except Exception:
  418. AIOHTTP_CLIENT_TIMEOUT_TOOL_SERVER_DATA = 10
  419. AIOHTTP_CLIENT_SESSION_TOOL_SERVER_SSL = (
  420. os.environ.get("AIOHTTP_CLIENT_SESSION_TOOL_SERVER_SSL", "True").lower() == "true"
  421. )
  422. ####################################
  423. # SENTENCE TRANSFORMERS
  424. ####################################
  425. SENTENCE_TRANSFORMERS_BACKEND = os.environ.get("SENTENCE_TRANSFORMERS_BACKEND", "")
  426. if SENTENCE_TRANSFORMERS_BACKEND == "":
  427. SENTENCE_TRANSFORMERS_BACKEND = "torch"
  428. SENTENCE_TRANSFORMERS_MODEL_KWARGS = os.environ.get(
  429. "SENTENCE_TRANSFORMERS_MODEL_KWARGS", ""
  430. )
  431. if SENTENCE_TRANSFORMERS_MODEL_KWARGS == "":
  432. SENTENCE_TRANSFORMERS_MODEL_KWARGS = None
  433. else:
  434. try:
  435. SENTENCE_TRANSFORMERS_MODEL_KWARGS = json.loads(
  436. SENTENCE_TRANSFORMERS_MODEL_KWARGS
  437. )
  438. except Exception:
  439. SENTENCE_TRANSFORMERS_MODEL_KWARGS = None
  440. SENTENCE_TRANSFORMERS_CROSS_ENCODER_BACKEND = os.environ.get(
  441. "SENTENCE_TRANSFORMERS_CROSS_ENCODER_BACKEND", ""
  442. )
  443. if SENTENCE_TRANSFORMERS_CROSS_ENCODER_BACKEND == "":
  444. SENTENCE_TRANSFORMERS_CROSS_ENCODER_BACKEND = "torch"
  445. SENTENCE_TRANSFORMERS_CROSS_ENCODER_MODEL_KWARGS = os.environ.get(
  446. "SENTENCE_TRANSFORMERS_CROSS_ENCODER_MODEL_KWARGS", ""
  447. )
  448. if SENTENCE_TRANSFORMERS_CROSS_ENCODER_MODEL_KWARGS == "":
  449. SENTENCE_TRANSFORMERS_CROSS_ENCODER_MODEL_KWARGS = None
  450. else:
  451. try:
  452. SENTENCE_TRANSFORMERS_CROSS_ENCODER_MODEL_KWARGS = json.loads(
  453. SENTENCE_TRANSFORMERS_CROSS_ENCODER_MODEL_KWARGS
  454. )
  455. except Exception:
  456. SENTENCE_TRANSFORMERS_CROSS_ENCODER_MODEL_KWARGS = None
  457. ####################################
  458. # OFFLINE_MODE
  459. ####################################
  460. ENABLE_VERSION_UPDATE_CHECK = (
  461. os.environ.get("ENABLE_VERSION_UPDATE_CHECK", "true").lower() == "true"
  462. )
  463. OFFLINE_MODE = os.environ.get("OFFLINE_MODE", "false").lower() == "true"
  464. if OFFLINE_MODE:
  465. os.environ["HF_HUB_OFFLINE"] = "1"
  466. ENABLE_VERSION_UPDATE_CHECK = False
  467. ####################################
  468. # AUDIT LOGGING
  469. ####################################
  470. # Where to store log file
  471. AUDIT_LOGS_FILE_PATH = f"{DATA_DIR}/audit.log"
  472. # Maximum size of a file before rotating into a new log file
  473. AUDIT_LOG_FILE_ROTATION_SIZE = os.getenv("AUDIT_LOG_FILE_ROTATION_SIZE", "10MB")
  474. # Comma separated list of logger names to use for audit logging
  475. # Default is "uvicorn.access" which is the access log for Uvicorn
  476. # You can add more logger names to this list if you want to capture more logs
  477. AUDIT_UVICORN_LOGGER_NAMES = os.getenv(
  478. "AUDIT_UVICORN_LOGGER_NAMES", "uvicorn.access"
  479. ).split(",")
  480. # METADATA | REQUEST | REQUEST_RESPONSE
  481. AUDIT_LOG_LEVEL = os.getenv("AUDIT_LOG_LEVEL", "NONE").upper()
  482. try:
  483. MAX_BODY_LOG_SIZE = int(os.environ.get("MAX_BODY_LOG_SIZE") or 2048)
  484. except ValueError:
  485. MAX_BODY_LOG_SIZE = 2048
  486. # Comma separated list for urls to exclude from audit
  487. AUDIT_EXCLUDED_PATHS = os.getenv("AUDIT_EXCLUDED_PATHS", "/chats,/chat,/folders").split(
  488. ","
  489. )
  490. AUDIT_EXCLUDED_PATHS = [path.strip() for path in AUDIT_EXCLUDED_PATHS]
  491. AUDIT_EXCLUDED_PATHS = [path.lstrip("/") for path in AUDIT_EXCLUDED_PATHS]
  492. ####################################
  493. # OPENTELEMETRY
  494. ####################################
  495. ENABLE_OTEL = os.environ.get("ENABLE_OTEL", "False").lower() == "true"
  496. ENABLE_OTEL_METRICS = os.environ.get("ENABLE_OTEL_METRICS", "False").lower() == "true"
  497. ENABLE_OTEL_LOGS = os.environ.get("ENABLE_OTEL_LOGS", "False").lower() == "true"
  498. OTEL_EXPORTER_OTLP_ENDPOINT = os.environ.get(
  499. "OTEL_EXPORTER_OTLP_ENDPOINT", "http://localhost:4317"
  500. )
  501. OTEL_METRICS_EXPORTER_OTLP_ENDPOINT = os.environ.get(
  502. "OTEL_METRICS_EXPORTER_OTLP_ENDPOINT", OTEL_EXPORTER_OTLP_ENDPOINT
  503. )
  504. OTEL_LOGS_EXPORTER_OTLP_ENDPOINT = os.environ.get(
  505. "OTEL_LOGS_EXPORTER_OTLP_ENDPOINT", OTEL_EXPORTER_OTLP_ENDPOINT
  506. )
  507. OTEL_EXPORTER_OTLP_INSECURE = (
  508. os.environ.get("OTEL_EXPORTER_OTLP_INSECURE", "False").lower() == "true"
  509. )
  510. OTEL_METRICS_EXPORTER_OTLP_INSECURE = (
  511. os.environ.get("OTEL_METRICS_EXPORTER_OTLP_INSECURE", "False").lower() == "true"
  512. )
  513. OTEL_LOGS_EXPORTER_OTLP_INSECURE = (
  514. os.environ.get("OTEL_LOGS_EXPORTER_OTLP_INSECURE", "False").lower() == "true"
  515. )
  516. OTEL_SERVICE_NAME = os.environ.get("OTEL_SERVICE_NAME", "open-webui")
  517. OTEL_RESOURCE_ATTRIBUTES = os.environ.get(
  518. "OTEL_RESOURCE_ATTRIBUTES", ""
  519. ) # e.g. key1=val1,key2=val2
  520. OTEL_TRACES_SAMPLER = os.environ.get(
  521. "OTEL_TRACES_SAMPLER", "parentbased_always_on"
  522. ).lower()
  523. OTEL_BASIC_AUTH_USERNAME = os.environ.get("OTEL_BASIC_AUTH_USERNAME", "")
  524. OTEL_BASIC_AUTH_PASSWORD = os.environ.get("OTEL_BASIC_AUTH_PASSWORD", "")
  525. OTEL_METRICS_BASIC_AUTH_USERNAME = os.environ.get(
  526. "OTEL_METRICS_BASIC_AUTH_USERNAME", OTEL_BASIC_AUTH_USERNAME
  527. )
  528. OTEL_METRICS_BASIC_AUTH_PASSWORD = os.environ.get(
  529. "OTEL_METRICS_BASIC_AUTH_PASSWORD", OTEL_BASIC_AUTH_PASSWORD
  530. )
  531. OTEL_LOGS_BASIC_AUTH_USERNAME = os.environ.get(
  532. "OTEL_LOGS_BASIC_AUTH_USERNAME", OTEL_BASIC_AUTH_USERNAME
  533. )
  534. OTEL_LOGS_BASIC_AUTH_PASSWORD = os.environ.get(
  535. "OTEL_LOGS_BASIC_AUTH_PASSWORD", OTEL_BASIC_AUTH_PASSWORD
  536. )
  537. OTEL_OTLP_SPAN_EXPORTER = os.environ.get(
  538. "OTEL_OTLP_SPAN_EXPORTER", "grpc"
  539. ).lower() # grpc or http
  540. OTEL_METRICS_OTLP_SPAN_EXPORTER = os.environ.get(
  541. "OTEL_METRICS_OTLP_SPAN_EXPORTER", OTEL_OTLP_SPAN_EXPORTER
  542. ).lower() # grpc or http
  543. OTEL_LOGS_OTLP_SPAN_EXPORTER = os.environ.get(
  544. "OTEL_LOGS_OTLP_SPAN_EXPORTER", OTEL_OTLP_SPAN_EXPORTER
  545. ).lower() # grpc or http
  546. ####################################
  547. # TOOLS/FUNCTIONS PIP OPTIONS
  548. ####################################
  549. PIP_OPTIONS = os.getenv("PIP_OPTIONS", "").split()
  550. PIP_PACKAGE_INDEX_OPTIONS = os.getenv("PIP_PACKAGE_INDEX_OPTIONS", "").split()
  551. ####################################
  552. # PROGRESSIVE WEB APP OPTIONS
  553. ####################################
  554. EXTERNAL_PWA_MANIFEST_URL = os.environ.get("EXTERNAL_PWA_MANIFEST_URL")