@mistralai/mistralai 2.0.0 → 2.1.0

package/src/models/components/chatcompletionstreamrequest.ts

@@ -11,11 +11,31 @@ import {
   AssistantMessage$Outbound,
   AssistantMessage$outboundSchema,
 } from "./assistantmessage.js";
+import {
+  CodeInterpreterTool,
+  CodeInterpreterTool$Outbound,
+  CodeInterpreterTool$outboundSchema,
+} from "./codeinterpretertool.js";
+import {
+  CustomConnector,
+  CustomConnector$Outbound,
+  CustomConnector$outboundSchema,
+} from "./customconnector.js";
+import {
+  DocumentLibraryTool,
+  DocumentLibraryTool$Outbound,
+  DocumentLibraryTool$outboundSchema,
+} from "./documentlibrarytool.js";
 import {
   GuardrailConfig,
   GuardrailConfig$Outbound,
   GuardrailConfig$outboundSchema,
 } from "./guardrailconfig.js";
+import {
+  ImageGenerationTool,
+  ImageGenerationTool$Outbound,
+  ImageGenerationTool$outboundSchema,
+} from "./imagegenerationtool.js";
 import {
   MistralPromptMode,
   MistralPromptMode$outboundSchema,
@@ -59,6 +79,16 @@ import {
   UserMessage$Outbound,
   UserMessage$outboundSchema,
 } from "./usermessage.js";
+import {
+  WebSearchPremiumTool,
+  WebSearchPremiumTool$Outbound,
+  WebSearchPremiumTool$outboundSchema,
+} from "./websearchpremiumtool.js";
+import {
+  WebSearchTool,
+  WebSearchTool$Outbound,
+  WebSearchTool$outboundSchema,
+} from "./websearchtool.js";
 
 /**
  * Stop generation if this token is detected. Or if one of these tokens is detected when providing an array
@@ -71,6 +101,30 @@ export type ChatCompletionStreamRequestMessage =
   | ToolMessage
   | UserMessage;
 
+export type ChatCompletionStreamRequestTools1 =
+  | DocumentLibraryTool
+  | CustomConnector
+  | Tool
+  | WebSearchTool
+  | WebSearchPremiumTool
+  | CodeInterpreterTool
+  | ImageGenerationTool;
+
+/**
+ * A list of tools the model may call. Use this to provide a list of functions the model may generate JSON inputs for.
+ */
+export type ChatCompletionStreamRequestTools2 =
+  | Array<
+    | DocumentLibraryTool
+    | CustomConnector
+    | Tool
+    | WebSearchTool
+    | WebSearchPremiumTool
+    | CodeInterpreterTool
+    | ImageGenerationTool
+  >
+  | Array<string>;
+
 /**
  * Controls which (if any) tool is called by the model. `none` means the model will not call any tool and instead generates a message. `auto` means the model can pick between generating a message or calling one or more tools. `any` or `required` means the model must call one or more tools. Specifying a particular tool via `{"type": "function", "function": {"name": "my_function"}}` forces the model to call that tool.
  */
@@ -119,7 +173,19 @@ export type ChatCompletionStreamRequest = {
   /**
    * A list of tools the model may call. Use this to provide a list of functions the model may generate JSON inputs for.
    */
-  tools?: Array<Tool> | null | undefined;
+  tools?:
+    | Array<
+      | DocumentLibraryTool
+      | CustomConnector
+      | Tool
+      | WebSearchTool
+      | WebSearchPremiumTool
+      | CodeInterpreterTool
+      | ImageGenerationTool
+    >
+    | Array<string>
+    | null
+    | undefined;
   /**
    * Controls which (if any) tool is called by the model. `none` means the model will not call any tool and instead generates a message. `auto` means the model can pick between generating a message or calling one or more tools. `any` or `required` means the model must call one or more tools. Specifying a particular tool via `{"type": "function", "function": {"name": "my_function"}}` forces the model to call that tool.
    */
@@ -154,6 +220,7 @@ export type ChatCompletionStreamRequest = {
    * Whether to inject a safety prompt before all conversations.
    */
   safePrompt?: boolean | undefined;
+  [additionalProperties: string]: unknown;
 };
 
 /** @internal */
@@ -205,6 +272,82 @@ export function chatCompletionStreamRequestMessageToJSON(
   );
 }
 
+/** @internal */
+export type ChatCompletionStreamRequestTools1$Outbound =
+  | DocumentLibraryTool$Outbound
+  | CustomConnector$Outbound
+  | Tool$Outbound
+  | WebSearchTool$Outbound
+  | WebSearchPremiumTool$Outbound
+  | CodeInterpreterTool$Outbound
+  | ImageGenerationTool$Outbound;
+
+/** @internal */
+export const ChatCompletionStreamRequestTools1$outboundSchema: z.ZodType<
+  ChatCompletionStreamRequestTools1$Outbound,
+  ChatCompletionStreamRequestTools1
+> = smartUnion([
+  DocumentLibraryTool$outboundSchema,
+  CustomConnector$outboundSchema,
+  Tool$outboundSchema,
+  WebSearchTool$outboundSchema,
+  WebSearchPremiumTool$outboundSchema,
+  CodeInterpreterTool$outboundSchema,
+  ImageGenerationTool$outboundSchema,
+]);
+
+export function chatCompletionStreamRequestTools1ToJSON(
+  chatCompletionStreamRequestTools1: ChatCompletionStreamRequestTools1,
+): string {
+  return JSON.stringify(
+    ChatCompletionStreamRequestTools1$outboundSchema.parse(
+      chatCompletionStreamRequestTools1,
+    ),
+  );
+}
+
+/** @internal */
+export type ChatCompletionStreamRequestTools2$Outbound =
+  | Array<
+    | DocumentLibraryTool$Outbound
+    | CustomConnector$Outbound
+    | Tool$Outbound
+    | WebSearchTool$Outbound
+    | WebSearchPremiumTool$Outbound
+    | CodeInterpreterTool$Outbound
+    | ImageGenerationTool$Outbound
+  >
+  | Array<string>;
+
+/** @internal */
+export const ChatCompletionStreamRequestTools2$outboundSchema: z.ZodType<
+  ChatCompletionStreamRequestTools2$Outbound,
+  ChatCompletionStreamRequestTools2
+> = smartUnion([
+  z.array(
+    smartUnion([
+      DocumentLibraryTool$outboundSchema,
+      CustomConnector$outboundSchema,
+      Tool$outboundSchema,
+      WebSearchTool$outboundSchema,
+      WebSearchPremiumTool$outboundSchema,
+      CodeInterpreterTool$outboundSchema,
+      ImageGenerationTool$outboundSchema,
+    ]),
+  ),
+  z.array(z.string()),
+]);
+
+export function chatCompletionStreamRequestTools2ToJSON(
+  chatCompletionStreamRequestTools2: ChatCompletionStreamRequestTools2,
+): string {
+  return JSON.stringify(
+    ChatCompletionStreamRequestTools2$outboundSchema.parse(
+      chatCompletionStreamRequestTools2,
+    ),
+  );
+}
+
 /** @internal */
 export type ChatCompletionStreamRequestToolChoice$Outbound =
   | ToolChoice$Outbound
@@ -243,7 +386,19 @@ export type ChatCompletionStreamRequest$Outbound = {
     | UserMessage$Outbound
   >;
   response_format?: ResponseFormat$Outbound | undefined;
-  tools?: Array<Tool$Outbound> | null | undefined;
+  tools?:
+    | Array<
+      | DocumentLibraryTool$Outbound
+      | CustomConnector$Outbound
+      | Tool$Outbound
+      | WebSearchTool$Outbound
+      | WebSearchPremiumTool$Outbound
+      | CodeInterpreterTool$Outbound
+      | ImageGenerationTool$Outbound
+    >
+    | Array<string>
+    | null
+    | undefined;
   tool_choice?: ToolChoice$Outbound | string | undefined;
   presence_penalty?: number | undefined;
   frequency_penalty?: number | undefined;
@@ -254,6 +409,7 @@ export type ChatCompletionStreamRequest$Outbound = {
   prompt_mode?: string | null | undefined;
   guardrails?: Array<GuardrailConfig$Outbound> | null | undefined;
   safe_prompt?: boolean | undefined;
+  [additionalProperties: string]: unknown;
 };
 
 /** @internal */
@@ -280,7 +436,22 @@ export const ChatCompletionStreamRequest$outboundSchema: z.ZodType<
     ]),
   ),
   responseFormat: ResponseFormat$outboundSchema.optional(),
