schema.py

from pydantic import BaseModel, Field
from typing import List

class PokemonEmeraldResponse(BaseModel):
    answer: str = Field(
        description="The concise answer to the user's question."
    )
    citations: List[str] = Field(
        description="Direct quotes from the manual used to answer the question. Make sure the quotes are long enough and at least a few sentences long."
    )
    confidence_score: int = Field(
        description="0 to 100 score of how well the context answered the question. Don't give 100 confidence unless you are absolutely certain."
    )