src/genai/orchestration_agent/utils/state.py

from pydantic import BaseModel, Field
from typing import List


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



class UserReferenceResponseFormatter(BaseModel):
  video_idea: str = Field(description="Returns the video idea as it is, otherwise return null")
  video_story: str = Field(description="Returns the video story as it is, otherwise return 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.")