Initial commit of eigent-main

backend/app/utils/toolkit/google_gmail_mcp_toolkit.py

@@ -0,0 +1,53 @@
+from camel.toolkits import BaseToolkit, FunctionTool, MCPToolkit
+from app.component.environment import env, env_or_fail
+from app.component.command import bun
+from app.service.task import Agents
+from app.utils.toolkit.abstract_toolkit import AbstractToolkit
+
+
+class GoogleGmailMCPToolkit(BaseToolkit, AbstractToolkit):
+    agent_name: str = Agents.social_medium_agent
+
+    def __init__(
+        self,
+        api_task_id: str,
+        credentials_path: str | None = None,
+        timeout: float | None = None,
+        input_env: dict[str, str] | None = None,
+    ):
+        super().__init__(timeout)
+        self.api_task_id = api_task_id
+        credentials_path = credentials_path or env("GMAIL_CREDENTIALS_PATH")
+        self._mcp_toolkit = MCPToolkit(
+            config_dict={
+                "mcpServers": {
+                    "gmail": {
+                        "command": bun(),
+                        "args": ["x", "-y", "@gongrzhe/server-gmail-autoauth-mcp"],
+                        "env": {"GMAIL_CREDENTIALS_PATH": credentials_path, **(input_env or {})},
+                    }
+                }
+            },
+            timeout=timeout,
+        )
+
+    async def connect(self):
+        await self._mcp_toolkit.connect()
+
+    async def disconnect(self):
+        await self._mcp_toolkit.disconnect()
+
+    def get_tools(self) -> list[FunctionTool]:
+        return self._mcp_toolkit.get_tools()
+
+    @classmethod
+    async def get_can_use_tools(cls, api_task_id: str, input_env: dict[str, str] | None = None) -> list[FunctionTool]:
+        if env("GMAIL_CREDENTIALS_PATH") is None:
+            return []
+        toolkit = cls(api_task_id, env_or_fail("GMAIL_CREDENTIALS_PATH"), 180, input_env)
+        await toolkit.connect()
+        tools = []
+        for item in toolkit.get_tools():
+            setattr(item, "_toolkit_name", cls.__name__)
+            tools.append(item)
+        return tools