@core-ai/mistral 0.18.0 → 0.19.0

package/dist/index.d.ts

@@ -14,7 +14,7 @@ type MistralProvider = {
 declare function createMistral(options?: MistralProviderOptions$1): MistralProvider;
 
 type MistralModelCapabilities = ModelCapabilities;
-declare function getMistralModelCapabilities(_modelId: string): MistralModelCapabilities;
+declare function getMistralModelCapabilities(modelId: string): MistralModelCapabilities;
 
 declare const mistralGenerateProviderOptionsSchema: z.ZodObject<{
     stopSequences: z.ZodOptional<z.ZodArray<z.ZodString>>;

package/dist/index.js

@@ -15,6 +15,8 @@ import {
   asObject,
   getProviderMetadata,
   safeParseJsonObject,
+  ValidationError,
+  validateInputModalities,
   zodSchemaToJsonSchema
 } from "@core-ai/core-ai";
 
@@ -51,16 +53,82 @@ function parseMistralEmbedProviderOptions(providerOptions) {
 var mistralProviderOptionsSchema = mistralGenerateProviderOptionsSchema;
 
 // src/model-capabilities.ts
-var NO_REASONING_CAPABILITIES = {
-  reasoning: {
-    mode: "unsupported",
-    supportedEfforts: [],
-    restrictsSamplingParams: false,
-    supportedToolChoices: ["auto", "none", "required", "tool"]
-  }
+import {
+  getRegisteredModelCapabilities,
+  MULTIMODAL_INPUT_MODALITIES,
+  stripModelDateSuffix,
+  TEXT_ONLY_MODALITIES,
+  UNKNOWN_MODEL
+} from "@core-ai/core-ai";
+var MISTRAL_VERSION_SUFFIX_PATTERN = /-(?:latest|\d{4})$/;
+function createCapabilities(modalities) {
+  return {
+    reasoning: {
+      mode: "unsupported",
+      supportedEfforts: [],
+      restrictsSamplingParams: false,
+      supportedToolChoices: ["auto", "none", "required", "tool"]
+    },
+    modalities
+  };
+}
+var VISION_CAPABILITIES = createCapabilities(MULTIMODAL_INPUT_MODALITIES);
+var TEXT_ONLY_CAPABILITIES = createCapabilities(TEXT_ONLY_MODALITIES);
+var FAMILY_CAPABILITIES = {
+  "mistral-large": VISION_CAPABILITIES,
+  "mistral-medium": VISION_CAPABILITIES,
+  "mistral-small": VISION_CAPABILITIES,
+  "magistral-medium": VISION_CAPABILITIES,
+  "magistral-small": VISION_CAPABILITIES,
+  "pixtral-large": VISION_CAPABILITIES,
+  "pixtral-12b": VISION_CAPABILITIES,
+  "ministral-3b": VISION_CAPABILITIES,
+  "ministral-8b": VISION_CAPABILITIES,
+  "ministral-14b": VISION_CAPABILITIES,
+  codestral: TEXT_ONLY_CAPABILITIES,
+  "codestral-mamba": TEXT_ONLY_CAPABILITIES,
+  devstral: TEXT_ONLY_CAPABILITIES,
+  "devstral-medium": TEXT_ONLY_CAPABILITIES,
+  "devstral-small": TEXT_ONLY_CAPABILITIES,
+  "open-mistral-7b": TEXT_ONLY_CAPABILITIES,
+  "open-mistral-nemo": TEXT_ONLY_CAPABILITIES,
+  "open-mixtral-8x7b": TEXT_ONLY_CAPABILITIES,
+  "open-mixtral-8x22b": TEXT_ONLY_CAPABILITIES,
+  [UNKNOWN_MODEL]: VISION_CAPABILITIES
+};
+var VERSIONED_CAPABILITIES = {
+  // Vision arrived with Mistral Large 3 (`-2512`).
+  "mistral-large-2402": TEXT_ONLY_CAPABILITIES,
+  "mistral-large-2407": TEXT_ONLY_CAPABILITIES,
+  "mistral-large-2411": TEXT_ONLY_CAPABILITIES,
+  // Vision arrived with Mistral Medium 3 (`-2505`).
+  "mistral-medium-2312": TEXT_ONLY_CAPABILITIES,
+  // Vision arrived with Mistral Small 3.1 (`-2503`).
+  "mistral-small-2402": TEXT_ONLY_CAPABILITIES,
+  "mistral-small-2409": TEXT_ONLY_CAPABILITIES,
+  "mistral-small-2501": TEXT_ONLY_CAPABILITIES,
+  // Vision arrived with Magistral 1.2 (`-2509`).
+  "magistral-medium-2506": TEXT_ONLY_CAPABILITIES,
+  "magistral-medium-2507": TEXT_ONLY_CAPABILITIES,
+  "magistral-small-2506": TEXT_ONLY_CAPABILITIES,
+  "magistral-small-2507": TEXT_ONLY_CAPABILITIES,
+  // Vision arrived with Ministral 3 (`-2512`).
+  "ministral-3b-2410": TEXT_ONLY_CAPABILITIES,
+  "ministral-8b-2410": TEXT_ONLY_CAPABILITIES
+};
+var MISTRAL_MODEL_CAPABILITIES = {
+  ...VERSIONED_CAPABILITIES,
+  ...FAMILY_CAPABILITIES
 };
-function getMistralModelCapabilities(_modelId) {
-  return NO_REASONING_CAPABILITIES;
+function getMistralModelCapabilities(modelId) {
+  const registry = MISTRAL_MODEL_CAPABILITIES;
+  return registry[modelId] ?? getRegisteredModelCapabilities(registry, normalizeModelId(modelId));
+}
+function normalizeModelId(modelId) {
+  return stripModelDateSuffix(modelId).replace(
+    MISTRAL_VERSION_SUFFIX_PATTERN,
+    ""
+  );
 }
 
 // src/chat-adapter.ts
@@ -153,6 +221,9 @@ function convertUserContentPart(part) {
       }
     };
   }
+  if (part.type === "audio") {
+    throw new ValidationError("Mistral does not support audio input");
+  }
   return {
     type: "document_url",
     documentUrl: `data:${part.mimeType};base64,${part.data}`,
@@ -208,27 +279,34 @@ function createStructuredOutputOptions(options) {
     signal: options.signal
   };
 }
-function createGenerateRequest(modelId, options) {
+function createGenerateRequest(modelId, options, adapterOptions = {}) {
   const mistralOptions = parseMistralGenerateProviderOptions(
     options.providerOptions
   );
   const baseRequest = {
-    ...createRequestBase(modelId, options)
+    ...createRequestBase(modelId, options, adapterOptions)
   };
   return mapMistralProviderOptionsToRequest(baseRequest, mistralOptions);
 }
-function createStreamRequest(modelId, options) {
+function createStreamRequest(modelId, options, adapterOptions = {}) {
   const mistralOptions = parseMistralGenerateProviderOptions(
     options.providerOptions
   );
   const baseRequest = {
-    ...createRequestBase(modelId, options),
+    ...createRequestBase(modelId, options, adapterOptions),
     stream: true
   };
   return mapMistralProviderOptionsToRequest(baseRequest, mistralOptions);
 }
-function createRequestBase(modelId, options) {
-  const capabilities = getMistralModelCapabilities(modelId);
+function createRequestBase(modelId, options, adapterOptions) {
+  const capabilities = adapterOptions.capabilities ?? getMistralModelCapabilities(modelId);
+  const providerId = adapterOptions.providerId ?? "mistral";
+  validateInputModalities({
+    messages: options.messages,
+    capabilities,
+    modelId,
+    providerId
+  });
   return {
     model: modelId,
     messages: convertMessages(options.messages, {
@@ -747,6 +825,8 @@ function parseErrorBody(error) {
 // src/chat-model.ts
 function createMistralChatModel(client, modelId) {
   const provider = "mistral";
+  const capabilities = getMistralModelCapabilities(modelId);
+  const adapterOptions = { capabilities, providerId: provider };
   async function callMistralChatApi(call) {
     try {
       return await call();
@@ -755,14 +835,14 @@ function createMistralChatModel(client, modelId) {
     }
   }
   async function generateChat(options) {
-    const request = createGenerateRequest(modelId, options);
+    const request = createGenerateRequest(modelId, options, adapterOptions);
     const response = await callMistralChatApi(
       () => client.chat.complete(request, { signal: options.signal })
     );
     return mapGenerateResponse(response);
   }
   async function streamChat(options) {
-    const request = createStreamRequest(modelId, options);
+    const request = createStreamRequest(modelId, options, adapterOptions);
     return createChatStream(
       async () => transformStream(
         await callMistralChatApi(
@@ -777,7 +857,7 @@ function createMistralChatModel(client, modelId) {
   return {
     provider,
     modelId,
-    capabilities: getMistralModelCapabilities(modelId),
+    capabilities,
     generate: generateChat,
     stream: streamChat,
     async generateObject(options) {

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@core-ai/mistral",
-    "version": "0.18.0",
+    "version": "0.19.0",
     "description": "Mistral provider package for @core-ai/core-ai",
     "license": "MIT",
     "author": "Omnifact (https://omnifact.ai)",
@@ -43,7 +43,7 @@
         "test:watch": "vitest"
     },
     "dependencies": {
-        "@core-ai/core-ai": "^0.18.0",
+        "@core-ai/core-ai": "^0.19.0",
         "@mistralai/mistralai": "^2.4.1"
     },
     "peerDependencies": {