FlowiseAI/Flowise
1
0
Fork 0
mirror of https://github.com/FlowiseAI/Flowise.git synced 2026-10-12 02:02:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add dynamic headers option to MCP core for tool call-time (#6044)

Browse Source 
* add dynamic headers option to MCP core
This commit is contained in:
michaell-workday
2026-03-24 11:01:41 -07:00
committed by GitHub
parent f7d86912fa
commit c93d6c0617
1 changed files with 16 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/components/nodes/tools/MCP/core.ts
+16 -8
View File
@@ -15,14 +15,19 @@ export class MCPToolkit extends BaseToolkit {
client: Client | null = null
serverParams: StdioServerParameters | any
transportType: 'stdio' | 'sse'
/** Per-invocation HTTP headers injected at tools/call time; overrides static toolkit headers for the same names. */
getToolCallHeaders?: () => Promise<Record<string, string>>
constructor(serverParams: StdioServerParameters | any, transportType: 'stdio' | 'sse') {
super()
this.serverParams = serverParams
this.transportType = transportType
}
// Method to create a new client with transport
async createClient(): Promise<Client> {
/**
* Creates a new MCP client and connects it via the configured transport.
* @param injectHeaders - Additional HTTP headers merged over static `serverParams.headers` for this connection. Used to pass per-invocation headers (e.g. from {@link getToolCallHeaders}) into SSE/HTTP transports.
*/
async createClient(injectHeaders: Record<string, string> = {}): Promise<Client> {
const client = new Client(
{
name: 'flowise-client',
@@ -54,11 +59,13 @@ export class MCPToolkit extends BaseToolkit {
const baseUrl = new URL(this.serverParams.url)
await checkDenyList(this.serverParams.url)
const mergedHeaders = { ...this.serverParams?.headers, ...injectHeaders }
const headers = Object.keys(mergedHeaders).length > 0 ? mergedHeaders : undefined
try {
if (this.serverParams.headers) {
if (headers) {
transport = new StreamableHTTPClientTransport(baseUrl, {
requestInit: {
headers: this.serverParams.headers
headers
}
})
} else {
@@ -66,16 +73,16 @@ export class MCPToolkit extends BaseToolkit {
}
await client.connect(transport)
} catch (error) {
if (this.serverParams.headers) {
if (headers) {
transport = new SSEClientTransport(baseUrl, {
requestInit: {
headers: this.serverParams.headers
headers
},
eventSourceInit: {
fetch: async (url, init) => {
return secureFetch(url.toString(), {
...(init as any),
headers: this.serverParams.headers
headers
}) as any
}
}
@@ -148,7 +155,8 @@ export async function MCPTool({
return tool(
async (input): Promise<string> => {
// Create a new client for this request
const client = await toolkit.createClient()
const toolCallHeaders = await toolkit.getToolCallHeaders?.()
const client = await toolkit.createClient(toolCallHeaders)
try {
const req: CallToolRequest = { method: 'tools/call', params: { name: name, arguments: input as any } }