video_decode/utils/params.py

from pydantic import BaseModel
from typing import List, Optional



class TaskStatusParam(BaseModel):
    task_id: str

class EvaluateStatusParam(BaseModel):
    evaluate_id: str
class DecodeParam(BaseModel):
    channel_content_id: str
    video: str
    title: str

class DecodeWorkflowParam(BaseModel):
    task_id: Optional[str] = None
    video_id: Optional[str] = None
    video_url: Optional[str] = None
    title: Optional[str] = None
    task_status: Optional[int] = None


class DecodeListParam(BaseModel):
    video_list: List[DecodeParam]

class TopicListParam(BaseModel):
    video_list: List[DecodeWorkflowParam]
    type: Optional[int] = 1

class EvaluateWorkflowParam(BaseModel):
    video_id: str
    channel_content_id: str
    task_id: str
    status: int
    result: str
class EvaluateListParam(BaseModel):
    evaluate_list: List[EvaluateWorkflowParam]