1、增加调用 AI 视频生成能力和虚拟素材库管理的对外api

2、增加后台apikkey管理 3、增加apikey单独的模型定价 4、增加apikey调用情况 5、完善所有数据的注释增加
2026-08-06 13:13:28 +08:00
parent a55d4d649c
commit 0c511f3451
102 changed files with 13986 additions and 41 deletions
@@ -0,0 +1,140 @@
+from datetime import datetime
+
+from pydantic import BaseModel, ConfigDict, Field, field_validator
+
+
+# 保留供其他地方使用
+def _empty_to_null(value):
+    """将空字符串转为 None，避免 Pydantic 校验失败。"""
+    if value == "" or value == "null":
+        return None
+    return value
+
+
+class ApiKeyCallableModel(BaseModel):
+    """API Key 可调用模型配置。支持 camelCase 和 snake_case 两种字段名。"""
+
+    model_config = ConfigDict(populate_by_name=True)
+
+    engine_type: str = Field(..., description="video | image", alias="engineType")
+    engine_id: str = Field(..., description="引擎ID", alias="engineId")
+    model_name: str = Field(..., description="模型名称", alias="modelName")
+
+
+class ApiKeyCreateRequest(BaseModel):
+    """创建 API Key 请求。支持 camelCase 和 snake_case 两种字段名。"""
+
+    model_config = ConfigDict(populate_by_name=True)
+
+    company_name: str = Field(..., max_length=128, description="公司名称", alias="companyName")
+    description: str | None = Field(None, description="备注")
+    callable_models: list[ApiKeyCallableModel] = Field(
+        default_factory=list, description="可调用模型列表", alias="callableModels",
+    )
+    quota_limit: float | None = Field(None, description="配额总量，NULL=无限", alias="quotaLimit")
+    quota_cycle: str | None = Field(None, description="daily | monthly | one_time | NULL=无限", alias="quotaCycle")
+    valid_from: datetime | None = Field(None, description="生效时间", alias="validFrom")
+    valid_until: datetime | None = Field(None, description="过期时间", alias="validUntil")
+    max_concurrent_video_tasks: int | None = Field(
+        None, description="最大并发视频任务数", alias="maxConcurrentVideoTasks",
+    )
+
+    @field_validator("valid_from", "valid_until", mode="before")
+    @classmethod
+    def empty_str_to_none(cls, v):
+        if v == "" or v == "null" or v == 0:
+            return None
+        return v
+
+
+class ApiKeyUpdateRequest(BaseModel):
+    """更新 API Key 请求。支持 camelCase 和 snake_case 两种字段名。"""
+
+    model_config = ConfigDict(populate_by_name=True)
+
+    company_name: str | None = Field(None, max_length=128, alias="companyName")
+    description: str | None = None
+    callable_models: list[ApiKeyCallableModel] | None = Field(None, alias="callableModels")
+    quota_limit: float | None = Field(None, alias="quotaLimit")
+    quota_cycle: str | None = Field(None, alias="quotaCycle")
+    valid_from: datetime | None = Field(None, alias="validFrom")
+    valid_until: datetime | None = Field(None, alias="validUntil")
+    max_concurrent_video_tasks: int | None = Field(None, alias="maxConcurrentVideoTasks")
+    is_active: bool | None = Field(None, alias="isActive")
+
+    @field_validator("valid_from", "valid_until", mode="before")
+    @classmethod
+    def empty_str_to_none(cls, v):
+        if v == "" or v == "null" or v == 0:
+            return None
+        return v
+
+
+class ApiKeyResponse(BaseModel):
+    """API Key 详情响应。"""
+
+    id: str
+    company_name: str
+    api_key_prefix: str = Field(..., description="Key 前缀，如 vk_xxxx****")
+    description: str | None
+    callable_models: list[ApiKeyCallableModel]
+    quota_limit: float | None
+    quota_cycle: str | None
+    quota_used: float
+    valid_from: datetime | None
+    valid_until: datetime | None
+    max_concurrent_video_tasks: int | None
+    is_active: bool
+    last_used_at: datetime | None
+    created_at: datetime
+    updated_at: datetime
+
+    model_config = ConfigDict(from_attributes=True)
+
+
+class ApiKeyCreateResponse(BaseModel):
+    """创建 API Key 响应（包含完整明文 Key，仅此一次）。"""
+
+    id: str
+    company_name: str
+    api_key: str = Field(..., description="完整 API Key，仅创建时返回一次")
+    api_key_prefix: str
+    valid_until: datetime | None
+    created_at: datetime
+
+
+class ApiKeyRevealResponse(BaseModel):
+    """揭秘 API Key 响应（随时可获取明文）。"""
+
+    id: str
+    company_name: str
+    api_key: str = Field(..., description="完整 API Key")
+    api_key_prefix: str
+
+
+class ApiKeyListItem(BaseModel):
+    """API Key 列表项。"""
+
+    id: str
+    company_name: str
+    api_key_prefix: str
+    description: str | None
+    callable_models: list[ApiKeyCallableModel] = []
+    quota_limit: float | None
+    quota_cycle: str | None
+    quota_used: float
+    is_active: bool
+    valid_from: datetime | None
+    valid_until: datetime | None
+    max_concurrent_video_tasks: int | None
+    last_used_at: datetime | None
+    created_at: datetime
+
+    model_config = ConfigDict(from_attributes=True)
+
+
+class ApiKeyListOut(BaseModel):
+    """API Key 列表响应。"""
+
+    total: int
+    items: list[ApiKeyListItem]