@mistralai/mistralai 1.7.2 → 1.7.3

package/src/models/components/basemodelcard.ts

@@ -5,7 +5,6 @@
 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 {
@@ -15,11 +14,6 @@ import {
   ModelCapabilities$outboundSchema,
 } from "./modelcapabilities.js";
 
-export const Type = {
-  Base: "base",
-} as const;
-export type Type = ClosedEnum<typeof Type>;
-
 export type BaseModelCard = {
   id: string;
   object?: string | undefined;
@@ -31,30 +25,11 @@ export type BaseModelCard = {
   maxContextLength?: number | undefined;
   aliases?: Array<string> | undefined;
   deprecation?: Date | null | undefined;
+  deprecationReplacementModel?: string | null | undefined;
   defaultModelTemperature?: number | null | undefined;
   type?: "base" | undefined;
 };
 
-/** @internal */
-export const Type$inboundSchema: z.ZodNativeEnum<typeof Type> = z.nativeEnum(
-  Type,
-);
-
-/** @internal */
-export const Type$outboundSchema: z.ZodNativeEnum<typeof Type> =
-  Type$inboundSchema;
-
-/**
- * @internal
- * @deprecated This namespace will be removed in future versions. Use schemas and types that are exported directly from this module.
- */
-export namespace Type$ {
-  /** @deprecated use `Type$inboundSchema` instead. */
-  export const inboundSchema = Type$inboundSchema;
-  /** @deprecated use `Type$outboundSchema` instead. */
-  export const outboundSchema = Type$outboundSchema;
-}
-
 /** @internal */
 export const BaseModelCard$inboundSchema: z.ZodType<
   BaseModelCard,
@@ -73,12 +48,14 @@ export const BaseModelCard$inboundSchema: z.ZodType<
   deprecation: z.nullable(
     z.string().datetime({ offset: true }).transform(v => new Date(v)),
   ).optional(),
+  deprecation_replacement_model: z.nullable(z.string()).optional(),
   default_model_temperature: z.nullable(z.number()).optional(),
   type: z.literal("base").default("base"),
 }).transform((v) => {
   return remap$(v, {
     "owned_by": "ownedBy",
     "max_context_length": "maxContextLength",
+    "deprecation_replacement_model": "deprecationReplacementModel",
     "default_model_temperature": "defaultModelTemperature",
   });
 });
@@ -95,6 +72,7 @@ export type BaseModelCard$Outbound = {
   max_context_length: number;
   aliases?: Array<string> | undefined;
   deprecation?: string | null | undefined;
+  deprecation_replacement_model?: string | null | undefined;
   default_model_temperature?: number | null | undefined;
   type: "base";
 };
@@ -115,12 +93,14 @@ export const BaseModelCard$outboundSchema: z.ZodType<
   maxContextLength: z.number().int().default(32768),
   aliases: z.array(z.string()).optional(),
   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"),
+  type: z.literal("base").default("base" as const),
 }).transform((v) => {
   return remap$(v, {
     ownedBy: "owned_by",
     maxContextLength: "max_context_length",
+    deprecationReplacementModel: "deprecation_replacement_model",
     defaultModelTemperature: "default_model_temperature",
   });
 });

package/src/models/components/chatcompletionrequest.ts

