@mistralai/mistralai 2.2.1 → 2.2.6

package/packages/mistralai-azure/src/models/components/ocrconfidencescore.ts

@@ -0,0 +1,52 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ * @generated-id: f0ae261a8be7
+ */
+
+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 { SDKValidationError } from "../errors/sdkvalidationerror.js";
+
+/**
+ * Confidence score for a token or word in OCR output.
+ */
+export type OCRConfidenceScore = {
+  /**
+   * The word or text segment
+   */
+  text: string;
+  /**
+   * Confidence score (0-1)
+   */
+  confidence: number;
+  /**
+   * Start index of the text in the page markdown string
+   */
+  startIndex: number;
+};
+
+/** @internal */
+export const OCRConfidenceScore$inboundSchema: z.ZodType<
+  OCRConfidenceScore,
+  unknown
+> = z.object({
+  text: z.string(),
+  confidence: z.number(),
+  start_index: z.int(),
+}).transform((v) => {
+  return remap$(v, {
+    "start_index": "startIndex",
+  });
+});
+
+export function ocrConfidenceScoreFromJSON(
+  jsonString: string,
+): SafeParseResult<OCRConfidenceScore, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => OCRConfidenceScore$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'OCRConfidenceScore' from JSON`,
+  );
+}

package/packages/mistralai-azure/src/models/components/ocrpageconfidencescores.ts

@@ -0,0 +1,66 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ * @generated-id: decf753e462c
+ */
+
+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 { SDKValidationError } from "../errors/sdkvalidationerror.js";
+import {
+  OCRConfidenceScore,
+  OCRConfidenceScore$inboundSchema,
+} from "./ocrconfidencescore.js";
+
+/**
+ * Confidence scores for an OCR page at various granularities.
+ *
+ * @remarks
+ *
+ * Note on page-level stats:
+ * - For 'page' granularity: average/minimum are computed from per-token exp(logprob).
+ * - For 'word' granularity: average/minimum are computed from per-word confidence,
+ *   where each word's confidence is exp(mean(token_logprobs)) — a geometric mean
+ *   over the word's subword tokens.
+ */
+export type OCRPageConfidenceScores = {
+  /**
+   * Word-level confidence scores (populated only for 'word' granularity)
+   */
+  wordConfidenceScores?: Array<OCRConfidenceScore> | undefined;
+  /**
+   * Average confidence score for the page
+   */
+  averagePageConfidenceScore: number;
+  /**
+   * Minimum confidence score for the page
+   */
+  minimumPageConfidenceScore: number;
+};
+
+/** @internal */
+export const OCRPageConfidenceScores$inboundSchema: z.ZodType<
+  OCRPageConfidenceScores,
+  unknown
+> = z.object({
+  word_confidence_scores: z.array(OCRConfidenceScore$inboundSchema).optional(),
+  average_page_confidence_score: z.number(),
+  minimum_page_confidence_score: z.number(),
+}).transform((v) => {
+  return remap$(v, {
+    "word_confidence_scores": "wordConfidenceScores",
+    "average_page_confidence_score": "averagePageConfidenceScore",
+    "minimum_page_confidence_score": "minimumPageConfidenceScore",
+  });
+});
+
+export function ocrPageConfidenceScoresFromJSON(
+  jsonString: string,
+): SafeParseResult<OCRPageConfidenceScores, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => OCRPageConfidenceScores$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'OCRPageConfidenceScores' from JSON`,
+  );
+}

package/packages/mistralai-azure/src/models/components/ocrpageobject.ts

@@ -4,6 +4,7 @@
  */
 
 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 { SDKValidationError } from "../errors/sdkvalidationerror.js";
@@ -11,6 +12,10 @@ import {
   OCRImageObject,
   OCRImageObject$inboundSchema,
 } from "./ocrimageobject.js";
