batchwork 1.3.0 → 1.4.1

package/dist/{chunk-gwa0dkhj.js → chunk-n50t31ca.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;
   }
@@ -138,22 +138,48 @@ var assertByteCount = (label, bytes, maxBytes) => {
 var assertByteLength = (label, value, maxBytes) => {
   assertByteCount(label, byteLength(value), maxBytes);
 };
-var mapWithConcurrency = async (items, concurrency, mapper) => {
-  const results = [];
-  results.length = items.length;
-  let nextIndex = 0;
-  const workerCount = Math.min(concurrency, items.length);
-  const runNext = async () => {
-    const index = nextIndex;
-    nextIndex += 1;
-    if (index >= items.length) {
-      return;
+
+// src/util.ts
+var trimTrailingSlashes = (value) => {
+  let end = value.length;
+  while (end > 0 && value[end - 1] === "/") {
+    end -= 1;
+  }
+  return value.slice(0, end);
+};
+var asRecord = (value) => {
+  if (typeof value === "object" && value !== null) {
+    return value;
+  }
+  return {};
+};
+var asString = (value) => typeof value === "string" ? value : undefined;
+var asNumber = (value) => typeof value === "number" ? value : undefined;
+var asArray = (value) => Array.isArray(value) ? value : [];
+var asNumberArray = (value) => {
+  if (!Array.isArray(value) || value.length === 0) {
+    return;
+  }
+  const numbers = value.filter((item) => typeof item === "number");
+  return numbers.length === value.length ? numbers : undefined;
+};
+var omit = (obj, key) => {
+  const result = {};
+  for (const [k, v] of Object.entries(obj)) {
+    if (k !== key) {
+      result[k] = v;
     }
-    results[index] = await mapper(items[index]);
-    await runNext();
-  };
-  await Promise.all(Array.from({ length: workerCount }, () => runNext()));
-  return results;
+  }
+  return result;
+};
+var validDate = (date) => Number.isNaN(date.getTime()) ? undefined : date;
+var toDate = (value) => {
+  if (typeof value === "string") {
+    return validDate(new Date(value));
+  }
+  if (typeof value === "number") {
+    return validDate(new Date(value * 1000));
+  }
 };
 
 // src/http.ts
@@ -306,42 +332,6 @@ var encodeJsonArrayPayload = ({
   return `${prefix}${encodedItems.join(",")}${suffix}`;
 };
 
-// src/util.ts
-var asRecord = (value) => {
-  if (typeof value === "object" && value !== null) {
-    return value;
-  }
-  return {};
-};
-var asString = (value) => typeof value === "string" ? value : undefined;
-var asNumber = (value) => typeof value === "number" ? value : undefined;
-var asArray = (value) => Array.isArray(value) ? value : [];
-var asNumberArray = (value) => {
-  if (!Array.isArray(value) || value.length === 0) {
-    return;
-  }
-  const numbers = value.filter((item) => typeof item === "number");
-  return numbers.length === value.length ? numbers : undefined;
-};
-var omit = (obj, key) => {
-  const result = {};
-  for (const [k, v] of Object.entries(obj)) {
-    if (k !== key) {
-      result[k] = v;
-    }
-  }
-  return result;
-};
-var validDate = (date) => Number.isNaN(date.getTime()) ? undefined : date;
-var toDate = (value) => {
-  if (typeof value === "string") {
-    return validDate(new Date(value));
-  }
-  if (typeof value === "number") {
-    return validDate(new Date(value * 1000));
-  }
-};
-
 // src/providers/ids.ts
 var SIMPLE_PROVIDER_ID = /^[A-Za-z0-9_-]+$/u;
 var assertSimpleProviderId = (label, id) => {
@@ -525,237 +515,18 @@ async function* results(id, credentials) {
   }
 }
 var cancel = async (id, credentials) => {
-  const batchId = assertSimpleProviderId("Anthropic batch id", id);
-  await requestJson(`${baseUrl(credentials)}/v1/messages/batches/${batchId}/cancel`, {
-    headers: headers(credentials),
-    method: "POST"
-  });
-};
-var anthropicAdapter = {
-  cancel,
-  id: "anthropic",
-  results,
-  retrieve,
-  submit
-};
-
-// src/providers/google.ts
-var GOOGLE_BASE = "https://generativelanguage.googleapis.com/v1beta";
-var OPERATION_ID_LABEL = "Google operation id";
-var GOOGLE_BATCH_PREFIX = "batches";
-var apiKey2 = (credentials) => {
-  const key = credentials.apiKey ?? process.env.GOOGLE_GENERATIVE_AI_API_KEY ?? process.env.GEMINI_API_KEY;
-  if (!key) {
-    throw new BatchworkError("batchwork: missing Google Gemini API key. Set GOOGLE_GENERATIVE_AI_API_KEY (or GEMINI_API_KEY) or pass `apiKey`.");
-  }
-  return key;
-};
-var baseUrl2 = (credentials) => credentials.baseURL ?? GOOGLE_BASE;
-var headers2 = (credentials) => ({
-  "content-type": "application/json",
-  "x-goog-api-key": apiKey2(credentials),
-  ...credentials.headers
-});
-var mapState = (state, done) => {
-  if (state) {
-    if (state.endsWith("SUCCEEDED")) {
-      return "completed";
-    }
-    if (state.endsWith("FAILED")) {
-      return "failed";
-    }
-    if (state.endsWith("CANCELLED")) {
-      return "cancelled";
-    }
-    if (state.endsWith("EXPIRED")) {
-      return "expired";
-    }
-    if (state.endsWith("PENDING")) {
-      return "validating";
-    }
-    if (state.endsWith("RUNNING")) {
-      return "in_progress";
-    }
-  }
-  return done ? "completed" : "in_progress";
-};
-var inlinedResponses = (raw) => {
-  const obj = asRecord(raw);
-  const response = asRecord(obj.response);
-  const dest = asRecord(obj.dest);
-  const responseInline = response.inlinedResponses ?? response.inlined_responses;
-  const destInline = dest.inlinedResponses ?? dest.inlined_responses;
-  const nestedResponseInline = asRecord(responseInline);
-  const nestedDestInline = asRecord(destInline);
-  return [
-    ...asArray(responseInline),
-    ...asArray(nestedResponseInline.inlinedResponses),
-    ...asArray(nestedResponseInline.inlined_responses),
-    ...asArray(destInline),
-    ...asArray(nestedDestInline.inlinedResponses),
-    ...asArray(nestedDestInline.inlined_responses)
-  ];
-};
-var normalizeSnapshot2 = (raw) => {
-  const obj = asRecord(raw);
-  const items = inlinedResponses(raw);
-  const failed = items.filter((item) => asRecord(item).error).length;
-  const id = asString(obj.name) ?? "";
-  return {
-    id: id ? assertPrefixedProviderId(OPERATION_ID_LABEL, id, GOOGLE_BATCH_PREFIX) : "",
-    provider: "google",
-    raw,
-    requestCounts: {
-      completed: items.length - failed,
-      failed,
-      total: items.length
-    },
-    status: mapState(asString(obj.state) ?? asString(asRecord(obj.state).name) ?? asString(asRecord(obj.metadata).state), obj.done === true)
-  };
-};
-var textFromResponse = (response) => {
-  const candidate = asRecord(asArray(asRecord(response).candidates)[0]);
-  const text = asArray(asRecord(candidate.content).parts).map((part) => asString(asRecord(part).text) ?? "").join("");
-  return text.length > 0 ? text : undefined;
-};
-var embeddingFromResponse = (response) => asNumberArray(asRecord(asRecord(response).embedding).values);
-var imagesFromResponse = (response) => {
-  const candidate = asRecord(asArray(asRecord(response).candidates)[0]);
-  const images = [];
-  for (const part of asArray(asRecord(candidate.content).parts)) {
-    const partObj = asRecord(part);
-    const inline = asRecord(partObj.inlineData ?? partObj.inline_data);
-    const data = asString(inline.data);
-    const mediaType = asString(inline.mimeType) ?? asString(inline.mime_type);
-    if (data && mediaType?.startsWith("image/")) {
-      images.push({ data, mediaType });
-    }
-  }
-  return images.length > 0 ? images : undefined;
-};
-var usageFromResponse = (response) => {
-  const usage = asRecord(asRecord(response).usageMetadata);
-  const inputTokens = asNumber(usage.promptTokenCount);
-  const outputTokens = asNumber(usage.candidatesTokenCount);
-  const totalTokens = asNumber(usage.totalTokenCount);
-  if (inputTokens === undefined && outputTokens === undefined && totalTokens === undefined) {
-    return;
-  }
-  return {
-    inputTokens,
-    outputTokens,
-    totalTokens: totalTokens ?? (inputTokens ?? 0) + (outputTokens ?? 0)
-  };
-};
-var normalizeResult2 = (item) => {
-  const obj = asRecord(item);
-  const customId = asString(asRecord(obj.metadata).key) ?? asString(obj.key) ?? asString(obj.custom_id) ?? "";
-  if (obj.error) {
-    const error = asRecord(obj.error);
-    return {
-      customId,
-      error: {
-        code: asNumber(error.code) ?? asString(error.code),
-        message: asString(error.message) ?? "Request errored.",
-        type: asString(error.status)
-      },
-      response: obj.error,
-      status: "errored"
-    };
-  }
-  return {
-    customId,
-    embedding: embeddingFromResponse(obj.response),
-    images: imagesFromResponse(obj.response),
-    response: obj.response,
-    status: "succeeded",
-    text: textFromResponse(obj.response),
-    usage: usageFromResponse(obj.response)
-  };
-};
-var EMBED_CONFIG_KEYS = new Set([
-  "outputDimensionality",
-  "taskType",
-  "title"
-]);
-var toEmbedRequest = (body) => {
-  const request = {};
-  const config = {};
-  for (const [key, value] of Object.entries(body)) {
-    if (EMBED_CONFIG_KEYS.has(key)) {
-      config[key] = value;
-    } else {
-      request[key] = value;
-    }
-  }
-  if (Object.keys(config).length > 0) {
-    request.embedContentConfig = config;
-  }
-  return request;
-};
-var submit2 = async (input) => {
-  const limits = resolveBatchLimits(input.limits);
-  const isEmbedding = input.endpoint.toLowerCase().includes("embedcontent");
-  const method = isEmbedding ? "asyncBatchEmbedContent" : "batchGenerateContent";
-  const requests = input.built.map((item) => {
-    const payload = omit(item.body, "stream");
-    return {
-      metadata: { key: item.customId },
-      request: isEmbedding ? toEmbedRequest(payload) : payload
-    };
-  });
-  const body = encodeJsonArrayPayload({
-    items: requests,
-    label: "batch upload payload",
-    maxBytes: limits.maxUploadBytes,
-    prefix: '{"batch":{"display_name":"batchwork","input_config":{"requests":{"requests":[',
-    suffix: "]}}}}"
-  });
-  const raw = await requestJson(`${baseUrl2(input.credentials)}/models/${input.modelId}:${method}`, {
-    body,
-    headers: headers2(input.credentials),
-    method: "POST"
-  });
-  return normalizeSnapshot2(raw);
-};
-var retrieve2 = async (id, credentials) => {
-  const operationId = assertPrefixedProviderId(OPERATION_ID_LABEL, id, GOOGLE_BATCH_PREFIX);
-  const raw = await requestJson(`${baseUrl2(credentials)}/${operationId}`, {
-    headers: headers2(credentials)
-  });
-  return normalizeSnapshot2(raw);
-};
-var fileNameFrom = (value) => asString(asRecord(value).name) ?? asString(value);
-async function* results2(id, credentials) {
-  const snapshot = await retrieve2(id, credentials);
-  const raw = asRecord(snapshot.raw);
-  const response = asRecord(raw.response);
-  const dest = asRecord(raw.dest);
-  const responsesFile = fileNameFrom(response.responsesFile) ?? fileNameFrom(response.responses_file) ?? asString(dest.fileName) ?? asString(dest.file_name);
-  if (responsesFile) {
-    throw new BatchworkError(`batchwork: batch "${id}" returned file-mode results, which are not supported yet.`);
-  }
-  const items = inlinedResponses(raw);
-  if (items.length === 0) {
-    throw new BatchworkError(`batchwork: results are not ready for batch "${id}" (status: ${snapshot.status}).`);
-  }
-  for (const item of items) {
-    yield normalizeResult2(item);
-  }
-}
-var cancel2 = async (id, credentials) => {
-  const operationId = assertPrefixedProviderId(OPERATION_ID_LABEL, id, GOOGLE_BATCH_PREFIX);
-  await requestJson(`${baseUrl2(credentials)}/${operationId}:cancel`, {
-    headers: headers2(credentials),
+  const batchId = assertSimpleProviderId("Anthropic batch id", id);
+  await requestJson(`${baseUrl(credentials)}/v1/messages/batches/${batchId}/cancel`, {
+    headers: headers(credentials),
     method: "POST"
   });
 };
-var googleAdapter = {
-  cancel: cancel2,
-  id: "google",
-  results: results2,
-  retrieve: retrieve2,
-  submit: submit2
+var anthropicAdapter = {
+  cancel,
+  id: "anthropic",
+  results,
+  retrieve,
+  submit
 };
 
 // src/providers/shared.ts
@@ -777,7 +548,30 @@ var textFromBody = (body) => {
       return content;
     }
   }
-  return asString(obj.output_text) ?? asString(obj.text);
+  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);
@@ -818,11 +612,11 @@ var imagesFromBody = (body) => {
   return images.length > 0 ? images : undefined;
 };
 var moderationFromBody = (body) => {
-  const results3 = asArray(asRecord(body).results);
-  if (results3.length === 0) {
+  const results2 = asArray(asRecord(body).results);
+  if (results2.length === 0) {
     return;
   }
-  const first = asRecord(results3[0]);
+  const first = asRecord(results2[0]);
   const categories = {};
   for (const [key, value] of Object.entries(asRecord(first.categories))) {
     if (typeof value === "boolean") {
@@ -899,24 +693,24 @@ var normalizeOpenAIResult = (line) => {
     status: "errored"
   };
 };
-var uploadInputFile = async (jsonl, baseUrl3, headers3, options = {}) => {
+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(`${baseUrl3}/files`, {
+  const raw = await requestJson(`${baseUrl2}/files`, {
     body: form,
-    headers: headers3,
+    headers: headers2,
     method: "POST",
     redirect: "manual"
   });
   return raw.id;
 };
-async function* streamResultFile(fileId, baseUrl3, headers3) {
-  const stream = await requestStream(`${baseUrl3}/files/${fileId}/content`, {
-    headers: headers3,
+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)) {
@@ -944,7 +738,7 @@ var mapStatus2 = (status) => {
     }
   }
 };
-var normalizeSnapshot3 = (raw, provider) => {
+var normalizeSnapshot2 = (raw, provider) => {
   const outer = asRecord(raw);
   const obj = asRecord(outer.job);
   const source = Object.keys(obj).length > 0 ? obj : outer;
@@ -967,14 +761,19 @@ var normalizeSnapshot3 = (raw, provider) => {
 var createOpenAICompatibleAdapter = (config) => {
   const completionWindow = config.completionWindow ?? DEFAULT_COMPLETION_WINDOW;
   const lineFormat = config.lineFormat ?? "method-url";
-  const baseUrl3 = (credentials) => credentials.baseURL ?? config.baseUrl;
+  const baseUrl2 = (credentials) => config.resolveBaseUrl?.(credentials) ?? credentials.baseURL ?? config.baseUrl ?? (() => {
+    throw new BatchworkError(`batchwork: missing ${config.apiKeyLabel} base URL. Pass \`baseURL\`.`);
+  })();
   const authHeaders = (credentials) => ({
-    Authorization: `Bearer ${resolveApiKey(credentials, config.apiKeyEnv, config.apiKeyLabel)}`,
+    ...config.authHeaders ? config.authHeaders(credentials) : {
+      Authorization: `Bearer ${resolveApiKey(credentials, config.apiKeyEnv, config.apiKeyLabel)}`
+    },
     ...credentials.headers
   });
-  const submit3 = async (input) => {
+  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") {
@@ -991,53 +790,321 @@ var createOpenAICompatibleAdapter = (config) => {
         url: endpoint
       };
     }), { label: "batch upload JSONL", maxBytes: limits.maxUploadBytes });
-    const headers3 = authHeaders(input.credentials);
-    const url = baseUrl3(input.credentials);
+    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: headers3, jsonl, purpose }) : uploadInputFile(jsonl, url, headers3, { purpose }));
+    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,
+        endpoint: batchEndpoint,
         input_file_id: inputFileId,
         metadata: input.metadata
       }),
-      headers: { ...headers3, "content-type": "application/json" },
+      headers: { ...headers2, "content-type": "application/json" },
       method: "POST"
     });
-    return normalizeSnapshot3(raw, config.id);
+    return normalizeSnapshot2(raw, config.id);
   };
-  const retrieve3 = async (id, credentials) => {
+  const retrieve2 = async (id, credentials) => {
     const batchId = assertSimpleProviderId(`${config.id} batch id`, id);
-    const raw = await requestJson(`${baseUrl3(credentials)}/batches/${batchId}`, {
+    const raw = await requestJson(`${baseUrl2(credentials)}/batches/${batchId}`, {
       headers: authHeaders(credentials)
     });
-    return normalizeSnapshot3(raw, config.id);
+    return normalizeSnapshot2(raw, config.id);
   };
-  async function* results3(id, credentials) {
-    const snapshot = await retrieve3(id, credentials);
+  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 headers3 = authHeaders(credentials);
+    const headers2 = authHeaders(credentials);
     if (outputFileId) {
-      yield* streamResultFile(assertSimpleProviderId(`${config.id} output file id`, outputFileId), baseUrl3(credentials), headers3);
+      yield* streamResultFile(assertSimpleProviderId(`${config.id} output file id`, outputFileId), baseUrl2(credentials), headers2);
     }
     if (errorFileId) {
-      yield* streamResultFile(assertSimpleProviderId(`${config.id} error file id`, errorFileId), baseUrl3(credentials), headers3);
+      yield* streamResultFile(assertSimpleProviderId(`${config.id} error file id`, errorFileId), baseUrl2(credentials), headers2);
     }
   }
-  const cancel3 = async (id, credentials) => {
+  const cancel2 = async (id, credentials) => {
     const batchId = assertSimpleProviderId(`${config.id} batch id`, id);
-    await requestJson(`${baseUrl3(credentials)}/batches/${batchId}/cancel`, {
+    await requestJson(`${baseUrl2(credentials)}/batches/${batchId}/cancel`, {
       headers: authHeaders(credentials),
       method: "POST"
     });
   };
-  return { cancel: cancel3, id: config.id, results: results3, retrieve: retrieve3, submit: submit3 };
+  return { cancel: cancel2, id: config.id, results: results2, retrieve: retrieve2, submit: submit2 };
+};
+
+// src/providers/azure.ts
+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 = trimTrailingSlashes(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";
+var GOOGLE_BATCH_PREFIX = "batches";
+var apiKey2 = (credentials) => {
+  const key = credentials.apiKey ?? process.env.GOOGLE_GENERATIVE_AI_API_KEY ?? process.env.GEMINI_API_KEY;
+  if (!key) {
+    throw new BatchworkError("batchwork: missing Google Gemini API key. Set GOOGLE_GENERATIVE_AI_API_KEY (or GEMINI_API_KEY) or pass `apiKey`.");
+  }
+  return key;
+};
+var baseUrl3 = (credentials) => credentials.baseURL ?? GOOGLE_BASE;
+var headers2 = (credentials) => ({
+  "content-type": "application/json",
+  "x-goog-api-key": apiKey2(credentials),
+  ...credentials.headers
+});
+var mapState = (state, done) => {
+  if (state) {
+    if (state.endsWith("SUCCEEDED")) {
+      return "completed";
+    }
+    if (state.endsWith("FAILED")) {
+      return "failed";
+    }
+    if (state.endsWith("CANCELLED")) {
+      return "cancelled";
+    }
+    if (state.endsWith("EXPIRED")) {
+      return "expired";
+    }
+    if (state.endsWith("PENDING")) {
+      return "validating";
+    }
+    if (state.endsWith("RUNNING")) {
+      return "in_progress";
+    }
+  }
+  return done ? "completed" : "in_progress";
+};
+var inlinedResponses = (raw) => {
+  const obj = asRecord(raw);
+  const response = asRecord(obj.response);
+  const dest = asRecord(obj.dest);
+  const responseInline = response.inlinedResponses ?? response.inlined_responses;
+  const destInline = dest.inlinedResponses ?? dest.inlined_responses;
+  const nestedResponseInline = asRecord(responseInline);
+  const nestedDestInline = asRecord(destInline);
+  return [
+    ...asArray(responseInline),
+    ...asArray(nestedResponseInline.inlinedResponses),
+    ...asArray(nestedResponseInline.inlined_responses),
+    ...asArray(destInline),
+    ...asArray(nestedDestInline.inlinedResponses),
+    ...asArray(nestedDestInline.inlined_responses)
+  ];
+};
+var normalizeSnapshot3 = (raw) => {
+  const obj = asRecord(raw);
+  const items = inlinedResponses(raw);
+  const failed = items.filter((item) => asRecord(item).error).length;
+  const id = asString(obj.name) ?? "";
+  return {
+    id: id ? assertPrefixedProviderId(OPERATION_ID_LABEL, id, GOOGLE_BATCH_PREFIX) : "",
+    provider: "google",
+    raw,
+    requestCounts: {
+      completed: items.length - failed,
+      failed,
+      total: items.length
+    },
+    status: mapState(asString(obj.state) ?? asString(asRecord(obj.state).name) ?? asString(asRecord(obj.metadata).state), obj.done === true)
+  };
+};
+var textFromResponse = (response) => {
+  const candidate = asRecord(asArray(asRecord(response).candidates)[0]);
+  const text = asArray(asRecord(candidate.content).parts).map((part) => asString(asRecord(part).text) ?? "").join("");
+  return text.length > 0 ? text : undefined;
+};
+var embeddingFromResponse = (response) => asNumberArray(asRecord(asRecord(response).embedding).values);
+var imagesFromResponse = (response) => {
+  const candidate = asRecord(asArray(asRecord(response).candidates)[0]);
+  const images = [];
+  for (const part of asArray(asRecord(candidate.content).parts)) {
+    const partObj = asRecord(part);
+    const inline = asRecord(partObj.inlineData ?? partObj.inline_data);
+    const data = asString(inline.data);
+    const mediaType = asString(inline.mimeType) ?? asString(inline.mime_type);
+    if (data && mediaType?.startsWith("image/")) {
+      images.push({ data, mediaType });
+    }
+  }
+  return images.length > 0 ? images : undefined;
+};
+var usageFromResponse = (response) => {
+  const usage = asRecord(asRecord(response).usageMetadata);
+  const inputTokens = asNumber(usage.promptTokenCount);
+  const outputTokens = asNumber(usage.candidatesTokenCount);
+  const totalTokens = asNumber(usage.totalTokenCount);
+  if (inputTokens === undefined && outputTokens === undefined && totalTokens === undefined) {
+    return;
+  }
+  return {
+    inputTokens,
+    outputTokens,
+    totalTokens: totalTokens ?? (inputTokens ?? 0) + (outputTokens ?? 0)
+  };
+};
+var normalizeResult2 = (item) => {
+  const obj = asRecord(item);
+  const customId = asString(asRecord(obj.metadata).key) ?? asString(obj.key) ?? asString(obj.custom_id) ?? "";
+  if (obj.error) {
+    const error = asRecord(obj.error);
+    return {
+      customId,
+      error: {
+        code: asNumber(error.code) ?? asString(error.code),
+        message: asString(error.message) ?? "Request errored.",
+        type: asString(error.status)
+      },
+      response: obj.error,
+      status: "errored"
+    };
+  }
+  return {
+    customId,
+    embedding: embeddingFromResponse(obj.response),
+    images: imagesFromResponse(obj.response),
+    response: obj.response,
+    status: "succeeded",
+    text: textFromResponse(obj.response),
+    usage: usageFromResponse(obj.response)
+  };
+};
+var EMBED_CONFIG_KEYS = new Set([
+  "outputDimensionality",
+  "taskType",
+  "title"
+]);
+var toEmbedRequest = (body) => {
+  const request = {};
+  const config = {};
+  for (const [key, value] of Object.entries(body)) {
+    if (EMBED_CONFIG_KEYS.has(key)) {
+      config[key] = value;
+    } else {
+      request[key] = value;
+    }
+  }
+  if (Object.keys(config).length > 0) {
+    request.embedContentConfig = config;
+  }
+  return request;
+};
+var submit2 = async (input) => {
+  const limits = resolveBatchLimits(input.limits);
+  const isEmbedding = input.endpoint.toLowerCase().includes("embedcontent");
+  const method = isEmbedding ? "asyncBatchEmbedContent" : "batchGenerateContent";
+  const requests = input.built.map((item) => {
+    const payload = omit(item.body, "stream");
+    return {
+      metadata: { key: item.customId },
+      request: isEmbedding ? toEmbedRequest(payload) : payload
+    };
+  });
+  const body = encodeJsonArrayPayload({
+    items: requests,
+    label: "batch upload payload",
+    maxBytes: limits.maxUploadBytes,
+    prefix: '{"batch":{"display_name":"batchwork","input_config":{"requests":{"requests":[',
+    suffix: "]}}}}"
+  });
+  const raw = await requestJson(`${baseUrl3(input.credentials)}/models/${input.modelId}:${method}`, {
+    body,
+    headers: headers2(input.credentials),
+    method: "POST"
+  });
+  return normalizeSnapshot3(raw);
+};
+var retrieve2 = async (id, credentials) => {
+  const operationId = assertPrefixedProviderId(OPERATION_ID_LABEL, id, GOOGLE_BATCH_PREFIX);
+  const raw = await requestJson(`${baseUrl3(credentials)}/${operationId}`, {
+    headers: headers2(credentials)
+  });
+  return normalizeSnapshot3(raw);
+};
+var fileNameFrom = (value) => asString(asRecord(value).name) ?? asString(value);
+async function* results2(id, credentials) {
+  const snapshot = await retrieve2(id, credentials);
+  const raw = asRecord(snapshot.raw);
+  const response = asRecord(raw.response);
+  const dest = asRecord(raw.dest);
+  const responsesFile = fileNameFrom(response.responsesFile) ?? fileNameFrom(response.responses_file) ?? asString(dest.fileName) ?? asString(dest.file_name);
+  if (responsesFile) {
+    throw new BatchworkError(`batchwork: batch "${id}" returned file-mode results, which are not supported yet.`);
+  }
+  const items = inlinedResponses(raw);
+  if (items.length === 0) {
+    throw new BatchworkError(`batchwork: results are not ready for batch "${id}" (status: ${snapshot.status}).`);
+  }
+  for (const item of items) {
+    yield normalizeResult2(item);
+  }
+}
+var cancel2 = async (id, credentials) => {
+  const operationId = assertPrefixedProviderId(OPERATION_ID_LABEL, id, GOOGLE_BATCH_PREFIX);
+  await requestJson(`${baseUrl3(credentials)}/${operationId}:cancel`, {
+    headers: headers2(credentials),
+    method: "POST"
+  });
+};
+var googleAdapter = {
+  cancel: cancel2,
+  id: "google",
+  results: results2,
+  retrieve: retrieve2,
+  submit: submit2
 };
 
 // src/providers/groq.ts
@@ -1054,8 +1121,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
 });
@@ -1109,8 +1176,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],
@@ -1118,7 +1185,7 @@ var submit3 = async (input) => {
       model: input.modelId
     }),
     headers: {
-      ...authHeaders(input.credentials),
+      ...authHeaders2(input.credentials),
       "content-type": "application/json"
     },
     method: "POST"
@@ -1127,8 +1194,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);
 };
@@ -1137,12 +1204,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}).`);
@@ -1150,8 +1217,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"
   });
 };
@@ -1292,8 +1359,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
 });
@@ -1411,11 +1478,11 @@ var submit4 = async (input) => {
     method: "POST",
     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"
@@ -1424,21 +1491,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);
@@ -1448,8 +1515,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"
   });
 };
@@ -1464,6 +1531,7 @@ var xaiAdapter = {
 // src/providers/index.ts
 var adapters = {
   anthropic: anthropicAdapter,
+  azure: azureAdapter,
   google: googleAdapter,
   groq: groqAdapter,
   mistral: mistralAdapter,
@@ -1473,7 +1541,7 @@ var adapters = {
 };
 var getAdapter = (provider) => adapters[provider];
 
-export { BatchworkError, UnsupportedProviderError, MissingDependencyError, resolveBatchLimits, assertByteLength, mapWithConcurrency, isTerminalStatus, BatchJob, getAdapter };
+export { BatchworkError, UnsupportedProviderError, MissingDependencyError, resolveBatchLimits, assertByteLength, trimTrailingSlashes, isTerminalStatus, BatchJob, getAdapter };
 
-//# debugId=4C939B96519322C164756E2164756E21
-//# sourceMappingURL=chunk-gwa0dkhj.js.map
+//# debugId=C29D2DE3DBB0695664756E2164756E21
+//# sourceMappingURL=chunk-n50t31ca.js.map