Initial commit

backend/app/settings.py

@@ -0,0 +1,111 @@
+from __future__ import annotations
+
+import os
+from dataclasses import dataclass
+from pathlib import Path
+
+from dotenv import load_dotenv
+
+
+BACKEND_ROOT = Path(__file__).resolve().parents[1]
+PROJECT_ROOT = BACKEND_ROOT.parent
+
+load_dotenv(BACKEND_ROOT / ".env")
+
+
+@dataclass(frozen=True)
+class ProviderModel:
+    id: str
+    vision: bool = False
+
+
+@dataclass(frozen=True)
+class ProviderConfig:
+    id: str
+    label: str
+    base_url: str
+    api_key: str
+    models: tuple[ProviderModel, ...]
+
+    @property
+    def configured(self) -> bool:
+        return bool(self.base_url and self.api_key and self.models)
+
+    def model(self, model_id: str) -> ProviderModel | None:
+        return next((model for model in self.models if model.id == model_id), None)
+
+
+@dataclass(frozen=True)
+class Settings:
+    task_root: Path
+    conversation_root: Path
+    library_root: Path
+    engine_root: Path
+    llm_base_url: str
+    llm_api_key: str
+    llm_model: str
+    llm_timeout_s: float
+    default_provider_id: str
+    providers: tuple[ProviderConfig, ...]
+
+    @property
+    def llm_configured(self) -> bool:
+        return self.provider_for(self.default_provider_id) is not None
+
+    def provider_for(self, provider_id: str | None) -> ProviderConfig | None:
+        requested = str(provider_id or self.default_provider_id).strip().lower()
+        return next((provider for provider in self.providers if provider.id == requested and provider.configured), None)
+
+    def resolve_model(self, provider_id: str | None, model_id: str | None) -> tuple[ProviderConfig, ProviderModel]:
+        provider = self.provider_for(provider_id)
+        if provider is None:
+            raise ValueError("The selected model provider is not configured")
+        selected = str(model_id or "").strip() or provider.models[0].id
+        model = provider.model(selected)
+        if model is None:
+            raise ValueError("The selected model is not enabled for this provider")
+        return provider, model
+
+
+def _models(value: str, vision_value: str = "") -> tuple[ProviderModel, ...]:
+    vision_ids = {item.strip() for item in vision_value.split(",") if item.strip()}
+    return tuple(
+        ProviderModel(id=item, vision=item in vision_ids)
+        for item in (part.strip() for part in value.split(","))
+        if item
+    )
+
+
+def _provider(prefix: str, provider_id: str, label: str, default_base_url: str, default_model: str = "") -> ProviderConfig:
+    # The legacy CDSL_LLM_* variables remain the DeepSeek default so existing
+    # local installations continue to work without copying secrets.
+    legacy = provider_id == "deepseek"
+    base_url = os.getenv(f"CDSL_{prefix}_BASE_URL", os.getenv("CDSL_LLM_BASE_URL", default_base_url) if legacy else default_base_url).rstrip("/")
+    api_key = os.getenv(f"CDSL_{prefix}_API_KEY", os.getenv("CDSL_LLM_API_KEY", "") if legacy else "")
+    model_list = os.getenv(f"CDSL_{prefix}_MODELS", os.getenv("CDSL_LLM_MODEL", default_model) if legacy else default_model)
+    vision_models = os.getenv(f"CDSL_{prefix}_VISION_MODELS", "")
+    return ProviderConfig(provider_id, label, base_url, api_key, _models(model_list, vision_models))
+
+
+def get_settings() -> Settings:
+    data_root = BACKEND_ROOT / "data"
+    providers = (
+        _provider("DEEPSEEK", "deepseek", "DeepSeek", "https://api.deepseek.com/v1", "deepseek-chat"),
+        _provider("OPENAI", "openai", "OpenAI", "https://api.openai.com/v1"),
+        _provider("KIMI", "kimi", "Kimi", "https://api.moonshot.cn/v1"),
+    )
+    default_provider_id = os.getenv("CDSL_DEFAULT_PROVIDER", "deepseek").strip().lower() or "deepseek"
+    default_provider = next((item for item in providers if item.id == default_provider_id), providers[0])
+    default_model = os.getenv("CDSL_DEFAULT_MODEL", "").strip() or (default_provider.models[0].id if default_provider.models else "")
+    return Settings(
+        task_root=data_root / "tasks",
+        conversation_root=data_root / "conversations",
+        library_root=BACKEND_ROOT / "cdsl_library",
+        engine_root=BACKEND_ROOT / "engine" / "cdsl_engine",
+        llm_base_url=default_provider.base_url,
+        llm_api_key=default_provider.api_key,
+        llm_model=default_model,
+        llm_timeout_s=float(os.getenv("CDSL_LLM_TIMEOUT_S", "90")),
+        default_provider_id=default_provider_id,
+        providers=providers,
+    )