@mistralai/mistralai 2.0.0 → 2.1.0

package/src/models/components/agentscompletionstreamrequest.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,27 @@ export type AgentsCompletionStreamRequestMessage =
   | ToolMessage
   | UserMessage;
 
+export type AgentsCompletionStreamRequestTools1 =
+  | DocumentLibraryTool
+  | CustomConnector
+  | Tool
+  | WebSearchTool
+  | WebSearchPremiumTool
+  | CodeInterpreterTool
+  | ImageGenerationTool;
+
+export type AgentsCompletionStreamRequestTools2 =
+  | Array<
+    | DocumentLibraryTool
+    | CustomConnector
+    | Tool
+    | WebSearchTool
+    | WebSearchPremiumTool
+    | CodeInterpreterTool
+    | ImageGenerationTool
+  >
+  | Array<string>;
+
 export type AgentsCompletionStreamRequestToolChoice =
   | ToolChoice
   | ToolChoiceEnum;
@@ -103,7 +154,19 @@ export type AgentsCompletionStreamRequest = {
    * Specify the format that the model must output. By default it will use `{ "type": "text" }`. Setting to `{ "type": "json_object" }` enables JSON mode, which guarantees the message the model generates is in JSON. When using JSON mode you MUST also instruct the model to produce JSON yourself with a system or a user message. Setting to `{ "type": "json_schema" }` enables JSON schema mode, which guarantees the message the model generates is in JSON and follows the schema you provide.
    */
   responseFormat?: ResponseFormat | undefined;
-  tools?: Array<Tool> | null | undefined;
+  tools?:
+    | Array<
+      | DocumentLibraryTool
+      | CustomConnector
+      | Tool
+      | WebSearchTool
+      | WebSearchPremiumTool
+      | CodeInterpreterTool
+      | ImageGenerationTool
+    >
+    | Array<string>
+    | null
+    | undefined;
   toolChoice?: ToolChoice | ToolChoiceEnum | undefined;
   /**
    * The `presence_penalty` determines how much the model penalizes the repetition of words or phrases. A higher presence penalty encourages the model to use a wider variety of words and phrases, making the output more diverse and creative.
@@ -132,6 +195,7 @@ export type AgentsCompletionStreamRequest = {
    * The ID of the agent to use for this completion.
    */
   agentId: string;
+  [additionalProperties: string]: unknown;
 };
 
 /** @internal */
@@ -183,6 +247,82 @@ export function agentsCompletionStreamRequestMessageToJSON(
   );
 }
 
+/** @internal */
+export type AgentsCompletionStreamRequestTools1$Outbound =
+  | DocumentLibraryTool$Outbound
+  | CustomConnector$Outbound
+  | Tool$Outbound
+  | WebSearchTool$Outbound
+  | WebSearchPremiumTool$Outbound
+  | CodeInterpreterTool$Outbound
+  | ImageGenerationTool$Outbound;
+
+/** @internal */
+export const AgentsCompletionStreamRequestTools1$outboundSchema: z.ZodType<
+  AgentsCompletionStreamRequestTools1$Outbound,
+  AgentsCompletionStreamRequestTools1
+> = smartUnion([
+  DocumentLibraryTool$outboundSchema,
+  CustomConnector$outboundSchema,
+  Tool$outboundSchema,
+  WebSearchTool$outboundSchema,
+  WebSearchPremiumTool$outboundSchema,
+  CodeInterpreterTool$outboundSchema,
+  ImageGenerationTool$outboundSchema,
+]);
+
+export function agentsCompletionStreamRequestTools1ToJSON(
+  agentsCompletionStreamRequestTools1: AgentsCompletionStreamRequestTools1,
+): string {
+  return JSON.stringify(
+    AgentsCompletionStreamRequestTools1$outboundSchema.parse(
+      agentsCompletionStreamRequestTools1,
+    ),
+  );
+}
+
+/** @internal */
+export type AgentsCompletionStreamRequestTools2$Outbound =
+  | Array<
+    | DocumentLibraryTool$Outbound
+    | CustomConnector$Outbound
+    | Tool$Outbound
+    | WebSearchTool$Outbound
+    | WebSearchPremiumTool$Outbound
+    | CodeInterpreterTool$Outbound
+    | ImageGenerationTool$Outbound
+  >
+  | Array<string>;
+
+/** @internal */
+export const AgentsCompletionStreamRequestTools2$outboundSchema: z.ZodType<
+  AgentsCompletionStreamRequestTools2$Outbound,
+  AgentsCompletionStreamRequestTools2
+> = smartUnion([
+  z.array(
+    smartUnion([
+      DocumentLibraryTool$outboundSchema,
+      CustomConnector$outboundSchema,
+      Tool$outboundSchema,
+      WebSearchTool$outboundSchema,
+      WebSearchPremiumTool$outboundSchema,
+      CodeInterpreterTool$outboundSchema,
+      ImageGenerationTool$outboundSchema,
+    ]),
+  ),
+  z.array(z.string()),
+]);
+
+export function agentsCompletionStreamRequestTools2ToJSON(
+  agentsCompletionStreamRequestTools2: AgentsCompletionStreamRequestTools2,
+): string {
+  return JSON.stringify(
+    AgentsCompletionStreamRequestTools2$outboundSchema.parse(
+      agentsCompletionStreamRequestTools2,
+    ),
+  );
+}
+
 /** @internal */
 export type AgentsCompletionStreamRequestToolChoice$Outbound =
   | ToolChoice$Outbound
