ideation_agent/utils/state.py

from pydantic import BaseModel, Field
from typing import Optional
import operator
from typing import Annotated

class State(BaseModel):
  topic: list
  business_details: Optional[dict]
  ideator_response: Optional[str] = None
  critic_response: Optional[str]=None
  improver_response: Optional[str]=None
  validator1_response: Optional[str]=None
  validator2_response: Optional[str]=None
  validator3_response: Optional[str]=None
  validator4_response: Optional[str]=None
  disagreement_reason: Optional[str]=None


class QueryFormatter(BaseModel):
    video_topic: str = Field(description="The video topic that user passes to the agent")

class ValidationFormatter(BaseModel):
  result: str = Field(description="Returns **validated** if the story is validated. Returns **not validated** if story is not validated.")
  reason: str = Field(description="Returns the reason why the story is validated or not validated.")