batchwork 1.2.1 → 1.4.0

package/dist/{chunk-jvrfwjwq.js → chunk-vy4w8mpb.js}

@@ -9,7 +9,7 @@ class BatchworkError extends Error {
 class UnsupportedProviderError extends BatchworkError {
   provider;
   constructor(provider, detail) {
-    super(detail ?? `batchwork: provider "${provider}" is not supported yet. Supported providers: openai, anthropic, google, groq, mistral, together, xai.`);
+    super(detail ?? `batchwork: provider "${provider}" is not supported yet. Supported providers: openai, azure, anthropic, google, groq, mistral, together, xai.`);
     this.name = "UnsupportedProviderError";
     this.provider = provider;
   }
@@ -539,6 +539,366 @@ var anthropicAdapter = {
   submit
 };
 
+// src/providers/shared.ts
+var HTTP_OK_MIN = 200;
+var HTTP_OK_MAX = 300;
+var resolveApiKey = (credentials, envVar, label) => {
+  const key = credentials.apiKey ?? process.env[envVar];
+  if (!key) {
+    throw new BatchworkError(`batchwork: missing ${label} API key. Set ${envVar} or pass \`apiKey\`.`);
+  }
+  return key;
+};
+var textFromBody = (body) => {
+  const obj = asRecord(body);
+  const choices = asArray(obj.choices);
+  if (choices.length > 0) {
+    const content = asString(asRecord(asRecord(choices[0]).message).content);
+    if (content) {
+      return content;
+    }
+  }
+  const directText = asString(obj.output_text) ?? asString(obj.text);
+  if (directText !== undefined) {
+    return directText;
+  }
+  const parts = [];
+  for (const output of asArray(obj.output)) {
+    const item = asRecord(output);
+    if (item.type !== "message") {
+      continue;
+    }
+    for (const content of asArray(item.content)) {
+      const part = asRecord(content);
+      if (part.type !== "output_text") {
+        continue;
+      }
+      const text = asString(part.text);
+      if (text !== undefined) {
+        parts.push(text);
+      }
+    }
+  }
+  if (parts.length > 0) {
+    return parts.join("");
+  }
+};
+var segmentsFromBody = (body) => {
+  const obj = asRecord(body);
+  if (asString(obj.text) === undefined) {
+    return;
+  }
+  const segments = [];
+  for (const item of asArray(obj.segments)) {
+    const segment = asRecord(item);
+    const text = asString(segment.text);
+    if (text !== undefined) {
+      segments.push({
+        endSecond: asNumber(segment.end),
+        startSecond: asNumber(segment.start),
+        text
+      });
+    }
+  }
+  return segments.length > 0 ? segments : undefined;
+};
+var embeddingFromBody = (body) => {
+  const data = asArray(asRecord(body).data);
+  if (data.length === 0) {
+    return;
+  }
+  return asNumberArray(asRecord(data[0]).embedding);
+};
+var imagesFromBody = (body) => {
+  const obj = asRecord(body);
+  const mediaType = `image/${asString(obj.output_format) ?? "png"}`;
+  const images = [];
+  for (const item of asArray(obj.data)) {
+    const b64 = asString(asRecord(item).b64_json);
+    if (b64) {
+      images.push({ data: b64, mediaType });
+    }
+  }
+  return images.length > 0 ? images : undefined;
+};
+var moderationFromBody = (body) => {
+  const results2 = asArray(asRecord(body).results);
+  if (results2.length === 0) {
+    return;
+  }
+  const first = asRecord(results2[0]);
+  const categories = {};
+  for (const [key, value] of Object.entries(asRecord(first.categories))) {
+    if (typeof value === "boolean") {
+      categories[key] = value;
+    }
+  }
+  if (Object.keys(categories).length === 0) {
+    return;
+  }
+  const categoryScores = {};
+  for (const [key, value] of Object.entries(asRecord(first.category_scores))) {
+    const score = asNumber(value);
+    if (score !== undefined) {
+      categoryScores[key] = score;
+    }
+  }
+  const flagged = typeof first.flagged === "boolean" ? first.flagged : Object.values(categories).some(Boolean);
+  return { categories, categoryScores, flagged };
+};
+var usageFromBody = (body) => {
+  const usage = asRecord(asRecord(body).usage);
+  const inputTokens = asNumber(usage.prompt_tokens) ?? asNumber(usage.input_tokens);
+  const outputTokens = asNumber(usage.completion_tokens) ?? asNumber(usage.output_tokens);
+  const totalTokens = asNumber(usage.total_tokens);
+  if (inputTokens === undefined && outputTokens === undefined && totalTokens === undefined) {
+    return;
+  }
+  return {
+    inputTokens,
+    outputTokens,
+    totalTokens: totalTokens ?? (inputTokens ?? 0) + (outputTokens ?? 0)
+  };
+};
+var errorFromValue = (value, fallback) => {
+  const obj = asRecord(value);
+  const nested = asRecord(obj.error);
+  const source = nested.message ? nested : obj;
+  return {
+    code: asNumber(source.code) ?? asString(source.code),
+    message: asString(source.message) ?? fallback,
+    type: asString(source.type)
+  };
+};
+var normalizeOpenAIResult = (line) => {
+  const obj = asRecord(line);
+  const customId = asString(obj.custom_id) ?? "";
+  if (obj.error) {
+    return {
+      customId,
+      error: errorFromValue(obj.error, "Request errored."),
+      response: obj.error,
+      status: "errored"
+    };
+  }
+  const response = asRecord(obj.response);
+  const statusCode = asNumber(response.status_code) ?? 0;
+  if (statusCode >= HTTP_OK_MIN && statusCode < HTTP_OK_MAX) {
+    return {
+      customId,
+      embedding: embeddingFromBody(response.body),
+      images: imagesFromBody(response.body),
+      moderation: moderationFromBody(response.body),
+      response: response.body,
+      segments: segmentsFromBody(response.body),
+      status: "succeeded",
+      text: textFromBody(response.body),
+      usage: usageFromBody(response.body)
+    };
+  }
+  return {
+    customId,
+    error: errorFromValue(response.body, `Request failed with status ${statusCode}.`),
+    response: response.body,
+    status: "errored"
+  };
+};
+var uploadInputFile = async (jsonl, baseUrl2, headers2, options = {}) => {
+  const form = new FormData;
+  const purpose = options.purpose === undefined ? "batch" : options.purpose;
+  if (purpose !== null) {
+    form.append("purpose", purpose);
+  }
+  form.append("file", new Blob([jsonl], { type: "application/jsonl" }), "batchwork.jsonl");
+  const raw = await requestJson(`${baseUrl2}/files`, {
+    body: form,
+    headers: headers2,
+    method: "POST",
+    redirect: "manual"
+  });
+  return raw.id;
+};
+async function* streamResultFile(fileId, baseUrl2, headers2) {
+  const stream = await requestStream(`${baseUrl2}/files/${fileId}/content`, {
+    headers: headers2,
+    redirect: "manual"
+  });
+  for await (const line of streamJsonl(stream)) {
+    yield normalizeOpenAIResult(line);
+  }
+}
+
+// src/providers/openai-compatible.ts
+var DEFAULT_COMPLETION_WINDOW = "24h";
+var mapStatus2 = (status) => {
+  const normalized = status?.toLowerCase();
+  switch (normalized) {
+    case "validating":
+    case "in_progress":
+    case "finalizing":
+    case "completed":
+    case "failed":
+    case "expired":
+    case "cancelling":
+    case "cancelled": {
+      return normalized;
+    }
+    default: {
+      return "in_progress";
+    }
+  }
+};
+var normalizeSnapshot2 = (raw, provider) => {
+  const outer = asRecord(raw);
+  const obj = asRecord(outer.job);
+  const source = Object.keys(obj).length > 0 ? obj : outer;
+  const counts = asRecord(source.request_counts);
+  return {
+    completedAt: toDate(source.completed_at),
+    createdAt: toDate(source.created_at),
+    expiresAt: toDate(source.expires_at),
+    id: asString(source.id) ?? "",
+    provider,
+    raw: source,
+    requestCounts: {
+      completed: asNumber(counts.completed) ?? 0,
+      failed: asNumber(counts.failed) ?? 0,
+      total: asNumber(counts.total) ?? 0
+    },
+    status: mapStatus2(asString(source.status))
+  };
+};
+var createOpenAICompatibleAdapter = (config) => {
+  const completionWindow = config.completionWindow ?? DEFAULT_COMPLETION_WINDOW;
+  const lineFormat = config.lineFormat ?? "method-url";
+  const baseUrl2 = (credentials) => config.resolveBaseUrl?.(credentials) ?? credentials.baseURL ?? config.baseUrl ?? (() => {
+    throw new BatchworkError(`batchwork: missing ${config.apiKeyLabel} base URL. Pass \`baseURL\`.`);
+  })();
+  const authHeaders = (credentials) => ({
+    ...config.authHeaders ? config.authHeaders(credentials) : {
+      Authorization: `Bearer ${resolveApiKey(credentials, config.apiKeyEnv, config.apiKeyLabel)}`
+    },
+    ...credentials.headers
+  });
+  const submit2 = async (input) => {
+    const limits = resolveBatchLimits(input.limits);
+    const endpoint = config.normalizeEndpoint ? config.normalizeEndpoint(input.endpoint) : input.endpoint;
+    const batchEndpoint = config.batchEndpoint?.(endpoint) ?? endpoint;
+    const jsonl = encodeJsonl(input.built.map((item) => {
+      const body = omit(item.body, "stream");
+      if (lineFormat === "body-only") {
+        return {
+          body,
+          custom_id: item.customId,
+          ...config.lineExtras?.(endpoint)
+        };
+      }
+      return {
+        body,
+        custom_id: item.customId,
+        method: "POST",
+        url: endpoint
+      };
+    }), { label: "batch upload JSONL", maxBytes: limits.maxUploadBytes });
+    const headers2 = authHeaders(input.credentials);
+    const url = baseUrl2(input.credentials);
+    const purpose = config.filePurpose ?? "batch";
+    const inputFileId = await (config.uploadFile ? config.uploadFile({ baseUrl: url, headers: headers2, jsonl, purpose }) : uploadInputFile(jsonl, url, headers2, { purpose }));
+    const raw = await requestJson(`${url}/batches`, {
+      body: JSON.stringify({
+        completion_window: completionWindow,
+        endpoint: batchEndpoint,
+        input_file_id: inputFileId,
+        metadata: input.metadata
+      }),
+      headers: { ...headers2, "content-type": "application/json" },
+      method: "POST"
+    });
+    return normalizeSnapshot2(raw, config.id);
+  };
+  const retrieve2 = async (id, credentials) => {
+    const batchId = assertSimpleProviderId(`${config.id} batch id`, id);
+    const raw = await requestJson(`${baseUrl2(credentials)}/batches/${batchId}`, {
+      headers: authHeaders(credentials)
+    });
+    return normalizeSnapshot2(raw, config.id);
+  };
+  async function* results2(id, credentials) {
+    const snapshot = await retrieve2(id, credentials);
+    const raw = asRecord(snapshot.raw);
+    const outputFileId = asString(raw.output_file_id);
+    const errorFileId = asString(raw.error_file_id);
+    if (!(outputFileId || errorFileId)) {
+      throw new BatchworkError(`batchwork: results are not ready for batch "${id}" (status: ${snapshot.status}).`);
+    }
+    const headers2 = authHeaders(credentials);
+    if (outputFileId) {
+      yield* streamResultFile(assertSimpleProviderId(`${config.id} output file id`, outputFileId), baseUrl2(credentials), headers2);
+    }
+    if (errorFileId) {
+      yield* streamResultFile(assertSimpleProviderId(`${config.id} error file id`, errorFileId), baseUrl2(credentials), headers2);
+    }
+  }
+  const cancel2 = async (id, credentials) => {
+    const batchId = assertSimpleProviderId(`${config.id} batch id`, id);
+    await requestJson(`${baseUrl2(credentials)}/batches/${batchId}/cancel`, {
+      headers: authHeaders(credentials),
+      method: "POST"
+    });
+  };
+  return { cancel: cancel2, id: config.id, results: results2, retrieve: retrieve2, submit: submit2 };
+};
+
+// src/providers/azure.ts
+var withoutTrailingSlash = (value) => value.replace(/\/+$/u, "");
+var isAzureOpenAIUrl = (value) => {
+  try {
+    return new URL(value).hostname.endsWith(".openai.azure.com");
+  } catch {
+    return false;
+  }
+};
+var baseUrl2 = (credentials) => {
+  const configured = credentials.baseURL ?? (process.env.AZURE_RESOURCE_NAME ? `https://${process.env.AZURE_RESOURCE_NAME}.openai.azure.com/openai` : undefined);
+  if (!configured) {
+    throw new BatchworkError("batchwork: missing Azure OpenAI resource. Set AZURE_RESOURCE_NAME or pass `baseURL`.");
+  }
+  const normalized = withoutTrailingSlash(configured);
+  if (!isAzureOpenAIUrl(normalized)) {
+    return normalized;
+  }
+  if (normalized.endsWith("/v1")) {
+    return normalized;
+  }
+  return normalized.endsWith("/openai") ? `${normalized}/v1` : `${normalized}/openai/v1`;
+};
+var hasCallerAuth = (headers2) => Object.keys(headers2 ?? {}).some((name) => {
+  const normalized = name.toLowerCase();
+  return normalized === "api-key" || normalized === "authorization";
+});
+var authHeaders = (credentials) => {
+  if (credentials.apiKey) {
+    return { "api-key": credentials.apiKey };
+  }
+  if (hasCallerAuth(credentials.headers)) {
+    return {};
+  }
+  const apiKey2 = process.env.AZURE_API_KEY ?? process.env.AZURE_OPENAI_API_KEY;
+  if (apiKey2) {
+    return { "api-key": apiKey2 };
+  }
+  throw new BatchworkError("batchwork: missing Azure OpenAI API key. Set AZURE_API_KEY or pass `apiKey`.");
+};
+var azureAdapter = createOpenAICompatibleAdapter({
+  apiKeyEnv: "AZURE_API_KEY",
+  apiKeyLabel: "Azure OpenAI",
+  authHeaders,
+  batchEndpoint: () => "/v1/chat/completions",
+  id: "azure",
+  normalizeEndpoint: (endpoint) => endpoint.replace(/^\/openai/u, ""),
+  resolveBaseUrl: baseUrl2
+});
+
 // src/providers/google.ts
 var GOOGLE_BASE = "https://generativelanguage.googleapis.com/v1beta";
 var OPERATION_ID_LABEL = "Google operation id";
@@ -550,7 +910,7 @@ var apiKey2 = (credentials) => {
   }
   return key;
 };
-var baseUrl2 = (credentials) => credentials.baseURL ?? GOOGLE_BASE;
+var baseUrl3 = (credentials) => credentials.baseURL ?? GOOGLE_BASE;
 var headers2 = (credentials) => ({
   "content-type": "application/json",
   "x-goog-api-key": apiKey2(credentials),
@@ -596,7 +956,7 @@ var inlinedResponses = (raw) => {
     ...asArray(nestedDestInline.inlined_responses)
   ];
 };
-var normalizeSnapshot2 = (raw) => {
+var normalizeSnapshot3 = (raw) => {
   const obj = asRecord(raw);
   const items = inlinedResponses(raw);
   const failed = items.filter((item) => asRecord(item).error).length;
@@ -711,19 +1071,19 @@ var submit2 = async (input) => {
     prefix: '{"batch":{"display_name":"batchwork","input_config":{"requests":{"requests":[',
     suffix: "]}}}}"
   });
-  const raw = await requestJson(`${baseUrl2(input.credentials)}/models/${input.modelId}:${method}`, {
+  const raw = await requestJson(`${baseUrl3(input.credentials)}/models/${input.modelId}:${method}`, {
     body,
     headers: headers2(input.credentials),
     method: "POST"
   });
-  return normalizeSnapshot2(raw);
+  return normalizeSnapshot3(raw);
 };
 var retrieve2 = async (id, credentials) => {
   const operationId = assertPrefixedProviderId(OPERATION_ID_LABEL, id, GOOGLE_BATCH_PREFIX);
-  const raw = await requestJson(`${baseUrl2(credentials)}/${operationId}`, {
+  const raw = await requestJson(`${baseUrl3(credentials)}/${operationId}`, {
     headers: headers2(credentials)
   });
-  return normalizeSnapshot2(raw);
+  return normalizeSnapshot3(raw);
 };
 var fileNameFrom = (value) => asString(asRecord(value).name) ?? asString(value);
 async function* results2(id, credentials) {
@@ -745,7 +1105,7 @@ async function* results2(id, credentials) {
 }
 var cancel2 = async (id, credentials) => {
   const operationId = assertPrefixedProviderId(OPERATION_ID_LABEL, id, GOOGLE_BATCH_PREFIX);
-  await requestJson(`${baseUrl2(credentials)}/${operationId}:cancel`, {
+  await requestJson(`${baseUrl3(credentials)}/${operationId}:cancel`, {
     headers: headers2(credentials),
     method: "POST"
   });
@@ -758,238 +1118,6 @@ var googleAdapter = {
   submit: submit2
 };
 
-// src/providers/shared.ts
-var HTTP_OK_MIN = 200;
-var HTTP_OK_MAX = 300;
-var resolveApiKey = (credentials, envVar, label) => {
-  const key = credentials.apiKey ?? process.env[envVar];
-  if (!key) {
-    throw new BatchworkError(`batchwork: missing ${label} API key. Set ${envVar} or pass \`apiKey\`.`);
-  }
-  return key;
-};
-var textFromBody = (body) => {
-  const obj = asRecord(body);
-  const choices = asArray(obj.choices);
-  if (choices.length > 0) {
-    const content = asString(asRecord(asRecord(choices[0]).message).content);
-    if (content) {
-      return content;
-    }
-  }
-  return asString(obj.output_text);
-};
-var embeddingFromBody = (body) => {
-  const data = asArray(asRecord(body).data);
-  if (data.length === 0) {
-    return;
-  }
-  return asNumberArray(asRecord(data[0]).embedding);
-};
-var imagesFromBody = (body) => {
-  const obj = asRecord(body);
-  const mediaType = `image/${asString(obj.output_format) ?? "png"}`;
-  const images = [];
-  for (const item of asArray(obj.data)) {
-    const b64 = asString(asRecord(item).b64_json);
-    if (b64) {
-      images.push({ data: b64, mediaType });
-    }
-  }
-  return images.length > 0 ? images : undefined;
-};
-var usageFromBody = (body) => {
-  const usage = asRecord(asRecord(body).usage);
-  const inputTokens = asNumber(usage.prompt_tokens) ?? asNumber(usage.input_tokens);
-  const outputTokens = asNumber(usage.completion_tokens) ?? asNumber(usage.output_tokens);
-  const totalTokens = asNumber(usage.total_tokens);
-  if (inputTokens === undefined && outputTokens === undefined && totalTokens === undefined) {
-    return;
-  }
-  return {
-    inputTokens,
-    outputTokens,
-    totalTokens: totalTokens ?? (inputTokens ?? 0) + (outputTokens ?? 0)
-  };
-};
-var errorFromValue = (value, fallback) => {
-  const obj = asRecord(value);
-  const nested = asRecord(obj.error);
-  const source = nested.message ? nested : obj;
-  return {
-    code: asNumber(source.code) ?? asString(source.code),
-    message: asString(source.message) ?? fallback,
-    type: asString(source.type)
-  };
-};
-var normalizeOpenAIResult = (line) => {
-  const obj = asRecord(line);
-  const customId = asString(obj.custom_id) ?? "";
-  if (obj.error) {
-    return {
-      customId,
-      error: errorFromValue(obj.error, "Request errored."),
-      response: obj.error,
-      status: "errored"
-    };
-  }
-  const response = asRecord(obj.response);
-  const statusCode = asNumber(response.status_code) ?? 0;
-  if (statusCode >= HTTP_OK_MIN && statusCode < HTTP_OK_MAX) {
-    return {
-      customId,
-      embedding: embeddingFromBody(response.body),
-      images: imagesFromBody(response.body),
-      response: response.body,
-      status: "succeeded",
-      text: textFromBody(response.body),
-      usage: usageFromBody(response.body)
-    };
-  }
-  return {
-    customId,
-    error: errorFromValue(response.body, `Request failed with status ${statusCode}.`),
-    response: response.body,
-    status: "errored"
-  };
-};
-var uploadInputFile = async (jsonl, baseUrl3, headers3, options = {}) => {
-  const form = new FormData;
-  const purpose = options.purpose === undefined ? "batch" : options.purpose;
-  if (purpose !== null) {
-    form.append("purpose", purpose);
-  }
-  form.append("file", new Blob([jsonl], { type: "application/jsonl" }), "batchwork.jsonl");
-  const raw = await requestJson(`${baseUrl3}/files`, {
-    body: form,
-    headers: headers3,
-    method: "POST",
-    redirect: "manual"
-  });
-  return raw.id;
-};
-async function* streamResultFile(fileId, baseUrl3, headers3) {
-  const stream = await requestStream(`${baseUrl3}/files/${fileId}/content`, {
-    headers: headers3,
-    redirect: "manual"
-  });
-  for await (const line of streamJsonl(stream)) {
-    yield normalizeOpenAIResult(line);
-  }
-}
-
-// src/providers/openai-compatible.ts
-var DEFAULT_COMPLETION_WINDOW = "24h";
-var mapStatus2 = (status) => {
-  const normalized = status?.toLowerCase();
-  switch (normalized) {
-    case "validating":
-    case "in_progress":
-    case "finalizing":
-    case "completed":
-    case "failed":
-    case "expired":
-    case "cancelling":
-    case "cancelled": {
-      return normalized;
-    }
-    default: {
-      return "in_progress";
-    }
-  }
-};
-var normalizeSnapshot3 = (raw, provider) => {
-  const outer = asRecord(raw);
-  const obj = asRecord(outer.job);
-  const source = Object.keys(obj).length > 0 ? obj : outer;
-  const counts = asRecord(source.request_counts);
-  return {
-    completedAt: toDate(source.completed_at),
-    createdAt: toDate(source.created_at),
-    expiresAt: toDate(source.expires_at),
-    id: asString(source.id) ?? "",
-    provider,
-    raw: source,
-    requestCounts: {
-      completed: asNumber(counts.completed) ?? 0,
-      failed: asNumber(counts.failed) ?? 0,
-      total: asNumber(counts.total) ?? 0
-    },
-    status: mapStatus2(asString(source.status))
-  };
-};
-var createOpenAICompatibleAdapter = (config) => {
-  const completionWindow = config.completionWindow ?? DEFAULT_COMPLETION_WINDOW;
-  const lineFormat = config.lineFormat ?? "method-url";
-  const baseUrl3 = (credentials) => credentials.baseURL ?? config.baseUrl;
-  const authHeaders = (credentials) => ({
-    Authorization: `Bearer ${resolveApiKey(credentials, config.apiKeyEnv, config.apiKeyLabel)}`,
-    ...credentials.headers
-  });
-  const submit3 = async (input) => {
-    const limits = resolveBatchLimits(input.limits);
-    const endpoint = config.normalizeEndpoint ? config.normalizeEndpoint(input.endpoint) : input.endpoint;
-    const jsonl = encodeJsonl(input.built.map((item) => {
-      const body = omit(item.body, "stream");
-      if (lineFormat === "body-only") {
-        return { body, custom_id: item.customId };
-      }
-      return {
-        body,
-        custom_id: item.customId,
-        method: "POST",
-        url: endpoint
-      };
-    }), { label: "batch upload JSONL", maxBytes: limits.maxUploadBytes });
-    const headers3 = authHeaders(input.credentials);
-    const url = baseUrl3(input.credentials);
-    const purpose = config.filePurpose ?? "batch";
-    const inputFileId = await (config.uploadFile ? config.uploadFile({ baseUrl: url, headers: headers3, jsonl, purpose }) : uploadInputFile(jsonl, url, headers3, { purpose }));
-    const raw = await requestJson(`${url}/batches`, {
-      body: JSON.stringify({
-        completion_window: completionWindow,
-        endpoint,
-        input_file_id: inputFileId,
-        metadata: input.metadata
-      }),
-      headers: { ...headers3, "content-type": "application/json" },
-      method: "POST"
-    });
-    return normalizeSnapshot3(raw, config.id);
-  };
-  const retrieve3 = async (id, credentials) => {
-    const batchId = assertSimpleProviderId(`${config.id} batch id`, id);
-    const raw = await requestJson(`${baseUrl3(credentials)}/batches/${batchId}`, {
-      headers: authHeaders(credentials)
-    });
-    return normalizeSnapshot3(raw, config.id);
-  };
-  async function* results3(id, credentials) {
-    const snapshot = await retrieve3(id, credentials);
-    const raw = asRecord(snapshot.raw);
-    const outputFileId = asString(raw.output_file_id);
-    const errorFileId = asString(raw.error_file_id);
-    if (!(outputFileId || errorFileId)) {
-      throw new BatchworkError(`batchwork: results are not ready for batch "${id}" (status: ${snapshot.status}).`);
-    }
-    const headers3 = authHeaders(credentials);
-    if (outputFileId) {
-      yield* streamResultFile(assertSimpleProviderId(`${config.id} output file id`, outputFileId), baseUrl3(credentials), headers3);
-    }
-    if (errorFileId) {
-      yield* streamResultFile(assertSimpleProviderId(`${config.id} error file id`, errorFileId), baseUrl3(credentials), headers3);
-    }
-  }
-  const cancel3 = async (id, credentials) => {
-    const batchId = assertSimpleProviderId(`${config.id} batch id`, id);
-    await requestJson(`${baseUrl3(credentials)}/batches/${batchId}/cancel`, {
-      headers: authHeaders(credentials),
-      method: "POST"
-    });
-  };
-  return { cancel: cancel3, id: config.id, results: results3, retrieve: retrieve3, submit: submit3 };
-};
-
 // src/providers/groq.ts
 var groqAdapter = createOpenAICompatibleAdapter({
   apiKeyEnv: "GROQ_API_KEY",
@@ -1004,8 +1132,8 @@ var groqAdapter = createOpenAICompatibleAdapter({
 var MISTRAL_BASE = "https://api.mistral.ai/v1";
 var JOB_ID_LABEL = "Mistral job id";
 var apiKey3 = (credentials) => resolveApiKey(credentials, "MISTRAL_API_KEY", "Mistral");
-var baseUrl3 = (credentials) => credentials.baseURL ?? MISTRAL_BASE;
-var authHeaders = (credentials) => ({
+var baseUrl4 = (credentials) => credentials.baseURL ?? MISTRAL_BASE;
+var authHeaders2 = (credentials) => ({
   Authorization: `Bearer ${apiKey3(credentials)}`,
   ...credentials.headers
 });
@@ -1059,8 +1187,8 @@ var submit3 = async (input) => {
     body: omit(omit(item.body, "stream"), "model"),
     custom_id: item.customId
   })), { label: "batch upload JSONL", maxBytes: limits.maxUploadBytes });
-  const inputFileId = await uploadInputFile(jsonl, baseUrl3(input.credentials), authHeaders(input.credentials));
-  const raw = await requestJson(`${baseUrl3(input.credentials)}/batch/jobs`, {
+  const inputFileId = await uploadInputFile(jsonl, baseUrl4(input.credentials), authHeaders2(input.credentials));
+  const raw = await requestJson(`${baseUrl4(input.credentials)}/batch/jobs`, {
     body: JSON.stringify({
       endpoint: input.endpoint,
       input_files: [inputFileId],
@@ -1068,7 +1196,7 @@ var submit3 = async (input) => {
       model: input.modelId
     }),
     headers: {
-      ...authHeaders(input.credentials),
+      ...authHeaders2(input.credentials),
       "content-type": "application/json"
     },
     method: "POST"
@@ -1077,8 +1205,8 @@ var submit3 = async (input) => {
 };
 var retrieve3 = async (id, credentials) => {
   const jobId = assertSimpleProviderId(JOB_ID_LABEL, id);
-  const raw = await requestJson(`${baseUrl3(credentials)}/batch/jobs/${jobId}`, {
-    headers: authHeaders(credentials)
+  const raw = await requestJson(`${baseUrl4(credentials)}/batch/jobs/${jobId}`, {
+    headers: authHeaders2(credentials)
   });
   return normalizeSnapshot4(raw);
 };
@@ -1087,12 +1215,12 @@ async function* results3(id, credentials) {
   const raw = asRecord(snapshot.raw);
   const outputFileId = asString(raw.output_file);
   const errorFileId = asString(raw.error_file);
-  const headers3 = authHeaders(credentials);
+  const headers3 = authHeaders2(credentials);
   if (outputFileId) {
-    yield* streamResultFile(assertSimpleProviderId("Mistral output file id", outputFileId), baseUrl3(credentials), headers3);
+    yield* streamResultFile(assertSimpleProviderId("Mistral output file id", outputFileId), baseUrl4(credentials), headers3);
   }
   if (errorFileId) {
-    yield* streamResultFile(assertSimpleProviderId("Mistral error file id", errorFileId), baseUrl3(credentials), headers3);
+    yield* streamResultFile(assertSimpleProviderId("Mistral error file id", errorFileId), baseUrl4(credentials), headers3);
   }
   if (!(outputFileId || errorFileId)) {
     throw new BatchworkError(`batchwork: results are not ready for batch "${id}" (status: ${snapshot.status}).`);
@@ -1100,8 +1228,8 @@ async function* results3(id, credentials) {
 }
 var cancel3 = async (id, credentials) => {
   const jobId = assertSimpleProviderId(JOB_ID_LABEL, id);
-  await requestJson(`${baseUrl3(credentials)}/batch/jobs/${jobId}/cancel`, {
-    headers: authHeaders(credentials),
+  await requestJson(`${baseUrl4(credentials)}/batch/jobs/${jobId}/cancel`, {
+    headers: authHeaders2(credentials),
     method: "POST"
   });
 };
@@ -1232,6 +1360,7 @@ var togetherAdapter = createOpenAICompatibleAdapter({
   baseUrl: "https://api.together.xyz/v1",
   filePurpose: "batch-api",
   id: "together",
+  lineExtras: (endpoint) => endpoint.startsWith("/v1/audio/") ? { method: "FILE" } : undefined,
   lineFormat: "body-only",
   uploadFile: uploadTogetherFile
 });
@@ -1241,8 +1370,8 @@ var XAI_BASE = "https://api.x.ai/v1";
 var BATCH_ID_LABEL = "xAI batch id";
 var RESULTS_PAGE_SIZE = 100;
 var apiKey4 = (credentials) => resolveApiKey(credentials, "XAI_API_KEY", "xAI");
-var baseUrl4 = (credentials) => credentials.baseURL ?? XAI_BASE;
-var authHeaders2 = (credentials) => ({
+var baseUrl5 = (credentials) => credentials.baseURL ?? XAI_BASE;
+var authHeaders3 = (credentials) => ({
   Authorization: `Bearer ${apiKey4(credentials)}`,
   ...credentials.headers
 });
@@ -1303,6 +1432,25 @@ var imagesFromXaiCompletion = (completion) => {
   }
   return images.length > 0 ? images : undefined;
 };
+var videosFromXaiCompletion = (completion) => {
+  const obj = asRecord(completion);
+  const entries = asArray(obj.data);
+  const sources = entries.length > 0 ? entries : [obj];
+  const videos = [];
+  for (const source of sources) {
+    const record = asRecord(source);
+    const video = asRecord(record.video);
+    const url = asString(video.url) ?? asString(record.url);
+    if (url) {
+      const duration = asNumber(video.duration) ?? asNumber(record.duration);
+      videos.push({
+        ...duration === undefined ? {} : { durationSeconds: duration },
+        url
+      });
+    }
+  }
+  return videos.length > 0 ? videos : undefined;
+};
 var normalizeResult3 = (item) => {
   const obj = asRecord(item);
   const customId = asString(obj.batch_request_id) ?? "";
@@ -1321,10 +1469,12 @@ var normalizeResult3 = (item) => {
     };
   }
   const response = asRecord(batchResult.response);
+  const [opKey] = Object.keys(response);
   const completion = response.chat_get_completion ?? Object.values(response)[0];
+  const isVideo = response.chat_get_completion === undefined && opKey !== undefined && opKey.includes("video");
   return {
     customId,
-    images: imagesFromXaiCompletion(completion),
+    ...isVideo ? { videos: videosFromXaiCompletion(completion) } : { images: imagesFromXaiCompletion(completion) },
     response: completion,
     status: "succeeded",
     text: textFromBody(completion),
@@ -1337,13 +1487,13 @@ var submit4 = async (input) => {
     body: omit(item.body, "stream"),
     custom_id: item.customId,
     method: "POST",
-    url: input.endpoint
+    url: item.endpoint || input.endpoint
   })), { label: "batch upload JSONL", maxBytes: limits.maxUploadBytes });
-  const inputFileId = await uploadInputFile(jsonl, baseUrl4(input.credentials), authHeaders2(input.credentials), { purpose: null });
-  const raw = await requestJson(`${baseUrl4(input.credentials)}/batches`, {
+  const inputFileId = await uploadInputFile(jsonl, baseUrl5(input.credentials), authHeaders3(input.credentials), { purpose: null });
+  const raw = await requestJson(`${baseUrl5(input.credentials)}/batches`, {
     body: JSON.stringify({ input_file_id: inputFileId, name: "batchwork" }),
     headers: {
-      ...authHeaders2(input.credentials),
+      ...authHeaders3(input.credentials),
       "content-type": "application/json"
     },
     method: "POST"
@@ -1352,21 +1502,21 @@ var submit4 = async (input) => {
 };
 var retrieve4 = async (id, credentials) => {
   const batchId = assertSimpleProviderId(BATCH_ID_LABEL, id);
-  const raw = await requestJson(`${baseUrl4(credentials)}/batches/${batchId}`, {
-    headers: authHeaders2(credentials)
+  const raw = await requestJson(`${baseUrl5(credentials)}/batches/${batchId}`, {
+    headers: authHeaders3(credentials)
   });
   return normalizeSnapshot5(raw);
 };
 async function* results4(id, credentials) {
   const batchId = assertSimpleProviderId(BATCH_ID_LABEL, id);
-  const headers3 = authHeaders2(credentials);
+  const headers3 = authHeaders3(credentials);
   let token;
   do {
     const query = new URLSearchParams({ limit: String(RESULTS_PAGE_SIZE) });
     if (token) {
       query.set("pagination_token", token);
     }
-    const raw = await requestJson(`${baseUrl4(credentials)}/batches/${batchId}/results?${query.toString()}`, { headers: headers3 });
+    const raw = await requestJson(`${baseUrl5(credentials)}/batches/${batchId}/results?${query.toString()}`, { headers: headers3 });
     const page = asRecord(raw);
     for (const item of Array.isArray(page.results) ? page.results : []) {
       yield normalizeResult3(item);
@@ -1376,8 +1526,8 @@ async function* results4(id, credentials) {
 }
 var cancel4 = async (id, credentials) => {
   const batchId = assertSimpleProviderId(BATCH_ID_LABEL, id);
-  await requestJson(`${baseUrl4(credentials)}/batches/${batchId}:cancel`, {
-    headers: authHeaders2(credentials),
+  await requestJson(`${baseUrl5(credentials)}/batches/${batchId}:cancel`, {
+    headers: authHeaders3(credentials),
     method: "POST"
   });
 };
@@ -1392,6 +1542,7 @@ var xaiAdapter = {
 // src/providers/index.ts
 var adapters = {
   anthropic: anthropicAdapter,
+  azure: azureAdapter,
   google: googleAdapter,
   groq: groqAdapter,
   mistral: mistralAdapter,
@@ -1403,5 +1554,5 @@ var getAdapter = (provider) => adapters[provider];
 
 export { BatchworkError, UnsupportedProviderError, MissingDependencyError, resolveBatchLimits, assertByteLength, mapWithConcurrency, isTerminalStatus, BatchJob, getAdapter };
 
-//# debugId=032D6B4F83FC26F964756E2164756E21
-//# sourceMappingURL=chunk-jvrfwjwq.js.map
+//# debugId=C9B199FFE835E11B64756E2164756E21
+//# sourceMappingURL=chunk-vy4w8mpb.js.map