from typing import Any, Dict, List, Optional

from pydantic import UUID4, BaseModel

from .base import ORMBase


class EnricherCreate(BaseModel):
    name: str
    description: Optional[str] = None
    category: Optional[List[str]] = None


class EnricherRead(ORMBase):
    id: UUID4
    name: str
    class_name: str
    description: Optional[str]
    category: Optional[List[str]]
    flow_schema: Optional[Dict[str, Any]]


class EnricherUpdate(BaseModel):
    name: Optional[str] = None
    class_name: Optional[str] = None
    description: Optional[str] = None
    category: Optional[List[str]] = None