@@ -134,6 +134,9 @@ export type ChatCompletionRequest = {
   n?: number | null | undefined;
   prediction?: Prediction | undefined;
   parallelToolCalls?: boolean | 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;
   /**
    * Whether to inject a safety prompt before all conversations.

package/src/models/components/chatcompletionstreamrequest.ts

@@ -131,6 +131,9 @@ export type ChatCompletionStreamRequest = {
   n?: number | null | undefined;
   prediction?: Prediction | undefined;
   parallelToolCalls?: boolean | 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;
   /**
    * Whether to inject a safety prompt before all conversations.

package/src/models/components/contentchunk.ts

@@ -12,6 +12,12 @@ import {
   DocumentURLChunk$Outbound,
   DocumentURLChunk$outboundSchema,
 } from "./documenturlchunk.js";
+import {
+  FileChunk,
+  FileChunk$inboundSchema,
+  FileChunk$Outbound,
+  FileChunk$outboundSchema,
+} from "./filechunk.js";
 import {
   ImageURLChunk,
   ImageURLChunk$inboundSchema,
@@ -35,6 +41,7 @@ export type ContentChunk =
   | (ImageURLChunk & { type: "image_url" })
   | (TextChunk & { type: "text" })
   | (ReferenceChunk & { type: "reference" })
+  | (FileChunk & { type: "file" })
   | (DocumentURLChunk & { type: "document_url" });
 
 /** @internal */
@@ -56,6 +63,9 @@ export const ContentChunk$inboundSchema: z.ZodType<
       type: v.type,
     })),
   ),
+  FileChunk$inboundSchema.and(
+    z.object({ type: z.literal("file") }).transform((v) => ({ type: v.type })),
+  ),
   DocumentURLChunk$inboundSchema.and(
     z.object({ type: z.literal("document_url") }).transform((v) => ({
       type: v.type,
@@ -68,6 +78,7 @@ export type ContentChunk$Outbound =
   | (ImageURLChunk$Outbound & { type: "image_url" })
   | (TextChunk$Outbound & { type: "text" })
   | (ReferenceChunk$Outbound & { type: "reference" })
+  | (FileChunk$Outbound & { type: "file" })
   | (DocumentURLChunk$Outbound & { type: "document_url" });
 
 /** @internal */
@@ -89,6 +100,9 @@ export const ContentChunk$outboundSchema: z.ZodType<
       type: v.type,
     })),
   ),
+  FileChunk$outboundSchema.and(
+    z.object({ type: z.literal("file") }).transform((v) => ({ type: v.type })),
+  ),
   DocumentURLChunk$outboundSchema.and(
     z.object({ type: z.literal("document_url") }).transform((v) => ({
       type: v.type,

package/src/models/components/conversationrestartrequest.ts

@@ -40,11 +40,11 @@ export type ConversationRestartRequest = {
    */
   store?: boolean | undefined;
   handoffExecution?: ConversationRestartRequestHandoffExecution | undefined;
-  fromEntryId: string;
   /**
    * White-listed arguments from the completion API
    */
   completionArgs?: CompletionArgs | undefined;
+  fromEntryId: string;
 };
 
 /** @internal */
@@ -81,13 +81,13 @@ export const ConversationRestartRequest$inboundSchema: z.ZodType<
   store: z.boolean().default(true),
   handoff_execution: ConversationRestartRequestHandoffExecution$inboundSchema
     .default("server"),
-  from_entry_id: z.string(),
   completion_args: CompletionArgs$inboundSchema.optional(),
+  from_entry_id: z.string(),
 }).transform((v) => {
   return remap$(v, {
     "handoff_execution": "handoffExecution",
-    "from_entry_id": "fromEntryId",
     "completion_args": "completionArgs",
+    "from_entry_id": "fromEntryId",
   });
 });
 
@@ -97,8 +97,8 @@ export type ConversationRestartRequest$Outbound = {
   stream: boolean;
   store: boolean;
   handoff_execution: string;
-  from_entry_id: string;
   completion_args?: CompletionArgs$Outbound | undefined;
+  from_entry_id: string;
 };
 
 /** @internal */
@@ -112,13 +112,13 @@ export const ConversationRestartRequest$outboundSchema: z.ZodType<
   store: z.boolean().default(true),
   handoffExecution: ConversationRestartRequestHandoffExecution$outboundSchema
     .default("server"),
-  fromEntryId: z.string(),
   completionArgs: CompletionArgs$outboundSchema.optional(),
+  fromEntryId: z.string(),
 }).transform((v) => {
   return remap$(v, {
     handoffExecution: "handoff_execution",
-    fromEntryId: "from_entry_id",
     completionArgs: "completion_args",
+    fromEntryId: "from_entry_id",
   });
 });
 

