ideation_agent/utils/state.py

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

class State(BaseModel):
  topic: Annotated[list[str], operator.add]
  business_details : Annotated[list[dict],operator.add] = []
  ideator_response: Annotated[list[str],operator.add] = []
  critic_response: Annotated[list[str],operator.add] = []
  improver_response: Annotated[list[str],operator.add] = []
  validator1_response: Annotated[list[str],operator.add] = []
  validator2_response: Annotated[list[str],operator.add] = []
  validator3_response: Annotated[list[str],operator.add] = []
  validator4_response: Annotated[list[str],operator.add] = []
  disagreement_reason: Annotated[list[str],operator.add] = []


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.")