src/genai/analytics_chatbot/utils/schemas.py

from pydantic import BaseModel, Field
from typing import Optional , Dict , Any

class ResponseFormatter(BaseModel):
    endpoint: str = Field(description='Return the exact endpoint from the knowledge base of endpoints.')
    method: str = Field(description='Return the exact request type from the knowledge base of endpoints. GET or POST ')
    parameters: Optional[Dict[str, Optional[Any]]] = Field(
    None,
    description="Return the dictionary of parameters to pass to the endpoint. Each parameter value can be None if not specified."
    )

class CompareBodyFormatter(BaseModel):
    names: list
    frequency: str

class LatestMessageFormatter(BaseModel):
    latest_message:str

class ParameterFormatter(BaseModel):
    parameters_values:Optional[Dict[str, Optional[Any]]] = Field(None)

class EndpointFormatter(BaseModel):
    endpoint: str