src/genai/analytics_chatbot/utils/schemas.py

from pydantic import BaseModel, Field

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: dict = Field(description='Return the dictionary of parameters to pass to the endpoint.')

class InfluencerNames(BaseModel):
    names: list