@shop-os/modelkit 0.4.5 → 0.4.6

package/CHANGELOG.md

@@ -9,6 +9,11 @@ header in the release PR.
 
 ## [Unreleased]
 
+## [0.4.6] - Unreleased
+
+### Added
+- `ModelKit.resumeVideo(req, jobId)` and `Adapter.resumeVideo` (ark/Seedance only): reconnect to a video job `generateVideo` submitted earlier, by its `jobId`, without submitting a new job. Other adapters throw `unsupported_option`.
+
 ## [0.4.5] - Unreleased
 
 ### Fixed

package/README.md

@@ -51,6 +51,10 @@ await kit.generateText({
 await clip.cleanup?.();
 ```
 
+A video job that outlives its caller can be rejoined with `resumeVideo`: pass the same
+request and the `jobId` from `generateVideo` and await `job.result()` as usual. Only the
+ark adapter (Seedance) supports it today; every other adapter throws `unsupported_option`.
+
 ### Video tools
 
 `video-tool` rows (Topaz upscale, frame interpolation, SDR to HDR on fal) take a clip and

package/dist/index.cjs

@@ -3770,6 +3770,11 @@ var arkAdapter = {
   async generateVideo(req, ctx) {
     throwIfAborted(ctx.signal);
     return generateSeedanceVideo(req, ctx);
+  },
+  resumeVideo(req, jobId, ctx) {
+    throwIfAborted(ctx.signal);
+    const key = arkKey(ctx);
+    return seedanceJob(ctx, key, jobId, seedanceUsage(ctx, req, seedanceInitialDuration(req)));
   }
 };
 function arkKey(ctx) {
@@ -3939,6 +3944,9 @@ function seedanceUsage(ctx, req, duration) {
   if (req.resolution !== void 0) usage2.resolution = req.resolution;
   return usage2;
 }
+function seedanceInitialDuration(req) {
+  return req.durationSeconds ?? 5;
+}
 function seedanceErrorMessage(error, status) {
   const asString = import_zod7.z.string().safeParse(error);
   if (asString.success) return asString.data;
@@ -3946,6 +3954,78 @@ function seedanceErrorMessage(error, status) {
   if (asObject.success && asObject.data.message !== void 0) return asObject.data.message;
   return `Seedance task ${status}`;
 }
+function seedanceJob(ctx, key, jobId, usage2) {
+  const modelId = ctx.model.id;
+  const fetchStatus = async () => {
+    const res = await arkFetch(ctx, `${arkBaseUrl(ctx)}${SUBMIT_PATH}/${jobId}`, {
+      headers: { Authorization: `Bearer ${key}` }
+    });
+    if (!res.ok) {
+      if (ctx.call.maxRetries !== 0 && (res.status === 429 || res.status >= 500))
+        return void 0;
+      const text2 = await res.text().catch(() => "");
+      throwArkError(res.status, text2, modelId);
+    }
+    const parsed = SeedanceResponseSchema.safeParse(await res.json());
+    if (!parsed.success) return {};
+    return parsed.data;
+  };
+  return {
+    jobId,
+    async poll() {
+      const polled = await fetchStatus();
+      if (!polled) return { status: "running" };
+      const status = (polled.status ?? "queued").toLowerCase();
+      if (status === "succeeded") return { status: "succeeded", raw: polled };
+      if (status === "failed" || status === "expired" || status === "cancelled") {
+        return { status: "failed", raw: polled };
+      }
+      if (status === "queued" || status === "pending")
+        return { status: "pending", raw: polled };
+      return { status: "running", raw: polled };
+    },
+    async result(opts) {
+      const optSignal = opts?.signal ?? ctx.signal;
+      const pollOpts = {
+        intervalMs: opts?.pollIntervalMs ?? SEEDANCE_POLL_INTERVAL_MS,
+        timeoutMs: opts?.timeoutMs ?? RESULT_TIMEOUT_MS
+      };
+      if (optSignal !== void 0) pollOpts.signal = optSignal;
+      return pollUntil(async () => {
+        const polled = await fetchStatus();
+        if (!polled) return void 0;
+        const status = (polled.status ?? "queued").toLowerCase();
+        if (status === "succeeded") {
+          const url = polled.content?.video_url;
+          if (!url)
+            throwArkError(
+              void 0,
+              "Seedance task succeeded but returned no video_url",
+              modelId
+            );
+          if (polled.duration !== void 0 && polled.duration > 0) {
+            usage2.quantity = polled.duration;
+            usage2.durationSeconds = polled.duration;
+          }
+          const tokens = polled.usage;
+          if (tokens?.total_tokens !== void 0 && tokens.completion_tokens !== void 0) {
+            usage2.inputTokens = tokens.total_tokens - tokens.completion_tokens;
+            usage2.outputTokens = tokens.completion_tokens;
+          }
+          return {
+            video: { url, expiresAt: new Date(Date.now() + VIDEO_URL_TTL_MS) },
+            usage: usage2,
+            raw: polled
+          };
+        }
+        if (status === "failed" || status === "expired" || status === "cancelled") {
+          throwArkError(void 0, seedanceErrorMessage(polled.error, status), modelId);
+        }
+        return void 0;
+      }, pollOpts);
+    }
+  };
+}
 async function generateSeedanceVideo(req, ctx) {
   const modelId = ctx.model.id;
   const taskType = req.taskType;
@@ -3960,7 +4040,7 @@ async function generateSeedanceVideo(req, ctx) {
     });
   }
   const key = arkKey(ctx);
-  const requestedDuration = req.durationSeconds ?? 5;
+  const requestedDuration = seedanceInitialDuration(req);
   let duration = requestedDuration;
   if (taskType === "edit") duration = -1;
   const resolution = req.resolution ?? "720p";
@@ -4041,75 +4121,7 @@ async function generateSeedanceVideo(req, ctx) {
   if (!submit.id) throwArkError(void 0, "Seedance task create returned no id", modelId);
   const jobId = submit.id;
   const usage2 = seedanceUsage(ctx, req, taskType === "edit" ? requestedDuration : duration);
-  const fetchStatus = async () => {
-    const res = await arkFetch(ctx, `${arkBaseUrl(ctx)}${SUBMIT_PATH}/${jobId}`, {
-      headers: { Authorization: `Bearer ${key}` }
-    });
-    if (!res.ok) {
-      if (ctx.call.maxRetries !== 0 && (res.status === 429 || res.status >= 500))
-        return void 0;
-      const text2 = await res.text().catch(() => "");
-      throwArkError(res.status, text2, modelId);
-    }
-    const parsed = SeedanceResponseSchema.safeParse(await res.json());
-    if (!parsed.success) return {};
-    return parsed.data;
-  };
-  return {
-    jobId,
-    async poll() {
-      const polled = await fetchStatus();
-      if (!polled) return { status: "running" };
-      const status = (polled.status ?? "queued").toLowerCase();
-      if (status === "succeeded") return { status: "succeeded", raw: polled };
-      if (status === "failed" || status === "expired" || status === "cancelled") {
-        return { status: "failed", raw: polled };
-      }
-      if (status === "queued" || status === "pending")
-        return { status: "pending", raw: polled };
-      return { status: "running", raw: polled };
-    },
-    async result(opts) {
-      const optSignal = opts?.signal ?? ctx.signal;
-      const pollOpts = {
-        intervalMs: opts?.pollIntervalMs ?? SEEDANCE_POLL_INTERVAL_MS,
-        timeoutMs: opts?.timeoutMs ?? RESULT_TIMEOUT_MS
-      };
-      if (optSignal !== void 0) pollOpts.signal = optSignal;
-      return pollUntil(async () => {
-        const polled = await fetchStatus();
-        if (!polled) return void 0;
-        const status = (polled.status ?? "queued").toLowerCase();
-        if (status === "succeeded") {
-          const url = polled.content?.video_url;
-          if (!url)
-            throwArkError(
-              void 0,
-              "Seedance task succeeded but returned no video_url",
-              modelId
-            );
-          if (polled.duration !== void 0 && polled.duration > 0) {
-            usage2.quantity = polled.duration;
-            usage2.durationSeconds = polled.duration;
-          }
-          const tokens = polled.usage;
-          if (tokens?.total_tokens !== void 0 && tokens.completion_tokens !== void 0) {
-            usage2.inputTokens = tokens.total_tokens - tokens.completion_tokens;
-            usage2.outputTokens = tokens.completion_tokens;
-          }
-          return {
-            video: { url, expiresAt: new Date(Date.now() + VIDEO_URL_TTL_MS) },
-            usage: usage2,
-            raw: polled
-          };
-        }
-        if (status === "failed" || status === "expired" || status === "cancelled") {
-          throwArkError(void 0, seedanceErrorMessage(polled.error, status), modelId);
-        }
-        return void 0;
-      }, pollOpts);
-    }
-  };
+  return seedanceJob(ctx, key, jobId, usage2);
 }
 
 // src/adapters/fal.ts
@@ -9689,6 +9701,33 @@ function createModelKit(config, adapters) {
       result: () => done
     };
   }
+  function wrapVideoJob(model, job, warnings, startedAt) {
+    const result = job.result.bind(job);
+    let settled;
+    return {
+      ...job,
+      warnings,
+      result(opts) {
+        if (settled) return settled;
+        const pending = result(opts).then(
+          async (out) => {
+            const usage2 = timed(
+              { ...out.usage, model: model.id, unit: model.usage.unit },
+              startedAt
+            );
+            await emit(usage2);
+            return { ...out, usage: usage2 };
+          },
+          (cause) => {
+            settled = void 0;
+            throw cause;
+          }
+        );
+        settled = pending;
+        return pending;
+      }
+    };
+  }
   return {
     catalog,
     describe: (id) => catalog.require(id),
@@ -9773,31 +9812,52 @@ function createModelKit(config, adapters) {
         });
       const startedAt = Date.now();
       const job = await adapter.generateVideo(request, ctx);
-      const result = job.result.bind(job);
-      let settled;
-      return {
-        ...job,
-        warnings,
-        result(opts) {
-          if (settled) return settled;
-          const pending = result(opts).then(
-            async (out) => {
-              const usage2 = timed(
-                { ...out.usage, model: model.id, unit: model.usage.unit },
-                startedAt
-              );
-              await emit(usage2);
-              return { ...out, usage: usage2 };
-            },
-            (cause) => {
-              settled = void 0;
-              throw cause;
-            }
-          );
-          settled = pending;
-          return pending;
-        }
-      };
+      return wrapVideoJob(model, job, warnings, startedAt);
+    },
+    async resumeVideo(req, jobId, call = {}) {
+      mediaRetryOptions(call);
+      if (!jobId) {
+        throw new ModelKitError({
+          code: "missing_input",
+          message: "resumeVideo needs a jobId",
+          retryable: false
+        });
+      }
+      if (!isCatalogId(req.model)) {
+        throw new ModelKitError({
+          code: "unsupported_option",
+          message: "resumeVideo needs a catalog model id"
+        });
+      }
+      const model = catalog.require(req.model);
+      if (model.type !== "video")
+        throw new ModelKitError({
+          code: "unsupported_option",
+          message: `${model.id} is not a video model`,
+          model: model.id
+        });
+      const { request: input, warnings: inputWarnings } = videoRequestOf({
+        ...req,
+        model: req.model
+      });
+      const { request, warnings } = resolveVideoOptions(
+        model,
+        input,
+        call.onUnsupported ?? "error"
+      );
+      warnings.push(...inputWarnings);
+      logCoercions(logger, model.id, warnings);
+      const signal = combineSignals(req.abortSignal, call.signal);
+      const { adapter, ctx } = context(model, signal ? { ...call, signal } : call, warnings);
+      if (!adapter.resumeVideo)
+        throw new ModelKitError({
+          code: "unsupported_option",
+          message: `${adapter.id} cannot resume a video job`,
+          model: model.id
+        });
+      const startedAt = Date.now();
+      const job = adapter.resumeVideo(request, jobId, ctx);
+      return wrapVideoJob(model, { ...job, jobId }, warnings, startedAt);
     },
     async generateAudio(req, call = {}) {
       mediaRetryOptions(call);