batchwork 1.2.1 → 1.4.0

package/dist/{chunk-sw8dg4sm.js → chunk-2ea62n95.js}

@@ -7,15 +7,29 @@ import {
   getAdapter,
   mapWithConcurrency,
   resolveBatchLimits
-} from "./chunk-jvrfwjwq.js";
+} from "./chunk-vy4w8mpb.js";
 import {
   __require
 } from "./chunk-v0bahtg2.js";
 
 // src/model.ts
 var CAPTURE_API_KEY = "batchwork-capture";
+var azureCaptureBaseUrl = (baseURL) => {
+  if (!baseURL) {
+    return;
+  }
+  const normalized = baseURL.replace(/\/+$/u, "");
+  try {
+    if (new URL(normalized).hostname.endsWith(".openai.azure.com")) {
+      const root = normalized.replace(/\/v1$/u, "");
+      return root.endsWith("/openai") ? root : `${root}/openai`;
+    }
+  } catch {}
+  return normalized;
+};
 var PACKAGE_BY_PROVIDER = {
   anthropic: { label: "Anthropic", specifier: "@ai-sdk/anthropic" },
+  azure: { label: "Azure OpenAI", specifier: "@ai-sdk/azure" },
   google: { label: "Google Gemini", specifier: "@ai-sdk/google" },
   groq: { label: "Groq", specifier: "@ai-sdk/groq" },
   mistral: { label: "Mistral", specifier: "@ai-sdk/mistral" },
@@ -25,6 +39,7 @@ var PACKAGE_BY_PROVIDER = {
 };
 var PROVIDER_BY_FAMILY = {
   anthropic: "anthropic",
+  azure: "azure",
   google: "google",
   groq: "groq",
   mistral: "mistral",
@@ -70,7 +85,7 @@ var resolveModel = (model) => {
   }
   const [family, suffix] = splitOnce(model.provider, ".");
   const provider = PROVIDER_BY_FAMILY[family];
-  if (provider === "openai") {
+  if (provider === "openai" || provider === "azure") {
     return { kind: openaiKind(suffix), modelId: model.modelId, provider };
   }
   if (provider) {
@@ -86,6 +101,9 @@ var importProvider = (provider) => {
     case "anthropic": {
       return import("@ai-sdk/anthropic");
     }
+    case "azure": {
+      return import("@ai-sdk/azure");
+    }
     case "google": {
       return import("@ai-sdk/google");
     }
@@ -139,6 +157,20 @@ var createCaptureModel = async (resolved, credentials, fetchImpl) => {
       }
       return provider.chat(resolved.modelId);
     }
+    case "azure": {
+      const { createAzure } = await loadProvider("azure");
+      const provider = createAzure({
+        ...settings,
+        baseURL: azureCaptureBaseUrl(credentials.baseURL)
+      });
+      if (resolved.kind === "responses") {
+        return provider.responses(resolved.modelId);
+      }
+      if (resolved.kind === "completion") {
+        return provider.completion(resolved.modelId);
+      }
+      return provider.chat(resolved.modelId);
+    }
     case "anthropic": {
       const { createAnthropic } = await loadProvider("anthropic");
       return createAnthropic(settings).messages(resolved.modelId);
@@ -199,6 +231,38 @@ var createCaptureEmbeddingModel = async (resolved, credentials, fetchImpl) => {
     }
   }
 };
+var IMAGE_EDIT_PROVIDERS = new Set(["openai", "xai"]);
+var unsupportedImageEditProvider = (provider) => new UnsupportedProviderError(provider, `batchwork: provider "${provider}" does not offer batch image editing. Image edits are supported for: openai, xai.`);
+var VIDEO_PROVIDERS = new Set(["xai"]);
+var unsupportedVideoProvider = (provider) => new UnsupportedProviderError(provider, `batchwork: provider "${provider}" does not offer batch video generation. Videos are supported for: xai.`);
+var createCaptureVideoModel = async (resolved, credentials, fetchImpl) => {
+  if (resolved.provider !== "xai") {
+    throw unsupportedVideoProvider(resolved.provider);
+  }
+  const { createXai } = await loadProvider("xai");
+  return createXai({
+    apiKey: credentials.apiKey ?? CAPTURE_API_KEY,
+    baseURL: credentials.baseURL,
+    fetch: fetchImpl,
+    headers: credentials.headers
+  }).video(resolved.modelId);
+};
+var TRANSLATION_PROVIDERS = new Set([
+  "groq",
+  "together"
+]);
+var unsupportedTranslationProvider = (provider) => new UnsupportedProviderError(provider, `batchwork: provider "${provider}" does not offer batch translation. Translations are supported for: groq, together.`);
+var MODERATION_PROVIDERS = new Set([
+  "mistral",
+  "openai"
+]);
+var unsupportedModerationProvider = (provider) => new UnsupportedProviderError(provider, `batchwork: provider "${provider}" does not offer batch moderation. Moderations are supported for: openai, mistral.`);
+var TRANSCRIPTION_PROVIDERS = new Set([
+  "groq",
+  "mistral",
+  "together"
+]);
+var unsupportedTranscriptionProvider = (provider) => new UnsupportedProviderError(provider, `batchwork: provider "${provider}" does not offer batch transcription. Transcriptions are supported for: groq, mistral, together.`);
 var IMAGE_PROVIDERS = new Set([
   "google",
   "openai",
@@ -232,7 +296,12 @@ var createCaptureImageModel = async (resolved, credentials, fetchImpl) => {
 };
 
 // src/body.ts
-import { embed, generateImage, generateText } from "ai";
+import {
+  embed,
+  experimental_generateVideo,
+  generateImage,
+  generateText
+} from "ai";
 var MAX_CAUSE_DEPTH = 10;
 
 class CaptureSignalError extends Error {
@@ -395,6 +464,99 @@ var buildEmbeddingBodies = async (resolved, requests, credentials, rawLimits) =>
     return built;
   });
 };
+var MODERATION_ENDPOINT = "/v1/moderations";
+var moderationInput = (request) => {
+  const imageUrls = request.imageUrls ?? [];
+  if (imageUrls.length === 0) {
+    return request.value;
+  }
+  return [
+    ...request.value === undefined ? [] : [{ text: request.value, type: "text" }],
+    ...imageUrls.map((url) => ({ image_url: { url }, type: "image_url" }))
+  ];
+};
+var moderationBody = (resolved, request, customId) => {
+  if (request.value === undefined && !request.imageUrls?.length) {
+    throw new BatchworkError(`batchwork: moderation request "${customId}" needs \`value\` or \`imageUrls\`.`);
+  }
+  const options = request.providerOptions?.[resolved.provider];
+  if (resolved.provider === "openai") {
+    return {
+      input: moderationInput(request),
+      model: resolved.modelId,
+      ...options
+    };
+  }
+  if (resolved.provider === "mistral") {
+    if (request.imageUrls?.length) {
+      throw new BatchworkError(`batchwork: moderation request "${customId}" has \`imageUrls\`, but Mistral moderation is text-only.`);
+    }
+    return { input: request.value, model: resolved.modelId, ...options };
+  }
+  throw unsupportedModerationProvider(resolved.provider);
+};
+var buildModerationBodies = (resolved, requests, rawLimits) => {
+  const limits = resolveBatchLimits(rawLimits);
+  if (requests.length > limits.maxRequests) {
+    throw new BatchworkError(`batchwork: requests length ${requests.length} exceeds the ${limits.maxRequests} request limit.`);
+  }
+  return assignCustomIds(requests).map((item) => {
+    const body = moderationBody(resolved, item.request, item.customId);
+    assertByteLength(`request "${item.customId}"`, JSON.stringify(body), limits.maxRequestBytes);
+    return { body, customId: item.customId, endpoint: MODERATION_ENDPOINT };
+  });
+};
+var TRANSCRIPTION_ENDPOINT = "/v1/audio/transcriptions";
+var transcriptionBody = (resolved, request) => {
+  const options = request.providerOptions?.[resolved.provider];
+  if (resolved.provider === "groq") {
+    return {
+      model: resolved.modelId,
+      url: request.audioUrl,
+      ...request.language ? { language: request.language } : {},
+      ...request.timestampGranularities ? {
+        response_format: "verbose_json",
+        timestamp_granularities: request.timestampGranularities
+      } : {},
+      ...options
+    };
+  }
+  if (resolved.provider === "mistral") {
+    return {
+      file_url: request.audioUrl,
+      model: resolved.modelId,
+      ...request.language ? { language: request.language } : {},
+      ...request.timestampGranularities ? { timestamp_granularities: request.timestampGranularities } : {},
+      ...options
+    };
+  }
+  if (resolved.provider === "together") {
+    return {
+      file: request.audioUrl,
+      model: resolved.modelId,
+      ...request.language ? { language: request.language } : {},
+      ...request.timestampGranularities ? {
+        response_format: "verbose_json",
+        timestamp_granularities: request.timestampGranularities
+      } : {},
+      ...options
+    };
+  }
+  throw unsupportedTranscriptionProvider(resolved.provider);
+};
+var buildAudioBodies = (resolved, requests, defaults, endpoint, rawLimits) => {
+  const limits = resolveBatchLimits(rawLimits);
+  if (requests.length > limits.maxRequests) {
+    throw new BatchworkError(`batchwork: requests length ${requests.length} exceeds the ${limits.maxRequests} request limit.`);
+  }
+  return assignCustomIds(requests).map((item) => {
+    const body = transcriptionBody(resolved, mergeDefaults(item.request, defaults));
+    assertByteLength(`request "${item.customId}"`, JSON.stringify(body), limits.maxRequestBytes);
+    return { body, customId: item.customId, endpoint };
+  });
+};
+var buildTranscriptionBodies = (resolved, requests, defaults, rawLimits) => buildAudioBodies(resolved, requests, defaults, TRANSCRIPTION_ENDPOINT, rawLimits);
+var buildTranslationBodies = (resolved, requests, defaults, rawLimits) => buildAudioBodies(resolved, requests, defaults, "/v1/audio/translations", rawLimits);
 var buildImageBodies = async (resolved, requests, defaults, credentials, rawLimits) => {
   const limits = resolveBatchLimits(rawLimits);
   if (requests.length > limits.maxRequests) {
@@ -408,6 +570,87 @@ var buildImageBodies = async (resolved, requests, defaults, credentials, rawLimi
     return built;
   });
 };
+var IMAGE_EDIT_ENDPOINT = "/v1/images/edits";
+var openaiImageRef = (ref) => ("fileId" in ref) ? { file_id: ref.fileId } : { image_url: ref.imageUrl };
+var imageEditBody = (resolved, request, customId) => {
+  if (request.images.length === 0) {
+    throw new BatchworkError(`batchwork: image-edit request "${customId}" needs at least one entry in \`images\`.`);
+  }
+  const options = request.providerOptions?.[resolved.provider];
+  if (resolved.provider === "openai") {
+    return {
+      images: request.images.map(openaiImageRef),
+      model: resolved.modelId,
+      prompt: request.prompt,
+      ...request.mask ? { mask: openaiImageRef(request.mask) } : {},
+      ...request.n === undefined ? {} : { n: request.n },
+      ...request.size ? { size: request.size } : {},
+      ...options
+    };
+  }
+  if (resolved.provider === "xai") {
+    if (request.mask) {
+      throw new BatchworkError(`batchwork: image-edit request "${customId}" has a \`mask\`, but xAI image edits do not support masks.`);
+    }
+    if (request.size) {
+      throw new BatchworkError(`batchwork: image-edit request "${customId}" has \`size\`, but xAI image edits take \`providerOptions.xai.aspect_ratio\` instead.`);
+    }
+    const urls = request.images.map((ref) => {
+      if ("fileId" in ref) {
+        throw new BatchworkError(`batchwork: image-edit request "${customId}" uses a \`fileId\` reference, but xAI image edits accept image URLs only.`);
+      }
+      return ref.imageUrl;
+    });
+    return {
+      model: resolved.modelId,
+      prompt: request.prompt,
+      ...urls.length === 1 ? { image: { url: urls[0] } } : { images: urls.map((url) => ({ url })) },
+      ...request.n === undefined ? {} : { n: request.n },
+      ...options
+    };
+  }
+  throw unsupportedImageEditProvider(resolved.provider);
+};
+var buildImageEditBodies = (resolved, requests, defaults, rawLimits) => {
+  const limits = resolveBatchLimits(rawLimits);
+  if (requests.length > limits.maxRequests) {
+    throw new BatchworkError(`batchwork: requests length ${requests.length} exceeds the ${limits.maxRequests} request limit.`);
+  }
+  return assignCustomIds(requests).map((item) => {
+    const body = imageEditBody(resolved, mergeDefaults(item.request, defaults), item.customId);
+    assertByteLength(`request "${item.customId}"`, JSON.stringify(body), limits.maxRequestBytes);
+    return { body, customId: item.customId, endpoint: IMAGE_EDIT_ENDPOINT };
+  });
+};
+var captureVideoOne = async (model, request, customId, maxRequestBytes) => {
+  try {
+    await experimental_generateVideo({
+      aspectRatio: request.aspectRatio,
+      duration: request.duration,
+      maxRetries: 0,
+      model,
+      prompt: request.prompt,
+      providerOptions: request.providerOptions,
+      resolution: request.resolution
+    });
+  } catch (error) {
+    return bodyFromCapture(error, customId, maxRequestBytes);
+  }
+  throw new BatchworkError("batchwork: the request was not intercepted while building the video body.");
+};
+var buildVideoBodies = async (resolved, requests, defaults, credentials, rawLimits) => {
+  const limits = resolveBatchLimits(rawLimits);
+  if (requests.length > limits.maxRequests) {
+    throw new BatchworkError(`batchwork: requests length ${requests.length} exceeds the ${limits.maxRequests} request limit.`);
+  }
+  const model = await createCaptureVideoModel(resolved, credentials, captureFetch);
+  const items = assignCustomIds(requests);
+  return await mapWithConcurrency(items, limits.captureConcurrency, async (item) => {
+    const built = await captureVideoOne(model, mergeDefaults(item.request, defaults), item.customId, limits.maxRequestBytes);
+    assertByteLength(`request "${item.customId}"`, JSON.stringify(built.body), limits.maxRequestBytes);
+    return built;
+  });
+};
 
 // src/batch.ts
 var EMPTY_REQUESTS_MESSAGE = "batchwork: `requests` must not be empty.";
@@ -488,10 +731,127 @@ var submitImages = async (options) => {
   });
   return new BatchJob(adapter, credentials, snapshot);
 };
+var submitImageEdits = async (options) => {
+  if (options.requests.length === 0) {
+    throw new BatchworkError(EMPTY_REQUESTS_MESSAGE);
+  }
+  const resolved = resolveModel(options.model);
+  if (!IMAGE_EDIT_PROVIDERS.has(resolved.provider)) {
+    throw unsupportedImageEditProvider(resolved.provider);
+  }
+  const credentials = pickCredentials(options);
+  const limits = resolveBatchLimits(options.limits);
+  const adapter = getAdapter(resolved.provider);
+  const built = buildImageEditBodies(resolved, options.requests, options.defaults, limits);
+  const snapshot = await adapter.submit({
+    built,
+    credentials,
+    endpoint: built[0]?.endpoint ?? "",
+    limits,
+    metadata: options.metadata,
+    modelId: resolved.modelId
+  });
+  return new BatchJob(adapter, credentials, snapshot);
+};
+var submitModerations = async (options) => {
+  if (options.requests.length === 0) {
+    throw new BatchworkError(EMPTY_REQUESTS_MESSAGE);
+  }
+  const resolved = resolveModel(options.model);
+  if (!MODERATION_PROVIDERS.has(resolved.provider)) {
+    throw unsupportedModerationProvider(resolved.provider);
+  }
+  const credentials = pickCredentials(options);
+  const limits = resolveBatchLimits(options.limits);
+  const adapter = getAdapter(resolved.provider);
+  const built = buildModerationBodies(resolved, options.requests, limits);
+  const snapshot = await adapter.submit({
+    built,
+    credentials,
+    endpoint: built[0]?.endpoint ?? "",
+    limits,
+    metadata: options.metadata,
+    modelId: resolved.modelId
+  });
+  return new BatchJob(adapter, credentials, snapshot);
+};
+var submitTranscriptions = async (options) => {
+  if (options.requests.length === 0) {
+    throw new BatchworkError(EMPTY_REQUESTS_MESSAGE);
+  }
+  const resolved = resolveModel(options.model);
+  if (!TRANSCRIPTION_PROVIDERS.has(resolved.provider)) {
+    throw unsupportedTranscriptionProvider(resolved.provider);
+  }
+  const credentials = pickCredentials(options);
+  const limits = resolveBatchLimits(options.limits);
+  const adapter = getAdapter(resolved.provider);
+  const built = buildTranscriptionBodies(resolved, options.requests, options.defaults, limits);
+  const snapshot = await adapter.submit({
+    built,
+    credentials,
+    endpoint: built[0]?.endpoint ?? "",
+    limits,
+    metadata: options.metadata,
+    modelId: resolved.modelId
+  });
+  return new BatchJob(adapter, credentials, snapshot);
+};
+var submitVideos = async (options) => {
+  if (options.requests.length === 0) {
+    throw new BatchworkError(EMPTY_REQUESTS_MESSAGE);
+  }
+  const resolved = resolveModel(options.model);
+  if (!VIDEO_PROVIDERS.has(resolved.provider)) {
+    throw unsupportedVideoProvider(resolved.provider);
+  }
+  const credentials = pickCredentials(options);
+  const limits = resolveBatchLimits(options.limits);
+  const adapter = getAdapter(resolved.provider);
+  const built = await buildVideoBodies(resolved, options.requests, options.defaults, credentials, limits);
+  const snapshot = await adapter.submit({
+    built,
+    credentials,
+    endpoint: built[0]?.endpoint ?? "",
+    limits,
+    metadata: options.metadata,
+    modelId: resolved.modelId
+  });
+  return new BatchJob(adapter, credentials, snapshot);
+};
+var submitTranslations = async (options) => {
+  if (options.requests.length === 0) {
+    throw new BatchworkError(EMPTY_REQUESTS_MESSAGE);
+  }
+  const resolved = resolveModel(options.model);
+  if (!TRANSLATION_PROVIDERS.has(resolved.provider)) {
+    throw unsupportedTranslationProvider(resolved.provider);
+  }
+  const credentials = pickCredentials(options);
+  const limits = resolveBatchLimits(options.limits);
+  const adapter = getAdapter(resolved.provider);
+  const built = buildTranslationBodies(resolved, options.requests, options.defaults, limits);
+  const snapshot = await adapter.submit({
+    built,
+    credentials,
+    endpoint: built[0]?.endpoint ?? "",
+    limits,
+    metadata: options.metadata,
+    modelId: resolved.modelId
+  });
+  return new BatchJob(adapter, credentials, snapshot);
+};
 var batch = Object.assign(submitText, {
   embeddings: submitEmbeddings,
-  images: submitImages,
-  text: submitText
+  images: Object.assign(submitImages, {
+    create: submitImages,
+    edit: submitImageEdits
+  }),
+  moderations: submitModerations,
+  text: submitText,
+  transcriptions: submitTranscriptions,
+  translations: submitTranslations,
+  videos: submitVideos
 });
 var batchEmbeddings = submitEmbeddings;
 var batchImages = submitImages;
@@ -512,5 +872,5 @@ var cancelBatch = async (ref) => {
 
 export { resolveModel, batch, batchEmbeddings, batchImages, getBatch, getBatchResults, cancelBatch };
 
-//# debugId=EA06C285349F9BD164756E2164756E21
-//# sourceMappingURL=chunk-sw8dg4sm.js.map
+//# debugId=1B451ACF41BE2A3C64756E2164756E21
+//# sourceMappingURL=chunk-2ea62n95.js.map