mirror of
https://github.com/farcasclaudiu/Flowise.git
synced 2026-06-28 19:00:59 +03:00
Feat/add headers to custom mcp (#4581)
* add headers to custom mcp * Refactor MCP fetch method and update CustomMCP documentation to include variable usage in headers
This commit is contained in:
@@ -53,10 +53,29 @@ export class MCPToolkit extends BaseToolkit {
|
||||
|
||||
const baseUrl = new URL(this.serverParams.url)
|
||||
try {
|
||||
transport = new StreamableHTTPClientTransport(baseUrl)
|
||||
if (this.serverParams.headers) {
|
||||
transport = new StreamableHTTPClientTransport(baseUrl, {
|
||||
requestInit: {
|
||||
headers: this.serverParams.headers
|
||||
}
|
||||
})
|
||||
} else {
|
||||
transport = new StreamableHTTPClientTransport(baseUrl)
|
||||
}
|
||||
await client.connect(transport)
|
||||
} catch (error) {
|
||||
transport = new SSEClientTransport(baseUrl)
|
||||
if (this.serverParams.headers) {
|
||||
transport = new SSEClientTransport(baseUrl, {
|
||||
requestInit: {
|
||||
headers: this.serverParams.headers
|
||||
},
|
||||
eventSourceInit: {
|
||||
fetch: (url, init) => fetch(url, { ...init, headers: this.serverParams.headers })
|
||||
}
|
||||
})
|
||||
} else {
|
||||
transport = new SSEClientTransport(baseUrl)
|
||||
}
|
||||
await client.connect(transport)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user