"""Pydantic schemas for the Entity Links plugin."""

from __future__ import annotations

from pydantic import BaseModel, Field


class EntityLinkRequest(BaseModel):
    entity_type: str = Field(..., pattern="^(company|contact)$")
    entity_id: str


class EntityLinkResponse(BaseModel):
    id: str
    file_id: str
    entity_type: str
    entity_id: str


class FileListResponse(BaseModel):
    file_id: str
    entity_type: str