19 lines
691 B
Python
19 lines
691 B
Python
import os
|
|
from pydantic import SecretStr
|
|
from pydantic_settings import BaseSettings, SettingsConfigDict
|
|
|
|
|
|
class Config(BaseSettings):
|
|
BASE_DIR: str = os.path.abspath(os.path.join(os.path.dirname(__file__)))
|
|
DOCS_PATH: str = os.path.join(BASE_DIR, "data", "docs")
|
|
PARSED_JSON_PATH: str = os.path.join(BASE_DIR, "data", "parsed_json")
|
|
DOCS_CHROMA_PATH: str = os.path.join(BASE_DIR, "data", "chroma_db")
|
|
DOCS_COLLECTION_NAME: str = "docs"
|
|
MAX_CHUNK_SIZE: int = 2048
|
|
CHUNK_OVERLAP: int = 50
|
|
LM_MODEL_NAME: str = "/models/Qwen3-Embedding-0.6B"
|
|
LOCAL_LLM_NAME: str = "/models/Qwen3-4B"
|
|
QWEN_MODEL_NAME: str = "Qwen3-8B"
|
|
|
|
|
|
settings = Config() # type: ignore |