fund-tracer/backend/app/schemas/assessment.py

from datetime import datetime
from uuid import UUID

from app.models.assessment import ConfidenceLevel, ReviewStatus
from app.schemas.base import CamelModel
from app.schemas.transaction import TransactionOut


class AssessmentOut(CamelModel):
    id: UUID
    case_id: UUID
    transaction_id: UUID
    transaction: TransactionOut | None = None
    confidence_level: ConfidenceLevel
    assessed_amount: float
    reason: str
    exclude_reason: str
    review_status: ReviewStatus
    review_note: str
    reviewed_by: str
    reviewed_at: datetime | None = None


class AssessmentListOut(CamelModel):
    items: list[AssessmentOut]
    total: int


class ReviewSubmit(CamelModel):
    review_status: ReviewStatus
    review_note: str = ""
    reviewed_by: str = "demo_user"


class InquirySuggestionOut(CamelModel):
    suggestions: list[str]