@@ -219,7 +359,19 @@ export type AgentsCompletionStreamRequest$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;
@@ -230,6 +382,7 @@ export type AgentsCompletionStreamRequest$Outbound = {
   prompt_mode?: string | null | undefined;
   guardrails?: Array<GuardrailConfig$Outbound> | null | undefined;
   agent_id: string;
+  [additionalProperties: string]: unknown;
 };
 
 /** @internal */
@@ -253,7 +406,22 @@ export const AgentsCompletionStreamRequest$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,
@@ -267,19 +435,21 @@ export const AgentsCompletionStreamRequest$outboundSchema: z.ZodType<
   promptMode: z.nullable(MistralPromptMode$outboundSchema).optional(),
   guardrails: z.nullable(z.array(GuardrailConfig$outboundSchema)).optional(),
   agentId: z.string(),
-}).transform((v) => {
-  return remap$(v, {
-    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",
-    agentId: "agent_id",
-  });
+}).catchall(z.any()).transform((v) => {
+  return {
+    ...remap$(v, {
+      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",
+      agentId: "agent_id",
+    }),
+  };
 });
 
 export function agentsCompletionStreamRequestToJSON(

package/src/models/components/basefielddefinition.ts

@@ -11,7 +11,7 @@ import { OpenEnum } from "../../types/enums.js";
 import { Result as SafeParseResult } from "../../types/fp.js";
 import { SDKValidationError } from "../errors/sdkvalidationerror.js";
 
-export const TypeEnum = {
+export const BaseFieldDefinitionType = {
   Enum: "ENUM",
   Text: "TEXT",
   Int: "INT",
@@ -20,7 +20,7 @@ export const TypeEnum = {
   Timestamp: "TIMESTAMP",
   Array: "ARRAY",
 } as const;
-export type TypeEnum = OpenEnum<typeof TypeEnum>;
+export type BaseFieldDefinitionType = OpenEnum<typeof BaseFieldDefinitionType>;
 
 export const SupportedOperator = {
   Lt: "lt",
@@ -48,14 +48,16 @@ export type SupportedOperator = OpenEnum<typeof SupportedOperator>;
 export type BaseFieldDefinition = {
   name: string;
   label: string;
-  type: TypeEnum;
+  type: BaseFieldDefinitionType;
   group?: string | null | undefined;
   supportedOperators: Array<SupportedOperator>;
 };
 
 /** @internal */
-export const TypeEnum$inboundSchema: z.ZodType<TypeEnum, unknown> = openEnums
-  .inboundSchema(TypeEnum);
+export const BaseFieldDefinitionType$inboundSchema: z.ZodType<
+  BaseFieldDefinitionType,
+  unknown
+> = openEnums.inboundSchema(BaseFieldDefinitionType);
 
 /** @internal */
 export const SupportedOperator$inboundSchema: z.ZodType<
@@ -70,7 +72,7 @@ export const BaseFieldDefinition$inboundSchema: z.ZodType<
 > = z.object({
   name: z.string(),
   label: z.string(),
-  type: TypeEnum$inboundSchema,
+  type: BaseFieldDefinitionType$inboundSchema,
   group: z.nullable(z.string()).optional(),
   supported_operators: z.array(SupportedOperator$inboundSchema),
 }).transform((v) => {

package/src/models/components/chatcompletionrequest.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 ChatCompletionRequestMessage =
   | ToolMessage
   | UserMessage;
 
+export type ChatCompletionRequestTools1 =
+  | 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 ChatCompletionRequestTools2 =
+  | 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.
  */
@@ -122,7 +176,19 @@ export type ChatCompletionRequest = {
   /**
    * 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.
    */
@@ -157,6 +223,7 @@ export type ChatCompletionRequest = {
    * Whether to inject a safety prompt before all conversations.
    */
   safePrompt?: boolean | undefined;
+  [additionalProperties: string]: unknown;
 };
 
 /** @internal */
@@ -206,6 +273,82 @@ export function chatCompletionRequestMessageToJSON(
   );
 }
 
+/** @internal */
+export type ChatCompletionRequestTools1$Outbound =
+  | DocumentLibraryTool$Outbound
+  | CustomConnector$Outbound
+  | Tool$Outbound
+  | WebSearchTool$Outbound
+  | WebSearchPremiumTool$Outbound
+  | CodeInterpreterTool$Outbound
+  | ImageGenerationTool$Outbound;
+
+/** @internal */
+export const ChatCompletionRequestTools1$outboundSchema: z.ZodType<
+  ChatCompletionRequestTools1$Outbound,
+  ChatCompletionRequestTools1
+> = smartUnion([
+  DocumentLibraryTool$outboundSchema,
+  CustomConnector$outboundSchema,
+  Tool$outboundSchema,
+  WebSearchTool$outboundSchema,
+  WebSearchPremiumTool$outboundSchema,
+  CodeInterpreterTool$outboundSchema,
+  ImageGenerationTool$outboundSchema,
+]);
+
+export function chatCompletionRequestTools1ToJSON(
+  chatCompletionRequestTools1: ChatCompletionRequestTools1,
+): string {
+  return JSON.stringify(
+    ChatCompletionRequestTools1$outboundSchema.parse(
+      chatCompletionRequestTools1,
+    ),
+  );
+}
+
+/** @internal */
+export type ChatCompletionRequestTools2$Outbound =
+  | Array<
+    | DocumentLibraryTool$Outbound
+    | CustomConnector$Outbound
+    | Tool$Outbound
+    | WebSearchTool$Outbound
+    | WebSearchPremiumTool$Outbound
+    | CodeInterpreterTool$Outbound
+    | ImageGenerationTool$Outbound
+  >
+  | Array<string>;
+
+/** @internal */
+export const ChatCompletionRequestTools2$outboundSchema: z.ZodType<
+  ChatCompletionRequestTools2$Outbound,
+  ChatCompletionRequestTools2
+> = smartUnion([
+  z.array(
+    smartUnion([
+      DocumentLibraryTool$outboundSchema,
+      CustomConnector$outboundSchema,
+      Tool$outboundSchema,
+      WebSearchTool$outboundSchema,
+      WebSearchPremiumTool$outboundSchema,
+      CodeInterpreterTool$outboundSchema,
+      ImageGenerationTool$outboundSchema,
+    ]),
+  ),
+  z.array(z.string()),
+]);
+
+export function chatCompletionRequestTools2ToJSON(
+  chatCompletionRequestTools2: ChatCompletionRequestTools2,
+): string {
+  return JSON.stringify(
+    ChatCompletionRequestTools2$outboundSchema.parse(
+      chatCompletionRequestTools2,
+    ),
+  );
+}
+
 /** @internal */
 export type ChatCompletionRequestToolChoice$Outbound =
   | ToolChoice$Outbound
@@ -244,7 +387,19 @@ export type ChatCompletionRequest$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;
@@ -255,6 +410,7 @@ export type ChatCompletionRequest$Outbound = {
   prompt_mode?: string | null | undefined;
   guardrails?: Array<GuardrailConfig$Outbound> | null | undefined;
   safe_prompt?: boolean | undefined;
+  [additionalProperties: string]: unknown;
 };
 
 /** @internal */
@@ -281,7 +437,22 @@ export const ChatCompletionRequest$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,
@@ -295,20 +466,22 @@ export const ChatCompletionRequest$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 chatCompletionRequestToJSON(