mirror of
https://github.com/Sea-Haven-Industries/open-swe.git
synced 2026-09-30 09:13:14 +00:00
860aee48ee
* feat: add user-scoped Notion MCP OAuth Co-authored-by: open-swe[bot] <open-swe@users.noreply.github.com> * fix: refresh Notion token per tool call Co-authored-by: open-swe[bot] <open-swe@users.noreply.github.com> * fix: normalize Notion wrapper response format Co-authored-by: open-swe[bot] <open-swe@users.noreply.github.com> --------- Co-authored-by: open-swe[bot] <open-swe@users.noreply.github.com>
101 lines
3.2 KiB
Python
101 lines
3.2 KiB
Python
"""Server-side Notion tools backed by Notion's hosted MCP server."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
import logging
|
|
from datetime import timedelta
|
|
from typing import Any
|
|
|
|
from langchain_core.tools import BaseTool
|
|
|
|
from ..dashboard.notion_oauth import NOTION_MCP_URL
|
|
from ..dashboard.user_credentials import get_notion_access_token
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
_MCP_TIMEOUT_SECONDS = 30.0
|
|
|
|
|
|
async def _build_mcp_tools(access_token: str) -> list[BaseTool]:
|
|
from langchain_mcp_adapters.client import MultiServerMCPClient
|
|
|
|
client = MultiServerMCPClient(
|
|
{
|
|
"notion": {
|
|
"transport": "streamable_http",
|
|
"url": NOTION_MCP_URL,
|
|
"headers": {
|
|
"Authorization": f"Bearer {access_token}",
|
|
},
|
|
"timeout": timedelta(seconds=_MCP_TIMEOUT_SECONDS),
|
|
}
|
|
}
|
|
)
|
|
return await client.get_tools()
|
|
|
|
|
|
async def _fresh_mcp_tool(login: str, tool_name: str) -> BaseTool:
|
|
access_token = await get_notion_access_token(login)
|
|
if not access_token:
|
|
raise RuntimeError(
|
|
"Notion MCP authorization unavailable; reconnect Notion in Profile Settings"
|
|
)
|
|
tools = await _build_mcp_tools(access_token)
|
|
for tool in tools:
|
|
if tool.name == tool_name:
|
|
return tool
|
|
raise RuntimeError(f"Notion MCP tool {tool_name!r} is no longer available")
|
|
|
|
|
|
def _tool_input(args: tuple[Any, ...], kwargs: dict[str, Any]) -> str | dict[str, Any]:
|
|
if args and kwargs:
|
|
raise TypeError("Notion MCP tool received both positional and keyword input")
|
|
if not args:
|
|
return kwargs
|
|
if len(args) == 1 and isinstance(args[0], str):
|
|
return args[0]
|
|
if len(args) == 1 and isinstance(args[0], dict):
|
|
return args[0]
|
|
raise TypeError("Notion MCP tool received invalid positional input")
|
|
|
|
|
|
class _RefreshingNotionMCPTool(BaseTool):
|
|
login: str
|
|
mcp_tool_name: str
|
|
|
|
def _run(self, *args: Any, **kwargs: Any) -> Any:
|
|
try:
|
|
asyncio.get_running_loop()
|
|
except RuntimeError:
|
|
return asyncio.run(self._arun(*args, **kwargs))
|
|
raise RuntimeError("Notion MCP tools must be called asynchronously")
|
|
|
|
async def _arun(self, *args: Any, **kwargs: Any) -> Any:
|
|
tool = await _fresh_mcp_tool(self.login, self.mcp_tool_name)
|
|
return await tool.ainvoke(_tool_input(args, kwargs))
|
|
|
|
|
|
def _refreshing_tool(login: str, tool: BaseTool) -> BaseTool:
|
|
return _RefreshingNotionMCPTool(
|
|
name=tool.name,
|
|
description=tool.description,
|
|
args_schema=tool.args_schema,
|
|
response_format="content",
|
|
login=login,
|
|
mcp_tool_name=tool.name,
|
|
)
|
|
|
|
|
|
async def load_notion_tools(login: str) -> list[BaseTool]:
|
|
"""Return Notion MCP tools for a connected user."""
|
|
access_token = await get_notion_access_token(login)
|
|
if not access_token:
|
|
return []
|
|
try:
|
|
tools = await _build_mcp_tools(access_token)
|
|
except Exception: # noqa: BLE001
|
|
logger.warning("Failed to load Notion MCP tools", exc_info=True)
|
|
return []
|
|
logger.info("Loaded %d Notion MCP tool(s) for %s", len(tools), login)
|
|
return [_refreshing_tool(login, tool) for tool in tools]
|