"""Schemas for import services."""

from typing import Dict, Optional

from pydantic import BaseModel


class ImportResult(BaseModel):
    """Common import result schema."""

    import_count: Dict[str, int]
    success: bool
    error_message: Optional[str] = None


class ChatImportResult(ImportResult):
    """Result schema for chat imports."""

    conversations: int = 0
    messages: int = 0


class ProjectImportResult(ImportResult):
    """Result schema for project imports."""

    documents: int = 0
    prompts: int = 0


class EntityImportResult(ImportResult):
    """Result schema for entity imports."""

    entities: int = 0
    relations: int = 0
    skipped_entities: int = 0