fuquan/nonebot-plugin-llmchat
1
0
Fork 0
mirror of https://github.com/FuQuan233/nonebot-plugin-llmchat.git synced 2026-02-05 19:48:06 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

✨ 参考VSCode的mcp服务器configuration-format新增对sse服务器自定义header的支持 #22

Browse source
This commit is contained in:
FuQuan233 2025-10-13 17:47:08 +08:00
parent 4bfcaf3a61
commit 2d61bd31ae
4 changed files with 15 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

3
nonebot_plugin_llmchat/mcpclient.py
View file

@ -22,7 +22,7 @@ class MCPClient:
for server_name, config in self.server_config.items():
logger.debug(f"正在连接服务器[{server_name}]")
if config.url:
sse_transport = await self.exit_stack.enter_async_context(sse_client(url=config.url))
sse_transport = await self.exit_stack.enter_async_context(sse_client(url=config.url, headers=config.headers))
read, write = sse_transport
self.sessions[server_name] = await self.exit_stack.enter_async_context(ClientSession(read, write))
await self.sessions[server_name].initialize()
@ -74,6 +74,7 @@ class MCPClient:
return response.content
def get_friendly_name(self, tool_name: str):
logger.debug(tool_name)
server_name, real_tool_name = tool_name.split("___")
return (self.server_config[server_name].friendly_name or server_name) + " - " + real_tool_name