agentmail-toolkit 0.1.0__py3-none-any.whl

agentmail_toolkit/openai.py

@@ -0,0 +1,32 @@
+from typing import Optional
+from agentmail import AgentMail
+from agents import FunctionTool, RunContextWrapper
+
+from .toolkit import Toolkit
+from .tools import Tool
+
+
+class AgentMailToolkit(Toolkit[FunctionTool]):
+    def __init__(self, client: Optional[AgentMail] = None):
+        super().__init__(client)
+
+    def _build_tool(self, tool: Tool):
+        async def on_invoke_tool(ctx: RunContextWrapper, input_str: str):
+            try:
+                result = self.call_method(
+                    tool.method_name,
+                    tool.params_schema.model_validate_json(input_str),
+                )
+                return result.model_dump_json()
+            except Exception as e:
+                return str(e)
+
+        params_json_schema = tool.params_schema.model_json_schema()
+        params_json_schema["additionalProperties"] = False
+
+        return FunctionTool(
+            name=tool.name,
+            description=tool.description,
+            params_json_schema=params_json_schema,
+            on_invoke_tool=on_invoke_tool,
+        )

agentmail_toolkit/schemas.py

@@ -0,0 +1,63 @@
+from typing import Optional, List
+from pydantic import BaseModel, Field
+
+
+class ListItemsParams(BaseModel):
+    limit: Optional[int] = Field(description="The maximum number of items to return")
+    last_key: Optional[str] = Field(description="The last key to use for pagination")
+
+
+class GetInboxParams(BaseModel):
+    inbox_id: str = Field(description="The ID of the inbox to get")
+
+
+class CreateInboxParams(BaseModel):
+    username: str = Field(description="The username of the inbox to create")
+    display_name: str = Field(description="The display name of the inbox to create")
+
+
+class ListThreadsParams(ListItemsParams):
+    inbox_id: str = Field(description="The ID of the inbox to list threads for")
+    received: bool = Field(description="Whether to filter by received threads")
+    sent: bool = Field(description="Whether to filter by sent threads")
+
+
+class GetThreadParams(BaseModel):
+    inbox_id: str = Field(description="The ID of the inbox to get the thread for")
+    thread_id: str = Field(description="The ID of the thread to get")
+
+
+class ListMessagesParams(ListItemsParams):
+    inbox_id: str = Field(description="The ID of the inbox to list messages for")
+
+
+class GetMessageParams(BaseModel):
+    inbox_id: str = Field(description="The ID of the inbox to get the message for")
+    message_id: str = Field(description="The ID of the message to get")
+
+
+class GetAttachmentParams(BaseModel):
+    inbox_id: str = Field(description="The ID of the inbox to get the attachment for")
+    message_id: str = Field(
+        description="The ID of the message to get the attachment for"
+    )
+    attachment_id: str = Field(description="The ID of the attachment to get")
+
+
+class SendMessageParams(BaseModel):
+    inbox_id: str = Field(description="The ID of the inbox to send the message from")
+    to: List[str] = Field(description="The list of recipients")
+    cc: Optional[List[str]] = Field(description="The list of CC recipients")
+    bcc: Optional[List[str]] = Field(description="The list of BCC recipients")
+    subject: Optional[str] = Field(description="The subject of the message")
+    text: Optional[str] = Field(description="The plain text body of the message")
+    html: Optional[str] = Field(description="The HTML body of the message")
+
+
+class ReplyToMessageParams(BaseModel):
+    inbox_id: str = Field(
+        description="The ID of the inbox to reply to the message from"
+    )
+    message_id: str = Field(description="The ID of the message to reply to")
+    text: Optional[str] = Field(description="The plain text body of the reply")
+    html: Optional[str] = Field(description="The HTML body of the reply")

agentmail_toolkit/toolkit.py

@@ -0,0 +1,36 @@
+from typing import TypeVar, Generic, List, Optional
+from abc import ABC, abstractmethod
+from pydantic import BaseModel
+from agentmail import AgentMail
+
+from .tools import Tool, tools
+
+
+T = TypeVar("T")
+
+
+class Toolkit(Generic[T], ABC):
+    _client: AgentMail = None
+    _tools: List[T] = None
+
+    def __init__(self, client: Optional[AgentMail] = None):
+        if client:
+            self._client = client
+        else:
+            self._client = AgentMail()
+
+        self._tools = [self._build_tool(tool) for tool in tools]
+
+    @abstractmethod
+    def _build_tool(self, tool: Tool) -> T:
+        pass
+
+    def call_method(self, method_name: str, args: BaseModel) -> BaseModel:
+        method = self._client
+        for part in method_name.split("."):
+            method = getattr(method, part)
+
+        return method(**args.model_dump())
+
+    def get_tools(self) -> List[T]:
+        return self._tools

