import os
from dataclasses import dataclass

@dataclass
class Config:
    pg_host: str = os.getenv("KB_PG_HOST", "192.168.1.114")
    pg_port: int = int(os.getenv("KB_PG_PORT", "5432"))
    pg_db: str = os.getenv("KB_PG_DB", "kb")
    pg_user: str = os.getenv("KB_PG_USER", "kb_user")
    pg_password: str = os.getenv("KB_PG_PASSWORD", "")
    
    ollama_url: str = os.getenv("KB_OLLAMA_URL", "http://192.168.1.116:11434")
    embed_model: str = os.getenv("KB_EMBED_MODEL", "nomic-embed-text")
    
    corpus_root: str = os.getenv("KB_CORPUS_ROOT", "/mnt/nas")
    
    chunk_size_tokens: int = 600
    chunk_overlap_tokens: int = 80
    embed_batch_size: int = 16

CONFIG = Config()