package/src/models/components/conversationrestartstreamrequest.ts

@@ -42,11 +42,11 @@ export type ConversationRestartStreamRequest = {
   handoffExecution?:
     | ConversationRestartStreamRequestHandoffExecution
     | undefined;
-  fromEntryId: string;
   /**
    * White-listed arguments from the completion API
    */
   completionArgs?: CompletionArgs | undefined;
+  fromEntryId: string;
 };
 
 /** @internal */
@@ -85,13 +85,13 @@ export const ConversationRestartStreamRequest$inboundSchema: z.ZodType<
     ConversationRestartStreamRequestHandoffExecution$inboundSchema.default(
       "server",
     ),
-  from_entry_id: z.string(),
   completion_args: CompletionArgs$inboundSchema.optional(),
+  from_entry_id: z.string(),
 }).transform((v) => {
   return remap$(v, {
     "handoff_execution": "handoffExecution",
-    "from_entry_id": "fromEntryId",
     "completion_args": "completionArgs",
+    "from_entry_id": "fromEntryId",
   });
 });
 
@@ -101,8 +101,8 @@ export type ConversationRestartStreamRequest$Outbound = {
   stream: boolean;
   store: boolean;
   handoff_execution: string;
-  from_entry_id: string;
   completion_args?: CompletionArgs$Outbound | undefined;
+  from_entry_id: string;
 };
 
 /** @internal */
@@ -118,13 +118,13 @@ export const ConversationRestartStreamRequest$outboundSchema: z.ZodType<
     ConversationRestartStreamRequestHandoffExecution$outboundSchema.default(
       "server",
     ),
-  fromEntryId: z.string(),
   completionArgs: CompletionArgs$outboundSchema.optional(),
+  fromEntryId: z.string(),
 }).transform((v) => {
   return remap$(v, {
     handoffExecution: "handoff_execution",
-    fromEntryId: "from_entry_id",
     completionArgs: "completion_args",
+    fromEntryId: "from_entry_id",
   });
 });
 

package/src/models/components/filechunk.ts

