364 lines
16 KiB
Python
364 lines
16 KiB
Python
from pathlib import Path
|
||
import os
|
||
|
||
from dotenv import load_dotenv
|
||
|
||
|
||
BASE_DIR = Path(__file__).resolve().parents[2]
|
||
load_dotenv(BASE_DIR / ".env")
|
||
|
||
|
||
def env(name: str, default: str | None = None) -> str | None:
|
||
return os.getenv(name, default)
|
||
|
||
|
||
def env_bool(name: str, default: bool = False) -> bool:
|
||
value = os.getenv(name)
|
||
if value is None:
|
||
return default
|
||
return value.lower() in {"1", "true", "yes", "on"}
|
||
|
||
|
||
def env_list(name: str, default: str = "") -> list[str]:
|
||
value = os.getenv(name, default)
|
||
return [item.strip() for item in value.split(",") if item.strip()]
|
||
|
||
|
||
def env_int(name: str, default: int) -> int:
|
||
"""读取整数环境变量；格式错误时给出可直接定位配置项的中文错误。"""
|
||
value = os.getenv(name)
|
||
if value is None or not value.strip():
|
||
return default
|
||
try:
|
||
return int(value)
|
||
except ValueError as exc:
|
||
raise ValueError(f"环境变量 {name} 必须是整数，当前值：{value!r}") from exc
|
||
|
||
|
||
def env_float(name: str, default: float) -> float:
|
||
"""读取小数环境变量；格式错误时给出可直接定位配置项的中文错误。"""
|
||
value = os.getenv(name)
|
||
if value is None or not value.strip():
|
||
return default
|
||
try:
|
||
return float(value)
|
||
except ValueError as exc:
|
||
raise ValueError(f"环境变量 {name} 必须是数字，当前值：{value!r}") from exc
|
||
|
||
|
||
def env_int_list(name: str, default: tuple[int, ...]) -> list[int]:
|
||
"""读取逗号分隔的整数列表；空字符串表示不重试。"""
|
||
value = os.getenv(name)
|
||
if value is None:
|
||
return list(default)
|
||
if not value.strip():
|
||
return []
|
||
try:
|
||
return [int(item.strip()) for item in value.split(",") if item.strip()]
|
||
except ValueError as exc:
|
||
raise ValueError(f"环境变量 {name} 必须是逗号分隔的整数列表，当前值：{value!r}") from exc
|
||
|
||
|
||
SECRET_KEY = env("DJANGO_SECRET_KEY", "airshelf-dev-insecure-key")
|
||
DEBUG = env_bool("DJANGO_DEBUG", False)
|
||
ALLOWED_HOSTS = env_list("DJANGO_ALLOWED_HOSTS", "localhost,127.0.0.1")
|
||
CSRF_TRUSTED_ORIGINS = env_list("DJANGO_CSRF_TRUSTED_ORIGINS")
|
||
|
||
INSTALLED_APPS = [
|
||
"django.contrib.admin",
|
||
"django.contrib.auth",
|
||
"django.contrib.contenttypes",
|
||
"django.contrib.sessions",
|
||
"django.contrib.messages",
|
||
"django.contrib.staticfiles",
|
||
"rest_framework",
|
||
"rest_framework.authtoken",
|
||
"corsheaders",
|
||
"apps.common",
|
||
"apps.accounts",
|
||
"apps.assets",
|
||
"apps.products",
|
||
"apps.projects",
|
||
"apps.ai",
|
||
"apps.billing",
|
||
"apps.ops",
|
||
"apps.adminpanel",
|
||
]
|
||
|
||
MIDDLEWARE = [
|
||
"django.middleware.security.SecurityMiddleware",
|
||
"corsheaders.middleware.CorsMiddleware",
|
||
"django.contrib.sessions.middleware.SessionMiddleware",
|
||
"django.middleware.common.CommonMiddleware",
|
||
"django.middleware.csrf.CsrfViewMiddleware",
|
||
"django.contrib.auth.middleware.AuthenticationMiddleware",
|
||
"django.contrib.messages.middleware.MessageMiddleware",
|
||
"django.middleware.clickjacking.XFrameOptionsMiddleware",
|
||
]
|
||
|
||
ROOT_URLCONF = "airshelf.urls"
|
||
|
||
TEMPLATES = [
|
||
{
|
||
"BACKEND": "django.template.backends.django.DjangoTemplates",
|
||
"DIRS": [],
|
||
"APP_DIRS": True,
|
||
"OPTIONS": {
|
||
"context_processors": [
|
||
"django.template.context_processors.debug",
|
||
"django.template.context_processors.request",
|
||
"django.contrib.auth.context_processors.auth",
|
||
"django.contrib.messages.context_processors.messages",
|
||
],
|
||
},
|
||
}
|
||
]
|
||
|
||
WSGI_APPLICATION = "airshelf.wsgi.application"
|
||
ASGI_APPLICATION = "airshelf.asgi.application"
|
||
|
||
if env("DB_ENGINE", "sqlite") == "mysql":
|
||
mysql_options = {
|
||
"charset": "utf8mb4",
|
||
"init_command": "SET sql_mode='STRICT_TRANS_TABLES'",
|
||
}
|
||
if env("DB_BIND_ADDRESS"):
|
||
mysql_options["bind_address"] = env("DB_BIND_ADDRESS")
|
||
DATABASES = {
|
||
"default": {
|
||
"ENGINE": "django.db.backends.mysql",
|
||
"NAME": env("DB_NAME", "airshelf"),
|
||
"USER": env("DB_USER", "airshelf"),
|
||
"PASSWORD": env("DB_PASSWORD", ""),
|
||
"HOST": env("DB_HOST", "127.0.0.1"),
|
||
"PORT": env("DB_PORT", "3306"),
|
||
"OPTIONS": mysql_options,
|
||
}
|
||
}
|
||
else:
|
||
DATABASES = {
|
||
"default": {
|
||
"ENGINE": "django.db.backends.sqlite3",
|
||
"NAME": BASE_DIR / "db.sqlite3",
|
||
}
|
||
}
|
||
|
||
# 连接复用(development + production 共用):连远程 MySQL 时,默认每请求重开连接
|
||
# (CONN_MAX_AGE=0),一次 TCP 握手 ~1.6s,登录时前端并发十几个接口逐个重握手 = 卡几十秒。
|
||
# 复用连接 + 健康检查 + 连接超时,放在 base 让两套环境都生效(sqlite 测试库不受影响)。
|
||
if DATABASES["default"]["ENGINE"].endswith("mysql"):
|
||
DATABASES["default"]["CONN_MAX_AGE"] = 300
|
||
DATABASES["default"]["CONN_HEALTH_CHECKS"] = True
|
||
DATABASES["default"].setdefault("OPTIONS", {})["connect_timeout"] = 5
|
||
|
||
AUTH_USER_MODEL = "accounts.User"
|
||
|
||
AUTH_PASSWORD_VALIDATORS = [
|
||
{"NAME": "django.contrib.auth.password_validation.UserAttributeSimilarityValidator"},
|
||
{"NAME": "django.contrib.auth.password_validation.MinimumLengthValidator"},
|
||
{"NAME": "django.contrib.auth.password_validation.CommonPasswordValidator"},
|
||
{"NAME": "django.contrib.auth.password_validation.NumericPasswordValidator"},
|
||
]
|
||
|
||
LANGUAGE_CODE = "zh-hans"
|
||
TIME_ZONE = "Asia/Shanghai"
|
||
USE_I18N = True
|
||
USE_TZ = True
|
||
|
||
STATIC_URL = "static/"
|
||
DEFAULT_AUTO_FIELD = "django.db.models.BigAutoField"
|
||
|
||
REST_FRAMEWORK = {
|
||
"DEFAULT_AUTHENTICATION_CLASSES": [
|
||
"rest_framework.authentication.TokenAuthentication",
|
||
"rest_framework.authentication.SessionAuthentication",
|
||
"rest_framework.authentication.BasicAuthentication",
|
||
],
|
||
"DEFAULT_PERMISSION_CLASSES": [
|
||
"rest_framework.permissions.IsAuthenticated",
|
||
],
|
||
"DEFAULT_PAGINATION_CLASS": "apps.common.pagination.DefaultPagination",
|
||
"DEFAULT_FILTER_BACKENDS": [
|
||
"rest_framework.filters.SearchFilter",
|
||
"rest_framework.filters.OrderingFilter",
|
||
],
|
||
"PAGE_SIZE": 20,
|
||
}
|
||
|
||
CORS_ALLOWED_ORIGINS = env_list("CORS_ALLOWED_ORIGINS")
|
||
CORS_ALLOW_CREDENTIALS = True
|
||
|
||
CACHES = {
|
||
"default": {
|
||
"BACKEND": "django.core.cache.backends.redis.RedisCache",
|
||
"LOCATION": env("REDIS_CACHE_URL", "redis://127.0.0.1:6379/0"),
|
||
}
|
||
}
|
||
|
||
CELERY_BROKER_URL = env("CELERY_BROKER_URL", "redis://127.0.0.1:6379/1")
|
||
CELERY_RESULT_BACKEND = env("CELERY_RESULT_BACKEND", "redis://127.0.0.1:6379/2")
|
||
# 本地端到端联调开关:true = 任务同进程内联执行,不碰共享 Redis 队列(线上 worker 拓扑不受影响)。
|
||
# 生产/集群绝不开;test.py 恒为 True。自重排类任务(poll_free_video_task)在 eager 下会跳过重排防递归。
|
||
CELERY_TASK_ALWAYS_EAGER = str(env("CELERY_TASK_ALWAYS_EAGER", "false")).lower() == "true"
|
||
CELERY_TASK_ACKS_LATE = True
|
||
CELERY_TASK_REJECT_ON_WORKER_LOST = True
|
||
CELERY_WORKER_PREFETCH_MULTIPLIER = 1
|
||
CELERY_TIMEZONE = TIME_ZONE
|
||
|
||
REDIS_LOCK_URL = env("REDIS_LOCK_URL", "redis://127.0.0.1:6379/3")
|
||
|
||
TOS = {
|
||
"endpoint": env("TOS_ENDPOINT"),
|
||
"bucket": env("TOS_BUCKET"),
|
||
"access_key_id": env("TOS_ACCESS_KEY_ID"),
|
||
"secret_access_key": env("TOS_SECRET_ACCESS_KEY"),
|
||
}
|
||
|
||
VOLCANO = {
|
||
"ark_api_key": env("VOLCANO_ARK_API_KEY"),
|
||
"ark_base_url": env("VOLCANO_ARK_BASE_URL", "https://ark.cn-beijing.volces.com/api/v3"),
|
||
}
|
||
|
||
# 豆包语音合成(旁白配音 TTS)。与 ARK 不是同一套钥匙:
|
||
# 火山引擎控制台 → 语音技术 → 语音合成大模型 → 创建应用,取 APPID 和 Access Token
|
||
VOLC_TTS = {
|
||
"appid": env("VOLC_TTS_APPID", ""),
|
||
"access_token": env("VOLC_TTS_ACCESS_TOKEN", ""),
|
||
"cluster": env("VOLC_TTS_CLUSTER", "volcano_tts"),
|
||
"base_url": env("VOLC_TTS_BASE_URL", "https://openspeech.bytedance.com/api/v1/tts"),
|
||
}
|
||
|
||
YUNQI = {
|
||
"api_key": env("YUNQI_API_KEY"),
|
||
"base_url": env("YUNQI_BASE_URL", "https://www.yunqiai.chat/v1"),
|
||
}
|
||
|
||
# tokenssr 中转站:一把 key 通吃 ~90 模型(gpt-image-2 参考图 / gpt-5.x / 全套 Gemini 文本&图像)。
|
||
TOKENSSR = {
|
||
"api_key": env("TOKENSSR_API_KEY", ""),
|
||
"base_url": env("TOKENSSR_BASE_URL", "https://king.tokenssr.com/v1"),
|
||
}
|
||
|
||
# 中转站凭证回退表(provider.name → .env)。可插拔解析顺序见 services.resolve_provider_credentials:
|
||
# DB 的 ModelProvider.base_url/api_key 优先,留空才回退到这里。密钥只在 .env,不写死、不强制进库。
|
||
PROVIDER_BASE_URLS = {
|
||
"volcengine": env("VOLCANO_ARK_BASE_URL", "https://ark.cn-beijing.volces.com/api/v3"),
|
||
"yunqi": env("YUNQI_BASE_URL", "https://www.yunqiai.chat/v1"),
|
||
# YunQi 对话(脚本助手):gpt-5.5 与 gemini-3.1-pro-preview 在 YunQi 同一地址、各有独立计费 key,
|
||
# 故拆成两个 provider(一 provider 一 key 是现有约定)。base_url 都指 YunQi。
|
||
"yunqi_gpt": env("YUNQI_BASE_URL", "https://www.yunqiai.chat/v1"),
|
||
"yunqi_gemini": env("YUNQI_BASE_URL", "https://www.yunqiai.chat/v1"),
|
||
"yunqi_gemini_guanzhuan": env("YUNQI_BASE_URL", "https://www.yunqiai.chat/v1"),
|
||
"tokenssr": env("TOKENSSR_BASE_URL", "https://king.tokenssr.com/v1"),
|
||
}
|
||
PROVIDER_KEYS = {
|
||
"volcengine": env("VOLCANO_ARK_API_KEY", ""),
|
||
"yunqi": env("YUNQI_API_KEY", ""),
|
||
"yunqi_gpt": env("YUNQI_GPT_API_KEY", ""),
|
||
"yunqi_gemini": env("YUNQI_GEMINI_API_KEY", ""),
|
||
"yunqi_gemini_guanzhuan": env("YUNQI_GEMINI_API_KEY", ""),
|
||
"tokenssr": env("TOKENSSR_API_KEY", ""),
|
||
}
|
||
# 临时:视频(Seedance)单独借 AirDrama 火山账号的 ARK key。真人素材库审核(ASSETS_API)建在 AirDrama 账号下,
|
||
# Seedance 引用 asset:// 素材必须同账号才解析得到(否则 asset not found)。只覆盖 video,图像/文本仍用自有 key。
|
||
# 待 AirShelf 自有账号开通人像素材库后,清空此项即回落默认 VOLCANO_ARK_API_KEY。
|
||
VIDEO_ARK_API_KEY = env("VIDEO_ARK_API_KEY", "")
|
||
# 中转站生图/改图端点的 Azure 风格 api-version(yunqi 的 images/edits 必须带,否则 404)。
|
||
# DB 的 ModelProvider.metadata.api_version 优先,留空回退到这里。
|
||
PROVIDER_API_VERSIONS = {
|
||
"yunqi": env("YUNQI_API_VERSION", "2025-04-01-preview"),
|
||
}
|
||
|
||
# 自由创作视频:团队在途任务并发上限(视频长时高价,限并发同时压 web 慢调用敞口)
|
||
FREE_VIDEO_MAX_CONCURRENT = int(env("FREE_VIDEO_MAX_CONCURRENT", "3"))
|
||
|
||
# 模型失败重试与动态 Fallback 的统一策略配置，已由全部 AI 业务入口与调用审计共用。
|
||
#
|
||
# 生效方式：修改环境变量后，需要同步重启 Django API 与 Celery Worker，确保两端读取同一策略。
|
||
# 时间单位：除 jitter_ratio 外，所有 timeout、cap 和 retry_delays 均为“秒”。
|
||
# 次数口径：max_calls 包含首次调用、原模型重试和所有候选模型调用；max_models 包含主模型。
|
||
MODEL_ROUTING_POLICY = {
|
||
# 单个用户逻辑任务最多尝试的不同模型数量；包含用户选择或系统默认的主模型。
|
||
# 合法范围 1～10。调大可提高成功率，但会增加等待时间和平台真实成本。
|
||
"max_models": env_int("MODEL_ROUTING_MAX_MODELS", 3),
|
||
# 单个用户逻辑任务最多发出的真实模型请求总数；包含首次调用、重试和 Fallback。
|
||
# 合法范围 1～20，且不得小于 max_models。用户侧仍只允许一次预留和一次最终结算。
|
||
"max_calls": env_int("MODEL_ROUTING_MAX_CALLS", 5),
|
||
# 退避时间随机抖动比例；0.20 表示在基础等待时间上随机浮动 ±20%。
|
||
# 合法范围 0～1。调大可减轻并发任务同时重试，但会让实际等待时间波动更明显。
|
||
"jitter_ratio": env_float("MODEL_ROUTING_JITTER_RATIO", 0.20),
|
||
"text": {
|
||
# 文本主模型失败后的重试等待时间；列表长度就是重试次数，默认 1 秒、3 秒（最多重试 2 次）。
|
||
"retry_delays": env_int_list("MODEL_ROUTING_TEXT_RETRY_DELAYS", (1, 3)),
|
||
# 普通非流式文本请求的单次最长等待时间。
|
||
"request_timeout": env_int("MODEL_ROUTING_TEXT_REQUEST_TIMEOUT", 120),
|
||
# 流式文本请求的单次最长等待时间；脚本 SSE 正常生成可能明显慢于普通文本。
|
||
"stream_timeout": env_int("MODEL_ROUTING_TEXT_STREAM_TIMEOUT", 300),
|
||
# 单个文本逻辑任务从首次调用 Provider 起允许的总执行时间，默认 8 分钟。
|
||
"total_timeout": env_int("MODEL_ROUTING_TEXT_TOTAL_TIMEOUT", 480),
|
||
# 429 响应 Retry-After 的最长接受时间；超过该值时不继续长时间等待，转入 Fallback 或最终失败。
|
||
"retry_after_cap": env_int("MODEL_ROUTING_TEXT_RETRY_AFTER_CAP", 15),
|
||
},
|
||
"image": {
|
||
# 图片主模型失败后的重试等待时间；默认等待 3 秒后重试 1 次。
|
||
"retry_delays": env_int_list("MODEL_ROUTING_IMAGE_RETRY_DELAYS", (3,)),
|
||
# 单次图片生成或编辑请求的最长等待时间；中转站真实出图可能超过 75 秒。
|
||
"request_timeout": env_int("MODEL_ROUTING_IMAGE_REQUEST_TIMEOUT", 300),
|
||
# 单张图片逻辑任务的总执行时间，默认 15 分钟。
|
||
"total_timeout": env_int("MODEL_ROUTING_IMAGE_TOTAL_TIMEOUT", 900),
|
||
# 图片 429 Retry-After 的最长接受时间。
|
||
"retry_after_cap": env_int("MODEL_ROUTING_IMAGE_RETRY_AFTER_CAP", 15),
|
||
},
|
||
"audio": {
|
||
# 配音主模型失败后的重试等待时间；默认等待 2 秒后重试 1 次。
|
||
"retry_delays": env_int_list("MODEL_ROUTING_AUDIO_RETRY_DELAYS", (2,)),
|
||
# 单次配音请求的最长等待时间。
|
||
"request_timeout": env_int("MODEL_ROUTING_AUDIO_REQUEST_TIMEOUT", 60),
|
||
# 单个配音逻辑任务的总执行时间，默认 3 分钟。
|
||
"total_timeout": env_int("MODEL_ROUTING_AUDIO_TOTAL_TIMEOUT", 180),
|
||
# 配音 429 Retry-After 的最长接受时间。
|
||
"retry_after_cap": env_int("MODEL_ROUTING_AUDIO_RETRY_AFTER_CAP", 15),
|
||
},
|
||
"video": {
|
||
# 视频尚未拿到 Provider 任务 ID 时的提交重试等待时间；默认等待 3 秒后重试 1 次。
|
||
"submit_retry_delays": env_int_list("MODEL_ROUTING_VIDEO_SUBMIT_RETRY_DELAYS", (3,)),
|
||
# 单次创建视频 Provider 任务的最长等待时间。
|
||
"submit_timeout": env_int("MODEL_ROUTING_VIDEO_SUBMIT_TIMEOUT", 120),
|
||
# 视频“提交与 Fallback”阶段总时限，默认 5 分钟；成功拿到 Provider 任务 ID 后结束该阶段。
|
||
"submit_total_timeout": env_int("MODEL_ROUTING_VIDEO_SUBMIT_TOTAL_TIMEOUT", 300),
|
||
# 单次查询视频任务状态的最长等待时间。
|
||
"poll_request_timeout": env_int("MODEL_ROUTING_VIDEO_POLL_REQUEST_TIMEOUT", 60),
|
||
# 视频提交遇到 429 时，Retry-After 的最长接受时间。
|
||
"retry_after_cap": env_int("MODEL_ROUTING_VIDEO_RETRY_AFTER_CAP", 30),
|
||
},
|
||
"postprocess": {
|
||
# 模型成功后的下载、上传和资产保存重试等待时间；不增加模型调用次数，也不触发模型 Fallback。
|
||
"retry_delays": env_int_list("MODEL_ROUTING_POSTPROCESS_RETRY_DELAYS", (2, 5, 10)),
|
||
},
|
||
}
|
||
|
||
# 模特库三视图真实生成与计费开关：必须在 API 与 Celery worker 同步部署后才显式开启。
|
||
# 关闭时提交接口直接拒绝；开启后允许预留积分并把生成任务投递给 worker。
|
||
MODEL_TRIVIEW_GENERATION_ENABLED = env_bool("MODEL_TRIVIEW_GENERATION_ENABLED", False)
|
||
|
||
# 模特上身图结构化提示词 V2.2 全量开关：
|
||
# - true：所有团队的 mode=model 商品上身图使用 V2.2 提示词；不替用户切换 Seedream/GPT，不改变数量、比例或扣费；
|
||
# - false：仅 CANARY_TEAM_IDS 命中的团队使用 V2.2；白名单也为空时全部沿用旧提示词；
|
||
# 修改后必须同步重启 API 与 Celery Worker；紧急回退设为 false 并清空白名单。
|
||
MODEL_TRYON_PROMPT_V2_ENABLED = env_bool("MODEL_TRYON_PROMPT_V2_ENABLED", False)
|
||
# Step 5.2 单团队试点：全局开关关闭时，仅命中该白名单的团队使用 V2；空值与无效值均不会误开全量。
|
||
MODEL_TRYON_PROMPT_V2_CANARY_TEAM_IDS = frozenset(
|
||
item.lower() for item in env_list("MODEL_TRYON_PROMPT_V2_CANARY_TEAM_IDS")
|
||
)
|
||
|
||
# 火山引擎人像素材库审核(真人资产绿/红盾)· AK/SK 暂借 AirDrama 已邀测账号,张业昌待换成 AirShelf 自有
|
||
ASSETS_API = {
|
||
"access_key": env("ASSETS_API_ACCESS_KEY", ""),
|
||
"secret_key": env("ASSETS_API_SECRET_KEY", ""),
|
||
"enabled": str(env("ASSETS_API_ENABLED", "false")).lower() == "true",
|
||
"project_name": env("ASSETS_API_PROJECT_NAME", "int_dev_Airlabs"),
|
||
}
|
||
|
||
# 开户赠送额度(积分)。商业决策(2026-07-03):不赠送(=0);genesis 流水仅 trial>0 才落(I8 天然满足)。
|
||
DEFAULT_TRIAL_CREDITS = env("DEFAULT_TRIAL_CREDITS", "0")
|