Mirrors/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-16 20:07:49 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #5160 from vikrantrathore/main

Browse Source 
fix: avoid overriding DATA_DIR and prevent errors when directories are the same
This commit is contained in:
Timothy Jaeryang Baek
2024-09-05 16:11:42 +02:00
committed by GitHub
parent 83574ccf6c 85fc35492d
commit fa786c8e05
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
backend/open_webui/env.py
View File

@@ -196,7 +196,7 @@ if PIP_INSTALL:
NEW_DATA_DIR.mkdir(parents=True, exist_ok=True) NEW_DATA_DIR.mkdir(parents=True, exist_ok=True)
# Check if the data directory exists in the package directory # Check if the data directory exists in the package directory
if DATA_DIR.exists(): if DATA_DIR.exists() and DATA_DIR != NEW_DATA_DIR:
log.info(f"Moving {DATA_DIR} to {NEW_DATA_DIR}") log.info(f"Moving {DATA_DIR} to {NEW_DATA_DIR}")
for item in DATA_DIR.iterdir(): for item in DATA_DIR.iterdir():
dest = NEW_DATA_DIR / item.name dest = NEW_DATA_DIR / item.name
@@ -205,10 +205,11 @@ if PIP_INSTALL:
else: else:
shutil.copy2(item, dest) shutil.copy2(item, dest)
DATA_DIR = OPEN_WEBUI_DIR / "data" DATA_DIR = Path(os.getenv("DATA_DIR", OPEN_WEBUI_DIR / "data"))
FRONTEND_BUILD_DIR = Path(os.getenv("FRONTEND_BUILD_DIR", BASE_DIR / "build")).resolve() FRONTEND_BUILD_DIR = Path(os.getenv("FRONTEND_BUILD_DIR", BASE_DIR / "build")).resolve()
if PIP_INSTALL: if PIP_INSTALL:
FRONTEND_BUILD_DIR = Path( FRONTEND_BUILD_DIR = Path(
os.getenv("FRONTEND_BUILD_DIR", OPEN_WEBUI_DIR / "frontend") os.getenv("FRONTEND_BUILD_DIR", OPEN_WEBUI_DIR / "frontend")