@alpic80/rivet-core 1.24.2-aidon.7 → 1.24.2-aidon.9

package/dist/esm/integrations/mcp/MCPBase.js

@@ -23,6 +23,13 @@ export const getMCPBaseInputs = (data) => {
             description: 'The endpoint URL for the MCP server',
         });
     }
+    if (data.transportType === 'http' && data.useHeadersInput) {
+        inputs.push({
+            dataType: 'object',
+            id: 'headers',
+            title: 'Headers',
+        });
+    }
     return inputs;
 };
 export const getMCPBaseEditors = async (context, data) => {
@@ -71,6 +78,12 @@ export const getMCPBaseEditors = async (context, data) => {
             dataKey: 'serverUrl',
             useInputToggleDataKey: 'useServerUrlInput',
             helperMessage: 'The endpoint URL for the MCP server to connect',
+        }, {
+            type: 'code',
+            label: 'Headers',
+            dataKey: 'headers',
+            useInputToggleDataKey: 'useHeadersInput',
+            language: 'json',
         });
     }
     else if (data.transportType === 'stdio') {

package/dist/esm/model/nodes/HttpCallNode.js

@@ -141,7 +141,7 @@ export class HttpCallNodeImpl extends NodeImpl {
         return dedent `
       ${this.data.useMethodInput ? '(Method Using Input)' : this.data.method} ${this.data.useUrlInput ? '(URL Using Input)' : this.data.url} ${this.data.useHeadersInput
             ? '\nHeaders: (Using Input)'
-            : this.data.headers.trim()
+            : this.data.headers?.trim()
                 ? `\nHeaders: ${this.data.headers}`
                 : ''}${this.data.useBodyInput ? '\nBody: (Using Input)' : this.data.body.trim() ? `\nBody: ${this.data.body}` : ''}${this.data.errorOnNon200 ? '\nError on non-200' : ''}
     `;
@@ -179,7 +179,7 @@ export class HttpCallNodeImpl extends NodeImpl {
                 headers = coerceType(headersInput, 'object');
             }
         }
-        else if (this.data.headers.trim()) {
+        else if (this.data.headers?.trim()) {
             headers = JSON.parse(this.data.headers);
         }
         let body;

package/dist/esm/model/nodes/MCPDiscoveryNode.js

@@ -2,7 +2,7 @@ import { nanoid } from 'nanoid';
 import { NodeImpl } from '../NodeImpl.js';
 import {} from '../../index.js';
 import { MCPError, MCPErrorType } from '../../integrations/mcp/MCPProvider.js';
-import { dedent, getInputOrData } from '../../utils/index.js';
+import { coerceType, dedent, getInputOrData } from '../../utils/index.js';
 import { nodeDefinition } from '../NodeDefinition.js';
 import { getServerHelperMessage, getServerOptions, loadMCPConfiguration } from '../../integrations/mcp/MCPUtils.js';
 import { getMCPBaseInputs } from '../../integrations/mcp/MCPBase.js';
@@ -22,6 +22,7 @@ class MCPDiscoveryNodeImpl extends NodeImpl {
                 version: '1.0.0',
                 transportType: 'stdio',
                 serverUrl: 'http://localhost:8080/mcp',
+                headers: '',
                 serverId: '',
                 useNameInput: false,
                 useVersionInput: false,
@@ -100,6 +101,12 @@ class MCPDiscoveryNodeImpl extends NodeImpl {
                 dataKey: 'serverUrl',
                 useInputToggleDataKey: 'useServerUrlInput',
                 helperMessage: 'The base URL endpoint for the MCP server with `/mcp`',
+            }, {
+                type: 'code',
+                label: 'Headers',
+                dataKey: 'headers',
+                useInputToggleDataKey: 'useHeadersInput',
+                language: 'json',
             });
         }
         else if (this.data.transportType === 'stdio') {
@@ -116,15 +123,21 @@ class MCPDiscoveryNodeImpl extends NodeImpl {
     }
     getBody(context) {
         let base;
+        let headers = '';
         if (this.data.transportType === 'http') {
             base = this.data.useServerUrlInput ? '(Using Server URL Input)' : this.data.serverUrl;
+            headers = this.data.useHeadersInput
+                ? '\nHeaders: (Using Input)'
+                : this.data.headers?.trim()
+                    ? `\nHeaders: ${this.data.headers}`
+                    : '';
         }
         else {
             base = `Server ID: ${this.data.serverId || '(None)'}`;
         }
         const namePart = `Name: ${this.data.name}`;
         const versionPart = `Version: ${this.data.version}`;
-        const parts = [namePart, versionPart, base];
+        const parts = [namePart, versionPart, base, headers];
         if (context.executor !== 'nodejs') {
             parts.push('(Requires Node Executor)');
         }
@@ -158,8 +171,24 @@ class MCPDiscoveryNodeImpl extends NodeImpl {
                 if (!serverUrl.includes('/mcp')) {
                     throw new MCPError(MCPErrorType.SERVER_COMMUNICATION_FAILED, 'Include /mcp in your server URL. For example: http://localhost:8080/mcp');
                 }
-                tools = await context.mcpProvider.getHTTPTools({ name, version }, serverUrl);
-                prompts = await context.mcpProvider.getHTTPrompts({ name, version }, serverUrl);
+                let headers;
+                if (this.data.useHeadersInput) {
+                    const headersInput = inputs['headers'];
+                    if (headersInput?.type === 'string') {
+                        headers = JSON.parse(headersInput.value);
+                    }
+                    else if (headersInput?.type === 'object') {
+                        headers = headersInput.value;
+                    }
+                    else {
+                        headers = coerceType(headersInput, 'object');
+                    }
+                }
+                else if (this.data.headers?.trim()) {
+                    headers = JSON.parse(this.data.headers);
+                }
+                tools = await context.mcpProvider.getHTTPTools({ name, version }, serverUrl, headers);
+                prompts = await context.mcpProvider.getHTTPrompts({ name, version }, serverUrl, headers);
             }
             else if (transportType === 'stdio') {
                 const serverId = this.data.serverId ?? '';

package/dist/esm/model/nodes/MCPGetPromptNode.js

@@ -3,7 +3,7 @@ import { nanoid } from 'nanoid/non-secure';
 import { NodeImpl } from '../NodeImpl.js';
 import { nodeDefinition } from '../NodeDefinition.js';
 import {} from '../GraphProcessor.js';
-import {} from '../../index.js';
+import { coerceType } from '../../index.js';
 import { MCPError, MCPErrorType } from '../../integrations/mcp/MCPProvider.js';
 import { coerceTypeOptional } from '../../utils/coerceType.js';
 import { dedent, getInputOrData } from '../../utils/index.js';
@@ -28,6 +28,7 @@ export class MCPGetPromptNodeImpl extends NodeImpl {
                 version: '1.0.0',
                 transportType: 'stdio',
                 serverUrl: 'http://localhost:8080/mcp',
+                headers: '',
                 serverId: '',
                 promptName: '',
                 promptArguments: dedent `
@@ -117,7 +118,13 @@ export class MCPGetPromptNodeImpl extends NodeImpl {
                 label: 'Server URL',
                 dataKey: 'serverUrl',
                 useInputToggleDataKey: 'useServerUrlInput',
-                helperMessage: 'The endpoint URL for the MCP server to connect',
+                helperMessage: 'The base URL endpoint for the MCP server with `/mcp`',
+            }, {
+                type: 'code',
+                label: 'Headers',
+                dataKey: 'headers',
+                useInputToggleDataKey: 'useHeadersInput',
+                language: 'json',
             });
         }
         else if (this.data.transportType === 'stdio') {
@@ -134,15 +141,21 @@ export class MCPGetPromptNodeImpl extends NodeImpl {
     }
     getBody(context) {
         let base;
+        let headers = '';
         if (this.data.transportType === 'http') {
             base = this.data.useServerUrlInput ? '(Using Server URL Input)' : this.data.serverUrl;
+            headers = this.data.useHeadersInput
+                ? '\nHeaders: (Using Input)'
+                : this.data.headers?.trim()
+                    ? `\nHeaders: ${this.data.headers}`
+                    : '';
         }
         else {
             base = `Server ID: ${this.data.serverId || '(None)'}`;
         }
         const namePart = `Name: ${this.data.name}`;
         const versionPart = `Version: ${this.data.version}`;
-        const parts = [namePart, versionPart, base];
+        const parts = [namePart, versionPart, base, headers];
         if (context.executor !== 'nodejs') {
             parts.push('(Requires Node Executor)');
         }
@@ -199,7 +212,23 @@ export class MCPGetPromptNodeImpl extends NodeImpl {
                 if (!serverUrl.includes('/mcp')) {
                     throw new MCPError(MCPErrorType.SERVER_COMMUNICATION_FAILED, 'Include /mcp in your server URL. For example: http://localhost:8080/mcp');
                 }
-                getPromptResponse = await context.mcpProvider.getHTTPrompt({ name, version }, serverUrl, getPromptRequest);
+                let headers;
+                if (this.data.useHeadersInput) {
+                    const headersInput = inputs['headers'];
+                    if (headersInput?.type === 'string') {
+                        headers = JSON.parse(headersInput.value);
+                    }
+                    else if (headersInput?.type === 'object') {
+                        headers = headersInput.value;
+                    }
+                    else {
+                        headers = coerceType(headersInput, 'object');
+                    }
+                }
+                else if (this.data.headers?.trim()) {
+                    headers = JSON.parse(this.data.headers);
+                }
+                getPromptResponse = await context.mcpProvider.getHTTPrompt({ name, version }, serverUrl, headers, getPromptRequest);
             }
             else if (transportType === 'stdio') {
                 const serverId = this.data.serverId ?? '';

package/dist/esm/model/nodes/MCPToolCallNode.js

@@ -3,7 +3,7 @@ import { nanoid } from 'nanoid/non-secure';
 import { NodeImpl } from '../NodeImpl.js';
 import { nodeDefinition } from '../NodeDefinition.js';
 import {} from '../GraphProcessor.js';
-import {} from '../../index.js';
+import { coerceType } from '../../index.js';
 import { MCPError, MCPErrorType } from '../../integrations/mcp/MCPProvider.js';
 import { coerceTypeOptional } from '../../utils/coerceType.js';
 import { getInputOrData } from '../../utils/index.js';
@@ -30,6 +30,7 @@ export class MCPToolCallNodeImpl extends NodeImpl {
                 version: '1.0.0',
                 transportType: 'stdio',
                 serverUrl: 'http://localhost:8080/mcp',
+                headers: '',
                 serverId: '',
                 toolName: '',
                 toolArguments: dedent `
@@ -131,7 +132,7 @@ export class MCPToolCallNodeImpl extends NodeImpl {
                 label: 'Tool ID',
                 dataKey: 'toolCallId',
                 useInputToggleDataKey: 'useToolCallIdInput',
-                helperMessage: 'The name for the MCP Tool Call',
+                helperMessage: 'The ID associated with the tool call',
             },
         ];
         if (this.data.transportType === 'http') {
@@ -140,7 +141,13 @@ export class MCPToolCallNodeImpl extends NodeImpl {
                 label: 'Server URL',
                 dataKey: 'serverUrl',
                 useInputToggleDataKey: 'useServerUrlInput',
-                helperMessage: 'The endpoint URL for the MCP server to connect',
+                helperMessage: 'The base URL endpoint for the MCP server with `/mcp`',
+            }, {
+                type: 'code',
+                label: 'Headers',
+                dataKey: 'headers',
+                useInputToggleDataKey: 'useHeadersInput',
+                language: 'json',
             });
         }
         else if (this.data.transportType === 'stdio') {
@@ -157,15 +164,21 @@ export class MCPToolCallNodeImpl extends NodeImpl {
     }
     getBody(context) {
         let base;
+        let headers = '';
         if (this.data.transportType === 'http') {
             base = this.data.useServerUrlInput ? '(Using Server URL Input)' : this.data.serverUrl;
+            headers = this.data.useHeadersInput
+                ? '\nHeaders: (Using Input)'
+                : this.data.headers?.trim()
+                    ? `\nHeaders: ${this.data.headers}`
+                    : '';
         }
         else {
             base = `Server ID: ${this.data.serverId || '(None)'}`;
         }
         const namePart = `Name: ${this.data.name}`;
         const versionPart = `Version: ${this.data.version}`;
-        const parts = [namePart, versionPart, base];
+        const parts = [namePart, versionPart, base, headers];
         if (context.executor !== 'nodejs') {
             parts.push('(Requires Node Executor)');
         }
@@ -223,7 +236,23 @@ export class MCPToolCallNodeImpl extends NodeImpl {
                 if (!serverUrl.includes('/mcp')) {
                     throw new MCPError(MCPErrorType.SERVER_COMMUNICATION_FAILED, 'Include /mcp in your server URL. For example: http://localhost:8080/mcp');
                 }
-                toolResponse = await context.mcpProvider.httpToolCall({ name, version }, serverUrl, toolCall);
+                let headers;
+                if (this.data.useHeadersInput) {
+                    const headersInput = inputs['headers'];
+                    if (headersInput?.type === 'string') {
+                        headers = JSON.parse(headersInput.value);
+                    }
+                    else if (headersInput?.type === 'object') {
+                        headers = headersInput.value;
+                    }
+                    else {
+                        headers = coerceType(headersInput, 'object');
+                    }
+                }
+                else if (this.data.headers?.trim()) {
+                    headers = JSON.parse(this.data.headers);
+                }
+                toolResponse = await context.mcpProvider.httpToolCall({ name, version }, serverUrl, headers, toolCall);
             }
             else if (transportType === 'stdio') {
                 const serverId = this.data.serverId ?? '';

package/dist/esm/plugins/aidon/plugin.js

@@ -17,25 +17,6 @@ export const aidonPlugin = {
             description: 'The API Key for the Aidon application.',
             helperText: 'API Key for the Aidon application.',
         },
-        fileBrowserURL: {
-            type: 'string',
-            label: 'FileBrowser URL',
-            description: 'The URL for the FileBrowser service.',
-            helperText: 'Defaults to https://ai-fb.aidon.ai. URL for the FileBrowser service.',
-            default: 'https://ai-fb.aidon.ai',
-        },
-        fileBrowserUsername: {
-            type: 'string',
-            label: 'FileBrowser Username',
-            description: 'The username for the FileBrowser service.',
-            helperText: 'Enter username given to access FileBrowser.',
-        },
-        fileBrowserPassword: {
-            type: 'secret',
-            label: 'FileBrowser password',
-            description: 'The password for the FileBrowser service.',
-            helperText: 'Enter passord given to access FileBrowser.',
-        },
     },
     register: (register) => {
         register(chatAidonNode());

package/dist/types/integrations/mcp/MCPBase.d.ts

@@ -8,6 +8,8 @@ export interface MCPBaseNodeData {
     transportType: MCP.TransportType;
     serverUrl?: string;
     serverId?: string;
+    headers: string;
+    useHeadersInput?: boolean;
     useNameInput?: boolean;
     useVersionInput?: boolean;
     useServerUrlInput?: boolean;

package/dist/types/integrations/mcp/MCPProvider.d.ts

@@ -121,7 +121,7 @@ export interface MCPProvider {
     getHTTPTools(clientConfig: {
         name: string;
         version: string;
-    }, serverUrl: string): Promise<MCP.Tool[]>;
+    }, serverUrl: string, headers: Record<string, string> | undefined): Promise<MCP.Tool[]>;
     getStdioTools(clientConfig: {
         name: string;
         version: string;
@@ -129,7 +129,7 @@ export interface MCPProvider {
     getHTTPrompts(clientConfig: {
         name: string;
         version: string;
-    }, serverUrl: string): Promise<MCP.Prompt[]>;
+    }, serverUrl: string, headers: Record<string, string> | undefined): Promise<MCP.Prompt[]>;
     getStdioPrompts(clientConfig: {
         name: string;
         version: string;
@@ -137,7 +137,7 @@ export interface MCPProvider {
     httpToolCall(clientConfig: {
         name: string;
         version: string;
-    }, serverUrl: string, toolCall: MCP.ToolCallRequest): Promise<MCP.ToolCallResponse>;
+    }, serverUrl: string, headers: Record<string, string> | undefined, toolCall: MCP.ToolCallRequest): Promise<MCP.ToolCallResponse>;
     stdioToolCall(clientConfig: {
         name: string;
         version: string;
@@ -145,7 +145,7 @@ export interface MCPProvider {
     getHTTPrompt(clientConfig: {
         name: string;
         version: string;
-    }, serverUrl: string, getPromptRequest: MCP.GetPromptRequest): Promise<MCP.GetPromptResponse>;
+    }, serverUrl: string, headers: Record<string, string> | undefined, getPromptRequest: MCP.GetPromptRequest): Promise<MCP.GetPromptResponse>;
     getStdioPrompt(clientConfig: {
         name: string;
         version: string;

package/package.json

@@ -2,7 +2,7 @@
   "name": "@alpic80/rivet-core",
   "license": "MIT",
   "repository": "https://github.com/castortech/rivet",
-  "version": "1.24.2-aidon.7",
+  "version": "1.24.2-aidon.9",
   "packageManager": "yarn@3.5.0",
   "main": "dist/cjs/bundle.cjs",
   "module": "dist/esm/index.js",
@@ -48,7 +48,7 @@
     "@google-cloud/vertexai": "^0.1.3",
     "@google/genai": "^0.12.0",
     "@huggingface/inference": "^4.13.0",
-    "@ironclad/rivet-core": "npm:@alpic80/rivet-core@1.24.2-aidon.7",
+    "@ironclad/rivet-core": "npm:@alpic80/rivet-core@1.24.2-aidon.9",
     "assemblyai": "^4.6.0",
     "autoevals": "^0.0.26",
     "cron-parser": "^4.9.0",