# api/schemas.py
from pydantic import BaseModel, Field
from typing import List, Optional

class AskRequest(BaseModel):
    prompt: str
    min_confidence: float = Field(0.8, ge=0.0, le=1.0)

class AskResponse(BaseModel):
    answer: Optional[str]
    confidence: float
    warning: Optional[str] = None
    weights: Optional[List[float]] = None

class HallucinationRequest(BaseModel):
    prompt: str

class HallucinationResponse(BaseModel):
    energy: float
    hallucination_risk: float
    grounded: bool

class RerankRequest(BaseModel):
    query: str
    docs: List[str]