agentmail_toolkit/tools.py

@@ -0,0 +1,86 @@
+from typing import List, Type
+from pydantic import BaseModel
+
+from .schemas import (
+    ListItemsParams,
+    GetInboxParams,
+    CreateInboxParams,
+    ListThreadsParams,
+    GetThreadParams,
+    ListMessagesParams,
+    GetMessageParams,
+    GetAttachmentParams,
+    SendMessageParams,
+    ReplyToMessageParams,
+)
+
+
+class Tool(BaseModel):
+    name: str
+    method_name: str
+    description: str
+    params_schema: Type[BaseModel]
+
+
+tools: List[Tool] = [
+    Tool(
+        name="list_inboxes",
+        method_name="inboxes.list",
+        description="List all inboxes",
+        params_schema=ListItemsParams,
+    ),
+    Tool(
+        name="get_inbox",
+        method_name="inboxes.get",
+        description="Get inbox by ID",
+        params_schema=GetInboxParams,
+    ),
+    Tool(
+        name="create_inbox",
+        method_name="inboxes.create",
+        description="Create a new inbox",
+        params_schema=CreateInboxParams,
+    ),
+    Tool(
+        name="list_threads",
+        method_name="threads.list",
+        description="List threads by inbox ID",
+        params_schema=ListThreadsParams,
+    ),
+    Tool(
+        name="get_thread",
+        method_name="threads.get",
+        description="Get thread by ID",
+        params_schema=GetThreadParams,
+    ),
+    Tool(
+        name="list_messages",
+        method_name="messages.list",
+        description="List messages by thread ID",
+        params_schema=ListMessagesParams,
+    ),
+    Tool(
+        name="get_message",
+        method_name="messages.get",
+        description="Get message by ID",
+        params_schema=GetMessageParams,
+    ),
+    Tool(
+        name="get_attachment",
+        method_name="attachments.get",
+        description="Get attachment by ID",
+        params_schema=GetAttachmentParams,
+    ),
+    Tool(
+        name="send_message",
+        method_name="messages.send",
+        description="Send a message",
+        params_schema=SendMessageParams,
+    ),
+    Tool(
+        name="reply_to_message",
+        method_name="messages.reply",
+        description="Reply to a message",
+        params_schema=ReplyToMessageParams,
+    ),
+]

agentmail_toolkit-0.1.0.dist-info/METADATA

@@ -0,0 +1,13 @@
+Metadata-Version: 2.4
+Name: agentmail-toolkit
+Version: 0.1.0
+Summary: AgentMail Toolkit
+Requires-Python: >=3.11
+Requires-Dist: agentmail>=0.0.16
+Requires-Dist: openai-agents>=0.0.3
+Requires-Dist: pydantic>=2.10.6
+Description-Content-Type: text/markdown
+
+# AgentMail Toolkit
+
+The AgentMail Toolkit integrates popular agent frameworks and protocols including OpenAI Agents SDK, Vercel AI SDK, and Model Context Protocol (MCP) with the AgentMail API.

agentmail_toolkit-0.1.0.dist-info/RECORD

@@ -0,0 +1,9 @@
+agentmail_toolkit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+agentmail_toolkit/openai.py,sha256=8SFyY80oT2XoJM_UG9ZuABl6E9aD-I4wXhTtNg4jd5M,1053
+agentmail_toolkit/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+agentmail_toolkit/schemas.py,sha256=RDlkjnYDhQfSSoaluTAZFCQjBVCFWAglA64cc9jVjQQ,2634
+agentmail_toolkit/toolkit.py,sha256=c9H7p8KnMLXouqPekks7aWZxCWFk_tIzwPpkFeJZLps,905
+agentmail_toolkit/tools.py,sha256=ErqM_e15iuP9BBdZ2ZSSujz0MnNegFC-Y48wCIgDZWo,2117
+agentmail_toolkit-0.1.0.dist-info/METADATA,sha256=k_dQ05zggFW5iS1Z9dc5Z8Gu9D_GmF8-NwHnWpNKsCs,448
+agentmail_toolkit-0.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+agentmail_toolkit-0.1.0.dist-info/RECORD,,

agentmail_toolkit-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.27.0
+Root-Is-Purelib: true
+Tag: py3-none-any