@mistralai/mistralai 2.2.0 → 2.2.5

package/packages/mistralai-gcp/src/models/components/referencechunk.ts

@@ -7,18 +7,46 @@ import * as z from "zod/v4";
 import { remap as remap$ } from "../../lib/primitives.js";
 import { safeParse } from "../../lib/schemas.js";
 import { Result as SafeParseResult } from "../../types/fp.js";
+import { smartUnion } from "../../types/smartUnion.js";
 import { SDKValidationError } from "../errors/sdkvalidationerror.js";
 
+export type ReferenceId = number | string;
+
 export type ReferenceChunk = {
   type?: "reference" | undefined;
-  referenceIds: Array<number>;
+  referenceIds: Array<number | string>;
 };
 
+/** @internal */
+export const ReferenceId$inboundSchema: z.ZodType<ReferenceId, unknown> =
+  smartUnion([z.int(), z.string()]);
+/** @internal */
+export type ReferenceId$Outbound = number | string;
+
+/** @internal */
+export const ReferenceId$outboundSchema: z.ZodType<
+  ReferenceId$Outbound,
+  ReferenceId
+> = smartUnion([z.int(), z.string()]);
+
+export function referenceIdToJSON(referenceId: ReferenceId): string {
+  return JSON.stringify(ReferenceId$outboundSchema.parse(referenceId));
+}
+export function referenceIdFromJSON(
+  jsonString: string,
+): SafeParseResult<ReferenceId, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => ReferenceId$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'ReferenceId' from JSON`,
+  );
+}
+
 /** @internal */
 export const ReferenceChunk$inboundSchema: z.ZodType<ReferenceChunk, unknown> =
   z.object({
     type: z.literal("reference").default("reference"),
-    reference_ids: z.array(z.int()),
+    reference_ids: z.array(smartUnion([z.int(), z.string()])),
   }).transform((v) => {
     return remap$(v, {
       "reference_ids": "referenceIds",
@@ -27,7 +55,7 @@ export const ReferenceChunk$inboundSchema: z.ZodType<ReferenceChunk, unknown> =
 /** @internal */
 export type ReferenceChunk$Outbound = {
   type: "reference";
-  reference_ids: Array<number>;
+  reference_ids: Array<number | string>;
 };
 
 /** @internal */
@@ -36,7 +64,7 @@ export const ReferenceChunk$outboundSchema: z.ZodType<
   ReferenceChunk
 > = z.object({
   type: z.literal("reference").default("reference" as const),
-  referenceIds: z.array(z.int()),
+  referenceIds: z.array(smartUnion([z.int(), z.string()])),
 }).transform((v) => {
   return remap$(v, {
     referenceIds: "reference_ids",

package/packages/mistralai-gcp/src/models/components/thinkchunk.ts

@@ -26,6 +26,10 @@ export type Thinking = TextChunk | ToolReferenceChunk | ReferenceChunk;
 export type ThinkChunk = {
   type: "thinking";
   thinking: Array<TextChunk | ToolReferenceChunk | ReferenceChunk>;
+  /**
+   * Signature to replay some reasoning blocks across turns.
+   */
+  signature?: string | null | undefined;
   /**
    * Whether the thinking chunk is closed or not. Currently only used for prefixing.
    */
@@ -56,6 +60,7 @@ export type ThinkChunk$Outbound = {
   thinking: Array<
     TextChunk$Outbound | ToolReferenceChunk$Outbound | ReferenceChunk$Outbound
   >;
+  signature?: string | null | undefined;
   closed?: boolean | undefined;
 };
 
@@ -72,6 +77,7 @@ export const ThinkChunk$outboundSchema: z.ZodType<
       ReferenceChunk$outboundSchema,
     ]),
   ),
+  signature: z.nullable(z.string()).optional(),
   closed: z.boolean().optional(),
 });
 

package/packages/mistralai-gcp/src/models/components/tool.ts

@@ -9,22 +9,21 @@ import {
   FunctionT$Outbound,
   FunctionT$outboundSchema,
 } from "./function.js";
-import { ToolTypes, ToolTypes$outboundSchema } from "./tooltypes.js";
 
 export type Tool = {
-  type?: ToolTypes | undefined;
+  type: "function";
   function: FunctionT;
 };
 
 /** @internal */
 export type Tool$Outbound = {
-  type?: string | undefined;
+  type: "function";
   function: FunctionT$Outbound;
 };
 
 /** @internal */
 export const Tool$outboundSchema: z.ZodType<Tool$Outbound, Tool> = z.object({
-  type: ToolTypes$outboundSchema.optional(),
+  type: z.literal("function"),
   function: FunctionT$outboundSchema,
 });
 

package/packages/mistralai-gcp/src/models/components/toolcall.ts

@@ -13,15 +13,10 @@ import {
   FunctionCall$Outbound,
   FunctionCall$outboundSchema,
 } from "./functioncall.js";
-import {
-  ToolTypes,
-  ToolTypes$inboundSchema,
-  ToolTypes$outboundSchema,
-} from "./tooltypes.js";
 
 export type ToolCall = {
   id?: string | undefined;
-  type?: ToolTypes | undefined;
+  type?: string | undefined;
   function: FunctionCall;
   index?: number | undefined;
 };
@@ -29,7 +24,7 @@ export type ToolCall = {
 /** @internal */
 export const ToolCall$inboundSchema: z.ZodType<ToolCall, unknown> = z.object({
   id: z.string().default("null"),
-  type: ToolTypes$inboundSchema.optional(),
+  type: z.string().optional(),
   function: FunctionCall$inboundSchema,
   index: z.int().default(0),
 });
@@ -45,7 +40,7 @@ export type ToolCall$Outbound = {
 export const ToolCall$outboundSchema: z.ZodType<ToolCall$Outbound, ToolCall> = z
   .object({
     id: z.string().default("null"),
-    type: ToolTypes$outboundSchema.optional(),
+    type: z.string().optional(),
     function: FunctionCall$outboundSchema,
     index: z.int().default(0),
   });

package/packages/mistralai-gcp/src/models/components/toolchoice.ts

@@ -9,13 +9,12 @@ import {
   FunctionName$Outbound,
   FunctionName$outboundSchema,
 } from "./functionname.js";
-import { ToolTypes, ToolTypes$outboundSchema } from "./tooltypes.js";
 
 /**
  * ToolChoice is either a ToolChoiceEnum or a ToolChoice
  */
 export type ToolChoice = {
-  type?: ToolTypes | undefined;
+  type?: string | undefined;
   /**
    * this restriction of `Function` is used to select a specific function to call
    */
@@ -33,7 +32,7 @@ export const ToolChoice$outboundSchema: z.ZodType<
   ToolChoice$Outbound,
   ToolChoice
 > = z.object({
-  type: ToolTypes$outboundSchema.optional(),
+  type: z.string().optional(),
   function: FunctionName$outboundSchema,
 });
 

package/packages/mistralai-gcp/src/models/components/toolconfiguration.ts

@@ -0,0 +1,42 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ * @generated-id: 9849ab330d6e
+ */
+
+import * as z from "zod/v4";
+import { remap as remap$ } from "../../lib/primitives.js";
+
+export type ToolConfiguration = {
+  exclude?: Array<string> | null | undefined;
+  include?: Array<string> | null | undefined;
+  requiresConfirmation?: Array<string> | null | undefined;
+};
+
+/** @internal */
+export type ToolConfiguration$Outbound = {
+  exclude?: Array<string> | null | undefined;
+  include?: Array<string> | null | undefined;
+  requires_confirmation?: Array<string> | null | undefined;
+};
+
+/** @internal */
+export const ToolConfiguration$outboundSchema: z.ZodType<
+  ToolConfiguration$Outbound,
+  ToolConfiguration
+> = z.object({
+  exclude: z.nullable(z.array(z.string())).optional(),
+  include: z.nullable(z.array(z.string())).optional(),
+  requiresConfirmation: z.nullable(z.array(z.string())).optional(),
+}).transform((v) => {
+  return remap$(v, {
+    requiresConfirmation: "requires_confirmation",
+  });
+});
+
+export function toolConfigurationToJSON(
+  toolConfiguration: ToolConfiguration,
+): string {
+  return JSON.stringify(
+    ToolConfiguration$outboundSchema.parse(toolConfiguration),
+  );
+}

package/packages/mistralai-gcp/src/models/components/toolreferencechunk.ts

@@ -10,7 +10,7 @@ import {
   BuiltInConnectors$outboundSchema,
 } from "./builtinconnectors.js";
 
-export type ToolUnion = BuiltInConnectors | string;
+export type ToolReferenceChunkTool = BuiltInConnectors | string;
 
 export type ToolReferenceChunk = {
   type?: "tool_reference" | undefined;
@@ -22,16 +22,20 @@ export type ToolReferenceChunk = {
 };
 
 /** @internal */
-export type ToolUnion$Outbound = string | string;
+export type ToolReferenceChunkTool$Outbound = string | string;
 
 /** @internal */
-export const ToolUnion$outboundSchema: z.ZodType<
-  ToolUnion$Outbound,
-  ToolUnion
+export const ToolReferenceChunkTool$outboundSchema: z.ZodType<
+  ToolReferenceChunkTool$Outbound,
+  ToolReferenceChunkTool
 > = smartUnion([BuiltInConnectors$outboundSchema, z.string()]);
 
-export function toolUnionToJSON(toolUnion: ToolUnion): string {
-  return JSON.stringify(ToolUnion$outboundSchema.parse(toolUnion));
+export function toolReferenceChunkToolToJSON(
+  toolReferenceChunkTool: ToolReferenceChunkTool,
+): string {
+  return JSON.stringify(
+    ToolReferenceChunkTool$outboundSchema.parse(toolReferenceChunkTool),
+  );
 }
 
 /** @internal */

package/packages/mistralai-gcp/src/models/components/websearchpremiumtool.ts

@@ -0,0 +1,44 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ * @generated-id: c03632bf7889
+ */
+
+import * as z from "zod/v4";
+import { remap as remap$ } from "../../lib/primitives.js";
+import {
+  ToolConfiguration,
+  ToolConfiguration$Outbound,
+  ToolConfiguration$outboundSchema,
+} from "./toolconfiguration.js";
+
+export type WebSearchPremiumTool = {
+  toolConfiguration?: ToolConfiguration | null | undefined;
+  type: "web_search_premium";
+};
+
+/** @internal */
+export type WebSearchPremiumTool$Outbound = {
+  tool_configuration?: ToolConfiguration$Outbound | null | undefined;
+  type: "web_search_premium";
+};
+
+/** @internal */
+export const WebSearchPremiumTool$outboundSchema: z.ZodType<
+  WebSearchPremiumTool$Outbound,
+  WebSearchPremiumTool
+> = z.object({
+  toolConfiguration: z.nullable(ToolConfiguration$outboundSchema).optional(),
+  type: z.literal("web_search_premium"),
+}).transform((v) => {
+  return remap$(v, {
+    toolConfiguration: "tool_configuration",
+  });
+});
+
+export function webSearchPremiumToolToJSON(
+  webSearchPremiumTool: WebSearchPremiumTool,
+): string {
+  return JSON.stringify(
+    WebSearchPremiumTool$outboundSchema.parse(webSearchPremiumTool),
+  );
+}

package/packages/mistralai-gcp/src/models/components/websearchtool.ts

@@ -0,0 +1,40 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ * @generated-id: 57ac372e9d2a
+ */
+
+import * as z from "zod/v4";
+import { remap as remap$ } from "../../lib/primitives.js";
+import {
+  ToolConfiguration,
+  ToolConfiguration$Outbound,
+  ToolConfiguration$outboundSchema,
+} from "./toolconfiguration.js";
+
+export type WebSearchTool = {
+  toolConfiguration?: ToolConfiguration | null | undefined;
+  type: "web_search";
+};
+
+/** @internal */
+export type WebSearchTool$Outbound = {
+  tool_configuration?: ToolConfiguration$Outbound | null | undefined;
+  type: "web_search";
+};
+
+/** @internal */
+export const WebSearchTool$outboundSchema: z.ZodType<
+  WebSearchTool$Outbound,
+  WebSearchTool
+> = z.object({
+  toolConfiguration: z.nullable(ToolConfiguration$outboundSchema).optional(),
+  type: z.literal("web_search"),
+}).transform((v) => {
+  return remap$(v, {
+    toolConfiguration: "tool_configuration",
+  });
+});
+
+export function webSearchToolToJSON(webSearchTool: WebSearchTool): string {
+  return JSON.stringify(WebSearchTool$outboundSchema.parse(webSearchTool));
+}

package/packages/mistralai-gcp/src/models/errors/httpvalidationerror.ts

@@ -5,7 +5,7 @@
 
 import * as z from "zod/v4";
 import * as components from "../components/index.js";
-import { MistralGCPError } from "./mistralgoogleclouderror.js";
+import { MistralGCPError } from "./mistralgcperror.js";
 
 export type HTTPValidationErrorData = {
   detail?: Array<components.ValidationError> | undefined;

package/packages/mistralai-gcp/src/models/errors/index.ts

@@ -5,7 +5,7 @@
 
 export * from "./httpclienterrors.js";
 export * from "./httpvalidationerror.js";
-export * from "./mistralgoogleclouderror.js";
+export * from "./mistralgcperror.js";
 export * from "./responsevalidationerror.js";
 export * from "./sdkerror.js";
 export * from "./sdkvalidationerror.js";

package/packages/mistralai-gcp/src/models/errors/responsevalidationerror.ts

@@ -4,7 +4,7 @@
  */
 
 import * as z from "zod/v4/core";
-import { MistralGCPError } from "./mistralgoogleclouderror.js";
+import { MistralGCPError } from "./mistralgcperror.js";
 import { formatZodError } from "./sdkvalidationerror.js";
 
 export class ResponseValidationError extends MistralGCPError {

package/packages/mistralai-gcp/src/models/errors/sdkerror.ts

@@ -3,7 +3,7 @@
  * @generated-id: 8e644875c472
  */
 
-import { MistralGCPError } from "./mistralgoogleclouderror.js";
+import { MistralGCPError } from "./mistralgcperror.js";
 
 /** The fallback error class if no more specific error class is matched */
 export class SDKError extends MistralGCPError {

package/packages/mistralai-gcp/src/types/async.ts

@@ -22,16 +22,17 @@ export type APICall =
 
 export class APIPromise<T> implements Promise<T> {
   readonly #promise: Promise<[T, APICall]>;
-  readonly #unwrapped: Promise<T>;
+  #unwrapped: Promise<T> | null;
 
   readonly [Symbol.toStringTag] = "APIPromise";
 
   constructor(p: [T, APICall] | Promise<[T, APICall]>) {
     this.#promise = p instanceof Promise ? p : Promise.resolve(p);
-    this.#unwrapped =
-      p instanceof Promise
-        ? this.#promise.then(([value]) => value)
-        : Promise.resolve(p[0]);
+    this.#unwrapped = p instanceof Promise ? null : Promise.resolve(p[0]);
+  }
+
+  #getUnwrapped(): Promise<T> {
+    return (this.#unwrapped ??= this.#promise.then(([value]) => value));
   }
 
   then<TResult1 = T, TResult2 = never>(
@@ -56,11 +57,11 @@ export class APIPromise<T> implements Promise<T> {
       | null
       | undefined,
   ): Promise<T | TResult> {
-    return this.#unwrapped.catch(onrejected);
+    return this.#getUnwrapped().catch(onrejected);
   }
 
   finally(onfinally?: (() => void) | null | undefined): Promise<T> {
-    return this.#unwrapped.finally(onfinally);
+    return this.#getUnwrapped().finally(onfinally);
   }
 
   $inspect(): Promise<[T, APICall]> {

package/src/extra/structChat.ts

@@ -118,23 +118,43 @@ export function convertToParsedChatCompletionResponse<T extends z.ZodTypeAny>(re
   for (const _choice of response.choices) {
     if (_choice.message === null || typeof _choice.message === 'undefined') {
       parsedChoices.push({..._choice, message: undefined});
+    } else if (
+      _choice.message.content !== null
+      && typeof _choice.message.content !== 'undefined'
+      && !Array.isArray(_choice.message.content)
+    ) {
+      let parsed: z.infer<T> | undefined;
+      try {
+        parsed = responseFormat.safeParse(JSON.parse(_choice.message.content)).data;
+      } catch {
+        // JSON.parse can throw if the model returns malformed JSON
+        // (e.g. truncated output when finish_reason is "length").
+        // Leave parsed as undefined so callers can detect the failure.
+        parsed = undefined;
+      }
+      parsedChoices.push({
+        ..._choice,
+        message: {
+          ..._choice.message,
+          parsed,
+        },
+      });
     } else {
-      if (_choice.message.content !== null && typeof _choice.message.content !== 'undefined' && !Array.isArray(_choice.message.content)) {
+      // content is null, undefined, or an array of content chunks;
+      // preserve the choice without a parsed field.
       parsedChoices.push({
         ..._choice,
         message: {
           ..._choice.message,
-          parsed: responseFormat.safeParse(JSON.parse(_choice.message.content)).data,
-        }
+          parsed: undefined,
+        },
       });
-      }
     }
   }
   return {
     ...response,
     choices: parsedChoices,
   };
-
 }
 
   // Function to convert Zod schema to strict JSON schema

package/src/funcs/agentsComplete.ts

@@ -5,6 +5,7 @@
 
 import { MistralCore } from "../core.js";
 import { encodeJSON } from "../lib/encodings.js";
+import { matchStatusCode } from "../lib/http.js";
 import * as M from "../lib/matchers.js";
 import { compactMap } from "../lib/primitives.js";
 import { safeParse } from "../lib/schemas.js";
@@ -120,7 +121,7 @@ async function $do(
     headers: headers,
     body: body,
     userAgent: client._options.userAgent,
-    timeoutMs: options?.timeoutMs || client._options.timeoutMs || 30000,
+    timeoutMs: options?.timeoutMs || client._options.timeoutMs || 60000,
   }, options);
   if (!requestRes.ok) {
     return [requestRes, { status: "invalid" }];
@@ -129,7 +130,8 @@ async function $do(
 
   const doResult = await client._do(req, {
     context,
-    errorCodes: ["422", "4XX", "5XX"],
+    isErrorStatusCode: (statusCode: number) =>
+      matchStatusCode({ status: statusCode } as Response, ["4XX", "5XX"]),
     retryConfig: context.retryConfig,
     retryCodes: context.retryCodes,
   });

package/src/funcs/agentsStream.ts

@@ -7,6 +7,7 @@ import * as z from "zod/v4";
 import { MistralCore } from "../core.js";
 import { encodeJSON } from "../lib/encodings.js";
 import { EventStream } from "../lib/event-streams.js";
+import { matchStatusCode } from "../lib/http.js";
 import * as M from "../lib/matchers.js";
 import { compactMap } from "../lib/primitives.js";
 import { safeParse } from "../lib/schemas.js";
@@ -126,7 +127,7 @@ async function $do(
     headers: headers,
     body: body,
     userAgent: client._options.userAgent,
-    timeoutMs: options?.timeoutMs || client._options.timeoutMs || 30000,
+    timeoutMs: options?.timeoutMs || client._options.timeoutMs || 60000,
   }, options);
   if (!requestRes.ok) {
     return [requestRes, { status: "invalid" }];
@@ -135,7 +136,8 @@ async function $do(
 
   const doResult = await client._do(req, {
     context,
-    errorCodes: ["422", "4XX", "5XX"],
+    isErrorStatusCode: (statusCode: number) =>
+      matchStatusCode({ status: statusCode } as Response, ["4XX", "5XX"]),
     retryConfig: context.retryConfig,
     retryCodes: context.retryCodes,
   });

package/src/funcs/audioSpeechComplete.ts

@@ -6,6 +6,7 @@
 import { MistralCore } from "../core.js";
 import { encodeJSON } from "../lib/encodings.js";
 import { EventStream } from "../lib/event-streams.js";
+import { matchStatusCode } from "../lib/http.js";
 import * as M from "../lib/matchers.js";
 import { compactMap } from "../lib/primitives.js";
 import { safeParse } from "../lib/schemas.js";
@@ -182,7 +183,7 @@ async function $do(
     headers: headers,
     body: body,
     userAgent: client._options.userAgent,
-    timeoutMs: options?.timeoutMs || client._options.timeoutMs || 30000,
+    timeoutMs: options?.timeoutMs || client._options.timeoutMs || 60000,
   }, options);
   if (!requestRes.ok) {
     return [requestRes, { status: "invalid" }];
@@ -191,7 +192,8 @@ async function $do(
 
   const doResult = await client._do(req, {
     context,
-    errorCodes: ["422", "4XX", "5XX"],
+    isErrorStatusCode: (statusCode: number) =>
+      matchStatusCode({ status: statusCode } as Response, ["4XX", "5XX"]),
     retryConfig: context.retryConfig,
     retryCodes: context.retryCodes,
   });

package/src/funcs/audioTranscriptionsComplete.ts

@@ -4,12 +4,13 @@
  */
 
 import { MistralCore } from "../core.js";
-import { appendForm } from "../lib/encodings.js";
+import { appendForm, normalizeBlob } from "../lib/encodings.js";
 import {
   bytesToBlob,
   getContentTypeFromFileName,
   readableStreamToArrayBuffer,
 } from "../lib/files.js";
+import { matchStatusCode } from "../lib/http.js";
 import * as M from "../lib/matchers.js";
 import { compactMap } from "../lib/primitives.js";
 import { safeParse } from "../lib/schemas.js";
@@ -99,8 +100,9 @@ async function $do(
   }
   if (payload.file !== undefined) {
     if (isBlobLike(payload.file)) {
-      const blob = payload.file;
-      const name = "name" in blob ? (blob.name as string) : undefined;
+      const file = payload.file;
+      const blob = await normalizeBlob(file);
+      const name = "name" in file ? (file.name as string) : undefined;
       appendForm(body, "file", blob, name);
     } else if (isReadableStream(payload.file.content)) {
       const buffer = await readableStreamToArrayBuffer(payload.file.content);
@@ -179,7 +181,7 @@ async function $do(
     headers: headers,
     body: body,
     userAgent: client._options.userAgent,
-    timeoutMs: options?.timeoutMs || client._options.timeoutMs || 30000,
+    timeoutMs: options?.timeoutMs || client._options.timeoutMs || 60000,
   }, options);
   if (!requestRes.ok) {
     return [requestRes, { status: "invalid" }];
@@ -188,7 +190,8 @@ async function $do(
 
   const doResult = await client._do(req, {
     context,
-    errorCodes: ["4XX", "5XX"],
+    isErrorStatusCode: (statusCode: number) =>
+      matchStatusCode({ status: statusCode } as Response, ["4XX", "5XX"]),
     retryConfig: context.retryConfig,
     retryCodes: context.retryCodes,
   });

package/src/funcs/audioTranscriptionsStream.ts

@@ -5,13 +5,14 @@
 
 import * as z from "zod/v4";
 import { MistralCore } from "../core.js";
-import { appendForm } from "../lib/encodings.js";
+import { appendForm, normalizeBlob } from "../lib/encodings.js";
 import { EventStream } from "../lib/event-streams.js";
 import {
   bytesToBlob,
   getContentTypeFromFileName,
   readableStreamToArrayBuffer,
 } from "../lib/files.js";
+import { matchStatusCode } from "../lib/http.js";
 import * as M from "../lib/matchers.js";
 import { compactMap } from "../lib/primitives.js";
 import { safeParse } from "../lib/schemas.js";
@@ -102,8 +103,9 @@ async function $do(
   }
   if (payload.file !== undefined) {
     if (isBlobLike(payload.file)) {
-      const blob = payload.file;
-      const name = "name" in blob ? (blob.name as string) : undefined;
+      const file = payload.file;
+      const blob = await normalizeBlob(file);
+      const name = "name" in file ? (file.name as string) : undefined;
       appendForm(body, "file", blob, name);
     } else if (isReadableStream(payload.file.content)) {
       const buffer = await readableStreamToArrayBuffer(payload.file.content);
@@ -182,7 +184,7 @@ async function $do(
     headers: headers,
     body: body,
     userAgent: client._options.userAgent,
-    timeoutMs: options?.timeoutMs || client._options.timeoutMs || 30000,
+    timeoutMs: options?.timeoutMs || client._options.timeoutMs || 60000,
   }, options);
   if (!requestRes.ok) {
     return [requestRes, { status: "invalid" }];
@@ -191,7 +193,8 @@ async function $do(
 
   const doResult = await client._do(req, {
     context,
-    errorCodes: ["4XX", "5XX"],
+    isErrorStatusCode: (statusCode: number) =>
+      matchStatusCode({ status: statusCode } as Response, ["4XX", "5XX"]),
     retryConfig: context.retryConfig,
     retryCodes: context.retryCodes,
   });