video_decode/utils/params.py

from pydantic import BaseModel
from typing import List, Optional
from enum import IntEnum


class SceneEnum(IntEnum):
    """业务场景枚举"""
    TOPIC = 0  # 选题
    CREATION = 1  # 创作
    PRODUCTION = 2  # 制作

class CapabilityEnum(IntEnum):
    """能力枚举"""
    DECODE = 0  # 解构
    PATTERN = 1  # 聚类
class ContentTypeEnum(IntEnum):
    """内容类型枚举"""
    TEXT = 1  # 文本
    IMAGE = 2  # 图片
    VIDEO = 3  # 视频


class ContentParam(BaseModel):
    channel_content_id: str  # 必传
    video_url: Optional[str] = None
    images: List[str] = []
    body_text: Optional[str] = None
    title: Optional[str] = None
    channel_account_id: Optional[str] = None
    channel_account_name: Optional[str] = None
    weight_score: Optional[float] = None  # 表现力分数，聚类必传


class DecodeContentParam(BaseModel):
    scene: SceneEnum  # 业务场景：0选题 1创作 2制作
    content_type: ContentTypeEnum  # 1 文本 2图片 3 视频
    content: ContentParam

class PatternContentParam(BaseModel):
    scene: SceneEnum  # 业务场景：0选题 1创作 2制作
    content_type: ContentTypeEnum  # 1 文本 2图片 3 视频
    pattern_name: str  # 聚类名称
    contents: List[ContentParam]