Скелет проекта

src/rag_agent/config.py

@@ -0,0 +1,57 @@
+from __future__ import annotations
+
+import os
+from dataclasses import dataclass
+from typing import Iterable, Sequence
+
+
+@dataclass(frozen=True)
+class AppConfig:
+    repo_path: str
+    db_dsn: str
+    chunk_size: int
+    chunk_overlap: int
+    embeddings_dim: int
+    embeddings_model: str
+    llm_model: str
+    allowed_extensions: Sequence[str]
+
+
+def _env_int(name: str, default: int) -> int:
+    value = os.getenv(name, "").strip()
+    if not value:
+        return default
+    try:
+        return int(value)
+    except ValueError as exc:
+        raise ValueError(f"Invalid integer for {name}: {value}") from exc
+
+
+def _env_list(name: str, default: Iterable[str]) -> list[str]:
+    value = os.getenv(name, "").strip()
+    if not value:
+        return list(default)
+    return [item.strip() for item in value.split(",") if item.strip()]
+
+
+def load_config() -> AppConfig:
+    repo_path = os.getenv("RAG_REPO_PATH", "").strip()
+    if not repo_path:
+        raise ValueError("RAG_REPO_PATH is required")
+
+    db_dsn = os.getenv("RAG_DB_DSN", "").strip()
+    if not db_dsn:
+        raise ValueError("RAG_DB_DSN is required")
+
+    return AppConfig(
+        repo_path=repo_path,
+        db_dsn=db_dsn,
+        chunk_size=_env_int("RAG_CHUNK_SIZE", 400),
+        chunk_overlap=_env_int("RAG_CHUNK_OVERLAP", 50),
+        embeddings_dim=_env_int("RAG_EMBEDDINGS_DIM", 1536),
+        embeddings_model=os.getenv("RAG_EMBEDDINGS_MODEL", "stub-embeddings"),
+        llm_model=os.getenv("RAG_LLM_MODEL", "stub-llm"),
+        allowed_extensions=tuple(
+            _env_list("RAG_ALLOWED_EXTENSIONS", [".md", ".txt", ".rst"])
+        ),
+    )