brainstroming_agent/utils/validators.py

from pydantic import BaseModel, ConfigDict, Field


# Pydantic model for extracted business info
class DetailsFormatter(BaseModel):
    business_type: str = Field(description="The type of the business")
    platform: str = Field(description="The platform used for the business")
    target_audience: str = Field(description="The target audience of the business")
    business_goals: str = Field(description="The business goals of the business")
    offerings: str = Field(description="The offerings of the business")
    Challenges_faced: str = Field(description="The challenges faced by the business")


class BrainstromTopicFormatter(BaseModel):
    topic1:str=Field(description="First brainstorming topic of the story")
    topic2:str=Field(description="Second brainstorming topic of the story")
    topic3:str=Field(description="Third brainstorming topic of the story")
    topic4:str=Field(description="Fourth brainstorming topic of the story")

class QueryFormatter(BaseModel):
    messages:str = Field(description="The user query")
    business_details: dict = Field(description="The details of the business of that user.")