MCP 客户端功能

This commit is contained in:
2025-09-05 20:37:15 +08:00
parent 7eb22743d8
commit b2914be3c1
18 changed files with 978 additions and 123 deletions

View File

@ -27,6 +27,7 @@ from nonebot_plugin_argot.extension import ArgotExtension # type: ignore
from .config import config
from .constants import INTRODUCTION, SUPPORT_IMAGE_MODELS
from .extensions.mcp_extension.client import get_mcp_list
from .handler import MarshoHandler
from .hooks import * # noqa: F403
from .instances import client, context, model_name, target_list, tools
@ -263,8 +264,10 @@ async def marsho(
usermsg = await handler.process_user_input(text, model_name)
tools_lists = tools.tools_list + list(
map(lambda v: v.data(), get_function_calls().values())
tools_lists = (
tools.tools_list
+ list(map(lambda v: v.data(), get_function_calls().values()))
+ await get_mcp_list()
)
logger.info(f"正在获取回答,模型:{model_name}")
await message_reaction(Emoji("66"))