93 lines
3.0 KiB
Python
93 lines
3.0 KiB
Python
"""Environment-backed application settings."""
|
|
|
|
from functools import lru_cache
|
|
from typing import Annotated
|
|
|
|
from pydantic import AliasChoices, Field, SecretStr, field_validator
|
|
from pydantic_settings import BaseSettings, NoDecode, SettingsConfigDict
|
|
|
|
|
|
class Settings(BaseSettings):
|
|
app_name: str = "Dieya Risk Prediction Backend"
|
|
app_version: str = "0.1.0"
|
|
environment: str = "development"
|
|
debug: bool = False
|
|
database_url: str = Field(
|
|
default="postgresql+asyncpg://postgres:postgres@localhost:5432/dieya_risk_prediction",
|
|
repr=False,
|
|
)
|
|
log_level: str = "INFO"
|
|
jwt_secret_key: SecretStr = SecretStr("CHANGE_THIS_IN_PRODUCTION")
|
|
jwt_algorithm: str = "HS256"
|
|
jwt_expire_minutes: int = Field(default=480, gt=0)
|
|
dingtalk_app_key: str = ""
|
|
dingtalk_app_secret: SecretStr | None = Field(default=None, repr=False)
|
|
dingtalk_corp_id: str = ""
|
|
dingtalk_agent_id: str = ""
|
|
llm_api_key: SecretStr | None = Field(
|
|
default=None,
|
|
repr=False,
|
|
validation_alias=AliasChoices(
|
|
"RISK_LLM_API_KEY",
|
|
"OPENAI_API_KEY",
|
|
"YUNWU_API_KEY",
|
|
),
|
|
)
|
|
llm_base_url: str = Field(
|
|
default="https://api.openai.com/v1",
|
|
validation_alias=AliasChoices("RISK_LLM_BASE_URL", "LLM_BASE_URL", "OPENAI_API_BASE"),
|
|
)
|
|
llm_model: str = Field(
|
|
default="gpt-4o-mini",
|
|
validation_alias=AliasChoices("RISK_LLM_MODEL", "LLM_MODEL"),
|
|
)
|
|
llm_timeout_seconds: int = Field(
|
|
default=120,
|
|
gt=0,
|
|
le=600,
|
|
validation_alias=AliasChoices("RISK_LLM_TIMEOUT_SECONDS", "LLM_TIMEOUT"),
|
|
)
|
|
llm_temperature: float = Field(
|
|
default=0.2,
|
|
ge=0,
|
|
le=2,
|
|
validation_alias=AliasChoices("RISK_LLM_TEMPERATURE", "LLM_TEMPERATURE"),
|
|
)
|
|
cors_allow_origins: Annotated[tuple[str, ...], NoDecode] = (
|
|
"http://localhost:39158",
|
|
"http://127.0.0.1:39158",
|
|
)
|
|
|
|
model_config = SettingsConfigDict(
|
|
env_file=".env",
|
|
env_prefix="RISK_",
|
|
extra="ignore",
|
|
case_sensitive=False,
|
|
)
|
|
|
|
@field_validator("database_url")
|
|
@classmethod
|
|
def validate_database_url(cls, value: str) -> str:
|
|
if not value.startswith("postgresql+asyncpg://"):
|
|
raise ValueError("database_url must use the postgresql+asyncpg:// scheme")
|
|
return value
|
|
|
|
@field_validator("llm_base_url")
|
|
@classmethod
|
|
def normalize_llm_base_url(cls, value: str) -> str:
|
|
return value.rstrip("/")
|
|
|
|
@field_validator("cors_allow_origins", mode="before")
|
|
@classmethod
|
|
def split_cors_origins(cls, value: object) -> object:
|
|
if isinstance(value, str):
|
|
value = tuple(item.strip() for item in value.split(",") if item.strip())
|
|
if isinstance(value, (list, tuple)) and "*" in value:
|
|
raise ValueError("cors_allow_origins must not contain '*' wildcard")
|
|
return value
|
|
|
|
|
|
@lru_cache(maxsize=1)
|
|
def get_settings() -> Settings:
|
|
return Settings()
|