from typing import Any, TypedDict

from pydantic import BaseModel


class Error(BaseModel):
    message: str
    type: str
    code: int
    fbtrace_id: str
    error_subcode: int | None = None
    error_user_title: str | None = None
    error_user_msg: str | None = None
    error_data: str | dict[str, Any] | None = None


class ErrorResponse(BaseModel):
    error: Error


class Audience(BaseModel):
    id: str


class UploadUsersSession(TypedDict):
    session_id: int
    batch_seq: int
    last_batch_flag: bool
    estimated_num_total: int


class UploadUsersPayload(TypedDict):
    schema: Any
    data: Any


class UploadUsersResponse(BaseModel):
    audience_id: str
    session_id: int
    num_received: int
    num_invalid_entries: int