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

backend: make the data directory and the artifacts from the frontend customizable using environment variables

Browse Source 
Signed-off-by: lucasew <lucas59356@gmail.com>
This commit is contained in:
lucasew
2024-01-19 17:13:09 -03:00
parent e3503d6617
commit 5b26d2a686
4 changed files with 16 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
backend/config.py
View File

@@ -24,10 +24,12 @@ except ImportError:
# File Upload
####################################
DATA_DIR = Path(os.getenv("DATA_DIR", './data')).resolve()
UPLOAD_DIR = "./data/uploads"
Path(UPLOAD_DIR).mkdir(parents=True, exist_ok=True)
UPLOAD_DIR = DATA_DIR / "uploads"
UPLOAD_DIR.mkdir(parents=True, exist_ok=True)
WEB_DIR = Path(os.getenv("WEB_DIR", "../build"))
####################################
# ENV (dev,test,prod)
@@ -82,10 +84,10 @@ if WEBUI_AUTH and WEBUI_JWT_SECRET_KEY == "":
# RAG
####################################
CHROMA_DATA_PATH = "./data/vector_db"
CHROMA_DATA_PATH = DATA_DIR / "vector_db"
EMBED_MODEL = "all-MiniLM-L6-v2"
CHROMA_CLIENT = chromadb.PersistentClient(
path=CHROMA_DATA_PATH, settings=Settings(allow_reset=True)
path=str(CHROMA_DATA_PATH), settings=Settings(allow_reset=True)
)
CHUNK_SIZE = 1500
CHUNK_OVERLAP = 100