schemas/schema.py

from pydantic import BaseModel
from typing import List, Optional

class SMSRequest(BaseModel):
    messages: List[str]

class EmbeddingResponse(BaseModel):
    dimensions: int
    embeddings: List[List[float]]

    class Config:
        
        json_schema_extra = {
            "example": {
                "dimensions": 768,
                "embeddings": [
                    [0.1, 0.2, 0.3, ...],  # 768 values for each message
                    [0.4, 0.5, 0.6, ...]
                ]
            }
        }