From 3a0696e1b03f4181508ce3cfeb788e3d18b7c75e Mon Sep 17 00:00:00 2001
From: Timothy Jaeryang Baek <tim@openwebui.com>
Date: Sun, 8 Jun 2025 21:21:51 +0400
Subject: [PATCH] refac

---
 backend/open_webui/env.py  | 2 ++
 backend/open_webui/main.py | 3 ++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/backend/open_webui/env.py b/backend/open_webui/env.py
index fcfccaedf..760174837 100644
--- a/backend/open_webui/env.py
+++ b/backend/open_webui/env.py
@@ -5,6 +5,7 @@ import os
 import pkgutil
 import sys
 import shutil
+from uuid import uuid4
 from pathlib import Path
 
 import markdown
@@ -130,6 +131,7 @@ else:
         PACKAGE_DATA = {"version": "0.0.0"}
 
 VERSION = PACKAGE_DATA["version"]
+INSTANCE_ID = os.environ.get("INSTANCE_ID", str(uuid4()))
 
 
 # Function to parse each section
diff --git a/backend/open_webui/main.py b/backend/open_webui/main.py
index ec29a5cbf..1577b0170 100644
--- a/backend/open_webui/main.py
+++ b/backend/open_webui/main.py
@@ -394,6 +394,7 @@ from open_webui.env import (
     SAFE_MODE,
     SRC_LOG_LEVELS,
     VERSION,
+    INSTANCE_ID,
     WEBUI_BUILD_HASH,
     WEBUI_SECRET_KEY,
     WEBUI_SESSION_COOKIE_SAME_SITE,
@@ -490,7 +491,7 @@ https://github.com/open-webui/open-webui
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
-    app.state.instance_id = os.environ.get("INSTANCE_ID", str(uuid4()))
+    app.state.instance_id = INSTANCE_ID
     start_logger()
 
     if RESET_CONFIG_ON_START: