|
|
|
|
|
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] |
|
|
|
