Skip to content

赵月辉 / fastgpt-migrated

Go to a project
Commit ca8adbbf by Archer Committed by GitHub
Options

feat: streamable http mcp (#4695) 

* feat: streamable http mcp

* mcp api path

* fix: ts
parent d91b2ae3
Showing with 550 additions and 373 deletions
packages/service/package.json
...@@ -3,7 +3,7 @@ ...@@ -3,7 +3,7 @@
"version": "1.0.0", "version": "1.0.0",
"dependencies": { "dependencies": {
"@fastgpt/global": "workspace:*", "@fastgpt/global": "workspace:*",
"@modelcontextprotocol/sdk": "^1.10.0", "@modelcontextprotocol/sdk": "^1.10.2",
"@node-rs/jieba": "2.0.1", "@node-rs/jieba": "2.0.1",
"@xmldom/xmldom": "^0.8.10", "@xmldom/xmldom": "^0.8.10",
"@zilliz/milvus2-sdk-node": "2.4.2", "@zilliz/milvus2-sdk-node": "2.4.2",
......
packages/web/i18n/en/dashboard_mcp.json
...@@ -12,9 +12,11 @@ ...@@ -12,9 +12,11 @@
"mcp_apps": "Number of associated applications", "mcp_apps": "Number of associated applications",
"mcp_endpoints": "Access address", "mcp_endpoints": "Access address",
"mcp_json_config": "Access script", "mcp_json_config": "Access script",
"mcp_link_way": "Access method",
"mcp_name": "MCP service name", "mcp_name": "MCP service name",
"mcp_server": "MCP Services", "mcp_server": "MCP Services",
"mcp_server_description": "Allows you to select some applications to provide external use with the MCP protocol. \nDue to the immaturity of the MCP protocol, this feature is still in the beta stage.", "mcp_server_description": "Allows you to select some applications to provide external use with the MCP protocol. \nDue to the immaturity of the MCP protocol, this feature is still in the beta stage.",
"not_sse_server": "The system does not configure SSE access service",
"search_app": "Search for apps", "search_app": "Search for apps",
"select_app": "Application selection", "select_app": "Application selection",
"start_use": "Get started", "start_use": "Get started",
......
packages/web/i18n/zh-CN/dashboard_mcp.json
...@@ -12,9 +12,11 @@ ...@@ -12,9 +12,11 @@
"mcp_apps": "关联应用数量", "mcp_apps": "关联应用数量",
"mcp_endpoints": "接入地址", "mcp_endpoints": "接入地址",
"mcp_json_config": "接入脚本", "mcp_json_config": "接入脚本",
"mcp_link_way": "接入方式",
"mcp_name": "MCP 服务名", "mcp_name": "MCP 服务名",
"mcp_server": "MCP 服务", "mcp_server": "MCP 服务",
"mcp_server_description": "允许你选择部分应用，以 MCP 的协议对外提供使用。由于 MCP 协议的不成熟，该功能仍处于测试阶段。", "mcp_server_description": "允许你选择部分应用，以 MCP 的协议对外提供使用。由于 MCP 协议的不成熟，该功能仍处于测试阶段。",
"not_sse_server": "系统未配置 SSE 接入服务",
"search_app": "搜索应用", "search_app": "搜索应用",
"select_app": "应用选择", "select_app": "应用选择",
"start_use": "开始使用", "start_use": "开始使用",
......
packages/web/i18n/zh-Hant/dashboard_mcp.json
...@@ -12,9 +12,11 @@ ...@@ -12,9 +12,11 @@
"mcp_apps": "關聯應用數量", "mcp_apps": "關聯應用數量",
"mcp_endpoints": "接入地址", "mcp_endpoints": "接入地址",
"mcp_json_config": "接入腳本", "mcp_json_config": "接入腳本",
"mcp_link_way": "接入方式",
"mcp_name": "MCP 服務名", "mcp_name": "MCP 服務名",
"mcp_server": "MCP 服務", "mcp_server": "MCP 服務",
"mcp_server_description": "允許你選擇部分應用，以 MCP 的協議對外提供使用。\n由於 MCP 協議的不成熟，該功能仍處於測試階段。", "mcp_server_description": "允許你選擇部分應用，以 MCP 的協議對外提供使用。\n由於 MCP 協議的不成熟，該功能仍處於測試階段。",
"not_sse_server": "系統未配置 SSE 接入服務",
"search_app": "搜索應用", "search_app": "搜索應用",
"select_app": "應用選擇", "select_app": "應用選擇",
"start_use": "開始使用", "start_use": "開始使用",
......
projects/app/package.json
...@@ -24,7 +24,7 @@ ...@@ -24,7 +24,7 @@
"@fastgpt/templates": "workspace:*", "@fastgpt/templates": "workspace:*",
"@fastgpt/web": "workspace:*", "@fastgpt/web": "workspace:*",
"@fortaine/fetch-event-source": "^3.0.6", "@fortaine/fetch-event-source": "^3.0.6",
"@modelcontextprotocol/sdk": "^1.10.0", "@modelcontextprotocol/sdk": "^1.10.2",
"@node-rs/jieba": "2.0.1", "@node-rs/jieba": "2.0.1",
"@tanstack/react-query": "^4.24.10", "@tanstack/react-query": "^4.24.10",
"ahooks": "^3.7.11", "ahooks": "^3.7.11",
......
projects/app/src/pageComponents/dashboard/Container.tsx
...@@ -184,16 +184,12 @@ const DashboardContainer = ({ ...@@ -184,16 +184,12 @@ const DashboardContainer = ({
: []) : [])
] ]
}, },
...(feConfigs?.mcpServerProxyEndpoint {
? [ groupId: TabEnum.mcp_server,
{ groupAvatar: 'key',
groupId: TabEnum.mcp_server, groupName: t('common:mcp_server'),
groupAvatar: 'key', children: []
groupName: t('common:mcp_server'), }
children: []
}
]
: [])
]; ];
}, [currentType, feConfigs.appTemplateCourse, pluginGroups, t, templateList, templateTags]); }, [currentType, feConfigs.appTemplateCourse, pluginGroups, t, templateList, templateTags]);
......
projects/app/src/pageComponents/dashboard/mcp/usageWay.tsx
import { McpKeyType } from '@fastgpt/global/support/mcp/type'; import { McpKeyType } from '@fastgpt/global/support/mcp/type';
import MyModal from '@fastgpt/web/components/common/MyModal'; import MyModal from '@fastgpt/web/components/common/MyModal';
import React from 'react'; import React, { useState } from 'react';
import { useTranslation } from 'next-i18next'; import { useTranslation } from 'next-i18next';
import { Box, Flex, HStack, ModalBody } from '@chakra-ui/react'; import { Box, Flex, HStack, ModalBody } from '@chakra-ui/react';
import { useSystemStore } from '@/web/common/system/useSystemStore'; import { useSystemStore } from '@/web/common/system/useSystemStore';
import FormLabel from '@fastgpt/web/components/common/MyBox/FormLabel'; import FormLabel from '@fastgpt/web/components/common/MyBox/FormLabel';
import CopyBox from '@fastgpt/web/components/common/String/CopyBox'; import CopyBox from '@fastgpt/web/components/common/String/CopyBox';
import MyIconButton from '@fastgpt/web/components/common/Icon/button'; import MyIconButton from '@fastgpt/web/components/common/Icon/button';
import LightRowTabs from '@fastgpt/web/components/common/Tabs/LightRowTabs';
type LinkWay = 'sse' | 'http';
const UsageWay = ({ mcp, onClose }: { mcp: McpKeyType; onClose: () => void }) => { const UsageWay = ({ mcp, onClose }: { mcp: McpKeyType; onClose: () => void }) => {
const { t } = useTranslation(); const { t } = useTranslation();
const { feConfigs } = useSystemStore(); const { feConfigs } = useSystemStore();
const [linkWay, setLinkWay] = useState<LinkWay>('http');
const sseUrl = `${feConfigs?.mcpServerProxyEndpoint}/${mcp.key}/sse`; const { url, jsonConfig } = (() => {
const jsonConfig = `{ if (linkWay === 'http') {
const baseUrl = feConfigs?.customApiDomain || `${location.origin}/api`;
const url = `${baseUrl}/mcp/app/${mcp.key}/mcp`;
const jsonConfig = `{
"mcpServers": { "mcpServers": {
"${feConfigs?.systemTitle}-mcp-${mcp._id}": { "${feConfigs?.systemTitle}-mcp-${mcp._id}": {
"url": "${sseUrl}" "url": "${url}"
} }
} }
}`; }`;
return {
url,
jsonConfig
};
}
const url = feConfigs?.mcpServerProxyEndpoint
? `${feConfigs?.mcpServerProxyEndpoint}/${mcp.key}/sse`
: '';
const jsonConfig = `{
"mcpServers": {
"${feConfigs?.systemTitle}-mcp-${mcp._id}": {
"url": "${url}"
}
}
}`;
return {
url,
jsonConfig
};
})();
return ( return (
<MyModal isOpen title={t('dashboard_mcp:usage_way')} onClose={onClose}> <MyModal iconSrc="key" isOpen title={t('dashboard_mcp:usage_way')} onClose={onClose}>
<ModalBody> <ModalBody>
<Box> <Flex>
<FormLabel>{t('dashboard_mcp:mcp_endpoints')}</FormLabel> <LightRowTabs<LinkWay>
<HStack mt={0.5} bg={'myGray.50'} px={2} py={1} borderRadius={'md'} fontSize={'sm'}> m={'auto'}
<Box userSelect={'all'} flex={'1 0 0'} whiteSpace={'pre-wrap'} wordBreak={'break-all'}> w={'100%'}
{sseUrl} list={[
{ label: 'Streamable HTTP', value: 'http' },
{ label: 'SSE', value: 'sse' }
]}
value={linkWay}
onChange={setLinkWay}
/>
</Flex>
{url ? (
<>
<Box mt={4}>
<FormLabel>{t('dashboard_mcp:mcp_endpoints')}</FormLabel>
<HStack mt={0.5} bg={'myGray.50'} px={2} py={1} borderRadius={'md'} fontSize={'sm'}>
<Box
userSelect={'all'}
flex={'1 0 0'}
whiteSpace={'pre-wrap'}
wordBreak={'break-all'}
>
{url}
</Box>
<CopyBox value={url}>
<MyIconButton icon="copy" />
</CopyBox>
</HStack>
</Box> </Box>
<CopyBox value={sseUrl}> <Box mt={4}>
<MyIconButton icon="copy" /> <Box borderRadius={'md'} bg={'myGray.100'} overflow={'hidden'} fontSize={'sm'}>
</CopyBox> <Flex
</HStack> p={3}
</Box> bg={'myWhite.500'}
<Box mt={4}> border={'base'}
<Box borderRadius={'md'} bg={'myGray.100'} overflow={'hidden'} fontSize={'sm'}> borderTopLeftRadius={'md'}
<Flex borderTopRightRadius={'md'}
p={3} >
bg={'myWhite.500'} <Box flex={1}>{t('dashboard_mcp:mcp_json_config')}</Box>
border={'base'} <CopyBox value={jsonConfig}>
borderTopLeftRadius={'md'} <MyIconButton icon="copy" />
borderTopRightRadius={'md'} </CopyBox>
> </Flex>
<Box flex={1}>{t('dashboard_mcp:mcp_json_config')}</Box> <Box whiteSpace={'pre-wrap'} wordBreak={'break-all'} p={3} overflowX={'auto'}>
<CopyBox value={jsonConfig}> {jsonConfig}
<MyIconButton icon="copy" /> </Box>
</CopyBox> </Box>
</Flex>
<Box whiteSpace={'pre-wrap'} wordBreak={'break-all'} p={3} overflowX={'auto'}>
{jsonConfig}
</Box> </Box>
</Box> </>
</Box> ) : (
<Flex h={'200px'} justifyContent={'center'} alignItems={'center'}>
{t('dashboard_mcp:not_sse_server')}
</Flex>
)}
</ModalBody> </ModalBody>
</MyModal> </MyModal>
); );
......
projects/app/src/pages/api/mcp/app/[key]/mcp.ts 0 → 100644
import type { ApiRequestProps, ApiResponseType } from '@fastgpt/service/type/next';
import { StreamableHTTPServerTransport } from '@modelcontextprotocol/sdk/server/streamableHttp.js';
import { Server } from '@modelcontextprotocol/sdk/server/index.js';
import { addLog } from '@fastgpt/service/common/system/log';
import {
CallToolRequestSchema,
CallToolResult,
ListToolsRequestSchema
} from '@modelcontextprotocol/sdk/types';
import { callMcpServerTool, getMcpServerTools } from '@/service/support/mcp/utils';
import { toolCallProps } from '@/service/support/mcp/type';
import { getErrText } from '@fastgpt/global/common/error/utils';
export type mcpQuery = { key: string };
export type mcpBody = toolCallProps;
const handlePost = async (req: ApiRequestProps<mcpBody, mcpQuery>, res: ApiResponseType<any>) => {
const key = req.query.key;
const server = new Server(
{
name: 'fastgpt-mcp-server-http-streamable',
version: '1.0.0'
},
{
capabilities: {
tools: {}
}
}
);
const transport: StreamableHTTPServerTransport = new StreamableHTTPServerTransport({
sessionIdGenerator: undefined
});
res.on('close', () => {
addLog.debug('[MCP server] Close connection');
transport.close();
server.close();
});
try {
const tools = await getMcpServerTools(key);
// Register list tools
server.setRequestHandler(ListToolsRequestSchema, () => ({
tools
}));
// Register call tool
const handleToolCall = async (
name: string,
args: Record<string, any>
): Promise<CallToolResult> => {
try {
addLog.debug(`Call tool: ${name} with args: ${JSON.stringify(args)}`);
const result = await callMcpServerTool({ key, toolName: name, inputs: args });
return {
content: [
{
type: 'text',
text: typeof result === 'string' ? result : JSON.stringify(result)
}
],
isError: false
};
} catch (error) {
return {
message: getErrText(error),
content: [],
isError: true
};
}
};
server.setRequestHandler(CallToolRequestSchema, async (request) => {
return handleToolCall(request.params.name, request.params.arguments ?? {});
});
// Connect to transport
await server.connect(transport);
await transport.handleRequest(req, res, req.body);
} catch (error) {
addLog.error('[MCP server] Error handling MCP request:', error);
if (!res.writableFinished) {
res.status(500).json({
jsonrpc: '2.0',
error: {
code: -32603,
message: 'Internal server error'
},
id: null
});
}
}
};
async function handler(req: ApiRequestProps<mcpBody, mcpQuery>, res: ApiResponseType<any>) {
const method = req.method;
if (method === 'POST') {
return handlePost(req, res);
}
res.writeHead(405).end(
JSON.stringify({
jsonrpc: '2.0',
error: {
code: -32000,
message: 'Method not allowed.'
},
id: null
})
);
return;
}
export default handler;
projects/app/src/pages/api/support/mcp/server/toolCall.ts
import type { ApiRequestProps, ApiResponseType } from '@fastgpt/service/type/next'; import type { ApiRequestProps, ApiResponseType } from '@fastgpt/service/type/next';
import { NextAPI } from '@/service/middleware/entry'; import { NextAPI } from '@/service/middleware/entry';
import { MongoMcpKey } from '@fastgpt/service/support/mcp/schema'; import { toolCallProps } from '@/service/support/mcp/type';
import { CommonErrEnum } from '@fastgpt/global/common/error/code/common'; import { callMcpServerTool } from '@/service/support/mcp/utils';
import { MongoApp } from '@fastgpt/service/core/app/schema';
import { AppTypeEnum } from '@fastgpt/global/core/app/constants';
import { AppSchema } from '@fastgpt/global/core/app/type';
import { getUserChatInfoAndAuthTeamPoints } from '@fastgpt/service/support/permission/auth/team';
import { getAppLatestVersion } from '@fastgpt/service/core/app/version/controller';
import { getNanoid } from '@fastgpt/global/common/string/tools';
import { AIChatItemType, UserChatItemType } from '@fastgpt/global/core/chat/type';
import {
getPluginRunUserQuery,
updatePluginInputByVariables
} from '@fastgpt/global/core/workflow/utils';
import { getPluginInputsFromStoreNodes } from '@fastgpt/global/core/app/plugin/utils';
import {
ChatItemValueTypeEnum,
ChatRoleEnum,
ChatSourceEnum
} from '@fastgpt/global/core/chat/constants';
import {
getWorkflowEntryNodeIds,
storeEdges2RuntimeEdges,
storeNodes2RuntimeNodes
} from '@fastgpt/global/core/workflow/runtime/utils';
import { WORKFLOW_MAX_RUN_TIMES } from '@fastgpt/service/core/workflow/constants';
import { dispatchWorkFlow } from '@fastgpt/service/core/workflow/dispatch';
import { getChatTitleFromChatMessage, removeEmptyUserInput } from '@fastgpt/global/core/chat/utils';
import { saveChat } from '@fastgpt/service/core/chat/saveChat';
import { DispatchNodeResponseKeyEnum } from '@fastgpt/global/core/workflow/runtime/constants';
import { createChatUsage } from '@fastgpt/service/support/wallet/usage/controller';
import { UsageSourceEnum } from '@fastgpt/global/support/wallet/usage/constants';
import { FlowNodeTypeEnum } from '@fastgpt/global/core/workflow/node/constant';
export type toolCallQuery = {}; export type toolCallQuery = {};
export type toolCallBody = { export type toolCallBody = toolCallProps;
key: string;
toolName: string;
inputs: Record<string, any>;
};
export type toolCallResponse = {}; export type toolCallResponse = {};
const dispatchApp = async (app: AppSchema, variables: Record<string, any>) => {
const isPlugin = app.type === AppTypeEnum.plugin;
const { timezone, externalProvider } = await getUserChatInfoAndAuthTeamPoints(app.tmbId);
// Get app latest version
const { nodes, edges, chatConfig } = await getAppLatestVersion(app._id, app);
const userQuestion: UserChatItemType = (() => {
if (isPlugin) {
return getPluginRunUserQuery({
pluginInputs: getPluginInputsFromStoreNodes(nodes || app.modules),
variables
});
}
return {
obj: ChatRoleEnum.Human,
value: [
{
type: ChatItemValueTypeEnum.text,
text: {
content: variables.question
}
}
]
};
})();
let runtimeNodes = storeNodes2RuntimeNodes(nodes, getWorkflowEntryNodeIds(nodes));
if (isPlugin) {
// Assign values to runtimeNodes using variables
runtimeNodes = updatePluginInputByVariables(runtimeNodes, variables);
// Plugin runtime does not need global variables(It has been injected into the pluginInputNode)
variables = {};
} else {
delete variables.question;
variables.system_fileUrlList = variables.fileUrlList;
delete variables.fileUrlList;
}
const chatId = getNanoid();
const { flowUsages, assistantResponses, newVariables, flowResponses, durationSeconds } =
await dispatchWorkFlow({
chatId,
timezone,
externalProvider,
mode: 'chat',
runningAppInfo: {
id: String(app._id),
teamId: String(app.teamId),
tmbId: String(app.tmbId)
},
runningUserInfo: {
teamId: String(app.teamId),
tmbId: String(app.tmbId)
},
uid: String(app.tmbId),
runtimeNodes,
runtimeEdges: storeEdges2RuntimeEdges(edges),
variables,
query: removeEmptyUserInput(userQuestion.value),
chatConfig,
histories: [],
stream: false,
maxRunTimes: WORKFLOW_MAX_RUN_TIMES
});
// Save chat
const aiResponse: AIChatItemType & { dataId?: string } = {
obj: ChatRoleEnum.AI,
value: assistantResponses,
[DispatchNodeResponseKeyEnum.nodeResponse]: flowResponses
};
const newTitle = isPlugin ? 'Mcp call' : getChatTitleFromChatMessage(userQuestion);
await saveChat({
chatId,
appId: app._id,
teamId: app.teamId,
tmbId: app.tmbId,
nodes,
appChatConfig: chatConfig,
variables: newVariables,
isUpdateUseTime: false, // owner update use time
newTitle,
source: ChatSourceEnum.mcp,
content: [userQuestion, aiResponse],
durationSeconds
});
// Push usage
createChatUsage({
appName: app.name,
appId: app._id,
teamId: app.teamId,
tmbId: app.tmbId,
source: UsageSourceEnum.mcp,
flowUsages
});
// Get MCP response type
const responseContent = (() => {
if (isPlugin) {
const output = flowResponses.find(
(item) => item.moduleType === FlowNodeTypeEnum.pluginOutput
);
if (output) {
return JSON.stringify(output.pluginOutput);
} else {
return 'Can not get response from plugin';
}
}
return assistantResponses
.map((item) => item?.text?.content)
.filter(Boolean)
.join('\n');
})();
return responseContent;
};
async function handler( async function handler(
req: ApiRequestProps<toolCallBody, toolCallQuery>, req: ApiRequestProps<toolCallBody, toolCallQuery>,
res: ApiResponseType<any> res: ApiResponseType<any>
): Promise<toolCallResponse> { ): Promise<toolCallResponse> {
const { key, toolName, inputs } = req.body; return callMcpServerTool(req.body);
const mcp = await MongoMcpKey.findOne({ key }, { apps: 1 }).lean();
if (!mcp) {
return Promise.reject(CommonErrEnum.invalidResource);
}
// Get app list
const appList = await MongoApp.find({
_id: { $in: mcp.apps.map((app) => app.appId) },
type: { $in: [AppTypeEnum.simple, AppTypeEnum.workflow, AppTypeEnum.plugin] }
}).lean();
const app = appList.find((app) => {
const mcpApp = mcp.apps.find((mcpApp) => String(mcpApp.appId) === String(app._id))!;
return toolName === mcpApp.toolName;
});
if (!app) {
return Promise.reject(CommonErrEnum.missingParams);
}
return await dispatchApp(app, inputs);
} }
export default NextAPI(handler); export default NextAPI(handler);
projects/app/src/pages/api/support/mcp/server/toolList.ts
import type { ApiRequestProps, ApiResponseType } from '@fastgpt/service/type/next'; import type { ApiRequestProps, ApiResponseType } from '@fastgpt/service/type/next';
import { NextAPI } from '@/service/middleware/entry'; import { NextAPI } from '@/service/middleware/entry';
import { MongoMcpKey } from '@fastgpt/service/support/mcp/schema';
import { CommonErrEnum } from '@fastgpt/global/common/error/code/common';
import { MongoApp } from '@fastgpt/service/core/app/schema';
import { authAppByTmbId } from '@fastgpt/service/support/permission/app/auth';
import { ReadPermissionVal } from '@fastgpt/global/support/permission/constant';
import { getAppLatestVersion } from '@fastgpt/service/core/app/version/controller';
import { Tool } from '@modelcontextprotocol/sdk/types'; import { Tool } from '@modelcontextprotocol/sdk/types';
import { FlowNodeTypeEnum } from '@fastgpt/global/core/workflow/node/constant'; import { getMcpServerTools } from '@/service/support/mcp/utils';
import { toolValueTypeList } from '@fastgpt/global/core/workflow/constants';
import { AppChatConfigType } from '@fastgpt/global/core/app/type';
import { AppTypeEnum } from '@fastgpt/global/core/app/constants';
import { FlowNodeInputItemType } from '@fastgpt/global/core/workflow/type/io';
export type listToolsQuery = { key: string }; export type listToolsQuery = { key: string };
...@@ -19,139 +9,13 @@ export type listToolsBody = {}; ...@@ -19,139 +9,13 @@ export type listToolsBody = {};
export type listToolsResponse = {}; export type listToolsResponse = {};
export const pluginNodes2InputSchema = (
nodes: { flowNodeType: FlowNodeTypeEnum; inputs: FlowNodeInputItemType[] }[]
) => {
const pluginInput = nodes.find((node) => node.flowNodeType === FlowNodeTypeEnum.pluginInput);
const schema: Tool['inputSchema'] = {
type: 'object',
properties: {},
required: []
};
pluginInput?.inputs.forEach((input) => {
const jsonSchema = (
toolValueTypeList.find((type) => type.value === input.valueType) || toolValueTypeList[0]
)?.jsonSchema;
schema.properties![input.key] = {
...jsonSchema,
description: input.description,
enum: input.enum?.split('\n').filter(Boolean) || undefined
};
if (input.required) {
// @ts-ignore
schema.required.push(input.key);
}
});
return schema;
};
export const workflow2InputSchema = (chatConfig?: {
fileSelectConfig?: AppChatConfigType['fileSelectConfig'];
variables?: AppChatConfigType['variables'];
}) => {
const schema: Tool['inputSchema'] = {
type: 'object',
properties: {
question: {
type: 'string',
description: 'Question from user'
},
...(chatConfig?.fileSelectConfig?.canSelectFile || chatConfig?.fileSelectConfig?.canSelectImg
? {
fileUrlList: {
type: 'array',
items: {
type: 'string'
},
description: 'File linkage'
}
}
: {})
},
required: ['question']
};
chatConfig?.variables?.forEach((item) => {
const jsonSchema = (
toolValueTypeList.find((type) => type.value === item.valueType) || toolValueTypeList[0]
)?.jsonSchema;
schema.properties![item.key] = {
...jsonSchema,
description: item.description,
enum: item.enums?.map((enumItem) => enumItem.value) || undefined
};
if (item.required) {
// @ts-ignore
schema.required!.push(item.key);
}
});
return schema;
};
async function handler( async function handler(
req: ApiRequestProps<listToolsBody, listToolsQuery>, req: ApiRequestProps<listToolsBody, listToolsQuery>,
res: ApiResponseType<any> res: ApiResponseType<any>
): Promise<Tool[]> { ): Promise<Tool[]> {
const { key } = req.query; const { key } = req.query;
const mcp = await MongoMcpKey.findOne({ key }, { apps: 1 }).lean(); return getMcpServerTools(key);
if (!mcp) {
return Promise.reject(CommonErrEnum.invalidResource);
}
// Get app list
const appList = await MongoApp.find(
{
_id: { $in: mcp.apps.map((app) => app.appId) },
type: { $in: [AppTypeEnum.simple, AppTypeEnum.workflow, AppTypeEnum.plugin] }
},
{ name: 1, intro: 1 }
).lean();
// Filter not permission app
const permissionAppList = await Promise.all(
appList.filter(async (app) => {
try {
await authAppByTmbId({ tmbId: mcp.tmbId, appId: app._id, per: ReadPermissionVal });
return true;
} catch (error) {
return false;
}
})
);
// Get latest version
const versionList = await Promise.all(
permissionAppList.map((app) => getAppLatestVersion(app._id, app))
);
// Compute mcp tools
const tools = versionList.map<Tool>((version, index) => {
const app = permissionAppList[index];
const mcpApp = mcp.apps.find((mcpApp) => String(mcpApp.appId) === String(app._id))!;
const isPlugin = !!version.nodes.find(
(node) => node.flowNodeType === FlowNodeTypeEnum.pluginInput
);
return {
name: mcpApp.toolName,
description: mcpApp.description,
inputSchema: isPlugin
? pluginNodes2InputSchema(version.nodes)
: workflow2InputSchema(version.chatConfig)
};
});
return tools;
} }
export default NextAPI(handler); export default NextAPI(handler);
projects/app/src/service/support/mcp/type.d.ts 0 → 100644
export type toolCallProps = {
key: string;
toolName: string;
inputs: Record<string, any>;
};
projects/mcp_server/package.json
...@@ -14,7 +14,7 @@ ...@@ -14,7 +14,7 @@
"mcp_test": "npx @modelcontextprotocol/inspector" "mcp_test": "npx @modelcontextprotocol/inspector"
}, },
"dependencies": { "dependencies": {
"@modelcontextprotocol/sdk": "1.9.0", "@modelcontextprotocol/sdk": "^1.10.2",
"axios": "^1.8.2", "axios": "^1.8.2",
"chalk": "^5.3.0", "chalk": "^5.3.0",
"dayjs": "^1.11.7", "dayjs": "^1.11.7",
......
test/cases/pages/api/support/mcp/server/toolList.test.ts
import { describe, it, expect, vi } from 'vitest'; import { describe, it, expect, vi } from 'vitest';
import { import { pluginNodes2InputSchema, workflow2InputSchema } from '@/service/support/mcp/utils';
pluginNodes2InputSchema,
workflow2InputSchema
} from '@/pages/api/support/mcp/server/toolList';
import { FlowNodeTypeEnum } from '@fastgpt/global/core/workflow/node/constant'; import { FlowNodeTypeEnum } from '@fastgpt/global/core/workflow/node/constant';
import { import {
VariableInputEnum, VariableInputEnum,
...@@ -22,7 +19,8 @@ vi.mock('@fastgpt/service/core/app/schema', () => ({ ...@@ -22,7 +19,8 @@ vi.mock('@fastgpt/service/core/app/schema', () => ({
find: vi.fn().mockReturnValue({ find: vi.fn().mockReturnValue({
lean: vi.fn() lean: vi.fn()
}) })
} },
AppCollectionName: 'apps'
})); }));
vi.mock('@fastgpt/service/core/app/version/controller', () => ({ vi.mock('@fastgpt/service/core/app/version/controller', () => ({
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment