V4.9.6 feature (#4565)

* Dashboard submenu (#4545) * add app submenu (#4452) * add app submenu * fix * width & i18n * optimize submenu code (#4515) * optimize submenu code * fix * fix * fix * fix ts * perf: dashboard sub menu * doc --------- Co-authored-by: heheer <heheer@sealos.io> * feat: value format test * doc * Mcp export (#4555) * feat: mcp server * feat: mcp server * feat: mcp server build * update doc * perf: path selector (#4556) * perf: path selector * fix: docker file path * perf: add image endpoint to dataset search (#4557) * perf: add image endpoint to dataset search * fix: mcp_server url * human in loop (#4558) * Support interactive nodes for loops, and enhance the function of merging nested and loop node history messages. (#4552) * feat: add LoopInteractive definition * feat: Support LoopInteractive type and update related logic * fix: Refactor loop handling logic and improve output value initialization * feat: Add mergeSignId to dispatchLoop and dispatchRunAppNode responses * feat: Enhance mergeChatResponseData to recursively merge plugin details and improve response handling * refactor: Remove redundant comments in mergeChatResponseData for clarity * perf: loop interactive * perf: human in loop --------- Co-authored-by: Theresa <63280168+sd0ric4@users.noreply.github.com> * mcp server ui * integrate mcp (#4549) * integrate mcp * delete unused code * fix ts * bug fix * fix * support whole mcp tools * add try catch * fix * fix * fix ts * fix test * fix ts * fix: interactive in v1 completions * doc * fix: router path * fix mcp integrate (#4563) * fix mcp integrate * fix ui * fix: mcp ux * feat: mcp call title * remove repeat loading * fix mcp tools avatar (#4564) * fix * fix avatar * fix update version * update doc * fix: value format * close server and remove cache * perf: avatar --------- Co-authored-by: heheer <heheer@sealos.io> Co-authored-by: Theresa <63280168+sd0ric4@users.noreply.github.com>
2025-04-16 22:18:51 +08:00
parent ab799e13cd
commit 952412f648
166 changed files with 6318 additions and 1263 deletions
--- a/projects/mcp_server/src/index.ts
+++ b/projects/mcp_server/src/index.ts
@@ -0,0 +1,108 @@
+#!/usr/bin/env node
+import './init.js';
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import {
+  CallToolRequestSchema,
+  ListToolsRequestSchema,
+  CallToolResult
+} from '@modelcontextprotocol/sdk/types.js';
+import { SSEServerTransport } from '@modelcontextprotocol/sdk/server/sse.js';
+import express from 'express';
+
+import { callTool, getTools } from './api/fastgpt.js';
+import { addLog } from './utils/log.js';
+import { getErrText } from './utils/error.js';
+
+const app = express();
+
+const transportMap: Record<string, SSEServerTransport> = {};
+
+app.get('/:key/sse', async (req, res) => {
+  const { key } = req.params;
+
+  const transport = new SSEServerTransport(`/${key}/messages`, res);
+
+  transportMap[transport.sessionId] = transport;
+
+  // Create server
+  const server = new Server(
+    {
+      name: 'fastgpt-mcp-server',
+      version: '1.0.0'
+    },
+    {
+      capabilities: {
+        tools: {}
+      }
+    }
+  );
+
+  transport.onclose = () => {
+    addLog.info(`Transport ${transport.sessionId} closed`);
+    delete transportMap[transport.sessionId];
+  };
+  transport.onerror = (err) => {
+    addLog.error(`Transport ${transport.sessionId} error`, err);
+  };
+  server.onclose = () => {
+    addLog.info(`Server ${transport.sessionId} closed`);
+    delete transportMap[transport.sessionId];
+  };
+  server.onerror = (err) => {
+    addLog.error(`Server ${transport.sessionId} error`, err);
+  };
+
+  server.setRequestHandler(ListToolsRequestSchema, async () => ({
+    tools: await getTools(key)
+  }));
+
+  const handleToolCall = async (
+    name: string,
+    args: Record<string, any>
+  ): Promise<CallToolResult> => {
+    try {
+      addLog.info(`Call tool: ${name} with args: ${JSON.stringify(args)}`);
+      const result = await callTool({ 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) =>
+    handleToolCall(request.params.name, request.params.arguments ?? {})
+  );
+
+  await server.connect(transport);
+  addLog.info(`Server connected: ${transport.sessionId}`);
+});
+
+app.post('/:key/messages', (req, res) => {
+  const { sessionId } = req.query as { sessionId: string };
+
+  const transport = transportMap[sessionId];
+  if (transport) {
+    transport.handlePostMessage(req, res);
+  }
+});
+
+const PORT = process.env.PORT || 3000;
+app
+  .listen(PORT, () => {
+    addLog.info(`Server is running on port ${PORT}`);
+  })
+  .on('error', (err) => {
+    addLog.error(`Server error`, err);
+  });