@workflow/world-testing 4.0.1-beta.15 → 4.0.1-beta.17

package/dist/.well-known/workflow/v1/flow.js

@@ -43854,7 +43854,8 @@ __name(deserializeChunk, "deserializeChunk");
 function createStreamer(basedir) {
     const streamEmitter = new import_node_events15.EventEmitter();
     return {
-        async writeToStream(name, chunk) {
+        async writeToStream(name, _runId, chunk) {
+            await _runId;
             const chunkId = `strm_${monotonicUlid2()}`;
             if (typeof chunk === "string") {
                 chunk = new TextEncoder().encode(chunk);
@@ -43872,7 +43873,8 @@ function createStreamer(basedir) {
                 chunkId
             });
         },
-        async closeStream(name) {
+        async closeStream(name, _runId) {
+            await _runId;
             const chunkId = `strm_${monotonicUlid2()}`;
             const chunkPath = import_node_path8.default.join(basedir, "streams", "chunks", `${name}-${chunkId}.json`);
             await write(chunkPath, serializeChunk({ chunk: Buffer.from([]), eof: true }));
@@ -45484,7 +45486,7 @@ __name(send, "send");
 var import_node_os4 = __toESM(require("node:os"), 1);
 var import_oidc2 = __toESM(require_dist(), 1);
 // ../world-vercel/dist/version.js
-var version2 = "4.0.1-beta.8";
+var version2 = "4.0.1-beta.9";
 // ../world-vercel/dist/utils.js
 var DEFAULT_RESOLVE_DATA_OPTION2 = "all";
 function dateToStringReplacer(_key, value) {
@@ -45836,7 +45838,9 @@ var WorkflowRunWireWithRefsSchema = WorkflowRunWireBaseSchema.omit({
     inputRef: external_exports.any().optional(),
     outputRef: external_exports.any().optional(),
     input: external_exports.array(external_exports.any()).optional(),
-    output: external_exports.any().optional()
+    output: external_exports.any().optional(),
+    blobStorageBytes: external_exports.number().optional(),
+    streamStorageBytes: external_exports.number().optional()
 });
 function filterRunData2(run, resolveData) {
     if (resolveData === "none") {
@@ -46144,32 +46148,37 @@ function createStorage2(config3) {
 }
 __name(createStorage2, "createStorage");
 // ../world-vercel/dist/streamer.js
-function getStreamUrl(name, httpConfig) {
+function getStreamUrl(name, runId, httpConfig) {
+    if (runId) {
+        return new URL(`${httpConfig.baseUrl}/v1/runs/${runId}/stream/${encodeURIComponent(name)}`);
+    }
     return new URL(`${httpConfig.baseUrl}/v1/stream/${encodeURIComponent(name)}`);
 }
 __name(getStreamUrl, "getStreamUrl");
 function createStreamer2(config3) {
     return {
-        async writeToStream(name, chunk) {
+        async writeToStream(name, runId, chunk) {
+            const resolvedRunId = await runId;
             const httpConfig = await getHttpConfig(config3);
-            await fetch(getStreamUrl(name, httpConfig), {
+            await fetch(getStreamUrl(name, resolvedRunId, httpConfig), {
                 method: "PUT",
                 body: chunk,
                 headers: httpConfig.headers,
                 duplex: "half"
             });
         },
-        async closeStream(name) {
+        async closeStream(name, runId) {
+            const resolvedRunId = await runId;
             const httpConfig = await getHttpConfig(config3);
             httpConfig.headers.set("X-Stream-Done", "true");
-            await fetch(getStreamUrl(name, httpConfig), {
+            await fetch(getStreamUrl(name, resolvedRunId, httpConfig), {
                 method: "PUT",
                 headers: httpConfig.headers
             });
         },
         async readFromStream(name, startIndex) {
             const httpConfig = await getHttpConfig(config3);
-            const url2 = getStreamUrl(name, httpConfig);
+            const url2 = getStreamUrl(name, void 0, httpConfig);
             if (typeof startIndex === "number") {
                 url2.searchParams.set("startIndex", String(startIndex));
             }
@@ -46784,17 +46793,22 @@ var WorkflowServerWritableStream = class extends WritableStream {
     static {
         __name(this, "WorkflowServerWritableStream");
     }
-    constructor(name) {
+    constructor(name, runId) {
+        if (typeof runId !== "string" && !(runId instanceof Promise)) {
+            throw new Error(`"runId" must be a string or a promise that resolves to a string, got "${typeof runId}"`);
+        }
         if (typeof name !== "string" || name.length === 0) {
             throw new Error(`"name" is required, got "${name}"`);
         }
         const world = getWorld();
         super({
             async write(chunk) {
-                await world.writeToStream(name, chunk);
+                const _runId = await runId;
+                await world.writeToStream(name, _runId, chunk);
             },
             async close() {
-                await world.closeStream(name);
+                const _runId = await runId;
+                await world.closeStream(name, _runId);
             }
         });
     }