@mistralai/mistralai 2.2.1 → 2.2.6

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

@@ -11,6 +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,
@@ -20,6 +45,10 @@ import {
   Prediction$Outbound,
   Prediction$outboundSchema,
 } from "./prediction.js";
+import {
+  ReasoningEffort,
+  ReasoningEffort$outboundSchema,
+} from "./reasoningeffort.js";
 import {
   ResponseFormat,
   ResponseFormat$Outbound,
@@ -50,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
@@ -62,6 +101,15 @@ export type ChatCompletionRequestMessage =
   | ToolMessage
   | UserMessage;
 
+export type ChatCompletionRequestTool =
+  | Tool
+  | WebSearchTool
+  | WebSearchPremiumTool
+  | CodeInterpreterTool
+  | ImageGenerationTool
+  | DocumentLibraryTool
+  | CustomConnector;
+
 /**
  * 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.
  */
@@ -79,7 +127,7 @@ export type ChatCompletionRequest = {
   /**
    * Nucleus sampling, where the model considers the results of the tokens with `top_p` probability mass. So 0.1 means only the tokens comprising the top 10% probability mass are considered. We generally recommend altering this or `temperature` but not both.
    */
-  topP?: number | undefined;
+  topP?: number | null | undefined;
   /**
    * The maximum number of tokens to generate in the completion. The token count of your prompt plus `max_tokens` cannot exceed the model's context length.
    */
@@ -91,7 +139,7 @@ export type ChatCompletionRequest = {
   /**
    * Stop generation if this token is detected. Or if one of these tokens is detected when providing an array
    */
-  stop?: string | Array<string> | undefined;
+  stop?: string | Array<string> | null | undefined;
   /**
    * The seed to use for random sampling. If set, different calls will generate deterministic results.
    */
@@ -113,7 +161,18 @@ 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<
+      | Tool
+      | WebSearchTool
+      | WebSearchPremiumTool
+      | CodeInterpreterTool
+      | ImageGenerationTool
+      | DocumentLibraryTool
+      | CustomConnector
+    >
+    | 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.
    */
@@ -121,11 +180,11 @@ export type ChatCompletionRequest = {
   /**
    * 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.
    */
-  presencePenalty?: number | undefined;
+  presencePenalty?: number | null | undefined;
   /**
    * The `frequency_penalty` penalizes the repetition of words based on their frequency in the generated text. A higher frequency penalty discourages the model from repeating words that have already appeared frequently in the output, promoting diversity and reducing repetition.
    */
-  frequencyPenalty?: number | undefined;
+  frequencyPenalty?: number | null | undefined;
   /**
    * Number of completions to return for each request, input tokens are only billed once.
    */
@@ -138,10 +197,12 @@ export type ChatCompletionRequest = {
    * Whether to enable parallel function calling during tool use, when enabled the model can call multiple tools in parallel.
    */
   parallelToolCalls?: boolean | undefined;
+  reasoningEffort?: ReasoningEffort | null | undefined;
   /**
    * Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used.
    */
   promptMode?: MistralPromptMode | null | undefined;
+  guardrails?: Array<GuardrailConfig> | null | undefined;
 };
 
 /** @internal */
@@ -191,6 +252,38 @@ export function chatCompletionRequestMessageToJSON(
   );
 }
 
+/** @internal */
+export type ChatCompletionRequestTool$Outbound =
+  | Tool$Outbound
+  | WebSearchTool$Outbound
+  | WebSearchPremiumTool$Outbound
+  | CodeInterpreterTool$Outbound
+  | ImageGenerationTool$Outbound
+  | DocumentLibraryTool$Outbound
+  | CustomConnector$Outbound;
+
+/** @internal */
+export const ChatCompletionRequestTool$outboundSchema: z.ZodType<
+  ChatCompletionRequestTool$Outbound,
+  ChatCompletionRequestTool
+> = z.union([
+  Tool$outboundSchema,
+  WebSearchTool$outboundSchema,
+  WebSearchPremiumTool$outboundSchema,
+  CodeInterpreterTool$outboundSchema,
+  ImageGenerationTool$outboundSchema,
+  DocumentLibraryTool$outboundSchema,
+  CustomConnector$outboundSchema,
+]);
+
+export function chatCompletionRequestToolToJSON(
+  chatCompletionRequestTool: ChatCompletionRequestTool,
+): string {
+  return JSON.stringify(
+    ChatCompletionRequestTool$outboundSchema.parse(chatCompletionRequestTool),
+  );
+}
+
 /** @internal */
 export type ChatCompletionRequestToolChoice$Outbound =
   | ToolChoice$Outbound
@@ -216,10 +309,10 @@ export function chatCompletionRequestToolChoiceToJSON(
 export type ChatCompletionRequest$Outbound = {
   model: string;
   temperature?: number | null | undefined;
-  top_p?: number | undefined;
+  top_p?: number | null | undefined;
   max_tokens?: number | null | undefined;
   stream: boolean;
-  stop?: string | Array<string> | undefined;
+  stop?: string | Array<string> | null | undefined;
   random_seed?: number | null | undefined;
   metadata?: { [k: string]: any } | null | undefined;
   messages: Array<
@@ -229,14 +322,27 @@ export type ChatCompletionRequest$Outbound = {
     | UserMessage$Outbound
   >;
   response_format?: ResponseFormat$Outbound | undefined;
-  tools?: Array<Tool$Outbound> | null | undefined;
+  tools?:
+    | Array<
+      | Tool$Outbound
+      | WebSearchTool$Outbound
+      | WebSearchPremiumTool$Outbound
+      | CodeInterpreterTool$Outbound
+      | ImageGenerationTool$Outbound
+      | DocumentLibraryTool$Outbound
+      | CustomConnector$Outbound
+    >
+    | null
+    | undefined;
   tool_choice?: ToolChoice$Outbound | string | undefined;
-  presence_penalty?: number | undefined;
-  frequency_penalty?: number | undefined;
+  presence_penalty?: number | null | undefined;
+  frequency_penalty?: number | null | undefined;
   n?: number | null | undefined;
   prediction?: Prediction$Outbound | undefined;
   parallel_tool_calls?: boolean | undefined;
+  reasoning_effort?: string | null | undefined;
   prompt_mode?: string | null | undefined;
+  guardrails?: Array<GuardrailConfig$Outbound> | null | undefined;
 };
 
 /** @internal */
@@ -246,10 +352,10 @@ export const ChatCompletionRequest$outboundSchema: z.ZodType<
 > = z.object({
   model: z.string(),
   temperature: z.nullable(z.number()).optional(),
-  topP: z.number().optional(),
+  topP: z.nullable(z.number()).optional(),
   maxTokens: z.nullable(z.int()).optional(),
   stream: z.boolean().default(false),
-  stop: smartUnion([z.string(), z.array(z.string())]).optional(),
+  stop: z.nullable(smartUnion([z.string(), z.array(z.string())])).optional(),
   randomSeed: z.nullable(z.int()).optional(),
   metadata: z.nullable(z.record(z.string(), z.any())).optional(),
   messages: z.array(
@@ -263,17 +369,31 @@ export const ChatCompletionRequest$outboundSchema: z.ZodType<
     ]),
   ),
   responseFormat: ResponseFormat$outboundSchema.optional(),
-  tools: z.nullable(z.array(Tool$outboundSchema)).optional(),
+  tools: z.nullable(
+    z.array(
+      z.union([
+        Tool$outboundSchema,
+        WebSearchTool$outboundSchema,
+        WebSearchPremiumTool$outboundSchema,
+        CodeInterpreterTool$outboundSchema,
+        ImageGenerationTool$outboundSchema,
+        DocumentLibraryTool$outboundSchema,
+        CustomConnector$outboundSchema,
+      ]),
+    ),
+  ).optional(),
   toolChoice: smartUnion([
     ToolChoice$outboundSchema,
     ToolChoiceEnum$outboundSchema,
   ]).optional(),
-  presencePenalty: z.number().optional(),
-  frequencyPenalty: z.number().optional(),
+  presencePenalty: z.nullable(z.number()).optional(),
+  frequencyPenalty: z.nullable(z.number()).optional(),
   n: z.nullable(z.int()).optional(),
   prediction: Prediction$outboundSchema.optional(),
   parallelToolCalls: z.boolean().optional(),
+  reasoningEffort: z.nullable(ReasoningEffort$outboundSchema).optional(),
   promptMode: z.nullable(MistralPromptMode$outboundSchema).optional(),
+  guardrails: z.nullable(z.array(GuardrailConfig$outboundSchema)).optional(),
 }).transform((v) => {
   return remap$(v, {
     topP: "top_p",
@@ -284,6 +404,7 @@ export const ChatCompletionRequest$outboundSchema: z.ZodType<
     presencePenalty: "presence_penalty",
     frequencyPenalty: "frequency_penalty",
     parallelToolCalls: "parallel_tool_calls",
+    reasoningEffort: "reasoning_effort",
     promptMode: "prompt_mode",
   });
 });

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

@@ -11,6 +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,
@@ -20,6 +45,10 @@ import {
   Prediction$Outbound,
   Prediction$outboundSchema,
 } from "./prediction.js";
+import {
+  ReasoningEffort,
+  ReasoningEffort$outboundSchema,
+} from "./reasoningeffort.js";
 import {
   ResponseFormat,
   ResponseFormat$Outbound,
@@ -50,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
@@ -62,6 +101,15 @@ export type ChatCompletionStreamRequestMessage =
   | ToolMessage
   | UserMessage;
 
+export type ChatCompletionStreamRequestTool =
+  | Tool
+  | WebSearchTool
+  | WebSearchPremiumTool
+  | CodeInterpreterTool
+  | ImageGenerationTool
+  | DocumentLibraryTool
+  | CustomConnector;
+
 /**
  * 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.
  */
@@ -79,7 +127,7 @@ export type ChatCompletionStreamRequest = {
   /**
    * Nucleus sampling, where the model considers the results of the tokens with `top_p` probability mass. So 0.1 means only the tokens comprising the top 10% probability mass are considered. We generally recommend altering this or `temperature` but not both.
    */
-  topP?: number | undefined;
+  topP?: number | null | undefined;
   /**
    * The maximum number of tokens to generate in the completion. The token count of your prompt plus `max_tokens` cannot exceed the model's context length.
    */
@@ -88,7 +136,7 @@ export type ChatCompletionStreamRequest = {
   /**
    * Stop generation if this token is detected. Or if one of these tokens is detected when providing an array
    */
-  stop?: string | Array<string> | undefined;
+  stop?: string | Array<string> | null | undefined;
   /**
    * The seed to use for random sampling. If set, different calls will generate deterministic results.
    */
@@ -110,7 +158,18 @@ 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<
+      | Tool
+      | WebSearchTool
+      | WebSearchPremiumTool
+      | CodeInterpreterTool
+      | ImageGenerationTool
+      | DocumentLibraryTool
+      | CustomConnector
+    >
+    | 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.
    */
@@ -118,11 +177,11 @@ export type ChatCompletionStreamRequest = {
   /**
    * 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.
    */
-  presencePenalty?: number | undefined;
+  presencePenalty?: number | null | undefined;
   /**
    * The `frequency_penalty` penalizes the repetition of words based on their frequency in the generated text. A higher frequency penalty discourages the model from repeating words that have already appeared frequently in the output, promoting diversity and reducing repetition.
    */
-  frequencyPenalty?: number | undefined;
+  frequencyPenalty?: number | null | undefined;
   /**
    * Number of completions to return for each request, input tokens are only billed once.
    */
@@ -135,10 +194,12 @@ export type ChatCompletionStreamRequest = {
    * Whether to enable parallel function calling during tool use, when enabled the model can call multiple tools in parallel.
    */
   parallelToolCalls?: boolean | undefined;
+  reasoningEffort?: ReasoningEffort | null | undefined;
   /**
    * Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used.
    */
   promptMode?: MistralPromptMode | null | undefined;
+  guardrails?: Array<GuardrailConfig> | null | undefined;
 };
 
 /** @internal */
@@ -190,6 +251,40 @@ export function chatCompletionStreamRequestMessageToJSON(
   );
 }
 
+/** @internal */
+export type ChatCompletionStreamRequestTool$Outbound =
+  | Tool$Outbound
+  | WebSearchTool$Outbound
+  | WebSearchPremiumTool$Outbound
+  | CodeInterpreterTool$Outbound
+  | ImageGenerationTool$Outbound
+  | DocumentLibraryTool$Outbound
+  | CustomConnector$Outbound;
+
+/** @internal */
+export const ChatCompletionStreamRequestTool$outboundSchema: z.ZodType<
+  ChatCompletionStreamRequestTool$Outbound,
+  ChatCompletionStreamRequestTool
+> = z.union([
+  Tool$outboundSchema,
+  WebSearchTool$outboundSchema,
+  WebSearchPremiumTool$outboundSchema,
+  CodeInterpreterTool$outboundSchema,
+  ImageGenerationTool$outboundSchema,
+  DocumentLibraryTool$outboundSchema,
+  CustomConnector$outboundSchema,
+]);
+
+export function chatCompletionStreamRequestToolToJSON(
+  chatCompletionStreamRequestTool: ChatCompletionStreamRequestTool,
+): string {
+  return JSON.stringify(
+    ChatCompletionStreamRequestTool$outboundSchema.parse(
+      chatCompletionStreamRequestTool,
+    ),
+  );
+}
+
 /** @internal */
 export type ChatCompletionStreamRequestToolChoice$Outbound =
   | ToolChoice$Outbound
@@ -215,10 +310,10 @@ export function chatCompletionStreamRequestToolChoiceToJSON(
 export type ChatCompletionStreamRequest$Outbound = {
   model: string;
   temperature?: number | null | undefined;
-  top_p?: number | undefined;
+  top_p?: number | null | undefined;
   max_tokens?: number | null | undefined;
   stream: boolean;
-  stop?: string | Array<string> | undefined;
+  stop?: string | Array<string> | null | undefined;
   random_seed?: number | null | undefined;
   metadata?: { [k: string]: any } | null | undefined;
   messages: Array<
@@ -228,14 +323,27 @@ export type ChatCompletionStreamRequest$Outbound = {
     | UserMessage$Outbound
   >;
   response_format?: ResponseFormat$Outbound | undefined;
-  tools?: Array<Tool$Outbound> | null | undefined;
+  tools?:
+    | Array<
+      | Tool$Outbound
+      | WebSearchTool$Outbound
+      | WebSearchPremiumTool$Outbound
+      | CodeInterpreterTool$Outbound
+      | ImageGenerationTool$Outbound
+      | DocumentLibraryTool$Outbound
+      | CustomConnector$Outbound
+    >
+    | null
+    | undefined;
   tool_choice?: ToolChoice$Outbound | string | undefined;
-  presence_penalty?: number | undefined;
-  frequency_penalty?: number | undefined;
+  presence_penalty?: number | null | undefined;
+  frequency_penalty?: number | null | undefined;
   n?: number | null | undefined;
   prediction?: Prediction$Outbound | undefined;
   parallel_tool_calls?: boolean | undefined;
+  reasoning_effort?: string | null | undefined;
   prompt_mode?: string | null | undefined;
+  guardrails?: Array<GuardrailConfig$Outbound> | null | undefined;
 };
 
 /** @internal */
@@ -245,10 +353,10 @@ export const ChatCompletionStreamRequest$outboundSchema: z.ZodType<
 > = z.object({
   model: z.string(),
   temperature: z.nullable(z.number()).optional(),
-  topP: z.number().optional(),
+  topP: z.nullable(z.number()).optional(),
   maxTokens: z.nullable(z.int()).optional(),
   stream: z.boolean().default(true),
-  stop: smartUnion([z.string(), z.array(z.string())]).optional(),
+  stop: z.nullable(smartUnion([z.string(), z.array(z.string())])).optional(),
   randomSeed: z.nullable(z.int()).optional(),
   metadata: z.nullable(z.record(z.string(), z.any())).optional(),
   messages: z.array(
@@ -262,17 +370,31 @@ export const ChatCompletionStreamRequest$outboundSchema: z.ZodType<
     ]),
   ),
   responseFormat: ResponseFormat$outboundSchema.optional(),
-  tools: z.nullable(z.array(Tool$outboundSchema)).optional(),
+  tools: z.nullable(
+    z.array(
+      z.union([
+        Tool$outboundSchema,
+        WebSearchTool$outboundSchema,
+        WebSearchPremiumTool$outboundSchema,
+        CodeInterpreterTool$outboundSchema,
+        ImageGenerationTool$outboundSchema,
+        DocumentLibraryTool$outboundSchema,
+        CustomConnector$outboundSchema,
+      ]),
+    ),
+  ).optional(),
   toolChoice: smartUnion([
     ToolChoice$outboundSchema,
     ToolChoiceEnum$outboundSchema,
   ]).optional(),
-  presencePenalty: z.number().optional(),
-  frequencyPenalty: z.number().optional(),
+  presencePenalty: z.nullable(z.number()).optional(),
+  frequencyPenalty: z.nullable(z.number()).optional(),
   n: z.nullable(z.int()).optional(),
   prediction: Prediction$outboundSchema.optional(),
   parallelToolCalls: z.boolean().optional(),
+  reasoningEffort: z.nullable(ReasoningEffort$outboundSchema).optional(),
   promptMode: z.nullable(MistralPromptMode$outboundSchema).optional(),
+  guardrails: z.nullable(z.array(GuardrailConfig$outboundSchema)).optional(),
 }).transform((v) => {
   return remap$(v, {
     topP: "top_p",
@@ -283,6 +405,7 @@ export const ChatCompletionStreamRequest$outboundSchema: z.ZodType<
     presencePenalty: "presence_penalty",
     frequencyPenalty: "frequency_penalty",
     parallelToolCalls: "parallel_tool_calls",
+    reasoningEffort: "reasoning_effort",
     promptMode: "prompt_mode",
   });
 });

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

@@ -0,0 +1,44 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ * @generated-id: d988a9992fe2
+ */
+
+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 CodeInterpreterTool = {
+  toolConfiguration?: ToolConfiguration | null | undefined;
+  type: "code_interpreter";
+};
+
+/** @internal */
+export type CodeInterpreterTool$Outbound = {
+  tool_configuration?: ToolConfiguration$Outbound | null | undefined;
+  type: "code_interpreter";
+};
+
+/** @internal */
+export const CodeInterpreterTool$outboundSchema: z.ZodType<
+  CodeInterpreterTool$Outbound,
+  CodeInterpreterTool
+> = z.object({
+  toolConfiguration: z.nullable(ToolConfiguration$outboundSchema).optional(),
+  type: z.literal("code_interpreter"),
+}).transform((v) => {
+  return remap$(v, {
+    toolConfiguration: "tool_configuration",
+  });
+});
+
+export function codeInterpreterToolToJSON(
+  codeInterpreterTool: CodeInterpreterTool,
+): string {
+  return JSON.stringify(
+    CodeInterpreterTool$outboundSchema.parse(codeInterpreterTool),
+  );
+}

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

@@ -0,0 +1,82 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ * @generated-id: e5811d3d23ad
+ */
+
+import * as z from "zod/v4";
+import { remap as remap$ } from "../../lib/primitives.js";
+import {
+  APIKeyAuth,
+  APIKeyAuth$Outbound,
+  APIKeyAuth$outboundSchema,
+} from "./apikeyauth.js";
+import {
+  OAuth2TokenAuth,
+  OAuth2TokenAuth$Outbound,
+  OAuth2TokenAuth$outboundSchema,
+} from "./oauth2tokenauth.js";
+import {
+  ToolConfiguration,
+  ToolConfiguration$Outbound,
+  ToolConfiguration$outboundSchema,
+} from "./toolconfiguration.js";
+
+export type Authorization = APIKeyAuth | OAuth2TokenAuth;
+
+export type CustomConnector = {
+  type: "connector";
+  connectorId: string;
+  authorization?: APIKeyAuth | OAuth2TokenAuth | null | undefined;
+  toolConfiguration?: ToolConfiguration | null | undefined;
+};
+
+/** @internal */
+export type Authorization$Outbound =
+  | APIKeyAuth$Outbound
+  | OAuth2TokenAuth$Outbound;
+
+/** @internal */
+export const Authorization$outboundSchema: z.ZodType<
+  Authorization$Outbound,
+  Authorization
+> = z.union([APIKeyAuth$outboundSchema, OAuth2TokenAuth$outboundSchema]);
+
+export function authorizationToJSON(authorization: Authorization): string {
+  return JSON.stringify(Authorization$outboundSchema.parse(authorization));
+}
+
+/** @internal */
+export type CustomConnector$Outbound = {
+  type: "connector";
+  connector_id: string;
+  authorization?:
+    | APIKeyAuth$Outbound
+    | OAuth2TokenAuth$Outbound
+    | null
+    | undefined;
+  tool_configuration?: ToolConfiguration$Outbound | null | undefined;
+};
+
+/** @internal */
+export const CustomConnector$outboundSchema: z.ZodType<
+  CustomConnector$Outbound,
+  CustomConnector
+> = z.object({
+  type: z.literal("connector"),
+  connectorId: z.string(),
+  authorization: z.nullable(
+    z.union([APIKeyAuth$outboundSchema, OAuth2TokenAuth$outboundSchema]),
+  ).optional(),
+  toolConfiguration: z.nullable(ToolConfiguration$outboundSchema).optional(),
+}).transform((v) => {
+  return remap$(v, {
+    connectorId: "connector_id",
+    toolConfiguration: "tool_configuration",
+  });
+});
+
+export function customConnectorToJSON(
+  customConnector: CustomConnector,
+): string {
+  return JSON.stringify(CustomConnector$outboundSchema.parse(customConnector));
+}