AgentMail

from typing import Optional, Type, Callable from pydantic import BaseModel from agentmail import AgentMail from .toolkit import Toolkit from .tools import Tool as BaseTool class Tool(BaseModel): name: str description: str params_schema: Type[BaseModel] func: Callable class AgentMailToolkit(Toolkit[Tool]): def __init__(self, client: Optional[AgentMail] = None): super().__init__(client) def _build_tool(self, tool: BaseTool): def func(**kwargs): return tool.func(self.client, kwargs) return Tool( name=tool.name, description=tool.description, params_schema=tool.params_schema, func=func, )