"Research JSON classes"

from pydantic import BaseModel, Field


class Questions(BaseModel):
    questions: list[str] = Field(..., title="List of Google search queries")


class ResearchLLM(BaseModel):
    answer: str = Field(..., title="The answer to the question as a verbose string")
    questions: Questions = Field(
        ..., title="The questions to search on Google only if needed"
    )


class ResearchLLM_AnswerOnly(BaseModel):
    answer: str = Field(..., title="The answer to the question")