Julian Carlson
22 lines
686 B
Python
22 lines
686 B
Python
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()
|