from pydantic import BaseModel, Field


class DoctorProfile(BaseModel):
    code_name: str
    source: str
    vector: list[float] = Field(default_factory=list)
    cluster_id: int | None = None
    traits: dict[str, float] = Field(default_factory=dict)