@@ -0,0 +1,75 @@
+/*
+ * 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 { Result as SafeParseResult } from "../../types/fp.js";
+import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+
+export type FileChunk = {
+  type?: "file" | undefined;
+  fileId: string;
+};
+
+/** @internal */
+export const FileChunk$inboundSchema: z.ZodType<
+  FileChunk,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  type: z.literal("file").default("file"),
+  file_id: z.string(),
+}).transform((v) => {
+  return remap$(v, {
+    "file_id": "fileId",
+  });
+});
+
+/** @internal */
+export type FileChunk$Outbound = {
+  type: "file";
+  file_id: string;
+};
+
+/** @internal */
+export const FileChunk$outboundSchema: z.ZodType<
+  FileChunk$Outbound,
+  z.ZodTypeDef,
+  FileChunk
+> = z.object({
+  type: z.literal("file").default("file" as const),
+  fileId: z.string(),
+}).transform((v) => {
+  return remap$(v, {
+    fileId: "file_id",
+  });
+});
+
+/**
+ * @internal
+ * @deprecated This namespace will be removed in future versions. Use schemas and types that are exported directly from this module.
+ */
+export namespace FileChunk$ {
+  /** @deprecated use `FileChunk$inboundSchema` instead. */
+  export const inboundSchema = FileChunk$inboundSchema;
+  /** @deprecated use `FileChunk$outboundSchema` instead. */
+  export const outboundSchema = FileChunk$outboundSchema;
+  /** @deprecated use `FileChunk$Outbound` instead. */
+  export type Outbound = FileChunk$Outbound;
+}
+
+export function fileChunkToJSON(fileChunk: FileChunk): string {
+  return JSON.stringify(FileChunk$outboundSchema.parse(fileChunk));
+}
+
+export function fileChunkFromJSON(
+  jsonString: string,
+): SafeParseResult<FileChunk, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => FileChunk$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'FileChunk' from JSON`,
+  );
+}

package/src/models/components/ftmodelcard.ts

@@ -5,7 +5,6 @@
 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 {
@@ -15,11 +14,6 @@ import {
   ModelCapabilities$outboundSchema,
 } from "./modelcapabilities.js";
 
-export const FTModelCardType = {
-  FineTuned: "fine-tuned",
-} as const;
-export type FTModelCardType = ClosedEnum<typeof FTModelCardType>;
-
 /**
  * Extra fields for fine-tuned models.
  */
@@ -34,6 +28,7 @@ export type FTModelCard = {
   maxContextLength?: number | undefined;
   aliases?: Array<string> | undefined;
   deprecation?: Date | null | undefined;
+  deprecationReplacementModel?: string | null | undefined;
   defaultModelTemperature?: number | null | undefined;
   type?: "fine-tuned" | undefined;
   job: string;
@@ -41,27 +36,6 @@ export type FTModelCard = {
   archived?: boolean | undefined;
 };
 
-/** @internal */
-export const FTModelCardType$inboundSchema: z.ZodNativeEnum<
-  typeof FTModelCardType
-> = z.nativeEnum(FTModelCardType);
-
-/** @internal */
-export const FTModelCardType$outboundSchema: z.ZodNativeEnum<
-  typeof FTModelCardType
-> = FTModelCardType$inboundSchema;
-
-/**
- * @internal
- * @deprecated This namespace will be removed in future versions. Use schemas and types that are exported directly from this module.
- */
-export namespace FTModelCardType$ {
-  /** @deprecated use `FTModelCardType$inboundSchema` instead. */
-  export const inboundSchema = FTModelCardType$inboundSchema;
-  /** @deprecated use `FTModelCardType$outboundSchema` instead. */
-  export const outboundSchema = FTModelCardType$outboundSchema;
-}
-
 /** @internal */
 export const FTModelCard$inboundSchema: z.ZodType<
   FTModelCard,
@@ -80,6 +54,7 @@ export const FTModelCard$inboundSchema: z.ZodType<
   deprecation: z.nullable(
     z.string().datetime({ offset: true }).transform(v => new Date(v)),
   ).optional(),
+  deprecation_replacement_model: z.nullable(z.string()).optional(),
   default_model_temperature: z.nullable(z.number()).optional(),
   type: z.literal("fine-tuned").default("fine-tuned"),
   job: z.string(),
@@ -89,6 +64,7 @@ export const FTModelCard$inboundSchema: z.ZodType<
   return remap$(v, {
     "owned_by": "ownedBy",
     "max_context_length": "maxContextLength",
+    "deprecation_replacement_model": "deprecationReplacementModel",
     "default_model_temperature": "defaultModelTemperature",
   });
 });
@@ -105,6 +81,7 @@ export type FTModelCard$Outbound = {
   max_context_length: number;
   aliases?: Array<string> | undefined;
   deprecation?: string | null | undefined;
+  deprecation_replacement_model?: string | null | undefined;
   default_model_temperature?: number | null | undefined;
   type: "fine-tuned";
   job: string;
@@ -128,8 +105,9 @@ export const FTModelCard$outboundSchema: z.ZodType<
   maxContextLength: z.number().int().default(32768),
   aliases: z.array(z.string()).optional(),
   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("fine-tuned").default("fine-tuned"),
+  type: z.literal("fine-tuned").default("fine-tuned" as const),
   job: z.string(),
   root: z.string(),
   archived: z.boolean().default(false),
@@ -137,6 +115,7 @@ export const FTModelCard$outboundSchema: z.ZodType<
   return remap$(v, {
     ownedBy: "owned_by",
     maxContextLength: "max_context_length",
+    deprecationReplacementModel: "deprecation_replacement_model",
     defaultModelTemperature: "default_model_temperature",
   });
 });

package/src/models/components/index.ts

@@ -72,6 +72,7 @@ export * from "./embeddingrequest.js";
 export * from "./embeddingresponse.js";
 export * from "./embeddingresponsedata.js";
 export * from "./eventout.js";
+export * from "./filechunk.js";
 export * from "./filepurpose.js";
 export * from "./fileschema.js";
 export * from "./filesignedurl.js";

package/src/models/components/inputentries.ts

@@ -6,6 +6,18 @@ 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 {
+  AgentHandoffEntry,
+  AgentHandoffEntry$inboundSchema,
+  AgentHandoffEntry$Outbound,
+  AgentHandoffEntry$outboundSchema,
+} from "./agenthandoffentry.js";
+import {
+  FunctionCallEntry,
+  FunctionCallEntry$inboundSchema,
+  FunctionCallEntry$Outbound,
+  FunctionCallEntry$outboundSchema,
+} from "./functioncallentry.js";
 import {
   FunctionResultEntry,
   FunctionResultEntry$inboundSchema,
@@ -18,8 +30,26 @@ import {
   MessageInputEntry$Outbound,
   MessageInputEntry$outboundSchema,
 } from "./messageinputentry.js";
+import {
+  MessageOutputEntry,
+  MessageOutputEntry$inboundSchema,
+  MessageOutputEntry$Outbound,
+  MessageOutputEntry$outboundSchema,
+} from "./messageoutputentry.js";
+import {
+  ToolExecutionEntry,
+  ToolExecutionEntry$inboundSchema,
+  ToolExecutionEntry$Outbound,
+  ToolExecutionEntry$outboundSchema,
+} from "./toolexecutionentry.js";
 
-export type InputEntries = MessageInputEntry | FunctionResultEntry;
+export type InputEntries =
+  | MessageInputEntry
+  | FunctionResultEntry
+  | ToolExecutionEntry
+  | FunctionCallEntry
+  | MessageOutputEntry
+  | AgentHandoffEntry;
 
 /** @internal */
 export const InputEntries$inboundSchema: z.ZodType<
@@ -29,12 +59,20 @@ export const InputEntries$inboundSchema: z.ZodType<
 > = z.union([
   MessageInputEntry$inboundSchema,
   FunctionResultEntry$inboundSchema,
+  ToolExecutionEntry$inboundSchema,
+  FunctionCallEntry$inboundSchema,
+  MessageOutputEntry$inboundSchema,
+  AgentHandoffEntry$inboundSchema,
 ]);
 
 /** @internal */
 export type InputEntries$Outbound =
   | MessageInputEntry$Outbound
-  | FunctionResultEntry$Outbound;
+  | FunctionResultEntry$Outbound
+  | ToolExecutionEntry$Outbound
+  | FunctionCallEntry$Outbound
+  | MessageOutputEntry$Outbound
+  | AgentHandoffEntry$Outbound;
 
 /** @internal */
 export const InputEntries$outboundSchema: z.ZodType<
@@ -44,6 +82,10 @@ export const InputEntries$outboundSchema: z.ZodType<
 > = z.union([
   MessageInputEntry$outboundSchema,
   FunctionResultEntry$outboundSchema,
+  ToolExecutionEntry$outboundSchema,
+  FunctionCallEntry$outboundSchema,
+  MessageOutputEntry$outboundSchema,
+  AgentHandoffEntry$outboundSchema,
 ]);
 
 /**

package/src/models/components/messageinputentry.ts

@@ -20,10 +20,10 @@ export const ObjectT = {
 } as const;
 export type ObjectT = ClosedEnum<typeof ObjectT>;
 
-export const MessageInputEntryType = {
+export const Type = {
   MessageInput: "message.input",
 } as const;
-export type MessageInputEntryType = ClosedEnum<typeof MessageInputEntryType>;
+export type Type = ClosedEnum<typeof Type>;
 
 export const MessageInputEntryRole = {
   Assistant: "assistant",
@@ -40,7 +40,7 @@ export type MessageInputEntryContent =
  */
 export type MessageInputEntry = {
   object?: ObjectT | undefined;
-  type?: MessageInputEntryType | undefined;
+  type?: Type | undefined;
   createdAt?: Date | undefined;
   completedAt?: Date | null | undefined;
   id?: string | undefined;
@@ -68,24 +68,23 @@ export namespace ObjectT$ {
 }
 
 /** @internal */
-export const MessageInputEntryType$inboundSchema: z.ZodNativeEnum<
-  typeof MessageInputEntryType
-> = z.nativeEnum(MessageInputEntryType);
+export const Type$inboundSchema: z.ZodNativeEnum<typeof Type> = z.nativeEnum(
+  Type,
+);
 
 /** @internal */
-export const MessageInputEntryType$outboundSchema: z.ZodNativeEnum<
-  typeof MessageInputEntryType
-> = MessageInputEntryType$inboundSchema;
+export const Type$outboundSchema: z.ZodNativeEnum<typeof Type> =
+  Type$inboundSchema;
 
 /**
  * @internal
  * @deprecated This namespace will be removed in future versions. Use schemas and types that are exported directly from this module.
  */
-export namespace MessageInputEntryType$ {
-  /** @deprecated use `MessageInputEntryType$inboundSchema` instead. */
-  export const inboundSchema = MessageInputEntryType$inboundSchema;
-  /** @deprecated use `MessageInputEntryType$outboundSchema` instead. */
-  export const outboundSchema = MessageInputEntryType$outboundSchema;
+export namespace Type$ {
+  /** @deprecated use `Type$inboundSchema` instead. */
+  export const inboundSchema = Type$inboundSchema;
+  /** @deprecated use `Type$outboundSchema` instead. */
+  export const outboundSchema = Type$outboundSchema;
 }
 
 /** @internal */
@@ -166,7 +165,7 @@ export const MessageInputEntry$inboundSchema: z.ZodType<
   unknown
 > = z.object({
   object: ObjectT$inboundSchema.default("entry"),
-  type: MessageInputEntryType$inboundSchema.default("message.input"),
+  type: Type$inboundSchema.default("message.input"),
   created_at: z.string().datetime({ offset: true }).transform(v => new Date(v))
     .optional(),
   completed_at: z.nullable(
@@ -203,7 +202,7 @@ export const MessageInputEntry$outboundSchema: z.ZodType<
   MessageInputEntry
 > = z.object({
   object: ObjectT$outboundSchema.default("entry"),
-  type: MessageInputEntryType$outboundSchema.default("message.input"),
+  type: Type$outboundSchema.default("message.input"),
   createdAt: z.date().transform(v => v.toISOString()).optional(),
   completedAt: z.nullable(z.date().transform(v => v.toISOString())).optional(),
   id: z.string().optional(),

package/src/models/components/modelcapabilities.ts

@@ -14,6 +14,7 @@ export type ModelCapabilities = {
   functionCalling?: boolean | undefined;
   fineTuning?: boolean | undefined;
   vision?: boolean | undefined;
+  classification?: boolean | undefined;
 };
 
 /** @internal */
@@ -27,6 +28,7 @@ export const ModelCapabilities$inboundSchema: z.ZodType<
   function_calling: z.boolean().default(true),
   fine_tuning: z.boolean().default(false),
   vision: z.boolean().default(false),
+  classification: z.boolean().default(false),
 }).transform((v) => {
   return remap$(v, {
     "completion_chat": "completionChat",
@@ -43,6 +45,7 @@ export type ModelCapabilities$Outbound = {
   function_calling: boolean;
   fine_tuning: boolean;
   vision: boolean;
+  classification: boolean;
 };
 
 /** @internal */
@@ -56,6 +59,7 @@ export const ModelCapabilities$outboundSchema: z.ZodType<
   functionCalling: z.boolean().default(true),
   fineTuning: z.boolean().default(false),
   vision: z.boolean().default(false),
+  classification: z.boolean().default(false),
 }).transform((v) => {
   return remap$(v, {
     completionChat: "completion_chat",