mirror of
https://github.com/AstrBotDevs/AstrBot
synced 2026-07-17 17:51:20 +08:00
f2c0c2a9de
This commit adds the _internal package structure for AstrBot's standardized MCP & Skills support: astrbot/_internal/mcp/: - MCPClient for MCP server connections - MCPTool wrapper for MCP tools - MCP configuration management astrbot/_internal/skills/: - SkillManager for skill lifecycle - Skill parser and loader - SkillToToolConverter for tool-based skills - Prompt builder for skills astrbot/_internal/tools/: - ToolSchema, FunctionTool, ToolSet base definitions - FunctionToolManager for tool registry - Builtin tools (cron, send_message, kb_query) - Tool providers (internal, plugin, computer) astrbot/api/: - Public API for tools (ToolRegistry, tool decorator) - Public API for MCP (get_mcp_servers, register_mcp_server) - Public API for skills (get_skill_manager, skill_to_tool)
41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
"""MCP tool wrapper."""
|
|
|
|
from datetime import timedelta
|
|
from typing import Generic
|
|
|
|
from astrbot._internal.tools.base import FunctionTool
|
|
from astrbot.core.agent.run_context import ContextWrapper, TContext
|
|
|
|
from .client import MCPClient
|
|
|
|
try:
|
|
import mcp
|
|
except (ModuleNotFoundError, ImportError):
|
|
mcp = None # type: ignore
|
|
|
|
|
|
class MCPTool(FunctionTool, Generic[TContext]):
|
|
"""A function tool that calls an MCP service."""
|
|
|
|
def __init__(
|
|
self, mcp_tool: mcp.Tool, mcp_client: MCPClient, mcp_server_name: str, **kwargs
|
|
) -> None:
|
|
super().__init__(
|
|
name=mcp_tool.name,
|
|
description=mcp_tool.description or "",
|
|
parameters=mcp_tool.inputSchema,
|
|
)
|
|
self.mcp_tool = mcp_tool
|
|
self.mcp_client = mcp_client
|
|
self.mcp_server_name = mcp_server_name
|
|
self.source = "mcp"
|
|
|
|
async def call(
|
|
self, context: ContextWrapper[TContext], **kwargs
|
|
) -> mcp.types.CallToolResult:
|
|
return await self.mcp_client.call_tool_with_reconnect(
|
|
tool_name=self.mcp_tool.name,
|
|
arguments=kwargs,
|
|
read_timeout_seconds=timedelta(seconds=context.tool_call_timeout),
|
|
)
|