@alpic80/rivet-core 1.24.2-aidon.1 → 1.24.2-aidon.10

package/dist/cjs/bundle.cjs

@@ -6244,6 +6244,7 @@ var SubGraphNodeImpl = class extends NodeImpl {
       }
       return outputs;
     } catch (err) {
+      console.error(`Error Processing subgraph: ${getError(err).message}`);
       if (!this.data.useErrorOutput) {
         throw err;
       }
@@ -10546,8 +10547,9 @@ var HttpCallNodeImpl = class extends NodeImpl {
     ];
   }
   getBody() {
+    var _a;
     return import_ts_dedent.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.useMethodInput ? "(Method Using Input)" : this.data.method} ${this.data.useUrlInput ? "(URL Using Input)" : this.data.url} ${this.data.useHeadersInput ? "\nHeaders: (Using Input)" : ((_a = this.data.headers) == null ? void 0 : _a.trim()) ? `
 Headers: ${this.data.headers}` : ""}${this.data.useBodyInput ? "\nBody: (Using Input)" : this.data.body.trim() ? `
 Body: ${this.data.body}` : ""}${this.data.errorOnNon200 ? "\nError on non-200" : ""}
     `;
@@ -10563,7 +10565,7 @@ Body: ${this.data.body}` : ""}${this.data.errorOnNon200 ? "\nError on non-200" :
     };
   }
   async process(inputs, context) {
-    var _a;
+    var _a, _b;
     const method = getInputOrData(this.data, inputs, "method", "string");
     const url = getInputOrData(this.data, inputs, "url", "string");
     try {
@@ -10581,7 +10583,7 @@ Body: ${this.data.body}` : ""}${this.data.errorOnNon200 ? "\nError on non-200" :
       } else {
         headers = coerceType(headersInput, "object");
       }
-    } else if (this.data.headers.trim()) {
+    } else if ((_a = this.data.headers) == null ? void 0 : _a.trim()) {
       headers = JSON.parse(this.data.headers);
     }
     let body;
@@ -10627,7 +10629,7 @@ Body: ${this.data.body}` : ""}${this.data.errorOnNon200 ? "\nError on non-200" :
           type: "string",
           value: responseText
         };
-        if ((_a = response.headers.get("content-type")) == null ? void 0 : _a.includes("application/json")) {
+        if ((_b = response.headers.get("content-type")) == null ? void 0 : _b.includes("application/json")) {
           const jsonData = JSON.parse(responseText);
           output["json"] = {
             type: "object",
@@ -13874,6 +13876,13 @@ var 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;
 };
 
@@ -13894,6 +13903,7 @@ var MCPDiscoveryNodeImpl = class extends NodeImpl {
         version: "1.0.0",
         transportType: "stdio",
         serverUrl: "http://localhost:8080/mcp",
+        headers: "",
         serverId: "",
         useNameInput: false,
         useVersionInput: false,
@@ -13966,13 +13976,22 @@ var MCPDiscoveryNodeImpl = class extends NodeImpl {
       }
     ];
     if (this.data.transportType === "http") {
-      editors.push({
-        type: "string",
-        label: "Server URL",
-        dataKey: "serverUrl",
-        useInputToggleDataKey: "useServerUrlInput",
-        helperMessage: "The base URL endpoint for the MCP server with `/mcp`"
-      });
+      editors.push(
+        {
+          type: "string",
+          label: "Server URL",
+          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") {
       const serverOptions = await getServerOptions(context);
       editors.push({
@@ -13986,15 +14005,19 @@ var MCPDiscoveryNodeImpl = class extends NodeImpl {
     return editors;
   }
   getBody(context) {
+    var _a;
     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)" : ((_a = this.data.headers) == null ? void 0 : _a.trim()) ? `
+Headers: ${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)");
     }
@@ -14011,6 +14034,7 @@ var MCPDiscoveryNodeImpl = class extends NodeImpl {
     };
   }
   async process(inputs, context) {
+    var _a;
     const name = getInputOrData(this.data, inputs, "name", "string");
     const version = getInputOrData(this.data, inputs, "version", "string");
     const transportType = getInputOrData(this.data, inputs, "transportType", "string");
@@ -14031,8 +14055,21 @@ var MCPDiscoveryNodeImpl = class extends NodeImpl {
             "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 == null ? void 0 : headersInput.type) === "string") {
+            headers = JSON.parse(headersInput.value);
+          } else if ((headersInput == null ? void 0 : headersInput.type) === "object") {
+            headers = headersInput.value;
+          } else {
+            headers = coerceType(headersInput, "object");
+          }
+        } else if ((_a = this.data.headers) == null ? void 0 : _a.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 ?? "";
         const mcpConfig = await loadMCPConfiguration(context);
@@ -14099,6 +14136,7 @@ var MCPToolCallNodeImpl = class extends NodeImpl {
         version: "1.0.0",
         transportType: "stdio",
         serverUrl: "http://localhost:8080/mcp",
+        headers: "",
         serverId: "",
         toolName: "",
         toolArguments: import_ts_dedent69.dedent`
@@ -14200,17 +14238,26 @@ var MCPToolCallNodeImpl = class 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") {
-      editors.push({
-        type: "string",
-        label: "Server URL",
-        dataKey: "serverUrl",
-        useInputToggleDataKey: "useServerUrlInput",
-        helperMessage: "The endpoint URL for the MCP server to connect"
-      });
+      editors.push(
+        {
+          type: "string",
+          label: "Server URL",
+          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") {
       const serverOptions = await getServerOptions(context);
       editors.push({
@@ -14224,15 +14271,19 @@ var MCPToolCallNodeImpl = class extends NodeImpl {
     return editors;
   }
   getBody(context) {
+    var _a;
     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)" : ((_a = this.data.headers) == null ? void 0 : _a.trim()) ? `
+Headers: ${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)");
     }
@@ -14249,6 +14300,7 @@ var MCPToolCallNodeImpl = class extends NodeImpl {
     };
   }
   async process(inputs, context) {
+    var _a;
     const name = getInputOrData(this.data, inputs, "name", "string");
     const version = getInputOrData(this.data, inputs, "version", "string");
     const toolName = getInputOrData(this.data, inputs, "toolName", "string");
@@ -14293,7 +14345,20 @@ var MCPToolCallNodeImpl = class extends NodeImpl {
             "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 == null ? void 0 : headersInput.type) === "string") {
+            headers = JSON.parse(headersInput.value);
+          } else if ((headersInput == null ? void 0 : headersInput.type) === "object") {
+            headers = headersInput.value;
+          } else {
+            headers = coerceType(headersInput, "object");
+          }
+        } else if ((_a = this.data.headers) == null ? void 0 : _a.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 ?? "";
         const mcpConfig = await loadMCPConfiguration(context);
@@ -14346,6 +14411,7 @@ var MCPGetPromptNodeImpl = class extends NodeImpl {
         version: "1.0.0",
         transportType: "stdio",
         serverUrl: "http://localhost:8080/mcp",
+        headers: "",
         serverId: "",
         promptName: "",
         promptArguments: import_ts_dedent.dedent`
@@ -14430,13 +14496,22 @@ var MCPGetPromptNodeImpl = class extends NodeImpl {
       }
     ];
     if (this.data.transportType === "http") {
-      editors.push({
-        type: "string",
-        label: "Server URL",
-        dataKey: "serverUrl",
-        useInputToggleDataKey: "useServerUrlInput",
-        helperMessage: "The endpoint URL for the MCP server to connect"
-      });
+      editors.push(
+        {
+          type: "string",
+          label: "Server URL",
+          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") {
       const serverOptions = await getServerOptions(context);
       editors.push({
@@ -14450,15 +14525,19 @@ var MCPGetPromptNodeImpl = class extends NodeImpl {
     return editors;
   }
   getBody(context) {
+    var _a;
     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)" : ((_a = this.data.headers) == null ? void 0 : _a.trim()) ? `
+Headers: ${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)");
     }
@@ -14475,6 +14554,7 @@ var MCPGetPromptNodeImpl = class extends NodeImpl {
     };
   }
   async process(inputs, context) {
+    var _a;
     const name = getInputOrData(this.data, inputs, "name", "string");
     const version = getInputOrData(this.data, inputs, "version", "string");
     const promptName = getInputOrData(this.data, inputs, "promptName", "string");
@@ -14518,7 +14598,20 @@ var MCPGetPromptNodeImpl = class extends NodeImpl {
             "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 == null ? void 0 : headersInput.type) === "string") {
+            headers = JSON.parse(headersInput.value);
+          } else if ((headersInput == null ? void 0 : headersInput.type) === "object") {
+            headers = headersInput.value;
+          } else {
+            headers = coerceType(headersInput, "object");
+          }
+        } else if ((_a = this.data.headers) == null ? void 0 : _a.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 ?? "";
         const mcpConfig = await loadMCPConfiguration(context);
@@ -15330,6 +15423,7 @@ var GraphProcessor = class _GraphProcessor {
       if (this.#running) {
         throw new Error("Cannot process graph while already processing");
       }
+      console.info(`Process graph called. Context:${context}, Inputs: ${JSON.stringify(inputs)}, Context Values: ${JSON.stringify(contextValues)}`);
       this.#initProcessState();
       this.#context = context;
       this.#graphInputs = inputs;
@@ -15339,8 +15433,11 @@ var GraphProcessor = class _GraphProcessor {
           this.#emitter.emit("error", { error });
         });
       }
+      console.info(`Process graph calling loadProjectReferences`);
       await this.#loadProjectReferences();
+      console.info(`Process graph called loadProjectReferences`);
       this.#preprocessGraph();
+      console.info(`Process graph called preprocessGraph`);
       if (!this.#isSubProcessor) {
         await this.#emitter.emit("start", {
           contextValues: this.#contextValues,
@@ -16609,7 +16706,7 @@ var ChatAidonNodeImpl = class extends ChatNodeImpl {
     }
     return path;
   }
-  async callToolGet(parsedArgs, schemaDetail, path, data) {
+  async callToolGet(schemaDetail, path, parsedArgs, data) {
     const queryParams = new URLSearchParams(
       parsedArgs.parameters
     ).toString();
@@ -16687,7 +16784,7 @@ var ChatAidonNodeImpl = class extends ChatNodeImpl {
         if (schemaDetail.requestInBody) {
           data = await this.callToolPost(schemaDetail, path, functionCall.arguments, data);
         } else {
-          data = await this.callToolGet(functionCall.arguments, schemaDetail, path, data);
+          data = await this.callToolGet(schemaDetail, path, functionCall.arguments, data);
         }
         messages["value"].push({
           type: "function",
@@ -16763,6 +16860,21 @@ function chatAidonNode() {
 var aidonPlugin = {
   id: "aidon",
   name: "Aidon",
+  configSpec: {
+    aidonURL: {
+      type: "string",
+      label: "Aidon URL",
+      description: "The URL for the Aidon application.",
+      helperText: "Defaults to https://app.aidon.ai. URL for the Aidon application.",
+      default: "https://app.aidon.ai"
+    },
+    aidonKey: {
+      type: "secret",
+      label: "Aidon API Key",
+      description: "The API Key for the Aidon application.",
+      helperText: "API Key for the Aidon application."
+    }
+  },
   register: (register) => {
     register(chatAidonNode());
   }
@@ -18928,7 +19040,7 @@ var ChatHuggingFaceNodeImpl = {
     const repetitionPenalty = getInputOrData(data, inputData, "repetitionPenalty", "number");
     const topP = getInputOrData(data, inputData, "topP", "number");
     const topK = getInputOrData(data, inputData, "topK", "number");
-    const hf = endpoint ? new import_inference.HfInferenceEndpoint(endpoint, accessToken) : new import_inference.HfInference(accessToken);
+    const hf = endpoint ? new import_inference.InferenceClient(accessToken, { endpointUrl: endpoint }) : new import_inference.InferenceClient(accessToken);
     const generationStream = hf.textGenerationStream({
       inputs: prompt,
       model,
@@ -19126,18 +19238,21 @@ var TextToImageHuggingFaceNodeImpl = {
     const negativePrompt = getInputOrData(data, inputData, "negativePrompt") || void 0;
     const guidanceScale = getInputOrData(data, inputData, "guidanceScale", "number");
     const numInferenceSteps = getInputOrData(data, inputData, "numInferenceSteps", "number");
-    const hf = endpoint ? new import_inference2.HfInferenceEndpoint(endpoint, accessToken) : new import_inference2.HfInference(accessToken);
-    const image = await hf.textToImage({
-      inputs: prompt,
-      model,
-      parameters: {
-        width,
-        height,
-        negative_prompt: negativePrompt,
-        guidance_scale: guidanceScale,
-        num_inference_steps: numInferenceSteps
-      }
-    });
+    const hf = endpoint ? new import_inference2.InferenceClient(accessToken, { endpointUrl: endpoint }) : new import_inference2.InferenceClient(accessToken);
+    const image = await hf.textToImage(
+      {
+        inputs: prompt,
+        model,
+        parameters: {
+          width,
+          height,
+          negative_prompt: negativePrompt,
+          guidance_scale: guidanceScale,
+          num_inference_steps: numInferenceSteps
+        }
+      },
+      { outputType: "blob" }
+    );
     return {
       ["output"]: {
         type: "image",
@@ -22971,11 +23086,13 @@ function nodeMatches(spec, event) {
 async function* getProcessorEvents(processor, spec) {
   const previousIndexes = /* @__PURE__ */ new Map();
   const usages = [];
+  let hasDelta = false;
   for await (const event of processor.events()) {
     if (event.type === "partialOutput") {
       if (spec.partialOutputs === true || nodeMatches(spec.partialOutputs, event)) {
         const currentOutput = coerceType(event.outputs["response"], "string");
         const delta = currentOutput.slice(previousIndexes.get(event.node.id) ?? 0);
+        hasDelta = true;
         yield {
           type: "partialOutput",
           nodeId: event.node.id,
@@ -23031,7 +23148,7 @@ async function* getProcessorEvents(processor, spec) {
           usages.push(usage);
         }
       }
-      if (spec.nodeFinish === true || nodeMatches(spec.nodeFinish, event)) {
+      if ((spec.nodeFinish === true || nodeMatches(spec.nodeFinish, event)) && !(spec.removeFinalOutput && hasDelta)) {
         yield {
           type: "nodeFinish",
           outputs: event.outputs,
@@ -23071,10 +23188,14 @@ data: ${JSON.stringify(data)}
   return new ReadableStream({
     async start(controller) {
       const userEventHandler = async (eventName, data) => {
-        sendEvent(controller, "event", {
-          name: eventName,
-          message: coerceType(data, "string")
-        });
+        const graphEvent = {
+          type: "event",
+          graphEvent: {
+            name: eventName,
+            message: coerceType(data, "string")
+          }
+        };
+        sendEvent(controller, "event", graphEvent);
       };
       const streamEvents = createOnStreamUserEvents(spec.userStreamEvents, userEventHandler);
       if (streamEvents) {
@@ -23108,6 +23229,21 @@ function getSingleNodeStream(processor, arg) {
     async start(controller) {
       var _a;
       try {
+        const userEventHandler = async (eventName, data) => {
+          const payload = {
+            name: eventName,
+            message: coerceType(data, "string")
+          };
+          controller.enqueue(`event: ${JSON.stringify(payload)}
+
+`);
+        };
+        const streamEvents = createOnStreamUserEvents(spec.userStreamEvents, userEventHandler);
+        if (streamEvents) {
+          for (const [name, fn] of Object.entries(streamEvents)) {
+            processor.onUserEvent(name, fn);
+          }
+        }
         for await (const event of getProcessorEvents(processor, spec)) {
           if (event.type === "partialOutput") {
             controller.enqueue(`data: ${JSON.stringify(event.delta)}