40 lines
817 B
Python
40 lines
817 B
Python
|
"""AtoCore configuration via environment variables."""
|
||
|
|
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
from pydantic_settings import BaseSettings
|
||
|
|
|
||
|
|
|
||
|
|
class Settings(BaseSettings):
|
||
|
|
debug: bool = False
|
||
|
|
data_dir: Path = Path("./data")
|
||
|
|
host: str = "127.0.0.1"
|
||
|
|
port: int = 8100
|
||
|
|
|
||
|
|
# Embedding
|
||
|
|
embedding_model: str = (
|
||
|
|
"sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2"
|
||
|
|
)
|
||
|
|
|
||
|
|
# Chunking
|
||
|
|
chunk_max_size: int = 800
|
||
|
|
chunk_overlap: int = 100
|
||
|
|
chunk_min_size: int = 50
|
||
|
|
|
||
|
|
# Context
|
||
|
|
context_budget: int = 3000
|
||
|
|
context_top_k: int = 15
|
||
|
|
|
||
|
|
model_config = {"env_prefix": "ATOCORE_"}
|
||
|
|
|
||
|
|
@property
|
||
|
|
def db_path(self) -> Path:
|
||
|
|
return self.data_dir / "atocore.db"
|
||
|
|
|
||
|
|
@property
|
||
|
|
def chroma_path(self) -> Path:
|
||
|
|
return self.data_dir / "chroma"
|
||
|
|
|
||
|
|
|
||
|
|
settings = Settings()
|