src/genai/orchestration_agent/utils/schemas.py

from pydantic import BaseModel, Field 
from typing import List , Optional

class ToolResponseFormatter(BaseModel):
  tools: List[str] = Field(description="Returns the name of the tool, tools, or an empty list.")

class UserReferenceResponseFormatter(BaseModel):
    video_idea: Optional[str] = Field(
        default=None,
        description="Returns the video idea as it is, otherwise null"
    )
    video_story: Optional[str] = Field(
        default=None,
        description="Returns the video story as it is, otherwise null"
    )

class ValidationFormatter(BaseModel):
    tool: List[str] = Field(description="Returns the name of the tool or tools as it is.")
    query_response: str = Field(description="Returns the reply of query as it is.")
    image_caption: str = Field(description="Returns the information of image as it is.")
    video_idea: str = Field(description="Returns the video idea as it is.")
    video_story: str = Field(description="Returns the video story as it is.")