from __future__ import annotations

from uuid import UUID

from pydantic import BaseModel, Field

from .operations import VectorSelectionBBox


class VectorPartitionSelectionRequest(BaseModel):
    dataset_ids: list[UUID] = Field(min_length=1, max_length=4096)
    bbox: VectorSelectionBBox
    area_id: UUID | None = None
    limit: int = Field(default=1000, ge=1, le=1000)