-  tools: z.nullable(z.array(Tool$outboundSchema)).optional(),
+  tools: z.nullable(
+    smartUnion([
+      z.array(
+        smartUnion([
+          DocumentLibraryTool$outboundSchema,
+          CustomConnector$outboundSchema,
+          Tool$outboundSchema,
+          WebSearchTool$outboundSchema,
+          WebSearchPremiumTool$outboundSchema,
+          CodeInterpreterTool$outboundSchema,
+          ImageGenerationTool$outboundSchema,
+        ]),
+      ),
+      z.array(z.string()),
+    ]),
+  ).optional(),
   toolChoice: smartUnion([
     ToolChoice$outboundSchema,
     ToolChoiceEnum$outboundSchema,
@@ -294,20 +465,22 @@ export const ChatCompletionStreamRequest$outboundSchema: z.ZodType<
   promptMode: z.nullable(MistralPromptMode$outboundSchema).optional(),
   guardrails: z.nullable(z.array(GuardrailConfig$outboundSchema)).optional(),
   safePrompt: z.boolean().optional(),
-}).transform((v) => {
-  return remap$(v, {
-    topP: "top_p",
-    maxTokens: "max_tokens",
-    randomSeed: "random_seed",
-    responseFormat: "response_format",
-    toolChoice: "tool_choice",
-    presencePenalty: "presence_penalty",
-    frequencyPenalty: "frequency_penalty",
-    parallelToolCalls: "parallel_tool_calls",
-    reasoningEffort: "reasoning_effort",
-    promptMode: "prompt_mode",
-    safePrompt: "safe_prompt",
-  });
+}).catchall(z.any()).transform((v) => {
+  return {
+    ...remap$(v, {
+      topP: "top_p",
+      maxTokens: "max_tokens",
+      randomSeed: "random_seed",
+      responseFormat: "response_format",
+      toolChoice: "tool_choice",
+      presencePenalty: "presence_penalty",
+      frequencyPenalty: "frequency_penalty",
+      parallelToolCalls: "parallel_tool_calls",
+      reasoningEffort: "reasoning_effort",
+      promptMode: "prompt_mode",
+      safePrompt: "safe_prompt",
+    }),
+  };
 });
 
 export function chatCompletionStreamRequestToJSON(

package/src/models/components/codeinterpretertool.ts

@@ -6,6 +6,7 @@
 import * as z from "zod/v4";
 import { remap as remap$ } from "../../lib/primitives.js";
 import { safeParse } from "../../lib/schemas.js";
+import { ClosedEnum } from "../../types/enums.js";
 import { Result as SafeParseResult } from "../../types/fp.js";
 import { SDKValidationError } from "../errors/sdkvalidationerror.js";
 import {
@@ -15,18 +16,34 @@ import {
   ToolConfiguration$outboundSchema,
 } from "./toolconfiguration.js";
 
+export const CodeInterpreterToolType = {
+  CodeInterpreter: "code_interpreter",
+} as const;
+export type CodeInterpreterToolType = ClosedEnum<
+  typeof CodeInterpreterToolType
+>;
+
 export type CodeInterpreterTool = {
   toolConfiguration?: ToolConfiguration | null | undefined;
-  type: "code_interpreter";
+  type?: CodeInterpreterToolType | undefined;
 };
 
+/** @internal */
+export const CodeInterpreterToolType$inboundSchema: z.ZodEnum<
+  typeof CodeInterpreterToolType
+> = z.enum(CodeInterpreterToolType);
+/** @internal */
+export const CodeInterpreterToolType$outboundSchema: z.ZodEnum<
+  typeof CodeInterpreterToolType
+> = CodeInterpreterToolType$inboundSchema;
+
 /** @internal */
 export const CodeInterpreterTool$inboundSchema: z.ZodType<
   CodeInterpreterTool,
   unknown
 > = z.object({
   tool_configuration: z.nullable(ToolConfiguration$inboundSchema).optional(),
-  type: z.literal("code_interpreter"),
+  type: CodeInterpreterToolType$inboundSchema.default("code_interpreter"),
 }).transform((v) => {
   return remap$(v, {
     "tool_configuration": "toolConfiguration",
@@ -35,7 +52,7 @@ export const CodeInterpreterTool$inboundSchema: z.ZodType<
 /** @internal */
 export type CodeInterpreterTool$Outbound = {
   tool_configuration?: ToolConfiguration$Outbound | null | undefined;
-  type: "code_interpreter";
+  type: string;
 };
 
 /** @internal */
@@ -44,7 +61,7 @@ export const CodeInterpreterTool$outboundSchema: z.ZodType<
   CodeInterpreterTool
 > = z.object({
   toolConfiguration: z.nullable(ToolConfiguration$outboundSchema).optional(),
-  type: z.literal("code_interpreter"),
+  type: CodeInterpreterToolType$outboundSchema.default("code_interpreter"),
 }).transform((v) => {
   return remap$(v, {
     toolConfiguration: "tool_configuration",

package/src/models/components/completionargs.ts

@@ -20,6 +20,11 @@ import {
   Prediction$Outbound,
   Prediction$outboundSchema,
 } from "./prediction.js";
+import {
+  ReasoningEffort,
+  ReasoningEffort$inboundSchema,
+  ReasoningEffort$outboundSchema,
+} from "./reasoningeffort.js";
 import {
   ResponseFormat,
   ResponseFormat$inboundSchema,
@@ -46,6 +51,7 @@ export type CompletionArgs = {
   prediction?: Prediction | null | undefined;
   responseFormat?: ResponseFormat | null | undefined;
   toolChoice?: ToolChoiceEnum | undefined;
+  reasoningEffort?: ReasoningEffort | null | undefined;
 };
 
 /** @internal */
@@ -61,6 +67,7 @@ export const CompletionArgs$inboundSchema: z.ZodType<CompletionArgs, unknown> =
     prediction: z.nullable(Prediction$inboundSchema).optional(),
     response_format: z.nullable(ResponseFormat$inboundSchema).optional(),
     tool_choice: ToolChoiceEnum$inboundSchema.optional(),
+    reasoning_effort: z.nullable(ReasoningEffort$inboundSchema).optional(),
   }).transform((v) => {
     return remap$(v, {
       "presence_penalty": "presencePenalty",
@@ -70,6 +77,7 @@ export const CompletionArgs$inboundSchema: z.ZodType<CompletionArgs, unknown> =
       "random_seed": "randomSeed",
       "response_format": "responseFormat",
       "tool_choice": "toolChoice",
+      "reasoning_effort": "reasoningEffort",
     });
   });
 /** @internal */
@@ -84,6 +92,7 @@ export type CompletionArgs$Outbound = {
   prediction?: Prediction$Outbound | null | undefined;
   response_format?: ResponseFormat$Outbound | null | undefined;
   tool_choice?: string | undefined;
+  reasoning_effort?: string | null | undefined;
 };
 
 /** @internal */
@@ -101,6 +110,7 @@ export const CompletionArgs$outboundSchema: z.ZodType<
   prediction: z.nullable(Prediction$outboundSchema).optional(),
   responseFormat: z.nullable(ResponseFormat$outboundSchema).optional(),
   toolChoice: ToolChoiceEnum$outboundSchema.optional(),
+  reasoningEffort: z.nullable(ReasoningEffort$outboundSchema).optional(),
 }).transform((v) => {
   return remap$(v, {
     presencePenalty: "presence_penalty",
@@ -110,6 +120,7 @@ export const CompletionArgs$outboundSchema: z.ZodType<
     randomSeed: "random_seed",
     responseFormat: "response_format",
     toolChoice: "tool_choice",
+    reasoningEffort: "reasoning_effort",
   });
 });
 

package/src/models/components/conversationrequest.ts

@@ -47,6 +47,11 @@ import {
   ImageGenerationTool$Outbound,
   ImageGenerationTool$outboundSchema,
 } from "./imagegenerationtool.js";
+import {
+  MetadataDict,
+  MetadataDict$Outbound,
+  MetadataDict$outboundSchema,
+} from "./metadatadict.js";
 import {
   WebSearchPremiumTool,
   WebSearchPremiumTool$Outbound,
@@ -67,13 +72,13 @@ export type ConversationRequestHandoffExecution = ClosedEnum<
 >;
 
 export type ConversationRequestTool =
-  | CodeInterpreterTool
-  | CustomConnector
-  | DocumentLibraryTool
+  | (CodeInterpreterTool & { type: "code_interpreter" })
+  | (CustomConnector & { type: "connector" })
+  | (DocumentLibraryTool & { type: "document_library" })
   | FunctionTool
-  | ImageGenerationTool
-  | WebSearchTool
-  | WebSearchPremiumTool;
+  | (ImageGenerationTool & { type: "image_generation" })
+  | (WebSearchTool & { type: "web_search" })
+  | (WebSearchPremiumTool & { type: "web_search_premium" });
 
 export type ConversationRequestAgentVersion = string | number;
 
@@ -85,13 +90,13 @@ export type ConversationRequest = {
   instructions?: string | null | undefined;
   tools?:
     | Array<
-      | CodeInterpreterTool
-      | CustomConnector
-      | DocumentLibraryTool
+      | (CodeInterpreterTool & { type: "code_interpreter" })
+      | (CustomConnector & { type: "connector" })
+      | (DocumentLibraryTool & { type: "document_library" })
       | FunctionTool
-      | ImageGenerationTool
-      | WebSearchTool
-      | WebSearchPremiumTool
+      | (ImageGenerationTool & { type: "image_generation" })
+      | (WebSearchTool & { type: "web_search" })
+      | (WebSearchPremiumTool & { type: "web_search_premium" })
     >
     | null
     | undefined;
@@ -99,7 +104,7 @@ export type ConversationRequest = {
   guardrails?: Array<GuardrailConfig> | null | undefined;
   name?: string | null | undefined;
   description?: string | null | undefined;
-  metadata?: { [k: string]: any } | null | undefined;
+  metadata?: MetadataDict | null | undefined;
   agentId?: string | null | undefined;
   agentVersion?: string | number | null | undefined;
   model?: string | null | undefined;
@@ -112,26 +117,36 @@ export const ConversationRequestHandoffExecution$outboundSchema: z.ZodEnum<
 
 /** @internal */
 export type ConversationRequestTool$Outbound =
-  | CodeInterpreterTool$Outbound
-  | CustomConnector$Outbound
-  | DocumentLibraryTool$Outbound
+  | (CodeInterpreterTool$Outbound & { type: "code_interpreter" })
+  | (CustomConnector$Outbound & { type: "connector" })
+  | (DocumentLibraryTool$Outbound & { type: "document_library" })
   | FunctionTool$Outbound
-  | ImageGenerationTool$Outbound
-  | WebSearchTool$Outbound
-  | WebSearchPremiumTool$Outbound;
+  | (ImageGenerationTool$Outbound & { type: "image_generation" })
+  | (WebSearchTool$Outbound & { type: "web_search" })
+  | (WebSearchPremiumTool$Outbound & { type: "web_search_premium" });
 
 /** @internal */
 export const ConversationRequestTool$outboundSchema: z.ZodType<
   ConversationRequestTool$Outbound,
   ConversationRequestTool
 > = z.union([
-  CodeInterpreterTool$outboundSchema,
-  CustomConnector$outboundSchema,
-  DocumentLibraryTool$outboundSchema,
+  CodeInterpreterTool$outboundSchema.and(
+    z.object({ type: z.literal("code_interpreter") }),
+  ),
+  CustomConnector$outboundSchema.and(
+    z.object({ type: z.literal("connector") }),
+  ),
+  DocumentLibraryTool$outboundSchema.and(
+    z.object({ type: z.literal("document_library") }),
+  ),
   FunctionTool$outboundSchema,
-  ImageGenerationTool$outboundSchema,
-  WebSearchTool$outboundSchema,
-  WebSearchPremiumTool$outboundSchema,
+  ImageGenerationTool$outboundSchema.and(
+    z.object({ type: z.literal("image_generation") }),
+  ),
+  WebSearchTool$outboundSchema.and(z.object({ type: z.literal("web_search") })),
+  WebSearchPremiumTool$outboundSchema.and(
+    z.object({ type: z.literal("web_search_premium") }),
+  ),
 ]);
 
 export function conversationRequestToolToJSON(
@@ -170,13 +185,13 @@ export type ConversationRequest$Outbound = {
   instructions?: string | null | undefined;
   tools?:
     | Array<
-      | CodeInterpreterTool$Outbound
-      | CustomConnector$Outbound
-      | DocumentLibraryTool$Outbound
+      | (CodeInterpreterTool$Outbound & { type: "code_interpreter" })
+      | (CustomConnector$Outbound & { type: "connector" })
+      | (DocumentLibraryTool$Outbound & { type: "document_library" })
       | FunctionTool$Outbound
-      | ImageGenerationTool$Outbound
-      | WebSearchTool$Outbound
-      | WebSearchPremiumTool$Outbound
+      | (ImageGenerationTool$Outbound & { type: "image_generation" })
+      | (WebSearchTool$Outbound & { type: "web_search" })
+      | (WebSearchPremiumTool$Outbound & { type: "web_search_premium" })
     >
     | null
     | undefined;
@@ -184,7 +199,7 @@ export type ConversationRequest$Outbound = {
   guardrails?: Array<GuardrailConfig$Outbound> | null | undefined;
   name?: string | null | undefined;
   description?: string | null | undefined;
-  metadata?: { [k: string]: any } | null | undefined;
+  metadata?: MetadataDict$Outbound | null | undefined;
   agent_id?: string | null | undefined;
   agent_version?: string | number | null | undefined;
   model?: string | null | undefined;
@@ -205,13 +220,25 @@ export const ConversationRequest$outboundSchema: z.ZodType<
   tools: z.nullable(
     z.array(
       z.union([
-        CodeInterpreterTool$outboundSchema,
-        CustomConnector$outboundSchema,
-        DocumentLibraryTool$outboundSchema,
+        CodeInterpreterTool$outboundSchema.and(
+          z.object({ type: z.literal("code_interpreter") }),
+        ),
+        CustomConnector$outboundSchema.and(
+          z.object({ type: z.literal("connector") }),
+        ),
+        DocumentLibraryTool$outboundSchema.and(
+          z.object({ type: z.literal("document_library") }),
+        ),
         FunctionTool$outboundSchema,
-        ImageGenerationTool$outboundSchema,
-        WebSearchTool$outboundSchema,
-        WebSearchPremiumTool$outboundSchema,
+        ImageGenerationTool$outboundSchema.and(
+          z.object({ type: z.literal("image_generation") }),
+        ),
+        WebSearchTool$outboundSchema.and(
+          z.object({ type: z.literal("web_search") }),
+        ),
+        WebSearchPremiumTool$outboundSchema.and(
+          z.object({ type: z.literal("web_search_premium") }),
+        ),
       ]),
     ),
   ).optional(),
@@ -219,7 +246,7 @@ export const ConversationRequest$outboundSchema: z.ZodType<
   guardrails: z.nullable(z.array(GuardrailConfig$outboundSchema)).optional(),
   name: z.nullable(z.string()).optional(),
   description: z.nullable(z.string()).optional(),
-  metadata: z.nullable(z.record(z.string(), z.any())).optional(),
+  metadata: z.nullable(MetadataDict$outboundSchema).optional(),
   agentId: z.nullable(z.string()).optional(),
   agentVersion: z.nullable(smartUnion([z.string(), z.int()])).optional(),
   model: z.nullable(z.string()).optional(),

package/src/models/components/conversationrestartrequest.ts

@@ -22,6 +22,11 @@ import {
   GuardrailConfig$Outbound,
   GuardrailConfig$outboundSchema,
 } from "./guardrailconfig.js";
+import {
+  MetadataDict,
+  MetadataDict$Outbound,
+  MetadataDict$outboundSchema,
+} from "./metadatadict.js";
 
 export const ConversationRestartRequestHandoffExecution = {
   Client: "client",
@@ -55,7 +60,7 @@ export type ConversationRestartRequest = {
   /**
    * Custom metadata for the conversation.
    */
-  metadata?: { [k: string]: any } | null | undefined;
+  metadata?: MetadataDict | null | undefined;
   fromEntryId: string;
   /**
    * Specific version of the agent to use when restarting. If not provided, uses the current version.
@@ -97,7 +102,7 @@ export type ConversationRestartRequest$Outbound = {
   handoff_execution: string;
   completion_args?: CompletionArgs$Outbound | undefined;
   guardrails?: Array<GuardrailConfig$Outbound> | null | undefined;
-  metadata?: { [k: string]: any } | null | undefined;
+  metadata?: MetadataDict$Outbound | null | undefined;
   from_entry_id: string;
   agent_version?: string | number | null | undefined;
 };
@@ -114,7 +119,7 @@ export const ConversationRestartRequest$outboundSchema: z.ZodType<
     .default("server"),
   completionArgs: CompletionArgs$outboundSchema.optional(),
   guardrails: z.nullable(z.array(GuardrailConfig$outboundSchema)).optional(),
-  metadata: z.nullable(z.record(z.string(), z.any())).optional(),
+  metadata: z.nullable(MetadataDict$outboundSchema).optional(),
   fromEntryId: z.string(),
   agentVersion: z.nullable(smartUnion([z.string(), z.int()])).optional(),
 }).transform((v) => {

package/src/models/components/conversationrestartstreamrequest.ts

@@ -22,6 +22,11 @@ import {
   GuardrailConfig$Outbound,
   GuardrailConfig$outboundSchema,
 } from "./guardrailconfig.js";
+import {
+  MetadataDict,
+  MetadataDict$Outbound,
+  MetadataDict$outboundSchema,
+} from "./metadatadict.js";
 
 export const ConversationRestartStreamRequestHandoffExecution = {
   Client: "client",
@@ -57,7 +62,7 @@ export type ConversationRestartStreamRequest = {
   /**
    * Custom metadata for the conversation.
    */
-  metadata?: { [k: string]: any } | null | undefined;
+  metadata?: MetadataDict | null | undefined;
   fromEntryId: string;
   /**
    * Specific version of the agent to use when restarting. If not provided, uses the current version.
@@ -102,7 +107,7 @@ export type ConversationRestartStreamRequest$Outbound = {
   handoff_execution: string;
   completion_args?: CompletionArgs$Outbound | undefined;
   guardrails?: Array<GuardrailConfig$Outbound> | null | undefined;
-  metadata?: { [k: string]: any } | null | undefined;
+  metadata?: MetadataDict$Outbound | null | undefined;
   from_entry_id: string;
   agent_version?: string | number | null | undefined;
 };
@@ -121,7 +126,7 @@ export const ConversationRestartStreamRequest$outboundSchema: z.ZodType<
     ),
   completionArgs: CompletionArgs$outboundSchema.optional(),
   guardrails: z.nullable(z.array(GuardrailConfig$outboundSchema)).optional(),
-  metadata: z.nullable(z.record(z.string(), z.any())).optional(),
+  metadata: z.nullable(MetadataDict$outboundSchema).optional(),
   fromEntryId: z.string(),
   agentVersion: z.nullable(smartUnion([z.string(), z.int()])).optional(),
 }).transform((v) => {