from abc import ABC, abstractmethod
from typing import Dict, Any, Union
from pydantic import BaseModel


class BaseAgent(ABC):
    @abstractmethod
    def run(self, *args, **kwargs) -> Union[Dict[str, Any], BaseModel]:
        pass