@workglow/ai 0.0.92 → 0.0.94

package/dist/node.js

@@ -1842,6 +1842,9 @@ Workflow9.prototype.textNamedEntityRecognition = CreateWorkflow9(TextNamedEntity
 import { CreateWorkflow as CreateWorkflow10, Workflow as Workflow10 } from "@workglow/task-graph";
 
 // src/task/base/StreamingAiTask.ts
+import {
+  getStreamingPorts
+} from "@workglow/task-graph";
 class StreamingAiTask extends AiTask {
   static type = "StreamingAiTask";
   async* executeStream(input, context) {
@@ -1852,10 +1855,18 @@ class StreamingAiTask extends AiTask {
       jobRunId: this.config.runnerId,
       input: jobInput
     });
-    yield* job.executeStream(jobInput, {
+    const ports = getStreamingPorts(this.outputSchema());
+    const defaultPort = ports.length > 0 ? ports[0].port : "text";
+    for await (const event of job.executeStream(jobInput, {
       signal: context.signal,
       updateProgress: context.updateProgress.bind(this)
-    });
+    })) {
+      if (event.type === "text-delta") {
+        yield { ...event, port: event.port ?? defaultPort };
+      } else {
+        yield event;
+      }
+    }
   }
 }
 
@@ -5792,4 +5803,4 @@ export {
   AiJob
 };
 
-//# debugId=F363EF2B0833C2C664756E2164756E21
+//# debugId=BDC88B5889B3E75A64756E2164756E21