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

package/dist/cjs/bundle.cjs

@@ -22971,11 +22971,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 +23033,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 +23073,11 @@ data: ${JSON.stringify(data)}
   return new ReadableStream({
     async start(controller) {
       const userEventHandler = async (eventName, data) => {
-        sendEvent(controller, "event", {
+        const graphEvent = {
           name: eventName,
           message: coerceType(data, "string")
-        });
+        };
+        sendEvent(controller, "event", { graphEvent });
       };
       const streamEvents = createOnStreamUserEvents(spec.userStreamEvents, userEventHandler);
       if (streamEvents) {
@@ -23108,6 +23111,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)}