src/genai/human_refined_ideation/utils/state.py

from pydantic import BaseModel, Field
from typing import Optional

class State(BaseModel):
    final_ideation: list
    query:list
    result: Optional[str] = None
    business_details: dict

class IdeationFormatter(BaseModel):
    refined_idea: str = Field(description="The refined idea after human intervention.")