3f183d14db
配 zhipu.py(zhipu 模块)一起使用: - service.py: 新增 _zhipu_translator(),引擎选择链路 spark → zhipu → tencent → maas → agnes → local - config.py: 新增 zhipu_api_key / zhipu_base_url / zhipu_model / zhipu_interval_sec - .env.example: 补 ZHIPU_* 字段说明 留空 zhipu_api_key = spark 不可用时直接降级 tencent(向后兼容)
137 lines
4.2 KiB
Python
137 lines
4.2 KiB
Python
"""应用配置:从 .env / 环境变量读取,集中管理所有开关。"""
|
|
from __future__ import annotations
|
|
|
|
from functools import lru_cache
|
|
from pathlib import Path
|
|
|
|
from pydantic import Field, field_validator
|
|
from pydantic_settings import BaseSettings, SettingsConfigDict
|
|
|
|
|
|
class Settings(BaseSettings):
|
|
model_config = SettingsConfigDict(
|
|
env_file=".env",
|
|
env_file_encoding="utf-8",
|
|
case_sensitive=False,
|
|
extra="ignore",
|
|
)
|
|
|
|
# ===== 通用 =====
|
|
tz: str = "Asia/Hong_Kong"
|
|
log_level: str = "INFO"
|
|
|
|
# ===== 数据库 =====
|
|
postgres_user: str
|
|
postgres_password: str
|
|
postgres_db: str
|
|
postgres_host: str = "postgres"
|
|
postgres_port: int = 5432
|
|
|
|
@property
|
|
def database_url(self) -> str:
|
|
# asyncpg
|
|
return (
|
|
f"postgresql+asyncpg://{self.postgres_user}:{self.postgres_password}"
|
|
f"@{self.postgres_host}:{self.postgres_port}/{self.postgres_db}"
|
|
)
|
|
|
|
@property
|
|
def sync_database_url(self) -> str:
|
|
# alembic 用的同步 URL
|
|
return (
|
|
f"postgresql+psycopg2://{self.postgres_user}:{self.postgres_password}"
|
|
f"@{self.postgres_host}:{self.postgres_port}/{self.postgres_db}"
|
|
)
|
|
|
|
# ===== Redis =====
|
|
redis_host: str = "redis"
|
|
redis_port: int = 6379
|
|
redis_password: str
|
|
redis_db: int = 0
|
|
|
|
@property
|
|
def redis_url(self) -> str:
|
|
return (
|
|
f"redis://:{self.redis_password}@{self.redis_host}:{self.redis_port}/{self.redis_db}"
|
|
)
|
|
|
|
# ===== JWT =====
|
|
jwt_secret: str
|
|
jwt_algorithm: str = "HS256"
|
|
access_token_ttl_min: int = 60
|
|
refresh_token_ttl_day: int = 14
|
|
|
|
# ===== 腾讯云 TMT =====
|
|
tencentcloud_secret_id: str = ""
|
|
tencentcloud_secret_key: str = ""
|
|
tencentcloud_region: str = "ap-hongkong"
|
|
tencent_tmt_endpoint: str = "tmt.tencentcloudapi.com"
|
|
tencent_tmt_quota_month: int = 5_000_000
|
|
tencent_tmt_quota_buffer: float = 0.05
|
|
tencent_tmt_max_chars_per_req: int = 4500
|
|
|
|
# ===== 星火 Spark(优先翻译;Lite/免费)=====
|
|
# 留空 = 不启用星火(直接走腾讯 TMT)
|
|
spark_api_password: str = ""
|
|
spark_base_url: str = "https://spark-api-open.xf-yun.com/v1"
|
|
spark_model: str = "lite"
|
|
spark_interval_sec: float = 1.0
|
|
|
|
# ===== 智谱 GLM(第二序位翻译;GLM-4-Flash 免费)=====
|
|
# 留空 = 不启用智谱(spark 不可用时直接降级到 tencent)
|
|
zhipu_api_key: str = ""
|
|
zhipu_base_url: str = "https://open.bigmodel.cn/api/paas/v4"
|
|
zhipu_model: str = "glm-4-flash"
|
|
zhipu_interval_sec: float = 1.0
|
|
|
|
@field_validator("tencent_tmt_quota_buffer")
|
|
@classmethod
|
|
def _check_buffer(cls, v: float) -> float:
|
|
if not 0.0 <= v <= 0.5:
|
|
raise ValueError("buffer 必须在 0~0.5")
|
|
return v
|
|
|
|
# ===== 本地翻译 =====
|
|
local_translate_enabled: bool = False
|
|
local_translate_model: str = "nllb-200-distilled-600M"
|
|
local_translate_device: str = "cpu"
|
|
|
|
# ===== 腾讯 MaaS(OpenAI 兼容翻译备用通道)=====
|
|
# 用法:腾讯云 MaaS 提供的翻译模型,通过 OpenAI 协议调用
|
|
# 留空 api_key = 不启用该 provider
|
|
tencent_maas_api_key: str = ""
|
|
tencent_maas_base_url: str = "https://maas-api.hivoice.cn/v1"
|
|
tencent_maas_model: str = "u2"
|
|
# 每篇调用间隔(秒),与 LLM 客户端解耦
|
|
tencent_maas_interval_sec: float = 1.0
|
|
|
|
# ===== 抓取 =====
|
|
fetch_global_qps: int = 4
|
|
fetch_timeout: int = 20
|
|
fetch_fail_pause_threshold: int = 3
|
|
fetch_max_retries: int = 2
|
|
|
|
# ===== Caddy / 域名 =====
|
|
domain: str = ""
|
|
acme_email: str = ""
|
|
|
|
# ===== Agnes LLM(智能增强)=====
|
|
# 留空 = 不启用 LLM 增强(翻译后只走默认排版,提示词也不读)
|
|
agnes_api_key: str = ""
|
|
agnes_base_url: str = "https://apihub.agnes-ai.com/v1"
|
|
agnes_chat_model: str = "agnes-2.0-flash"
|
|
agnes_image_model: str = "agnes-image-2.1-flash"
|
|
# 全局 LLM 调用间隔(秒),避免被限流
|
|
llm_interval_sec: float = 2.0
|
|
|
|
# ===== 内部路径(部署后可调) =====
|
|
project_root: Path = Path(__file__).resolve().parents[2]
|
|
|
|
|
|
@lru_cache
|
|
def get_settings() -> Settings:
|
|
return Settings() # type: ignore[call-arg]
|
|
|
|
|
|
settings = get_settings()
|