modified: .env.example
modified: config.py modified: docker-compose.yml modified: requirements.txt modified: services/rag_service.py
This commit is contained in:
11
config.py
11
config.py
@@ -8,16 +8,16 @@ BASE_DIR = os.path.abspath(os.path.dirname(__file__))
|
||||
|
||||
class Config:
|
||||
SECRET_KEY = os.environ.get("SECRET_KEY", "change-me-in-production")
|
||||
SQLALCHEMY_DATABASE_URI = os.environ.get(
|
||||
"DATABASE_URI", f"sqlite:///{os.path.join(BASE_DIR, 'app.db')}"
|
||||
_db_uri = os.environ.get("DATABASE_URI", "")
|
||||
_default_uri = f"sqlite:///{os.path.join(BASE_DIR, 'app.db')}"
|
||||
# Fall back to SQLite if DATABASE_URI is empty or not a valid SQLAlchemy URL
|
||||
SQLALCHEMY_DATABASE_URI = (
|
||||
_db_uri if _db_uri and "://" in _db_uri else _default_uri
|
||||
)
|
||||
SQLALCHEMY_TRACK_MODIFICATIONS = False
|
||||
|
||||
UPLOAD_FOLDER = os.environ.get("UPLOAD_FOLDER", os.path.join(BASE_DIR, "uploads"))
|
||||
VECTORDB_PATH = os.environ.get("VECTORDB_PATH", os.path.join(BASE_DIR, "vectordb"))
|
||||
TRANSFORMERS_CACHE = os.environ.get(
|
||||
"TRANSFORMERS_CACHE", os.path.join(BASE_DIR, ".cache")
|
||||
)
|
||||
|
||||
ALLOWED_EXTENSIONS = {"pdf", "txt", "docx", "md"}
|
||||
MAX_CONTENT_LENGTH = 50 * 1024 * 1024 # 50 MB
|
||||
@@ -26,6 +26,7 @@ class Config:
|
||||
AI_PROVIDER = os.environ.get("AI_PROVIDER", "lmstudio")
|
||||
LM_STUDIO_URL = os.environ.get("LM_STUDIO_URL", "http://localhost:1234")
|
||||
LM_STUDIO_MODEL = os.environ.get("LM_STUDIO_MODEL", "local-model")
|
||||
LM_STUDIO_EMBEDDING_MODEL = os.environ.get("LM_STUDIO_EMBEDDING_MODEL", "local-model")
|
||||
OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY", "")
|
||||
OPENAI_MODEL = os.environ.get("OPENAI_MODEL", "gpt-4o")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user