+import {
+  OCRPageConfidenceScores,
+  OCRPageConfidenceScores$inboundSchema,
+} from "./ocrpageconfidencescores.js";
 import {
   OCRPageDimensions,
   OCRPageDimensions$inboundSchema,
@@ -53,6 +58,10 @@ export type OCRPageObject = {
    * The dimensions of the PDF Page's screenshot image
    */
   dimensions: OCRPageDimensions | null;
+  /**
+   * Confidence scores for the OCR page (populated when confidence_scores_granularity is set)
+   */
+  confidenceScores?: OCRPageConfidenceScores | null | undefined;
 };
 
 /** @internal */
@@ -66,6 +75,12 @@ export const OCRPageObject$inboundSchema: z.ZodType<OCRPageObject, unknown> = z
     header: z.nullable(z.string()).optional(),
     footer: z.nullable(z.string()).optional(),
     dimensions: z.nullable(OCRPageDimensions$inboundSchema),
+    confidence_scores: z.nullable(OCRPageConfidenceScores$inboundSchema)
+      .optional(),
+  }).transform((v) => {
+    return remap$(v, {
+      "confidence_scores": "confidenceScores",
+    });
   });
 
 export function ocrPageObjectFromJSON(

package/packages/mistralai-azure/src/models/components/ocrrequest.ts

@@ -33,23 +33,35 @@ import {
  */
 export type Document = FileChunk | DocumentURLChunk | ImageURLChunk;
 
+/**
+ * Specific pages to process. Accepts a list of integers or a string of comma-separated numbers and ranges (e.g. '0,1,2' or '0-5' or '0,2-4'). Page numbers start from 0.
+ */
+export type Pages = string | Array<number>;
+
 export const TableFormat = {
   Markdown: "markdown",
   Html: "html",
 } as const;
 export type TableFormat = ClosedEnum<typeof TableFormat>;
 
+export const ConfidenceScoresGranularity = {
+  Word: "word",
+  Page: "page",
+} as const;
+export type ConfidenceScoresGranularity = ClosedEnum<
+  typeof ConfidenceScoresGranularity
+>;
+
 export type OCRRequest = {
   model: string | null;
-  id?: string | undefined;
   /**
    * Document to run OCR on
    */
   document: FileChunk | DocumentURLChunk | ImageURLChunk;
   /**
-   * Specific pages user wants to process in various formats: single number, range, or list of both. Starts from 0
+   * Specific pages to process. Accepts a list of integers or a string of comma-separated numbers and ranges (e.g. '0,1,2' or '0-5' or '0,2-4'). Page numbers start from 0.
    */
-  pages?: Array<number> | null | undefined;
+  pages?: string | Array<number> | null | undefined;
   /**
    * Include image URLs in response
    */
@@ -77,6 +89,10 @@ export type OCRRequest = {
   tableFormat?: TableFormat | null | undefined;
   extractHeader?: boolean | undefined;
   extractFooter?: boolean | undefined;
+  /**
+   * Granularity for confidence scores: 'word' (per-word scores) or 'page' (aggregate only). Defaults to None (no confidence scores) to keep response payload small.
+   */
+  confidenceScoresGranularity?: ConfidenceScoresGranularity | null | undefined;
 };
 
 /** @internal */
@@ -97,20 +113,35 @@ export function documentToJSON(document: Document): string {
   return JSON.stringify(Document$outboundSchema.parse(document));
 }
 
+/** @internal */
+export type Pages$Outbound = string | Array<number>;
+
+/** @internal */
+export const Pages$outboundSchema: z.ZodType<Pages$Outbound, Pages> =
+  smartUnion([z.string(), z.array(z.int())]);
+
+export function pagesToJSON(pages: Pages): string {
+  return JSON.stringify(Pages$outboundSchema.parse(pages));
+}
+
 /** @internal */
 export const TableFormat$outboundSchema: z.ZodEnum<typeof TableFormat> = z.enum(
   TableFormat,
 );
 
+/** @internal */
+export const ConfidenceScoresGranularity$outboundSchema: z.ZodEnum<
+  typeof ConfidenceScoresGranularity
+> = z.enum(ConfidenceScoresGranularity);
+
 /** @internal */
 export type OCRRequest$Outbound = {
   model: string | null;
-  id?: string | undefined;
   document:
     | FileChunk$Outbound
     | DocumentURLChunk$Outbound
     | ImageURLChunk$Outbound;
-  pages?: Array<number> | null | undefined;
+  pages?: string | Array<number> | null | undefined;
   include_image_base64?: boolean | null | undefined;
   image_limit?: number | null | undefined;
   image_min_size?: number | null | undefined;
@@ -120,6 +151,7 @@ export type OCRRequest$Outbound = {
   table_format?: string | null | undefined;
   extract_header?: boolean | undefined;
   extract_footer?: boolean | undefined;
+  confidence_scores_granularity?: string | null | undefined;
 };
 
 /** @internal */
@@ -128,13 +160,12 @@ export const OCRRequest$outboundSchema: z.ZodType<
   OCRRequest
 > = z.object({
   model: z.nullable(z.string()),
-  id: z.string().optional(),
   document: smartUnion([
     FileChunk$outboundSchema,
     DocumentURLChunk$outboundSchema,
     ImageURLChunk$outboundSchema,
   ]),
-  pages: z.nullable(z.array(z.int())).optional(),
+  pages: z.nullable(smartUnion([z.string(), z.array(z.int())])).optional(),
   includeImageBase64: z.nullable(z.boolean()).optional(),
   imageLimit: z.nullable(z.int()).optional(),
   imageMinSize: z.nullable(z.int()).optional(),
@@ -145,6 +176,9 @@ export const OCRRequest$outboundSchema: z.ZodType<
   tableFormat: z.nullable(TableFormat$outboundSchema).optional(),
   extractHeader: z.boolean().optional(),
   extractFooter: z.boolean().optional(),
+  confidenceScoresGranularity: z.nullable(
+    ConfidenceScoresGranularity$outboundSchema,
+  ).optional(),
 }).transform((v) => {
   return remap$(v, {
     includeImageBase64: "include_image_base64",
@@ -156,6 +190,7 @@ export const OCRRequest$outboundSchema: z.ZodType<
     tableFormat: "table_format",
     extractHeader: "extract_header",
     extractFooter: "extract_footer",
+    confidenceScoresGranularity: "confidence_scores_granularity",
   });
 });
 

package/packages/mistralai-azure/src/models/components/ocrtableobject.ts

@@ -4,11 +4,16 @@
  */
 
 import * as z from "zod/v4";
+import { remap as remap$ } from "../../lib/primitives.js";
 import { safeParse } from "../../lib/schemas.js";
 import * as openEnums from "../../types/enums.js";
 import { OpenEnum } from "../../types/enums.js";
 import { Result as SafeParseResult } from "../../types/fp.js";
 import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+import {
+  OCRConfidenceScore,
+  OCRConfidenceScore$inboundSchema,
+} from "./ocrconfidencescore.js";
 
 /**
  * Format of the table
@@ -35,6 +40,10 @@ export type OCRTableObject = {
    * Format of the table
    */
   format: Format;
+  /**
+   * Per-word confidence scores for the table content. Returned when confidence_scores_granularity is set to 'word'.
+   */
+  wordConfidenceScores?: Array<OCRConfidenceScore> | null | undefined;
 };
 
 /** @internal */
@@ -47,6 +56,13 @@ export const OCRTableObject$inboundSchema: z.ZodType<OCRTableObject, unknown> =
     id: z.string(),
     content: z.string(),
     format: Format$inboundSchema,
+    word_confidence_scores: z.nullable(
+      z.array(OCRConfidenceScore$inboundSchema),
+    ).optional(),
+  }).transform((v) => {
+    return remap$(v, {
+      "word_confidence_scores": "wordConfidenceScores",
+    });
   });
 
 export function ocrTableObjectFromJSON(

package/packages/mistralai-azure/src/models/components/reasoningeffort.ts

@@ -0,0 +1,17 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ * @generated-id: 0c909c55bb1f
+ */
+
+import * as z from "zod/v4";
+import { ClosedEnum } from "../../types/enums.js";
+
+export const ReasoningEffort = {
+  None: "none",
+  High: "high",
+} as const;
+export type ReasoningEffort = ClosedEnum<typeof ReasoningEffort>;
+
+/** @internal */
+export const ReasoningEffort$outboundSchema: z.ZodEnum<typeof ReasoningEffort> =
+  z.enum(ReasoningEffort);

package/packages/mistralai-azure/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-azure/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-azure/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-azure/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-azure/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-azure/src/models/components/toolconfiguration.ts

@@ -0,0 +1,42 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ * @generated-id: 17d10c0ca54e
+ */
+
+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-azure/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-azure/src/models/components/websearchpremiumtool.ts

@@ -0,0 +1,44 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ * @generated-id: a83aca99f21a
+ */
+
+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),
+  );
+}