@hypit/hypit 0.2.3 → 0.2.5

package/packages/provider-tokendance/src/provider.ts

@@ -0,0 +1,268 @@
+import { requestDeadline } from "@hypit/runtime-kit";
+import type { AsyncEndpoint, EndpointCredential, EndpointInvocationContext, EndpointOutcome, ImmediateEndpointHandler } from "@hypit/endpoint-kit";
+import { defineEndpointPackage, wakeAfter } from "@hypit/endpoint-kit";
+import type { GenerationArtifactUrlResolver } from "@hypit/generation";
+import { canonicalize } from "@hypit/protocol";
+import type { BlobRef, CapabilityRef } from "@hypit/protocol";
+import { credentialRef } from "@hypit/runtime";
+import type { CredentialRef, ResourceStore } from "@hypit/runtime";
+import { tokenDanceRouteForCapability, tokenDanceRoutes } from "./routes.js";
+import type { TokenDanceRoute } from "./routes.js";
+import { TokenDanceHttpError, TokenDanceServiceError, tokenDanceTaskFailure } from "./errors.js";
+
+export const tokenDanceProviderModuleRef = { name: "@hypit/provider-tokendance", version: "1" } as const;
+
+export type CreateTokenDanceProviderOptions = {
+  readonly instance?: string;
+  readonly pool?: string;
+  readonly baseUrl?: string;
+  readonly apiKey?: CredentialRef;
+  readonly defaultConcurrency?: number;
+  readonly actionLimits?: import("@hypit/endpoint-kit").EndpointActionLimits;
+  readonly pollIntervalMs?: number;
+  readonly requestTimeoutMs?: number;
+  readonly operationTimeoutMs?: number;
+  readonly fetch?: typeof globalThis.fetch;
+  /** Publish a referenced Resource at a URL the service can fetch; replaces inline data URLs. */
+  readonly publicAssetUrl?: (artifact: BlobRef, artifacts: ResourceStore, fields?: Readonly<Record<string, string | number | boolean>>) => Promise<string>;
+};
+
+type Handle = {
+  readonly contract: "hypit.tokendance-operation@1";
+  readonly taskId: string;
+  readonly route: string;
+  readonly startedAt: number;
+  readonly url?: string;
+};
+
+function assert(condition: unknown, message: string): asserts condition { if (!condition) throw new Error(message); }
+function object(value: unknown, subject: string): Record<string, unknown> {
+  assert(value !== null && typeof value === "object" && !Array.isArray(value), `${subject} must be an object`);
+  return value as Record<string, unknown>;
+}
+function capabilityKey(capability: CapabilityRef): string { return `${capability.module.name}@${capability.module.version}#${capability.name}`; }
+function apiBaseUrl(value: string): string {
+  let trimmed = value.trim();
+  while (trimmed.endsWith("/")) trimmed = trimmed.slice(0, -1);
+  assert(trimmed.length > 0, "TokenDance base URL is empty");
+  return trimmed;
+}
+function apiKey(credentials: Readonly<Record<string, EndpointCredential>>): string {
+  const value = credentials.apiKey?.secret;
+  assert(typeof value === "string" && value.length > 0, "TokenDance apiKey credential is unavailable; store a TokenDance API key for this Endpoint");
+  return value;
+}
+function failureMessage(error: unknown): string {
+  return error instanceof Error ? error.message : String(error);
+}
+function failure(error: unknown): EndpointOutcome {
+  return { status: "failed", failure: { code: error instanceof TokenDanceServiceError ? error.code : "TOKENDANCE_ERROR", message: failureMessage(error) } };
+}
+function httpsUrl(value: unknown, subject: string): string {
+  assert(typeof value === "string" && /^https?:\/\//u.test(value), `${subject} has no download URL`);
+  return value;
+}
+
+class TokenDanceClient {
+  constructor(readonly baseUrl: string, readonly timeout: number, readonly fetcher: typeof globalThis.fetch) {}
+  async json(path: string, key: string, init: RequestInit = {}): Promise<Record<string, unknown>> {
+    const deadline = requestDeadline(this.timeout);
+    try {
+      const response = await deadline.wait(this.fetcher(`${this.baseUrl}${path}`, {
+        ...init, signal: deadline.signal, headers: { authorization: `Bearer ${key}`, ...(init.headers ?? {}) },
+      }));
+      const text = await deadline.wait(response.text());
+      if (!response.ok) {
+        const input = typeof init.body === "string" ? JSON.parse(init.body) as Record<string, unknown> : undefined;
+        throw new TokenDanceHttpError(response.status, response, text, {
+          method: init.method ?? "GET", path, ...(typeof input?.model === "string" ? { model: input.model } : {}),
+        });
+      }
+      let body: unknown;
+      try { body = text.length === 0 ? {} : JSON.parse(text); } catch { throw new Error(`TokenDance returned invalid JSON (${response.status})`); }
+      return object(body, "TokenDance response");
+    } finally { deadline.finish(); }
+  }
+  /** MiniMax's file API through the gateway; the returned id is referenced as `mm_file://{file_id}`. */
+  async uploadMiniMaxInput(bytes: Uint8Array, artifact: BlobRef, key: string): Promise<string> {
+    const form = new FormData();
+    form.set("purpose", "video_generation_input");
+    form.set("file", new Blob([new Uint8Array(bytes)], { type: artifact.mediaType }), `${artifact.resource}.${artifact.mediaType.split("/")[1] ?? "bin"}`);
+    const response = await this.json("/minimax/v1/files/upload", key, { method: "POST", body: form });
+    const status = object(response.base_resp ?? {}, "TokenDance upload base_resp").status_code;
+    assert(status === undefined || status === 0, `TokenDance MiniMax upload rejected: ${String(object(response.base_resp, "TokenDance upload base_resp").status_msg ?? status)}`);
+    const id = object(response.file, "TokenDance upload file").file_id;
+    assert((typeof id === "string" && id.length > 0) || typeof id === "number", "TokenDance MiniMax upload returned no file_id");
+    return `mm_file://${String(id)}`;
+  }
+  async download(url: string): Promise<{ readonly bytes: Uint8Array; readonly mediaType: string }> {
+    const deadline = requestDeadline(this.timeout);
+    try {
+      const response = await deadline.wait(this.fetcher(url, { signal: deadline.signal }));
+      if (!response.ok) throw new Error(`TokenDance asset returned HTTP ${response.status}`);
+      return { bytes: new Uint8Array(await deadline.wait(response.arrayBuffer())), mediaType: response.headers.get("content-type")?.split(";", 1)[0] ?? "application/octet-stream" };
+    } finally { deadline.finish(); }
+  }
+}
+
+function mediaKind(mediaType: string): "image" | "video" | "audio" | undefined {
+  const kind = mediaType.split("/", 1)[0];
+  return kind === "image" || kind === "video" || kind === "audio" ? kind : undefined;
+}
+
+function resolverFor(client: TokenDanceClient, route: TokenDanceRoute, context: EndpointInvocationContext, publicAssetUrl: CreateTokenDanceProviderOptions["publicAssetUrl"]): GenerationArtifactUrlResolver {
+  const resolved = new Map<string, Promise<string>>();
+  return (artifact, fields) => {
+    const existing = resolved.get(artifact.resource);
+    if (existing !== undefined) return existing;
+    const promise = (async () => {
+      if (publicAssetUrl !== undefined) return await publicAssetUrl(artifact, context.resources, fields);
+      const kind = mediaKind(artifact.mediaType);
+      const limit = kind === undefined ? undefined : route.mediaLimits[kind];
+      assert(limit !== undefined,
+        `TokenDance ${route.protocol} accepts ${artifact.mediaType} references only by public URL; configure publicAssetUrl for this Endpoint`);
+      assert(artifact.size <= limit,
+        `TokenDance ${route.protocol} accepts ${kind} references up to ${limit / 1_000_000} MB; ${artifact.resource} is ${artifact.size} bytes`);
+      const bytes = await context.resources.get(artifact.resource);
+      assert(bytes !== undefined && bytes.byteLength === artifact.size, `Reference Resource ${artifact.resource} is unavailable or has changed`);
+      if (route.protocol === "minimax-video") return await client.uploadMiniMaxInput(bytes, artifact, apiKey(context.credentials));
+      return `data:${artifact.mediaType};base64,${Buffer.from(bytes).toString("base64")}`;
+    })();
+    resolved.set(artifact.resource, promise);
+    return promise;
+  };
+}
+
+async function prepare(client: TokenDanceClient, context: EndpointInvocationContext, publicAssetUrl: CreateTokenDanceProviderOptions["publicAssetUrl"]) {
+  const route = tokenDanceRouteForCapability(context.need.capability);
+  assert(route !== undefined, "TokenDance does not implement this exact capability");
+  const request = route.prepare(context.need.constraints);
+  await context.reportProgress?.({ phase: `Preparing TokenDance request: ${request.model}` });
+  let body: string;
+  try {
+    body = JSON.stringify(await request.compile(resolverFor(client, route, context, publicAssetUrl)));
+    const cap = route.mediaLimits.body;
+    assert(cap === undefined || Buffer.byteLength(body) <= cap,
+      `TokenDance ${route.protocol} accepts request bodies up to ${(cap ?? 0) / 1_000_000} MB; inline references make this one ${Buffer.byteLength(body)} bytes`);
+  } catch (error) {
+    throw new TokenDanceServiceError(error instanceof TokenDanceServiceError ? error.code : "TOKENDANCE_ERROR",
+      `TokenDance request preparation failed; model=${request.model}; generation not submitted: ${failureMessage(error)}`);
+  }
+  return { route, model: request.model, body };
+}
+
+async function store(client: TokenDanceClient, urls: readonly string[], resources: ResourceStore): Promise<BlobRef[]> {
+  const blobs: BlobRef[] = [];
+  for (const url of urls) {
+    const downloaded = await client.download(url);
+    blobs.push(await resources.put(downloaded.bytes, downloaded.mediaType));
+  }
+  return blobs;
+}
+
+const paths = {
+  "ark-video": { submit: "/ark/v3/generations/tasks", task: (id: string) => `/ark/v3/generations/tasks/${encodeURIComponent(id)}` },
+  "minimax-video": { submit: "/minimax/v2/video_generation", task: (id: string) => `/minimax/v2/query/video_generation/${encodeURIComponent(id)}` },
+} as const;
+
+function taskId(protocol: keyof typeof paths, response: Record<string, unknown>): string {
+  const id = protocol === "ark-video" ? response.id : response.task_id;
+  assert(typeof id === "string" && id.length > 0, "TokenDance response has no task id");
+  return id;
+}
+
+/** Ark answers the task itself; MiniMax wraps it in `task`. */
+function taskBody(protocol: keyof typeof paths, response: Record<string, unknown>): Record<string, unknown> {
+  return protocol === "ark-video" ? response : object(response.task, "TokenDance task");
+}
+
+function taskVideoUrl(protocol: keyof typeof paths, task: Record<string, unknown>): string {
+  const content = object(task.content, "TokenDance task content");
+  return httpsUrl(protocol === "ark-video" ? content.video_url : content.url, "TokenDance task");
+}
+
+function endpoint(client: TokenDanceClient, pollIntervalMs: number, maxOperationMs: number, publicAssetUrl: CreateTokenDanceProviderOptions["publicAssetUrl"]): AsyncEndpoint {
+  return {
+    async start(context) {
+      try {
+        const { route, model, body } = await prepare(client, context, publicAssetUrl);
+        assert(route.protocol !== "ark-image", "TokenDance image capabilities use an immediate endpoint");
+        await context.reportProgress?.({ phase: `Submitting TokenDance request: ${model}` });
+        const response = await client.json(paths[route.protocol].submit, apiKey(context.credentials), {
+          method: "POST", headers: { "content-type": "application/json" }, body,
+        });
+        const handle: Handle = { contract: "hypit.tokendance-operation@1", taskId: taskId(route.protocol, response), route: route.key, startedAt: Date.now() };
+        const receipt = { id: handle.taskId };
+        await context.checkpoint?.({ handle: canonicalize(handle), receipt });
+        return { ...wakeAfter(canonicalize(handle), pollIntervalMs, Date.now(), { phase: "submitted" }), receipt };
+      } catch (error) {
+        return failure(error);
+      }
+    },
+    async poll(context) {
+      try {
+        const handle = object(context.handle, "TokenDance handle") as unknown as Handle;
+        const route = tokenDanceRouteForCapability(context.need.capability);
+        assert(route !== undefined && handle.contract === "hypit.tokendance-operation@1" && handle.route === route.key && route.protocol !== "ark-image", "TokenDance handle is invalid");
+        const receipt = { id: handle.taskId };
+        if (Date.now() - handle.startedAt > maxOperationMs) {
+          return { status: "failed", receipt, failure: { code: "TOKENDANCE_OPERATION_TIMEOUT", message: `TokenDance task ${handle.taskId} exceeded this Provider's operationTimeoutMs (${maxOperationMs}); remote outcome is unknown` } };
+        }
+        const task = taskBody(route.protocol, await client.json(paths[route.protocol].task(handle.taskId), apiKey(context.credentials)));
+        const status = String(task.status);
+        if (status === "queued" || status === "running") return { ...wakeAfter(canonicalize(handle), pollIntervalMs, Date.now(), { phase: status }), receipt };
+        const rejected = tokenDanceTaskFailure(task, handle.taskId);
+        if (rejected !== undefined) return { ...failure(rejected), receipt };
+        assert(status === "succeeded", `TokenDance returned unknown task status ${status}`);
+        return { status: "ready", handle: canonicalize({ ...handle, url: taskVideoUrl(route.protocol, task) }), receipt };
+      } catch (error) {
+        return failure(error);
+      }
+    },
+    async collect(context) {
+      try {
+        const handle = object(context.handle, "TokenDance handle") as unknown as Handle;
+        const route = tokenDanceRouteForCapability(context.need.capability);
+        assert(route !== undefined && handle.route === route.key, "TokenDance collection route differs");
+        await context.reportProgress?.({ phase: "Receiving generated video" });
+        const blobs = await store(client, [httpsUrl(handle.url, "TokenDance handle")], context.resources);
+        return { status: "completed", result: { value: route.packageResult(blobs) }, receipt: { id: handle.taskId } };
+      } catch (error) {
+        return failure(error);
+      }
+    },
+  };
+}
+
+export function createTokenDanceProvider(options: CreateTokenDanceProviderOptions = {}) {
+  const requestTimeoutMs = options.requestTimeoutMs ?? 300_000;
+  const operationTimeoutMs = options.operationTimeoutMs ?? 30 * 60_000;
+  for (const [name, value] of Object.entries({ requestTimeoutMs, operationTimeoutMs })) {
+    assert(Number.isSafeInteger(value) && value > 0, `TokenDance ${name} must be a positive integer`);
+  }
+  const client = new TokenDanceClient(apiBaseUrl(options.baseUrl ?? "https://tokendance.space/gateway"), requestTimeoutMs, options.fetch ?? globalThis.fetch);
+  const asyncEndpoint = endpoint(client, options.pollIntervalMs ?? 10_000, operationTimeoutMs, options.publicAssetUrl);
+  const imageEndpoint: ImmediateEndpointHandler = async (context) => {
+    const { route, model, body } = await prepare(client, context, options.publicAssetUrl);
+    assert(route.protocol === "ark-image", "TokenDance video capabilities use an asynchronous endpoint");
+    await context.reportProgress?.({ phase: `Submitting TokenDance request: ${model}` });
+    const response = await client.json("/ark/v3/images/generations", apiKey(context.credentials), {
+      method: "POST", headers: { "content-type": "application/json" }, body,
+    });
+    assert(Array.isArray(response.data) && response.data.length > 0, "TokenDance image response has no data");
+    const urls = response.data.map((item, index) => httpsUrl(object(item, `TokenDance image ${index + 1}`).url, `TokenDance image ${index + 1}`));
+    await context.reportProgress?.({ phase: "Receiving generated images" });
+    return { value: route.packageResult(await store(client, urls, context.resources)) };
+  };
+  return defineEndpointPackage({
+    module: tokenDanceProviderModuleRef, facet: "gateway", instance: options.instance ?? "tokendance.default", pool: options.pool ?? options.instance ?? "tokendance.default",
+    pricing: { kind: "page", url: "https://tokendance.space/models" },
+    credentials: { apiKey: options.apiKey ?? credentialRef("os", "tokendance.api-key") },
+    credentialInputs: { apiKey: { label: "TokenDance API key" } },
+    defaultConcurrency: options.defaultConcurrency ?? 4,
+    ...(options.actionLimits === undefined ? {} : { actionLimits: options.actionLimits }),
+    capabilities: tokenDanceRoutes.map((route) => route.protocol === "ark-image"
+      ? { capability: route.capability, returns: route.returns, lifecycle: "immediate" as const, handler: imageEndpoint, capacity: route.capability.name, supports: route.supports }
+      : { capability: route.capability, returns: route.returns, lifecycle: "asynchronous" as const, endpoint: asyncEndpoint, capacity: route.capability.name, supports: route.supports }),
+  });
+}

package/packages/provider-tokendance/src/routes.ts

@@ -0,0 +1,192 @@
+import {
+  compileWireRequest,
+  selectWireModelForRequest,
+  generationTypes,
+  sealGeneratedImageSet,
+  sealGeneratedVideoSet,
+} from "@hypit/generation";
+import type { GenerationArtifactUrlResolver, GenerationRequest, GenerationWireMapping } from "@hypit/generation";
+import { canonicalize } from "@hypit/protocol";
+import type { BlobRef, CapabilityRef, CanonicalValue, StoredValue, TypeRef } from "@hypit/protocol";
+import type { EndpointRequest, EndpointSupport } from "@hypit/endpoint-kit";
+import { tokenDanceMappings } from "./mapping.js";
+
+/** Which TokenDance protocol a capability submits through. */
+export type TokenDanceProtocol = "ark-video" | "ark-image" | "minimax-video";
+
+export type TokenDancePreparedRequest = {
+  readonly model: string;
+  readonly protocol: TokenDanceProtocol;
+  readonly compile: (resolve: GenerationArtifactUrlResolver) => Promise<Record<string, unknown>>;
+};
+
+/** Documented byte limits for one protocol's media inputs; an absent kind is not accepted inline. */
+export type TokenDanceMediaLimits = {
+  readonly image?: number;
+  readonly video?: number;
+  readonly audio?: number;
+  /** Documented cap on the whole JSON request body, when inline media count toward it. */
+  readonly body?: number;
+};
+
+const MB = 1_000_000;
+
+/**
+ * Ark: images under 30 MB and audio at most 15 MB may travel as Base64 within a 64 MB request body;
+ * videos take URLs only. MiniMax: every kind is uploaded through its file API within these sizes.
+ */
+export const tokenDanceMediaLimits: Readonly<Record<TokenDanceProtocol, TokenDanceMediaLimits>> = {
+  "ark-video": { image: 30 * MB, audio: 15 * MB, body: 64 * MB },
+  "ark-image": { image: 30 * MB },
+  "minimax-video": { image: 30 * MB, video: 50 * MB, audio: 15 * MB },
+};
+
+export type TokenDanceRoute = GenerationWireMapping & {
+  readonly key: string;
+  readonly returns: TypeRef;
+  readonly protocol: TokenDanceProtocol;
+  readonly mediaLimits: TokenDanceMediaLimits;
+  readonly supports: (request: EndpointRequest) => EndpointSupport;
+  readonly prepare: (constraints: CanonicalValue) => TokenDancePreparedRequest;
+  readonly packageResult: (artifacts: readonly BlobRef[]) => StoredValue;
+};
+
+function scalar(request: GenerationRequest, port: string): string | number | boolean | undefined {
+  const value = request.ports[port]?.[0];
+  return typeof value === "string" || typeof value === "number" || typeof value === "boolean" ? value : undefined;
+}
+function present(request: GenerationRequest, port: string): boolean {
+  return (request.ports[port]?.length ?? 0) > 0;
+}
+function strings(value: unknown): readonly string[] {
+  return Array.isArray(value) ? value.filter((item): item is string => typeof item === "string" && item.length > 0) : [];
+}
+
+/** Seedream 5.0 lite pixel sizes for each resolution tier and aspect ratio, from the Ark image API reference. */
+const SEEDREAM_SIZES: Readonly<Record<string, Readonly<Record<string, string>>>> = {
+  basic: { "1:1": "2048x2048", "4:3": "2304x1728", "3:4": "1728x2304", "16:9": "2848x1600", "9:16": "1600x2848", "3:2": "2496x1664", "2:3": "1664x2496", "21:9": "3136x1344" },
+  high: { "1:1": "3072x3072", "4:3": "3456x2592", "3:4": "2592x3456", "16:9": "4096x2304", "9:16": "2304x4096", "3:2": "3744x2496", "2:3": "2496x3744", "21:9": "4704x2016" },
+  ultra: { "1:1": "4096x4096", "4:3": "4704x3520", "3:4": "3520x4704", "16:9": "5504x3040", "9:16": "3040x5504", "3:2": "4992x3328", "2:3": "3328x4992", "21:9": "6240x2656" },
+};
+
+function rejection(mapping: GenerationWireMapping, request: GenerationRequest): string | undefined {
+  const ratio = scalar(request, "aspectRatio");
+  if (mapping.capability.module.name === "@hypit/seedance") {
+    if (mapping.capability.name === "seedance-2.5" && present(request, "firstFrame") && ratio !== "adaptive") {
+      return `Seedance 2.5 frame mode on TokenDance takes only aspect-ratio adaptive, not ${String(ratio)}`;
+    }
+    return undefined;
+  }
+  if (mapping.capability.module.name === "@hypit/seedream") {
+    const quality = String(scalar(request, "quality"));
+    if (SEEDREAM_SIZES[quality]?.[String(ratio)] === undefined) {
+      return `TokenDance Seedream 5.0 lite has no size for quality ${quality} at ${String(ratio)}`;
+    }
+    return undefined;
+  }
+  if (mapping.capability.module.name === "@hypit/minimax-h3") {
+    const hasMedia = ["referenceImage", "referenceVideo", "referenceAudio", "firstFrame", "lastFrame"]
+      .some((port) => present(request, port));
+    if (!hasMedia && (ratio === undefined || ratio === "adaptive")) {
+      return "MiniMax H3 text-to-video on TokenDance requires an explicit aspect-ratio";
+    }
+  }
+  return undefined;
+}
+
+function contentItem(type: "image_url" | "video_url" | "audio_url", url: string, role: string) {
+  return { type, [type]: { url }, role };
+}
+
+/** Ark and MiniMax both take one `content` array of typed, role-tagged items. */
+function contentArray(input: Record<string, unknown>) {
+  const first = input.first_frame;
+  const last = input.last_frame;
+  return [
+    { type: "text", text: input.text },
+    ...(typeof first === "string" ? [contentItem("image_url", first, "first_frame")] : []),
+    ...(typeof last === "string" ? [contentItem("image_url", last, "last_frame")] : []),
+    ...strings(input.reference_image).map((url) => contentItem("image_url", url, "reference_image")),
+    ...strings(input.reference_video).map((url) => contentItem("video_url", url, "reference_video")),
+    ...strings(input.reference_audio).map((url) => contentItem("audio_url", url, "reference_audio")),
+  ];
+}
+
+function arkVideoBody(model: string, input: Record<string, unknown>): Record<string, unknown> {
+  return {
+    model,
+    content: contentArray(input),
+    resolution: input.resolution,
+    ratio: input.ratio,
+    duration: input.duration,
+    generate_audio: input.generate_audio,
+    ...(input.web_search === true ? { tools: [{ type: "web_search" }] } : {}),
+  };
+}
+
+function arkImageBody(model: string, input: Record<string, unknown>): Record<string, unknown> {
+  const images = strings(input.image);
+  return {
+    model,
+    prompt: input.prompt,
+    ...(images.length === 1 ? { image: images[0] } : images.length > 1 ? { image: images } : {}),
+    size: SEEDREAM_SIZES[String(input.quality)]![String(input.aspect_ratio)]!,
+    output_format: input.output_format,
+    response_format: "url",
+    watermark: false,
+  };
+}
+
+function minimaxVideoBody(model: string, input: Record<string, unknown>): Record<string, unknown> {
+  return {
+    model,
+    content: contentArray(input),
+    resolution: input.resolution,
+    duration: input.duration,
+    ...(typeof input.ratio === "string" ? { ratio: input.ratio } : {}),
+  };
+}
+
+function capabilityKey(capability: CapabilityRef): string {
+  return `${capability.module.name}@${capability.module.version}#${capability.name}`;
+}
+
+export const tokenDanceRoutes: readonly TokenDanceRoute[] = tokenDanceMappings.map((mapping) => {
+  const protocol: TokenDanceProtocol = mapping.capability.module.name === "@hypit/minimax-h3" ? "minimax-video"
+    : mapping.result === "image" ? "ark-image" : "ark-video";
+  const body = protocol === "ark-video" ? arkVideoBody : protocol === "ark-image" ? arkImageBody : minimaxVideoBody;
+  return {
+    ...mapping,
+    key: capabilityKey(mapping.capability),
+    returns: mapping.result === "image" ? generationTypes.imageSet : generationTypes.videoSet,
+    protocol,
+    mediaLimits: tokenDanceMediaLimits[protocol],
+    supports: (request) => {
+      const reason = rejection(mapping, request.constraints as unknown as GenerationRequest);
+      return reason === undefined ? { status: "supported" } : { status: "unsupported", reason };
+    },
+    prepare: (constraints) => {
+      const request = constraints as unknown as GenerationRequest;
+      const reason = rejection(mapping, request);
+      if (reason !== undefined) throw new Error(reason);
+      const model = selectWireModelForRequest(mapping, request);
+      return {
+        model,
+        protocol,
+        compile: async (resolve) => body(model, (await compileWireRequest(mapping, request, resolve)).input as Record<string, unknown>),
+      };
+    },
+    packageResult: (artifacts) => ({
+      kind: "inline",
+      value: canonicalize(mapping.result === "image"
+        ? sealGeneratedImageSet({ images: artifacts })
+        : sealGeneratedVideoSet({ videos: artifacts })),
+    }),
+  };
+});
+
+const byCapability = new Map(tokenDanceRoutes.map((route) => [route.key, route]));
+
+export function tokenDanceRouteForCapability(capability: CapabilityRef): TokenDanceRoute | undefined {
+  return byCapability.get(capabilityKey(capability));
+}

package/packages/video-cli/package.json

@@ -23,7 +23,11 @@
     "@hypit/package-loader-node": "workspace:*",
     "@hypit/project-context-node": "workspace:*",
     "@hypit/protocol": "workspace:*",
+    "@hypit/provider-hiapi": "workspace:*",
     "@hypit/provider-hypihub": "workspace:*",
+    "@hypit/provider-monid": "workspace:*",
+    "@hypit/provider-pollo": "workspace:*",
+    "@hypit/provider-tokendance": "workspace:*",
     "@hypit/runtime": "workspace:*",
     "@hypit/runtime-kit": "workspace:*",
     "@hypit/runtime-host-node": "workspace:*",