mnote/services/ingest_service/app/core/config.py

from __future__ import annotations

from functools import lru_cache
from pathlib import Path
from typing import Tuple, Type

from pydantic import Field
from pydantic_settings import BaseSettings, SettingsConfigDict
from pydantic_settings.sources import (
    DotEnvSettingsSource,
    EnvSettingsSource,
    PydanticBaseSettingsSource,
)


def _repo_root() -> Path:
    # services/ingest_service/app/core/config.py -> .../MNOTE
    return Path(__file__).resolve().parents[4]


class Settings(BaseSettings):
    """Ingest Service 配置（优先读取仓库根目录的 .env.local/.env）。"""

    model_config = SettingsConfigDict(
        env_file=(
            str(_repo_root() / ".env.local"),
            str(_repo_root() / ".env"),
        ),
        env_file_encoding="utf-8",
        case_sensitive=False,
        extra="ignore",
    )

    # 服务基础
    app_name: str = Field("AIMNOTE Ingest Service", env="INGEST_APP_NAME")
    environment: str = Field("development", env="INGEST_ENV")
    api_prefix: str = "/api"

    # LightRAG（独立服务 7777）
    lightrag_url: str = Field("http://127.0.0.1:7777", env="LIGHTRAG_URL")
    lightrag_api_key: str = Field("", env="LIGHTRAG_API_KEY")

    @classmethod
    def settings_customise_sources(
        cls,
        settings_cls: Type[BaseSettings],
        init_settings: PydanticBaseSettingsSource,
        env_settings: EnvSettingsSource,
        dotenv_settings: DotEnvSettingsSource,
        file_secret_settings: PydanticBaseSettingsSource,
    ) -> Tuple[PydanticBaseSettingsSource, ...]:
        # 关键点：优先使用 .env.local/.env 覆盖系统环境变量，避免被机器上的全局 env 污染。
        return (
            dotenv_settings,
            env_settings,
            init_settings,
            file_secret_settings,
        )


@lru_cache()
def get_settings() -> Settings:
    return Settings()