@mistralai/mistralai 1.7.3 → 1.7.5

package/src/models/components/audiotranscriptionrequest.ts

@@ -0,0 +1,133 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod";
+import { remap as remap$ } from "../../lib/primitives.js";
+import { safeParse } from "../../lib/schemas.js";
+import { blobLikeSchema } from "../../types/blobs.js";
+import { Result as SafeParseResult } from "../../types/fp.js";
+import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+import {
+  FileT,
+  FileT$inboundSchema,
+  FileT$Outbound,
+  FileT$outboundSchema,
+} from "./file.js";
+import {
+  TimestampGranularity,
+  TimestampGranularity$inboundSchema,
+  TimestampGranularity$outboundSchema,
+} from "./timestampgranularity.js";
+
+export type AudioTranscriptionRequest = {
+  model: string;
+  file?: FileT | Blob | undefined;
+  /**
+   * ID of a file uploaded to /v1/files
+   */
+  fileId?: string | null | undefined;
+  /**
+   * Url of a file to be transcribed
+   */
+  fileUrl?: string | null | undefined;
+  /**
+   * Language of the audio, e.g. 'en'. Providing the language can boost accuracy.
+   */
+  language?: string | null | undefined;
+  stream?: false | undefined;
+  temperature?: number | null | undefined;
+  /**
+   * Granularities of timestamps to include in the response.
+   */
+  timestampGranularities?: Array<TimestampGranularity> | undefined;
+};
+
+/** @internal */
+export const AudioTranscriptionRequest$inboundSchema: z.ZodType<
+  AudioTranscriptionRequest,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  model: z.string(),
+  file: FileT$inboundSchema.optional(),
+  file_id: z.nullable(z.string()).optional(),
+  file_url: z.nullable(z.string()).optional(),
+  language: z.nullable(z.string()).optional(),
+  stream: z.literal(false).default(false),
+  temperature: z.nullable(z.number()).optional(),
+  timestamp_granularities: z.array(TimestampGranularity$inboundSchema)
+    .optional(),
+}).transform((v) => {
+  return remap$(v, {
+    "file_id": "fileId",
+    "file_url": "fileUrl",
+    "timestamp_granularities": "timestampGranularities",
+  });
+});
+
+/** @internal */
+export type AudioTranscriptionRequest$Outbound = {
+  model: string;
+  file?: FileT$Outbound | Blob | undefined;
+  file_id?: string | null | undefined;
+  file_url?: string | null | undefined;
+  language?: string | null | undefined;
+  stream: false;
+  temperature?: number | null | undefined;
+  timestamp_granularities?: Array<string> | undefined;
+};
+
+/** @internal */
+export const AudioTranscriptionRequest$outboundSchema: z.ZodType<
+  AudioTranscriptionRequest$Outbound,
+  z.ZodTypeDef,
+  AudioTranscriptionRequest
+> = z.object({
+  model: z.string(),
+  file: FileT$outboundSchema.or(blobLikeSchema).optional(),
+  fileId: z.nullable(z.string()).optional(),
+  fileUrl: z.nullable(z.string()).optional(),
+  language: z.nullable(z.string()).optional(),
+  stream: z.literal(false).default(false as const),
+  temperature: z.nullable(z.number()).optional(),
+  timestampGranularities: z.array(TimestampGranularity$outboundSchema)
+    .optional(),
+}).transform((v) => {
+  return remap$(v, {
+    fileId: "file_id",
+    fileUrl: "file_url",
+    timestampGranularities: "timestamp_granularities",
+  });
+});
+
+/**
+ * @internal
+ * @deprecated This namespace will be removed in future versions. Use schemas and types that are exported directly from this module.
+ */
+export namespace AudioTranscriptionRequest$ {
+  /** @deprecated use `AudioTranscriptionRequest$inboundSchema` instead. */
+  export const inboundSchema = AudioTranscriptionRequest$inboundSchema;
+  /** @deprecated use `AudioTranscriptionRequest$outboundSchema` instead. */
+  export const outboundSchema = AudioTranscriptionRequest$outboundSchema;
+  /** @deprecated use `AudioTranscriptionRequest$Outbound` instead. */
+  export type Outbound = AudioTranscriptionRequest$Outbound;
+}
+
+export function audioTranscriptionRequestToJSON(
+  audioTranscriptionRequest: AudioTranscriptionRequest,
+): string {
+  return JSON.stringify(
+    AudioTranscriptionRequest$outboundSchema.parse(audioTranscriptionRequest),
+  );
+}
+
+export function audioTranscriptionRequestFromJSON(
+  jsonString: string,
+): SafeParseResult<AudioTranscriptionRequest, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => AudioTranscriptionRequest$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'AudioTranscriptionRequest' from JSON`,
+  );
+}

package/src/models/components/audiotranscriptionrequeststream.ts

@@ -0,0 +1,135 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod";
+import { remap as remap$ } from "../../lib/primitives.js";
+import { safeParse } from "../../lib/schemas.js";
+import { blobLikeSchema } from "../../types/blobs.js";
+import { Result as SafeParseResult } from "../../types/fp.js";
+import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+import {
+  FileT,
+  FileT$inboundSchema,
+  FileT$Outbound,
+  FileT$outboundSchema,
+} from "./file.js";
+import {
+  TimestampGranularity,
+  TimestampGranularity$inboundSchema,
+  TimestampGranularity$outboundSchema,
+} from "./timestampgranularity.js";
+
+export type AudioTranscriptionRequestStream = {
+  model: string;
+  file?: FileT | Blob | undefined;
+  /**
+   * ID of a file uploaded to /v1/files
+   */
+  fileId?: string | null | undefined;
+  /**
+   * Url of a file to be transcribed
+   */
+  fileUrl?: string | null | undefined;
+  /**
+   * Language of the audio, e.g. 'en'. Providing the language can boost accuracy.
+   */
+  language?: string | null | undefined;
+  stream?: true | undefined;
+  temperature?: number | null | undefined;
+  /**
+   * Granularities of timestamps to include in the response.
+   */
+  timestampGranularities?: Array<TimestampGranularity> | undefined;
+};
+
+/** @internal */
+export const AudioTranscriptionRequestStream$inboundSchema: z.ZodType<
+  AudioTranscriptionRequestStream,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  model: z.string(),
+  file: FileT$inboundSchema.optional(),
+  file_id: z.nullable(z.string()).optional(),
+  file_url: z.nullable(z.string()).optional(),
+  language: z.nullable(z.string()).optional(),
+  stream: z.literal(true).default(true),
+  temperature: z.nullable(z.number()).optional(),
+  timestamp_granularities: z.array(TimestampGranularity$inboundSchema)
+    .optional(),
+}).transform((v) => {
+  return remap$(v, {
+    "file_id": "fileId",
+    "file_url": "fileUrl",
+    "timestamp_granularities": "timestampGranularities",
+  });
+});
+
+/** @internal */
+export type AudioTranscriptionRequestStream$Outbound = {
+  model: string;
+  file?: FileT$Outbound | Blob | undefined;
+  file_id?: string | null | undefined;
+  file_url?: string | null | undefined;
+  language?: string | null | undefined;
+  stream: true;
+  temperature?: number | null | undefined;
+  timestamp_granularities?: Array<string> | undefined;
+};
+
+/** @internal */
+export const AudioTranscriptionRequestStream$outboundSchema: z.ZodType<
+  AudioTranscriptionRequestStream$Outbound,
+  z.ZodTypeDef,
+  AudioTranscriptionRequestStream
+> = z.object({
+  model: z.string(),
+  file: FileT$outboundSchema.or(blobLikeSchema).optional(),
+  fileId: z.nullable(z.string()).optional(),
+  fileUrl: z.nullable(z.string()).optional(),
+  language: z.nullable(z.string()).optional(),
+  stream: z.literal(true).default(true as const),
+  temperature: z.nullable(z.number()).optional(),
+  timestampGranularities: z.array(TimestampGranularity$outboundSchema)
+    .optional(),
+}).transform((v) => {
+  return remap$(v, {
+    fileId: "file_id",
+    fileUrl: "file_url",
+    timestampGranularities: "timestamp_granularities",
+  });
+});
+
+/**
+ * @internal
+ * @deprecated This namespace will be removed in future versions. Use schemas and types that are exported directly from this module.
+ */
+export namespace AudioTranscriptionRequestStream$ {
+  /** @deprecated use `AudioTranscriptionRequestStream$inboundSchema` instead. */
+  export const inboundSchema = AudioTranscriptionRequestStream$inboundSchema;
+  /** @deprecated use `AudioTranscriptionRequestStream$outboundSchema` instead. */
+  export const outboundSchema = AudioTranscriptionRequestStream$outboundSchema;
+  /** @deprecated use `AudioTranscriptionRequestStream$Outbound` instead. */
+  export type Outbound = AudioTranscriptionRequestStream$Outbound;
+}
+
+export function audioTranscriptionRequestStreamToJSON(
+  audioTranscriptionRequestStream: AudioTranscriptionRequestStream,
+): string {
+  return JSON.stringify(
+    AudioTranscriptionRequestStream$outboundSchema.parse(
+      audioTranscriptionRequestStream,
+    ),
+  );
+}
+
+export function audioTranscriptionRequestStreamFromJSON(
+  jsonString: string,
+): SafeParseResult<AudioTranscriptionRequestStream, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => AudioTranscriptionRequestStream$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'AudioTranscriptionRequestStream' from JSON`,
+  );
+}

