api/models.py

from typing import Any, Dict, List, Optional
from pydantic import BaseModel


class AnalyzeRequest(BaseModel):
    skip_citation_gap: bool = False
    skip_argumentation: bool = False


class AgentStatusInfo(BaseModel):
    status: str
    duration_s: Optional[float] = None
    error: Optional[str] = None


class JobStatusResponse(BaseModel):
    job_id: str
    status: str
    progress: float = 0.0
    agents: Dict[str, AgentStatusInfo] = {}
    scores: Dict[str, float] = {}
    report_path: Optional[str] = None
    duration_s: Optional[float] = None
    error: Optional[str] = None


class AnalyzeResponse(BaseModel):
    job_id: str
    status: str
    message: str