@workglow/ai 0.0.93 → 0.0.94

package/dist/bun.js

@@ -1843,6 +1843,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) {
@@ -1853,10 +1856,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;
+      }
+    }
   }
 }
 
@@ -5793,4 +5804,4 @@ export {
   AiJob
 };
 
-//# debugId=DA970B9D5234412964756E2164756E21
+//# debugId=E3FAD6BE2442A4DE64756E2164756E21