19 lines
621 B
Python
19 lines
621 B
Python
from pydantic_settings import BaseSettings
|
|
|
|
class Settings(BaseSettings):
|
|
DATABASE_URL: str = "postgresql+psycopg2://user:pass@db:5432/news"
|
|
REDIS_URL: str = "redis://redis:6379/0"
|
|
STORAGE_DIR: str = "/data/storage"
|
|
DPI: int = 300
|
|
LAYOUT_MODEL: str = "PubLayNet" # or DocLayNet / custom YOLOv11
|
|
OCR_LANGS: list = ["en", "hi", "te", "ta"]
|
|
LLM_PROVIDER: str = "gemini" # gemini | openai | local
|
|
LLM_API_KEY: str = ""
|
|
EMBED_MODEL: str = "sentence-transformers/all-MiniLM-L6-v2"
|
|
USE_GPU: bool = True
|
|
|
|
class Config:
|
|
env_file = ".env"
|
|
|
|
settings = Settings()
|