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

feat: chromadb vector store api

Browse Source
This commit is contained in:
Timothy J. Baek
2024-01-06 22:59:22 -08:00
parent b2c9f6dff8
commit 784b369cc9
4 changed files with 119 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
backend/config.py
View File

@@ -1,11 +1,11 @@
from dotenv import load_dotenv, find_dotenv
from constants import ERROR_MESSAGES
import os
import chromadb
from secrets import token_bytes
from base64 import b64encode
import os
from constants import ERROR_MESSAGES
load_dotenv(find_dotenv("../.env"))
@@ -19,8 +19,9 @@ ENV = os.environ.get("ENV", "dev")
# OLLAMA_API_BASE_URL
####################################
OLLAMA_API_BASE_URL = os.environ.get("OLLAMA_API_BASE_URL",
"http://localhost:11434/api")
OLLAMA_API_BASE_URL = os.environ.get(
"OLLAMA_API_BASE_URL", "http://localhost:11434/api"
)
if ENV == "prod":
if OLLAMA_API_BASE_URL == "/ollama/api":
@@ -56,3 +57,13 @@ WEBUI_JWT_SECRET_KEY = os.environ.get("WEBUI_JWT_SECRET_KEY", "t0p-s3cr3t")
if WEBUI_AUTH and WEBUI_JWT_SECRET_KEY == "":
raise ValueError(ERROR_MESSAGES.ENV_VAR_NOT_FOUND)
####################################
# RAG
####################################
CHROMA_DATA_PATH = "./data/vector_db"
EMBED_MODEL = "all-MiniLM-L6-v2"
CHROMA_CLIENT = chromadb.PersistentClient(path=CHROMA_DATA_PATH)
CHUNK_SIZE = 1500
CHUNK_OVERLAP = 100