seaislee1209andClaude Opus 4.8
99a442e883
索引(只 makemigrations 看 SQL,远程 MySQL apply 留用户低峰 / rule 3):
- Asset (team,category,-created_at)+(team,asset_type)、CreditLedger (team,-created_at)+
(team,ledger_type,created_at)、Project (team,-updated_at)、AITask (team,-created_at)、
Product (team,-created_at);迁移 assets0005/billing0002/projects0004/ai0008/products0002
消息中心 N+1(ops/views.py):
- type_counts 原 6 次 count(每请求)→ 一次 aggregate(Count(filter=...)) 条件聚合
- ensure_team_notifications 项目花费原逐项目 aggregate(N+1)→ 一次 values('project').annotate(Sum)
验证: apps.ops 测试 3/3 OK;全量 6fail+1err 均为预存 image provider 路由失败(stash 跑 baseline
复现)= 零新增回归;索引迁移在 sqlite 测试库成功 apply
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
110 lines
4.6 KiB
Python
110 lines
4.6 KiB
Python
from django.db import models
|
|
|
|
from apps.common.models import TeamOwnedModel, TimeStampedModel
|
|
|
|
|
|
class ModelProvider(TimeStampedModel):
|
|
class Status(models.TextChoices):
|
|
ACTIVE = "active", "Active"
|
|
DISABLED = "disabled", "Disabled"
|
|
|
|
name = models.CharField(max_length=64, unique=True)
|
|
display_name = models.CharField(max_length=128)
|
|
status = models.CharField(max_length=24, choices=Status.choices, default=Status.ACTIVE)
|
|
base_url = models.URLField(blank=True)
|
|
# 站级 API Key(中转站)。可插拔:换站 = 改这一行的 base_url + api_key,零改代码。
|
|
# 留空则 services 层按 provider.name 回退 settings.PROVIDER_KEYS(.env),避免密钥写死/进库。
|
|
api_key = models.CharField(max_length=255, blank=True)
|
|
metadata = models.JSONField(default=dict, blank=True)
|
|
|
|
def __str__(self) -> str:
|
|
return self.display_name
|
|
|
|
|
|
class ModelConfig(TimeStampedModel):
|
|
class Capability(models.TextChoices):
|
|
TEXT = "text", "Text"
|
|
IMAGE = "image", "Image"
|
|
VIDEO = "video", "Video"
|
|
VISION = "vision", "Vision"
|
|
AUDIO = "audio", "Audio"
|
|
EXPORT = "export", "Export"
|
|
|
|
class Status(models.TextChoices):
|
|
ACTIVE = "active", "Active"
|
|
DISABLED = "disabled", "Disabled"
|
|
|
|
provider = models.ForeignKey(ModelProvider, on_delete=models.CASCADE, related_name="models")
|
|
name = models.CharField(max_length=128)
|
|
display_name = models.CharField(max_length=128)
|
|
capability = models.CharField(max_length=32, choices=Capability.choices)
|
|
endpoint = models.CharField(max_length=255, blank=True)
|
|
unit_price = models.DecimalField(max_digits=12, decimal_places=4, default=0)
|
|
status = models.CharField(max_length=24, choices=Status.choices, default=Status.ACTIVE)
|
|
rate_limit_per_minute = models.PositiveIntegerField(default=60)
|
|
metadata = models.JSONField(default=dict, blank=True)
|
|
|
|
class Meta:
|
|
unique_together = [("provider", "name", "capability")]
|
|
|
|
def __str__(self) -> str:
|
|
return f"{self.provider.name}:{self.name}:{self.capability}"
|
|
|
|
|
|
class AITask(TeamOwnedModel):
|
|
class Type(models.TextChoices):
|
|
SCRIPT_GENERATION = "script_generation", "Script Generation"
|
|
SCRIPT_OPTIMIZATION = "script_optimization", "Script Optimization"
|
|
PRODUCT_IMAGE = "product_image", "Product Image"
|
|
PERSON_IMAGE = "person_image", "Person Image"
|
|
SCENE_IMAGE = "scene_image", "Scene Image"
|
|
STORYBOARD = "storyboard", "Storyboard"
|
|
VIDEO_SEGMENT = "video_segment", "Video Segment"
|
|
VOICEOVER = "voiceover", "Voiceover"
|
|
EXPORT = "export", "Export"
|
|
|
|
class Status(models.TextChoices):
|
|
CREATED = "created", "Created"
|
|
RESERVED = "reserved", "Reserved"
|
|
SUBMITTED = "submitted", "Submitted"
|
|
POLLING = "polling", "Polling"
|
|
POSTPROCESSING = "postprocessing", "Postprocessing"
|
|
SUCCEEDED = "succeeded", "Succeeded"
|
|
FAILED = "failed", "Failed"
|
|
COMPENSATING = "compensating", "Compensating"
|
|
CANCELLED = "cancelled", "Cancelled"
|
|
|
|
project = models.ForeignKey(
|
|
"projects.Project",
|
|
on_delete=models.CASCADE,
|
|
null=True,
|
|
blank=True,
|
|
related_name="ai_tasks",
|
|
)
|
|
task_type = models.CharField(max_length=48, choices=Type.choices)
|
|
status = models.CharField(max_length=32, choices=Status.choices, default=Status.CREATED)
|
|
model_config = models.ForeignKey(ModelConfig, on_delete=models.PROTECT, related_name="tasks")
|
|
idempotency_key = models.CharField(max_length=128, unique=True)
|
|
provider_task_id = models.CharField(max_length=255, blank=True)
|
|
request_payload = models.JSONField(default=dict, blank=True)
|
|
response_payload = models.JSONField(default=dict, blank=True)
|
|
estimated_cost = models.DecimalField(max_digits=12, decimal_places=4, default=0)
|
|
actual_cost = models.DecimalField(max_digits=12, decimal_places=4, default=0)
|
|
error_code = models.CharField(max_length=64, blank=True)
|
|
error_message = models.TextField(blank=True)
|
|
submitted_at = models.DateTimeField(null=True, blank=True)
|
|
completed_at = models.DateTimeField(null=True, blank=True)
|
|
|
|
class Meta:
|
|
indexes = [
|
|
models.Index(fields=["team", "status"]),
|
|
models.Index(fields=["project", "task_type"]),
|
|
models.Index(fields=["provider_task_id"]),
|
|
# 任务历史默认按团队 + 创建时间倒序(AI 工具页 / asset-factory)
|
|
models.Index(fields=["team", "-created_at"]),
|
|
]
|
|
|
|
def __str__(self) -> str:
|
|
return f"{self.task_type}:{self.status}:{self.id}"
|
|
|