from typing import Any

from pydantic import BaseModel


class CaseResponse(BaseModel):
    case_id: int
    case_name: str
    extracted_elements: dict[str, Any]
    case_profile: dict[str, Any]
    similar_cases: list[dict[str, Any]]


class SimilarCaseResponse(BaseModel):
    case_id: int
    similar_cases: list[dict[str, Any]]