app/schemas.py

# app/schemas.py
"""
Pydantic schemas.
"""

from pydantic import BaseModel, Field


class GenerationRequest(BaseModel):
    prompt: str = Field(..., min_length=1)
    temperature: float = Field(default=0.7, ge=0.0, le=2.0)
    max_tokens: int = Field(default=200, ge=1, le=1024)
    stream: bool = Field(default=True)


class GenerationResponse(BaseModel):
    text: str