Feature/MCP (Model Context Protocol) (#4134)

add mcp tools
2026-06-28 15:00:57 +03:00 · 2025-03-06 13:57:18 +00:00
parent 9c22bee991
commit 713ed26971
21 changed files with 1321 additions and 105 deletions
@@ -0,0 +1,114 @@
+import { Tool } from '@langchain/core/tools'
+import { ICommonObject, INode, INodeData, INodeOptionsValue, INodeParams } from '../../../../src/Interface'
+import { getCredentialData, getCredentialParam, getNodeModulesPackagePath } from '../../../../src/utils'
+import { MCPToolkit } from '../core'
+
+class Github_MCP implements INode {
+    label: string
+    name: string
+    version: number
+    description: string
+    type: string
+    icon: string
+    category: string
+    baseClasses: string[]
+    documentation: string
+    credential: INodeParams
+    inputs: INodeParams[]
+
+    constructor() {
+        this.label = 'Github MCP'
+        this.name = 'githubMCP'
+        this.version = 1.0
+        this.type = 'Github MCP Tool'
+        this.icon = 'github.svg'
+        this.category = 'Tools (MCP)'
+        this.description = 'MCP Server for the GitHub API'
+        this.documentation = 'https://github.com/modelcontextprotocol/servers/tree/main/src/github'
+        this.credential = {
+            label: 'Connect Credential',
+            name: 'credential',
+            type: 'credential',
+            credentialNames: ['githubApi']
+        }
+        this.inputs = [
+            {
+                label: 'Available Actions',
+                name: 'mcpActions',
+                type: 'asyncMultiOptions',
+                loadMethod: 'listActions',
+                refresh: true
+            }
+        ]
+        this.baseClasses = ['Tool']
+    }
+
+    //@ts-ignore
+    loadMethods = {
+        listActions: async (nodeData: INodeData, options: ICommonObject): Promise<INodeOptionsValue[]> => {
+            try {
+                const toolset = await this.getTools(nodeData, options)
+                toolset.sort((a: any, b: any) => a.name.localeCompare(b.name))
+
+                return toolset.map(({ name, ...rest }) => ({
+                    label: name.toUpperCase(),
+                    name: name,
+                    description: rest.description || name
+                }))
+            } catch (error) {
+                console.error('Error listing actions:', error)
+                return [
+                    {
+                        label: 'No Available Actions',
+                        name: 'error',
+                        description: 'No available actions, please check your Github Access Token and refresh'
+                    }
+                ]
+            }
+        }
+    }
+
+    async init(nodeData: INodeData, _: string, options: ICommonObject): Promise<any> {
+        const tools = await this.getTools(nodeData, options)
+
+        const _mcpActions = nodeData.inputs?.mcpActions
+        let mcpActions = []
+        if (_mcpActions) {
+            try {
+                mcpActions = typeof _mcpActions === 'string' ? JSON.parse(_mcpActions) : _mcpActions
+            } catch (error) {
+                console.error('Error parsing mcp actions:', error)
+            }
+        }
+
+        return tools.filter((tool: any) => mcpActions.includes(tool.name))
+    }
+
+    async getTools(nodeData: INodeData, options: ICommonObject): Promise<Tool[]> {
+        const credentialData = await getCredentialData(nodeData.credential ?? '', options)
+        const accessToken = getCredentialParam('accessToken', credentialData, nodeData)
+
+        if (!accessToken) {
+            throw new Error('Missing Github Access Token')
+        }
+
+        const packagePath = getNodeModulesPackagePath('@modelcontextprotocol/server-github/dist/index.js')
+
+        const serverParams = {
+            command: 'node',
+            args: [packagePath],
+            env: {
+                GITHUB_PERSONAL_ACCESS_TOKEN: accessToken
+            }
+        }
+
+        const toolkit = new MCPToolkit(serverParams, 'stdio')
+        await toolkit.initialize()
+
+        const tools = toolkit.tools ?? []
+
+        return tools as Tool[]
+    }
+}
+
+module.exports = { nodeClass: Github_MCP }