package/src/models/components/basemodelcard.ts

@@ -5,6 +5,7 @@
 import * as z from "zod";
 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 {
@@ -14,6 +15,11 @@ import {
   ModelCapabilities$outboundSchema,
 } from "./modelcapabilities.js";
 
+export const BaseModelCardType = {
+  Base: "base",
+} as const;
+export type BaseModelCardType = ClosedEnum<typeof BaseModelCardType>;
+
 export type BaseModelCard = {
   id: string;
   object?: string | undefined;
@@ -30,6 +36,27 @@ export type BaseModelCard = {
   type?: "base" | undefined;
 };
 
+/** @internal */
+export const BaseModelCardType$inboundSchema: z.ZodNativeEnum<
+  typeof BaseModelCardType
+> = z.nativeEnum(BaseModelCardType);
+
+/** @internal */
+export const BaseModelCardType$outboundSchema: z.ZodNativeEnum<
+  typeof BaseModelCardType
+> = BaseModelCardType$inboundSchema;
+
+/**
+ * @internal
+ * @deprecated This namespace will be removed in future versions. Use schemas and types that are exported directly from this module.
+ */
+export namespace BaseModelCardType$ {
+  /** @deprecated use `BaseModelCardType$inboundSchema` instead. */
+  export const inboundSchema = BaseModelCardType$inboundSchema;
+  /** @deprecated use `BaseModelCardType$outboundSchema` instead. */
+  export const outboundSchema = BaseModelCardType$outboundSchema;
+}
+
 /** @internal */
 export const BaseModelCard$inboundSchema: z.ZodType<
   BaseModelCard,
@@ -95,7 +122,7 @@ export const BaseModelCard$outboundSchema: z.ZodType<
   deprecation: z.nullable(z.date().transform(v => v.toISOString())).optional(),
   deprecationReplacementModel: z.nullable(z.string()).optional(),
   defaultModelTemperature: z.nullable(z.number()).optional(),
-  type: z.literal("base").default("base" as const),
+  type: z.literal("base").default("base"),
 }).transform((v) => {
   return remap$(v, {
     ownedBy: "owned_by",

package/src/models/components/batchjobin.ts

@@ -16,7 +16,8 @@ import {
 export type BatchJobIn = {
   inputFiles: Array<string>;
   endpoint: ApiEndpoint;
-  model: string;
+  model?: string | null | undefined;
+  agentId?: string | null | undefined;
   metadata?: { [k: string]: string } | null | undefined;
   timeoutHours?: number | undefined;
 };
@@ -29,12 +30,14 @@ export const BatchJobIn$inboundSchema: z.ZodType<
 > = z.object({
   input_files: z.array(z.string()),
   endpoint: ApiEndpoint$inboundSchema,
-  model: z.string(),
+  model: z.nullable(z.string()).optional(),
+  agent_id: z.nullable(z.string()).optional(),
   metadata: z.nullable(z.record(z.string())).optional(),
   timeout_hours: z.number().int().default(24),
 }).transform((v) => {
   return remap$(v, {
     "input_files": "inputFiles",
+    "agent_id": "agentId",
     "timeout_hours": "timeoutHours",
   });
 });
@@ -43,7 +46,8 @@ export const BatchJobIn$inboundSchema: z.ZodType<
 export type BatchJobIn$Outbound = {
   input_files: Array<string>;
   endpoint: string;
-  model: string;
+  model?: string | null | undefined;
+  agent_id?: string | null | undefined;
   metadata?: { [k: string]: string } | null | undefined;
   timeout_hours: number;
 };
@@ -56,12 +60,14 @@ export const BatchJobIn$outboundSchema: z.ZodType<
 > = z.object({
   inputFiles: z.array(z.string()),
   endpoint: ApiEndpoint$outboundSchema,
-  model: z.string(),
+  model: z.nullable(z.string()).optional(),
+  agentId: z.nullable(z.string()).optional(),
   metadata: z.nullable(z.record(z.string())).optional(),
   timeoutHours: z.number().int().default(24),
 }).transform((v) => {
   return remap$(v, {
     inputFiles: "input_files",
+    agentId: "agent_id",
     timeoutHours: "timeout_hours",
   });
 });

package/src/models/components/batchjobout.ts

@@ -31,7 +31,8 @@ export type BatchJobOut = {
   inputFiles: Array<string>;
   metadata?: { [k: string]: any } | null | undefined;
   endpoint: string;
-  model: string;
+  model?: string | null | undefined;
+  agentId?: string | null | undefined;
   outputFile?: string | null | undefined;
   errorFile?: string | null | undefined;
   errors: Array<BatchError>;
@@ -77,7 +78,8 @@ export const BatchJobOut$inboundSchema: z.ZodType<
   input_files: z.array(z.string()),
   metadata: z.nullable(z.record(z.any())).optional(),
   endpoint: z.string(),
-  model: z.string(),
+  model: z.nullable(z.string()).optional(),
+  agent_id: z.nullable(z.string()).optional(),
   output_file: z.nullable(z.string()).optional(),
   error_file: z.nullable(z.string()).optional(),
   errors: z.array(BatchError$inboundSchema),
@@ -92,6 +94,7 @@ export const BatchJobOut$inboundSchema: z.ZodType<
 }).transform((v) => {
   return remap$(v, {
     "input_files": "inputFiles",
+    "agent_id": "agentId",
     "output_file": "outputFile",
     "error_file": "errorFile",
     "created_at": "createdAt",
@@ -111,7 +114,8 @@ export type BatchJobOut$Outbound = {
   input_files: Array<string>;
   metadata?: { [k: string]: any } | null | undefined;
   endpoint: string;
-  model: string;
+  model?: string | null | undefined;
+  agent_id?: string | null | undefined;
   output_file?: string | null | undefined;
   error_file?: string | null | undefined;
   errors: Array<BatchError$Outbound>;
@@ -136,7 +140,8 @@ export const BatchJobOut$outboundSchema: z.ZodType<
   inputFiles: z.array(z.string()),
   metadata: z.nullable(z.record(z.any())).optional(),
   endpoint: z.string(),
-  model: z.string(),
+  model: z.nullable(z.string()).optional(),
+  agentId: z.nullable(z.string()).optional(),
   outputFile: z.nullable(z.string()).optional(),
   errorFile: z.nullable(z.string()).optional(),
   errors: z.array(BatchError$outboundSchema),
@@ -151,6 +156,7 @@ export const BatchJobOut$outboundSchema: z.ZodType<
 }).transform((v) => {
   return remap$(v, {
     inputFiles: "input_files",
+    agentId: "agent_id",
     outputFile: "output_file",
     errorFile: "error_file",
     createdAt: "created_at",

package/src/models/components/classifierftmodelout.ts

@@ -40,7 +40,9 @@ export type ClassifierFTModelOut = {
   object?: ClassifierFTModelOutObject | undefined;
   created: number;
   ownedBy: string;
+  workspaceId: string;
   root: string;
+  rootVersion: string;
   archived: boolean;
   name?: string | null | undefined;
   description?: string | null | undefined;
@@ -104,7 +106,9 @@ export const ClassifierFTModelOut$inboundSchema: z.ZodType<
   object: ClassifierFTModelOutObject$inboundSchema.default("model"),
   created: z.number().int(),
   owned_by: z.string(),
+  workspace_id: z.string(),
   root: z.string(),
+  root_version: z.string(),
   archived: z.boolean(),
   name: z.nullable(z.string()).optional(),
   description: z.nullable(z.string()).optional(),
@@ -117,6 +121,8 @@ export const ClassifierFTModelOut$inboundSchema: z.ZodType<
 }).transform((v) => {
   return remap$(v, {
     "owned_by": "ownedBy",
+    "workspace_id": "workspaceId",
+    "root_version": "rootVersion",
     "max_context_length": "maxContextLength",
     "classifier_targets": "classifierTargets",
     "model_type": "modelType",
@@ -129,7 +135,9 @@ export type ClassifierFTModelOut$Outbound = {
   object: string;
   created: number;
   owned_by: string;
+  workspace_id: string;
   root: string;
+  root_version: string;
   archived: boolean;
   name?: string | null | undefined;
   description?: string | null | undefined;
@@ -151,7 +159,9 @@ export const ClassifierFTModelOut$outboundSchema: z.ZodType<
   object: ClassifierFTModelOutObject$outboundSchema.default("model"),
   created: z.number().int(),
   ownedBy: z.string(),
+  workspaceId: z.string(),
   root: z.string(),
+  rootVersion: z.string(),
   archived: z.boolean(),
   name: z.nullable(z.string()).optional(),
   description: z.nullable(z.string()).optional(),
@@ -164,6 +174,8 @@ export const ClassifierFTModelOut$outboundSchema: z.ZodType<
 }).transform((v) => {
   return remap$(v, {
     ownedBy: "owned_by",
+    workspaceId: "workspace_id",
+    rootVersion: "root_version",
     maxContextLength: "max_context_length",
     classifierTargets: "classifier_targets",
     modelType: "model_type",

package/src/models/components/completionargs.ts

@@ -38,7 +38,7 @@ export type CompletionArgs = {
   stop?: CompletionArgsStop | null | undefined;
   presencePenalty?: number | null | undefined;
   frequencyPenalty?: number | null | undefined;
-  temperature?: number | undefined;
+  temperature?: number | null | undefined;
   topP?: number | null | undefined;
   maxTokens?: number | null | undefined;
   randomSeed?: number | null | undefined;
@@ -56,7 +56,7 @@ export const CompletionArgs$inboundSchema: z.ZodType<
   stop: z.nullable(CompletionArgsStop$inboundSchema).optional(),
   presence_penalty: z.nullable(z.number()).optional(),
   frequency_penalty: z.nullable(z.number()).optional(),
-  temperature: z.number().default(0.3),
+  temperature: z.nullable(z.number()).optional(),
   top_p: z.nullable(z.number()).optional(),
   max_tokens: z.nullable(z.number().int()).optional(),
   random_seed: z.nullable(z.number().int()).optional(),
@@ -80,7 +80,7 @@ export type CompletionArgs$Outbound = {
   stop?: CompletionArgsStop$Outbound | null | undefined;
   presence_penalty?: number | null | undefined;
   frequency_penalty?: number | null | undefined;
-  temperature: number;
+  temperature?: number | null | undefined;
   top_p?: number | null | undefined;
   max_tokens?: number | null | undefined;
   random_seed?: number | null | undefined;
@@ -98,7 +98,7 @@ export const CompletionArgs$outboundSchema: z.ZodType<
   stop: z.nullable(CompletionArgsStop$outboundSchema).optional(),
   presencePenalty: z.nullable(z.number()).optional(),
   frequencyPenalty: z.nullable(z.number()).optional(),
-  temperature: z.number().default(0.3),
+  temperature: z.nullable(z.number()).optional(),
   topP: z.nullable(z.number()).optional(),
   maxTokens: z.nullable(z.number().int()).optional(),
   randomSeed: z.nullable(z.number().int()).optional(),

package/src/models/components/completionftmodelout.ts

@@ -32,7 +32,9 @@ export type CompletionFTModelOut = {
   object?: CompletionFTModelOutObject | undefined;
   created: number;
   ownedBy: string;
+  workspaceId: string;
   root: string;
+  rootVersion: string;
   archived: boolean;
   name?: string | null | undefined;
   description?: string | null | undefined;
@@ -93,7 +95,9 @@ export const CompletionFTModelOut$inboundSchema: z.ZodType<
   object: CompletionFTModelOutObject$inboundSchema.default("model"),
   created: z.number().int(),
   owned_by: z.string(),
+  workspace_id: z.string(),
   root: z.string(),
+  root_version: z.string(),
   archived: z.boolean(),
   name: z.nullable(z.string()).optional(),
   description: z.nullable(z.string()).optional(),
@@ -105,6 +109,8 @@ export const CompletionFTModelOut$inboundSchema: z.ZodType<
 }).transform((v) => {
   return remap$(v, {
     "owned_by": "ownedBy",
+    "workspace_id": "workspaceId",
+    "root_version": "rootVersion",
     "max_context_length": "maxContextLength",
     "model_type": "modelType",
   });
@@ -116,7 +122,9 @@ export type CompletionFTModelOut$Outbound = {
   object: string;
   created: number;
   owned_by: string;
+  workspace_id: string;
   root: string;
+  root_version: string;
   archived: boolean;
   name?: string | null | undefined;
   description?: string | null | undefined;
@@ -137,7 +145,9 @@ export const CompletionFTModelOut$outboundSchema: z.ZodType<
   object: CompletionFTModelOutObject$outboundSchema.default("model"),
   created: z.number().int(),
   ownedBy: z.string(),
+  workspaceId: z.string(),
   root: z.string(),
+  rootVersion: z.string(),
   archived: z.boolean(),
   name: z.nullable(z.string()).optional(),
   description: z.nullable(z.string()).optional(),
@@ -149,6 +159,8 @@ export const CompletionFTModelOut$outboundSchema: z.ZodType<
 }).transform((v) => {
   return remap$(v, {
     ownedBy: "owned_by",
+    workspaceId: "workspace_id",
+    rootVersion: "root_version",
     maxContextLength: "max_context_length",
     modelType: "model_type",
   });

package/src/models/components/contentchunk.ts

@@ -6,6 +6,12 @@ import * as z from "zod";
 import { safeParse } from "../../lib/schemas.js";
 import { Result as SafeParseResult } from "../../types/fp.js";
 import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+import {
+  AudioChunk,
+  AudioChunk$inboundSchema,
+  AudioChunk$Outbound,
+  AudioChunk$outboundSchema,
+} from "./audiochunk.js";
 import {
   DocumentURLChunk,
   DocumentURLChunk$inboundSchema,
@@ -36,13 +42,21 @@ import {
   TextChunk$Outbound,
   TextChunk$outboundSchema,
 } from "./textchunk.js";
+import {
+  ThinkChunk,
+  ThinkChunk$inboundSchema,
+  ThinkChunk$Outbound,
+  ThinkChunk$outboundSchema,
+} from "./thinkchunk.js";
 
 export type ContentChunk =
   | (ImageURLChunk & { type: "image_url" })
   | (TextChunk & { type: "text" })
   | (ReferenceChunk & { type: "reference" })
   | (FileChunk & { type: "file" })
-  | (DocumentURLChunk & { type: "document_url" });
+  | (AudioChunk & { type: "input_audio" })
+  | (DocumentURLChunk & { type: "document_url" })
+  | (ThinkChunk & { type: "thinking" });
 
 /** @internal */
 export const ContentChunk$inboundSchema: z.ZodType<
@@ -66,11 +80,21 @@ export const ContentChunk$inboundSchema: z.ZodType<
   FileChunk$inboundSchema.and(
     z.object({ type: z.literal("file") }).transform((v) => ({ type: v.type })),
   ),
+  AudioChunk$inboundSchema.and(
+    z.object({ type: z.literal("input_audio") }).transform((v) => ({
+      type: v.type,
+    })),
+  ),
   DocumentURLChunk$inboundSchema.and(
     z.object({ type: z.literal("document_url") }).transform((v) => ({
       type: v.type,
     })),
   ),
+  ThinkChunk$inboundSchema.and(
+    z.object({ type: z.literal("thinking") }).transform((v) => ({
+      type: v.type,
+    })),
+  ),
 ]);
 
 /** @internal */
@@ -79,7 +103,9 @@ export type ContentChunk$Outbound =
   | (TextChunk$Outbound & { type: "text" })
   | (ReferenceChunk$Outbound & { type: "reference" })
   | (FileChunk$Outbound & { type: "file" })
-  | (DocumentURLChunk$Outbound & { type: "document_url" });
+  | (AudioChunk$Outbound & { type: "input_audio" })
+  | (DocumentURLChunk$Outbound & { type: "document_url" })
+  | (ThinkChunk$Outbound & { type: "thinking" });
 
 /** @internal */
 export const ContentChunk$outboundSchema: z.ZodType<
@@ -103,11 +129,21 @@ export const ContentChunk$outboundSchema: z.ZodType<
   FileChunk$outboundSchema.and(
     z.object({ type: z.literal("file") }).transform((v) => ({ type: v.type })),
   ),
+  AudioChunk$outboundSchema.and(
+    z.object({ type: z.literal("input_audio") }).transform((v) => ({
+      type: v.type,
+    })),
+  ),
   DocumentURLChunk$outboundSchema.and(
     z.object({ type: z.literal("document_url") }).transform((v) => ({
       type: v.type,
     })),
   ),
+  ThinkChunk$outboundSchema.and(
+    z.object({ type: z.literal("thinking") }).transform((v) => ({
+      type: v.type,
+    })),
+  ),
 ]);
 
 /**