@mistralai/mistralai 2.2.1 → 2.2.6

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/observability/formatting.ts

@@ -0,0 +1,194 @@
+/**
+ * Formatting helpers for converting Mistral API payloads to OTEL GenAI convention formats.
+ *
+ * These are pure functions with no OTEL dependencies - they transform objects to objects
+ * matching the GenAI semantic convention schemas for input/output messages and tool definitions.
+ * The caller is responsible for the final JSON serialization (single JSON.stringify on the whole
+ * collection) before setting span attributes.
+ *
+ * Schemas:
+ * - Input messages: https://github.com/open-telemetry/semantic-conventions/blob/main/docs/gen-ai/gen-ai-input-messages.json
+ * - Output messages: https://github.com/open-telemetry/semantic-conventions/blob/main/docs/gen-ai/gen-ai-output-messages.json
+ * - Tool definitions: https://github.com/Cirilla-zmh/semantic-conventions/blob/cc4d07e7e56b80e9aa5904a3d524c134699da37f/docs/gen-ai/gen-ai-tool-definitions.json
+ */
+
+/**
+ * Convert Mistral message content to OTEL parts array.
+ *
+ * Mistral content is either a string or an array of content chunks.
+ */
+function contentToParts(content: unknown): Array<Record<string, unknown>> {
+  if (content == null) {
+    return [];
+  }
+  if (typeof content === "string") {
+    return [{ type: "text", content }];
+  }
+  if (!Array.isArray(content)) {
+    return [];
+  }
+  // Content chunks array - map known Mistral types to OTEL part types
+  const parts: Array<Record<string, unknown>> = [];
+  for (const chunk of content) {
+    if (typeof chunk === "string") {
+      parts.push({ type: "text", content: chunk });
+    } else if (typeof chunk === "object" && chunk !== null) {
+      const c = chunk as Record<string, unknown>;
+      const chunkType = (c["type"] as string) || "";
+      if (chunkType === "text") {
+        parts.push({ type: "text", content: c["text"] || "" });
+      } else if (chunkType === "thinking") {
+        const thinking = c["thinking"];
+        let contentStr: string;
+        if (Array.isArray(thinking)) {
+          const textParts = thinking
+            .filter(
+              (sub): sub is Record<string, unknown> =>
+                typeof sub === "object" && sub !== null && (sub as Record<string, unknown>)["type"] === "text"
+            )
+            .map((sub) => (sub["text"] as string) || "");
+          contentStr = textParts.join("\n");
+        } else {
+          contentStr = String(thinking || "");
+        }
+        parts.push({ type: "reasoning", content: contentStr });
+      } else if (chunkType === "image_url") {
+        const url = c["image_url"];
+        const uri =
+          typeof url === "object" && url !== null
+            ? ((url as Record<string, unknown>)["url"] as string) || ""
+            : String(url || "");
+        parts.push({ type: "uri", modality: "image", uri });
+      } else {
+        // Catch-all for other content chunk types
+        parts.push({ type: chunkType });
+      }
+    }
+  }
+  return parts;
+}
+
+/**
+ * Convert Mistral tool_calls to OTEL ToolCallRequestPart entries.
+ */
+function toolCallsToParts(toolCalls: unknown): Array<Record<string, unknown>> {
+  if (!toolCalls || !Array.isArray(toolCalls)) {
+    return [];
+  }
+  const parts: Array<Record<string, unknown>> = [];
+  for (const tc of toolCalls) {
+    if (typeof tc !== "object" || tc === null) continue;
+    const tcObj = tc as Record<string, unknown>;
+    const func = (tcObj["function"] as Record<string, unknown>) || {};
+    const part: Record<string, unknown> = {
+      type: "tool_call",
+      name: func["name"] || "",
+    };
+    const tcId = tcObj["id"];
+    if (tcId != null) {
+      part["id"] = tcId;
+    }
+    const args = func["arguments"];
+    if (args != null) {
+      part["arguments"] = args;
+    }
+    parts.push(part);
+  }
+  return parts;
+}
+
+/**
+ * Format a single input message per the OTEL GenAI convention.
+ *
+ * Schema: https://github.com/open-telemetry/semantic-conventions/blob/main/docs/gen-ai/gen-ai-input-messages.json
+ * ChatMessage: {role (required), parts (required), name?}
+ *
+ * Conversation entry objects (e.g. function.result) don't carry a "role"
+ * field - they are detected via their "type" and mapped to the closest OTEL role.
+ */
+export function formatInputMessage(message: Record<string, unknown>): Record<string, unknown> {
+  const entryType = message["type"];
+
+  // Conversation entry: function.result to OTEL tool role
+  if (entryType === "function.result") {
+    const part: Record<string, unknown> = { type: "tool_call_response", response: message["result"] };
+    const toolCallId = message["tool_call_id"];
+    if (toolCallId != null) {
+      part["id"] = toolCallId;
+    }
+    return { role: "tool", parts: [part] };
+  }
+
+  // TODO: may need to handle other types for conversations (e.g. agent handoff)
+
+  const role = (message["role"] as string) || "unknown";
+  const parts: Array<Record<string, unknown>> = [];
+
+  if (role === "tool") {
+    // Tool messages are responses to tool calls
+    const toolPart: Record<string, unknown> = {
+      type: "tool_call_response",
+      response: message["content"],
+    };
+    const toolCallId = message["tool_call_id"];
+    if (toolCallId != null) {
+      toolPart["id"] = toolCallId;
+    }
+    parts.push(toolPart);
+  } else {
+    parts.push(...contentToParts(message["content"]));
+    parts.push(...toolCallsToParts(message["tool_calls"]));
+  }
+
+  return { role, parts };
+}
+
+/**
+ * Format a single output choice/message per the OTEL GenAI convention.
+ *
+ * Schema: https://github.com/open-telemetry/semantic-conventions/blob/main/docs/gen-ai/gen-ai-output-messages.json
+ * OutputMessage: {role (required), parts (required), finish_reason (required), name?}
+ */
+export function formatOutputMessage(choice: Record<string, unknown>): Record<string, unknown> {
+  const message = (choice["message"] as Record<string, unknown>) || {};
+  const parts: Array<Record<string, unknown>> = [];
+  parts.push(...contentToParts(message["content"]));
+  parts.push(...toolCallsToParts(message["tool_calls"]));
+
+  return {
+    role: message["role"] || "assistant",
+    parts,
+    finish_reason: choice["finish_reason"] || "",
+  };
+}
+
+/**
+ * Flatten a Mistral tool definition to the OTEL GenAI convention schema.
+ *
+ * Mistral format: {"type": "function", "function": {"name": ..., "description": ..., "parameters": ...}}
+ * OTEL format: {"type": "function", "name": ..., "description": ..., "parameters": ...}
+ *
+ * Schema, still under review: https://github.com/open-telemetry/semantic-conventions
+ */
+export function formatToolDefinition(tool: Record<string, unknown>): Record<string, unknown> | null {
+  // Early exit conditions: only functions supported for now, and name is required
+  const type = (tool["type"] as string) || "function";
+  const func = tool["function"] as Record<string, unknown> | undefined;
+  if (!func) {
+    return null;
+  }
+  const name = func["name"];
+  if (!name) {
+    return null;
+  }
+  const formatted: Record<string, unknown> = { type, name };
+  const description = func["description"];
+  if (description != null) {
+    formatted["description"] = description;
+  }
+  const parameters = func["parameters"];
+  if (parameters != null) {
+    formatted["parameters"] = parameters;
+  }
+  return formatted;
+}

package/src/extra/observability/index.ts

@@ -0,0 +1,2 @@
+export { registerTracerProvider, traceAsync } from "./otel.js";
+export { MISTRAL_SDK_OTEL_TRACER_NAME } from "./otel.js";