gitea/FastGPT
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

support mcp client streamable http (#4650)

Browse Source 
* support mcp streamable http

* fix

* fix

* remove deps
This commit is contained in:
heheer
2025-04-24 23:04:54 +08:00
committed by GitHub
parent 5c93545016
commit 2a54be4d91
16 changed files with 273 additions and 144 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
projects/app/src/pages/api/core/app/mcpTools/create.ts
View File

@@ -11,6 +11,7 @@ import {
getMCPToolRuntimeNode,
getMCPToolSetRuntimeNode
} from '@fastgpt/global/core/app/mcpTools/utils';
import { pushTrack } from '@fastgpt/service/common/middle/tracks/utils';
export type createMCPToolsQuery = {};
@@ -61,6 +62,13 @@ async function handler(
}
});
pushTrack.createApp({
type: AppTypeEnum.toolSet,
uid: userId,
teamId,
tmbId
});
return {};
}

43
projects/app/src/pages/api/core/app/mcpTools/getMCPTools.ts
View File

@@ -1,43 +0,0 @@
import { NextAPI } from '@/service/middleware/entry';
import { ToolType } from '@fastgpt/global/core/app/type';
import type { ApiRequestProps, ApiResponseType } from '@fastgpt/service/type/next';
import { Client } from '@modelcontextprotocol/sdk/client/index.js';
import { SSEClientTransport } from '@modelcontextprotocol/sdk/client/sse.js';
export type getMCPToolsQuery = {};
export type getMCPToolsBody = { url: string };
export type getMCPToolsResponse = ToolType[];
async function handler(
req: ApiRequestProps<getMCPToolsBody, getMCPToolsQuery>,
res: ApiResponseType<getMCPToolsResponse[]>
): Promise<getMCPToolsResponse> {
const { url } = req.body;
const client = new Client({
name: 'FastGPT-MCP-client',
version: '1.0.0'
});
const tools = await (async () => {
try {
const transport = new SSEClientTransport(new URL(url));
await client.connect(transport);
const response = await client.listTools();
return response.tools || [];
} catch (error) {
console.error('Error fetching MCP tools:', error);
return Promise.reject(error);
} finally {
await client.close();
}
})();
return tools as ToolType[];
}
export default NextAPI(handler);

45
projects/app/src/pages/api/core/app/mcpTools/runTest.ts
View File

@@ -1,45 +0,0 @@
import { SSEClientTransport } from '@modelcontextprotocol/sdk/client/sse.js';
import { Client } from '@modelcontextprotocol/sdk/client/index.js';
import { NextAPI } from '@/service/middleware/entry';
import type { ApiRequestProps, ApiResponseType } from '@fastgpt/service/type/next';
export type RunToolTestQuery = {};
export type RunToolTestBody = {
params: Record<string, any>;
url: string;
toolName: string;
};
export type RunToolTestResponse = any;
async function handler(
req: ApiRequestProps<RunToolTestBody, RunToolTestQuery>,
res: ApiResponseType<RunToolTestResponse>
): Promise<RunToolTestResponse> {
const { params, url, toolName } = req.body;
const client = new Client({
name: 'FastGPT-MCP-client',
version: '1.0.0'
});
const result = await (async () => {
try {
const transport = new SSEClientTransport(new URL(url));
await client.connect(transport);
return await client.callTool({
name: toolName,
arguments: params
});
} catch (error) {
console.error('Error running MCP tool test:', error);
return Promise.reject(error);
} finally {
await client.close();
}
})();
return result;
}
export default NextAPI(handler);

28
projects/app/src/pages/api/support/mcp/client/getTools.ts Normal file
View File

@@ -0,0 +1,28 @@
import { NextAPI } from '@/service/middleware/entry';
import { ToolType } from '@fastgpt/global/core/app/type';
import type { ApiRequestProps, ApiResponseType } from '@fastgpt/service/type/next';
import getMCPClient from '@fastgpt/service/core/app/mcp';
export type getMCPToolsQuery = {};
export type getMCPToolsBody = { url: string };
export type getMCPToolsResponse = ToolType[];
async function handler(
req: ApiRequestProps<getMCPToolsBody, getMCPToolsQuery>,
res: ApiResponseType<getMCPToolsResponse[]>
): Promise<getMCPToolsResponse> {
const { url } = req.body;
const mcpClient = getMCPClient({ url });
try {
const tools = await mcpClient.getTools();
return tools;
} catch (error) {
return Promise.reject(error);
}
}
export default NextAPI(handler);

31
projects/app/src/pages/api/support/mcp/client/runTool.ts Normal file
View File

@@ -0,0 +1,31 @@
import { NextAPI } from '@/service/middleware/entry';
import type { ApiRequestProps, ApiResponseType } from '@fastgpt/service/type/next';
import getMCPClient from '@fastgpt/service/core/app/mcp';
export type RunMCPToolQuery = {};
export type RunMCPToolBody = {
url: string;
toolName: string;
params: Record<string, any>;
};
export type RunMCPToolResponse = any;
async function handler(
req: ApiRequestProps<RunMCPToolBody, RunMCPToolQuery>,
res: ApiResponseType<RunMCPToolResponse>
): Promise<RunMCPToolResponse> {
const { url, toolName, params } = req.body;
const mcpClient = getMCPClient({ url });
try {
const result = await mcpClient.toolCall(toolName, params);
return result;
} catch (error) {
return Promise.reject(error);
}
}
export default NextAPI(handler);