@ai-sdk/mistral 1.2.7 → 2.0.0-alpha.1

package/dist/index.js

@@ -26,15 +26,15 @@ __export(src_exports, {
 module.exports = __toCommonJS(src_exports);
 
 // src/mistral-provider.ts
-var import_provider_utils5 = require("@ai-sdk/provider-utils");
+var import_provider4 = require("@ai-sdk/provider");
+var import_provider_utils4 = require("@ai-sdk/provider-utils");
 
 // src/mistral-chat-language-model.ts
-var import_provider_utils3 = require("@ai-sdk/provider-utils");
-var import_zod2 = require("zod");
+var import_provider_utils2 = require("@ai-sdk/provider-utils");
+var import_zod3 = require("zod");
 
 // src/convert-to-mistral-chat-messages.ts
 var import_provider = require("@ai-sdk/provider");
-var import_provider_utils = require("@ai-sdk/provider-utils");
 function convertToMistralChatMessages(prompt) {
   const messages = [];
   for (let i = 0; i < prompt.length; i++) {
@@ -49,36 +49,27 @@ function convertToMistralChatMessages(prompt) {
         messages.push({
           role: "user",
           content: content.map((part) => {
-            var _a;
             switch (part.type) {
               case "text": {
                 return { type: "text", text: part.text };
               }
-              case "image": {
-                return {
-                  type: "image_url",
-                  image_url: part.image instanceof URL ? part.image.toString() : `data:${(_a = part.mimeType) != null ? _a : "image/jpeg"};base64,${(0, import_provider_utils.convertUint8ArrayToBase64)(part.image)}`
-                };
-              }
               case "file": {
-                if (!(part.data instanceof URL)) {
+                if (part.mediaType.startsWith("image/")) {
+                  const mediaType = part.mediaType === "image/*" ? "image/jpeg" : part.mediaType;
+                  return {
+                    type: "image_url",
+                    image_url: part.data instanceof URL ? part.data.toString() : `data:${mediaType};base64,${part.data}`
+                  };
+                } else if (part.mediaType === "application/pdf") {
+                  return {
+                    type: "document_url",
+                    document_url: part.data.toString()
+                  };
+                } else {
                   throw new import_provider.UnsupportedFunctionalityError({
-                    functionality: "File content parts in user messages"
+                    functionality: "Only images and PDF file parts are supported"
                   });
                 }
-                switch (part.mimeType) {
-                  case "application/pdf": {
-                    return {
-                      type: "document_url",
-                      document_url: part.data.toString()
-                    };
-                  }
-                  default: {
-                    throw new import_provider.UnsupportedFunctionalityError({
-                      functionality: "Only PDF files are supported in user messages"
-                    });
-                  }
-                }
               }
             }
           })
@@ -135,6 +126,19 @@ function convertToMistralChatMessages(prompt) {
   return messages;
 }
 
+// src/get-response-metadata.ts
+function getResponseMetadata({
+  id,
+  model,
+  created
+}) {
+  return {
+    id: id != null ? id : void 0,
+    modelId: model != null ? model : void 0,
+    timestamp: created != null ? new Date(created * 1e3) : void 0
+  };
+}
+
 // src/map-mistral-finish-reason.ts
 function mapMistralFinishReason(finishReason) {
   switch (finishReason) {
@@ -150,42 +154,44 @@ function mapMistralFinishReason(finishReason) {
   }
 }
 
-// src/mistral-error.ts
-var import_provider_utils2 = require("@ai-sdk/provider-utils");
+// src/mistral-chat-options.ts
 var import_zod = require("zod");
-var mistralErrorDataSchema = import_zod.z.object({
-  object: import_zod.z.literal("error"),
-  message: import_zod.z.string(),
-  type: import_zod.z.string(),
-  param: import_zod.z.string().nullable(),
-  code: import_zod.z.string().nullable()
+var mistralProviderOptions = import_zod.z.object({
+  /**
+  Whether to inject a safety prompt before all conversations.
+  
+  Defaults to `false`.
+     */
+  safePrompt: import_zod.z.boolean().optional(),
+  documentImageLimit: import_zod.z.number().optional(),
+  documentPageLimit: import_zod.z.number().optional()
+});
+
+// src/mistral-error.ts
+var import_provider_utils = require("@ai-sdk/provider-utils");
+var import_zod2 = require("zod");
+var mistralErrorDataSchema = import_zod2.z.object({
+  object: import_zod2.z.literal("error"),
+  message: import_zod2.z.string(),
+  type: import_zod2.z.string(),
+  param: import_zod2.z.string().nullable(),
+  code: import_zod2.z.string().nullable()
 });
-var mistralFailedResponseHandler = (0, import_provider_utils2.createJsonErrorResponseHandler)({
+var mistralFailedResponseHandler = (0, import_provider_utils.createJsonErrorResponseHandler)({
   errorSchema: mistralErrorDataSchema,
   errorToMessage: (data) => data.message
 });
 
-// src/get-response-metadata.ts
-function getResponseMetadata({
-  id,
-  model,
-  created
-}) {
-  return {
-    id: id != null ? id : void 0,
-    modelId: model != null ? model : void 0,
-    timestamp: created != null ? new Date(created * 1e3) : void 0
-  };
-}
-
 // src/mistral-prepare-tools.ts
 var import_provider2 = require("@ai-sdk/provider");
-function prepareTools(mode) {
-  var _a;
-  const tools = ((_a = mode.tools) == null ? void 0 : _a.length) ? mode.tools : void 0;
+function prepareTools({
+  tools,
+  toolChoice
+}) {
+  tools = (tools == null ? void 0 : tools.length) ? tools : void 0;
   const toolWarnings = [];
   if (tools == null) {
-    return { tools: void 0, tool_choice: void 0, toolWarnings };
+    return { tools: void 0, toolChoice: void 0, toolWarnings };
   }
   const mistralTools = [];
   for (const tool of tools) {
@@ -202,29 +208,28 @@ function prepareTools(mode) {
       });
     }
   }
-  const toolChoice = mode.toolChoice;
   if (toolChoice == null) {
-    return { tools: mistralTools, tool_choice: void 0, toolWarnings };
+    return { tools: mistralTools, toolChoice: void 0, toolWarnings };
   }
   const type = toolChoice.type;
   switch (type) {
     case "auto":
     case "none":
-      return { tools: mistralTools, tool_choice: type, toolWarnings };
+      return { tools: mistralTools, toolChoice: type, toolWarnings };
     case "required":
-      return { tools: mistralTools, tool_choice: "any", toolWarnings };
+      return { tools: mistralTools, toolChoice: "any", toolWarnings };
     case "tool":
       return {
         tools: mistralTools.filter(
           (tool) => tool.function.name === toolChoice.toolName
         ),
-        tool_choice: "any",
+        toolChoice: "any",
         toolWarnings
       };
     default: {
       const _exhaustiveCheck = type;
       throw new import_provider2.UnsupportedFunctionalityError({
-        functionality: `Unsupported tool choice type: ${_exhaustiveCheck}`
+        functionality: `tool choice type: ${_exhaustiveCheck}`
       });
     }
   }
@@ -232,24 +237,20 @@ function prepareTools(mode) {
 
 // src/mistral-chat-language-model.ts
 var MistralChatLanguageModel = class {
-  constructor(modelId, settings, config) {
-    this.specificationVersion = "v1";
-    this.defaultObjectGenerationMode = "json";
-    this.supportsImageUrls = false;
+  constructor(modelId, config) {
+    this.specificationVersion = "v2";
+    this.supportedUrls = {
+      "application/pdf": [/^https:\/\/.*$/]
+    };
     this.modelId = modelId;
-    this.settings = settings;
     this.config = config;
   }
   get provider() {
     return this.config.provider;
   }
-  supportsUrl(url) {
-    return url.protocol === "https:";
-  }
-  getArgs({
-    mode,
+  async getArgs({
     prompt,
-    maxTokens,
+    maxOutputTokens,
     temperature,
     topP,
     topK,
@@ -258,11 +259,17 @@ var MistralChatLanguageModel = class {
     stopSequences,
     responseFormat,
     seed,
-    providerMetadata
+    providerOptions,
+    tools,
+    toolChoice
   }) {
-    var _a, _b;
-    const type = mode.type;
+    var _a;
     const warnings = [];
+    const options = (_a = await (0, import_provider_utils2.parseProviderOptions)({
+      provider: "mistral",
+      providerOptions,
+      schema: mistralProviderOptions
+    })) != null ? _a : {};
     if (topK != null) {
       warnings.push({
         type: "unsupported-setting",
@@ -298,126 +305,120 @@ var MistralChatLanguageModel = class {
       // model id:
       model: this.modelId,
       // model specific settings:
-      safe_prompt: this.settings.safePrompt,
+      safe_prompt: options.safePrompt,
       // standardized settings:
-      max_tokens: maxTokens,
+      max_tokens: maxOutputTokens,
       temperature,
       top_p: topP,
       random_seed: seed,
       // response format:
       response_format: (responseFormat == null ? void 0 : responseFormat.type) === "json" ? { type: "json_object" } : void 0,
       // mistral-specific provider options:
-      document_image_limit: (_a = providerMetadata == null ? void 0 : providerMetadata.mistral) == null ? void 0 : _a.documentImageLimit,
-      document_page_limit: (_b = providerMetadata == null ? void 0 : providerMetadata.mistral) == null ? void 0 : _b.documentPageLimit,
+      document_image_limit: options.documentImageLimit,
+      document_page_limit: options.documentPageLimit,
       // messages:
       messages: convertToMistralChatMessages(prompt)
     };
-    switch (type) {
-      case "regular": {
-        const { tools, tool_choice, toolWarnings } = prepareTools(mode);
-        return {
-          args: { ...baseArgs, tools, tool_choice },
-          warnings: [...warnings, ...toolWarnings]
-        };
-      }
-      case "object-json": {
-        return {
-          args: {
-            ...baseArgs,
-            response_format: { type: "json_object" }
-          },
-          warnings
-        };
-      }
-      case "object-tool": {
-        return {
-          args: {
-            ...baseArgs,
-            tool_choice: "any",
-            tools: [{ type: "function", function: mode.tool }]
-          },
-          warnings
-        };
-      }
-      default: {
-        const _exhaustiveCheck = type;
-        throw new Error(`Unsupported type: ${_exhaustiveCheck}`);
-      }
-    }
+    const {
+      tools: mistralTools,
+      toolChoice: mistralToolChoice,
+      toolWarnings
+    } = prepareTools({
+      tools,
+      toolChoice
+    });
+    return {
+      args: {
+        ...baseArgs,
+        tools: mistralTools,
+        tool_choice: mistralToolChoice
+      },
+      warnings: [...warnings, ...toolWarnings]
+    };
   }
   async doGenerate(options) {
-    var _a;
-    const { args, warnings } = this.getArgs(options);
+    const { args: body, warnings } = await this.getArgs(options);
     const {
       responseHeaders,
       value: response,
       rawValue: rawResponse
-    } = await (0, import_provider_utils3.postJsonToApi)({
+    } = await (0, import_provider_utils2.postJsonToApi)({
       url: `${this.config.baseURL}/chat/completions`,
-      headers: (0, import_provider_utils3.combineHeaders)(this.config.headers(), options.headers),
-      body: args,
+      headers: (0, import_provider_utils2.combineHeaders)(this.config.headers(), options.headers),
+      body,
       failedResponseHandler: mistralFailedResponseHandler,
-      successfulResponseHandler: (0, import_provider_utils3.createJsonResponseHandler)(
+      successfulResponseHandler: (0, import_provider_utils2.createJsonResponseHandler)(
         mistralChatResponseSchema
       ),
       abortSignal: options.abortSignal,
       fetch: this.config.fetch
     });
-    const { messages: rawPrompt, ...rawSettings } = args;
     const choice = response.choices[0];
+    const content = [];
     let text = extractTextContent(choice.message.content);
-    const lastMessage = rawPrompt[rawPrompt.length - 1];
+    const lastMessage = body.messages[body.messages.length - 1];
     if (lastMessage.role === "assistant" && (text == null ? void 0 : text.startsWith(lastMessage.content))) {
       text = text.slice(lastMessage.content.length);
     }
+    if (text != null && text.length > 0) {
+      content.push({ type: "text", text });
+    }
+    if (choice.message.tool_calls != null) {
+      for (const toolCall of choice.message.tool_calls) {
+        content.push({
+          type: "tool-call",
+          toolCallType: "function",
+          toolCallId: toolCall.id,
+          toolName: toolCall.function.name,
+          args: toolCall.function.arguments
+        });
+      }
+    }
     return {
-      text,
-      toolCalls: (_a = choice.message.tool_calls) == null ? void 0 : _a.map((toolCall) => ({
-        toolCallType: "function",
-        toolCallId: toolCall.id,
-        toolName: toolCall.function.name,
-        args: toolCall.function.arguments
-      })),
+      content,
       finishReason: mapMistralFinishReason(choice.finish_reason),
       usage: {
-        promptTokens: response.usage.prompt_tokens,
-        completionTokens: response.usage.completion_tokens
+        inputTokens: response.usage.prompt_tokens,
+        outputTokens: response.usage.completion_tokens,
+        totalTokens: response.usage.total_tokens
       },
-      rawCall: { rawPrompt, rawSettings },
-      rawResponse: {
+      request: { body },
+      response: {
+        ...getResponseMetadata(response),
         headers: responseHeaders,
         body: rawResponse
       },
-      request: { body: JSON.stringify(args) },
-      response: getResponseMetadata(response),
       warnings
     };
   }
   async doStream(options) {
-    const { args, warnings } = this.getArgs(options);
+    const { args, warnings } = await this.getArgs(options);
     const body = { ...args, stream: true };
-    const { responseHeaders, value: response } = await (0, import_provider_utils3.postJsonToApi)({
+    const { responseHeaders, value: response } = await (0, import_provider_utils2.postJsonToApi)({
       url: `${this.config.baseURL}/chat/completions`,
-      headers: (0, import_provider_utils3.combineHeaders)(this.config.headers(), options.headers),
+      headers: (0, import_provider_utils2.combineHeaders)(this.config.headers(), options.headers),
       body,
       failedResponseHandler: mistralFailedResponseHandler,
-      successfulResponseHandler: (0, import_provider_utils3.createEventSourceResponseHandler)(
+      successfulResponseHandler: (0, import_provider_utils2.createEventSourceResponseHandler)(
         mistralChatChunkSchema
       ),
       abortSignal: options.abortSignal,
       fetch: this.config.fetch
     });
-    const { messages: rawPrompt, ...rawSettings } = args;
     let finishReason = "unknown";
-    let usage = {
-      promptTokens: Number.NaN,
-      completionTokens: Number.NaN
+    const usage = {
+      inputTokens: void 0,
+      outputTokens: void 0,
+      totalTokens: void 0
     };
     let chunkNumber = 0;
     let trimLeadingSpace = false;
     return {
       stream: response.pipeThrough(
         new TransformStream({
+          start(controller) {
+            controller.enqueue({ type: "stream-start", warnings });
+          },
           transform(chunk, controller) {
             if (!chunk.success) {
               controller.enqueue({ type: "error", error: chunk.error });
@@ -432,10 +433,9 @@ var MistralChatLanguageModel = class {
               });
             }
             if (value.usage != null) {
-              usage = {
-                promptTokens: value.usage.prompt_tokens,
-                completionTokens: value.usage.completion_tokens
-              };
+              usage.inputTokens = value.usage.prompt_tokens;
+              usage.outputTokens = value.usage.completion_tokens;
+              usage.totalTokens = value.usage.total_tokens;
             }
             const choice = value.choices[0];
             if ((choice == null ? void 0 : choice.finish_reason) != null) {
@@ -447,7 +447,7 @@ var MistralChatLanguageModel = class {
             const delta = choice.delta;
             const textContent = extractTextContent(delta.content);
             if (chunkNumber <= 2) {
-              const lastMessage = rawPrompt[rawPrompt.length - 1];
+              const lastMessage = body.messages[body.messages.length - 1];
               if (lastMessage.role === "assistant" && textContent === lastMessage.content.trimEnd()) {
                 if (textContent.length < lastMessage.content.length) {
                   trimLeadingSpace = true;
@@ -457,8 +457,8 @@ var MistralChatLanguageModel = class {
             }
             if (textContent != null) {
               controller.enqueue({
-                type: "text-delta",
-                textDelta: trimLeadingSpace ? textContent.trimStart() : textContent
+                type: "text",
+                text: trimLeadingSpace ? textContent.trimStart() : textContent
               });
               trimLeadingSpace = false;
             }
@@ -486,10 +486,8 @@ var MistralChatLanguageModel = class {
           }
         })
       ),
-      rawCall: { rawPrompt, rawSettings },
-      rawResponse: { headers: responseHeaders },
-      request: { body: JSON.stringify(body) },
-      warnings
+      request: { body },
+      response: { headers: responseHeaders }
     };
   }
 };
@@ -518,105 +516,97 @@ function extractTextContent(content) {
   }
   return textContent.length ? textContent.join("") : void 0;
 }
-var mistralContentSchema = import_zod2.z.union([
-  import_zod2.z.string(),
-  import_zod2.z.array(
-    import_zod2.z.discriminatedUnion("type", [
-      import_zod2.z.object({
-        type: import_zod2.z.literal("text"),
-        text: import_zod2.z.string()
+var mistralContentSchema = import_zod3.z.union([
+  import_zod3.z.string(),
+  import_zod3.z.array(
+    import_zod3.z.discriminatedUnion("type", [
+      import_zod3.z.object({
+        type: import_zod3.z.literal("text"),
+        text: import_zod3.z.string()
       }),
-      import_zod2.z.object({
-        type: import_zod2.z.literal("image_url"),
-        image_url: import_zod2.z.union([
-          import_zod2.z.string(),
-          import_zod2.z.object({
-            url: import_zod2.z.string(),
-            detail: import_zod2.z.string().nullable()
+      import_zod3.z.object({
+        type: import_zod3.z.literal("image_url"),
+        image_url: import_zod3.z.union([
+          import_zod3.z.string(),
+          import_zod3.z.object({
+            url: import_zod3.z.string(),
+            detail: import_zod3.z.string().nullable()
           })
         ])
       }),
-      import_zod2.z.object({
-        type: import_zod2.z.literal("reference"),
-        reference_ids: import_zod2.z.array(import_zod2.z.number())
+      import_zod3.z.object({
+        type: import_zod3.z.literal("reference"),
+        reference_ids: import_zod3.z.array(import_zod3.z.number())
       })
     ])
   )
 ]).nullish();
-var mistralChatResponseSchema = import_zod2.z.object({
-  id: import_zod2.z.string().nullish(),
-  created: import_zod2.z.number().nullish(),
-  model: import_zod2.z.string().nullish(),
-  choices: import_zod2.z.array(
-    import_zod2.z.object({
-      message: import_zod2.z.object({
-        role: import_zod2.z.literal("assistant"),
+var mistralUsageSchema = import_zod3.z.object({
+  prompt_tokens: import_zod3.z.number(),
+  completion_tokens: import_zod3.z.number(),
+  total_tokens: import_zod3.z.number()
+});
+var mistralChatResponseSchema = import_zod3.z.object({
+  id: import_zod3.z.string().nullish(),
+  created: import_zod3.z.number().nullish(),
+  model: import_zod3.z.string().nullish(),
+  choices: import_zod3.z.array(
+    import_zod3.z.object({
+      message: import_zod3.z.object({
+        role: import_zod3.z.literal("assistant"),
         content: mistralContentSchema,
-        tool_calls: import_zod2.z.array(
-          import_zod2.z.object({
-            id: import_zod2.z.string(),
-            function: import_zod2.z.object({ name: import_zod2.z.string(), arguments: import_zod2.z.string() })
+        tool_calls: import_zod3.z.array(
+          import_zod3.z.object({
+            id: import_zod3.z.string(),
+            function: import_zod3.z.object({ name: import_zod3.z.string(), arguments: import_zod3.z.string() })
           })
         ).nullish()
       }),
-      index: import_zod2.z.number(),
-      finish_reason: import_zod2.z.string().nullish()
+      index: import_zod3.z.number(),
+      finish_reason: import_zod3.z.string().nullish()
     })
   ),
-  object: import_zod2.z.literal("chat.completion"),
-  usage: import_zod2.z.object({
-    prompt_tokens: import_zod2.z.number(),
-    completion_tokens: import_zod2.z.number()
-  })
+  object: import_zod3.z.literal("chat.completion"),
+  usage: mistralUsageSchema
 });
-var mistralChatChunkSchema = import_zod2.z.object({
-  id: import_zod2.z.string().nullish(),
-  created: import_zod2.z.number().nullish(),
-  model: import_zod2.z.string().nullish(),
-  choices: import_zod2.z.array(
-    import_zod2.z.object({
-      delta: import_zod2.z.object({
-        role: import_zod2.z.enum(["assistant"]).optional(),
+var mistralChatChunkSchema = import_zod3.z.object({
+  id: import_zod3.z.string().nullish(),
+  created: import_zod3.z.number().nullish(),
+  model: import_zod3.z.string().nullish(),
+  choices: import_zod3.z.array(
+    import_zod3.z.object({
+      delta: import_zod3.z.object({
+        role: import_zod3.z.enum(["assistant"]).optional(),
         content: mistralContentSchema,
-        tool_calls: import_zod2.z.array(
-          import_zod2.z.object({
-            id: import_zod2.z.string(),
-            function: import_zod2.z.object({ name: import_zod2.z.string(), arguments: import_zod2.z.string() })
+        tool_calls: import_zod3.z.array(
+          import_zod3.z.object({
+            id: import_zod3.z.string(),
+            function: import_zod3.z.object({ name: import_zod3.z.string(), arguments: import_zod3.z.string() })
           })
         ).nullish()
       }),
-      finish_reason: import_zod2.z.string().nullish(),
-      index: import_zod2.z.number()
+      finish_reason: import_zod3.z.string().nullish(),
+      index: import_zod3.z.number()
     })
   ),
-  usage: import_zod2.z.object({
-    prompt_tokens: import_zod2.z.number(),
-    completion_tokens: import_zod2.z.number()
-  }).nullish()
+  usage: mistralUsageSchema.nullish()
 });
 
 // src/mistral-embedding-model.ts
 var import_provider3 = require("@ai-sdk/provider");
-var import_provider_utils4 = require("@ai-sdk/provider-utils");
-var import_zod3 = require("zod");
+var import_provider_utils3 = require("@ai-sdk/provider-utils");
+var import_zod4 = require("zod");
 var MistralEmbeddingModel = class {
-  constructor(modelId, settings, config) {
-    this.specificationVersion = "v1";
+  constructor(modelId, config) {
+    this.specificationVersion = "v2";
+    this.maxEmbeddingsPerCall = 32;
+    this.supportsParallelCalls = false;
     this.modelId = modelId;
-    this.settings = settings;
     this.config = config;
   }
   get provider() {
     return this.config.provider;
   }
-  get maxEmbeddingsPerCall() {
-    var _a;
-    return (_a = this.settings.maxEmbeddingsPerCall) != null ? _a : 32;
-  }
-  get supportsParallelCalls() {
-    var _a;
-    return (_a = this.settings.supportsParallelCalls) != null ? _a : false;
-  }
   async doEmbed({
     values,
     abortSignal,
@@ -630,16 +620,20 @@ var MistralEmbeddingModel = class {
         values
       });
     }
-    const { responseHeaders, value: response } = await (0, import_provider_utils4.postJsonToApi)({
+    const {
+      responseHeaders,
+      value: response,
+      rawValue
+    } = await (0, import_provider_utils3.postJsonToApi)({
       url: `${this.config.baseURL}/embeddings`,
-      headers: (0, import_provider_utils4.combineHeaders)(this.config.headers(), headers),
+      headers: (0, import_provider_utils3.combineHeaders)(this.config.headers(), headers),
       body: {
         model: this.modelId,
         input: values,
         encoding_format: "float"
       },
       failedResponseHandler: mistralFailedResponseHandler,
-      successfulResponseHandler: (0, import_provider_utils4.createJsonResponseHandler)(
+      successfulResponseHandler: (0, import_provider_utils3.createJsonResponseHandler)(
         MistralTextEmbeddingResponseSchema
       ),
       abortSignal,
@@ -648,52 +642,55 @@ var MistralEmbeddingModel = class {
     return {
       embeddings: response.data.map((item) => item.embedding),
       usage: response.usage ? { tokens: response.usage.prompt_tokens } : void 0,
-      rawResponse: { headers: responseHeaders }
+      response: { headers: responseHeaders, body: rawValue }
     };
   }
 };
-var MistralTextEmbeddingResponseSchema = import_zod3.z.object({
-  data: import_zod3.z.array(import_zod3.z.object({ embedding: import_zod3.z.array(import_zod3.z.number()) })),
-  usage: import_zod3.z.object({ prompt_tokens: import_zod3.z.number() }).nullish()
+var MistralTextEmbeddingResponseSchema = import_zod4.z.object({
+  data: import_zod4.z.array(import_zod4.z.object({ embedding: import_zod4.z.array(import_zod4.z.number()) })),
+  usage: import_zod4.z.object({ prompt_tokens: import_zod4.z.number() }).nullish()
 });
 
 // src/mistral-provider.ts
 function createMistral(options = {}) {
   var _a;
-  const baseURL = (_a = (0, import_provider_utils5.withoutTrailingSlash)(options.baseURL)) != null ? _a : "https://api.mistral.ai/v1";
+  const baseURL = (_a = (0, import_provider_utils4.withoutTrailingSlash)(options.baseURL)) != null ? _a : "https://api.mistral.ai/v1";
   const getHeaders = () => ({
-    Authorization: `Bearer ${(0, import_provider_utils5.loadApiKey)({
+    Authorization: `Bearer ${(0, import_provider_utils4.loadApiKey)({
       apiKey: options.apiKey,
       environmentVariableName: "MISTRAL_API_KEY",
       description: "Mistral"
     })}`,
     ...options.headers
   });
-  const createChatModel = (modelId, settings = {}) => new MistralChatLanguageModel(modelId, settings, {
+  const createChatModel = (modelId) => new MistralChatLanguageModel(modelId, {
     provider: "mistral.chat",
     baseURL,
     headers: getHeaders,
     fetch: options.fetch
   });
-  const createEmbeddingModel = (modelId, settings = {}) => new MistralEmbeddingModel(modelId, settings, {
+  const createEmbeddingModel = (modelId) => new MistralEmbeddingModel(modelId, {
     provider: "mistral.embedding",
     baseURL,
     headers: getHeaders,
     fetch: options.fetch
   });
-  const provider = function(modelId, settings) {
+  const provider = function(modelId) {
     if (new.target) {
       throw new Error(
         "The Mistral model function cannot be called with the new keyword."
       );
     }
-    return createChatModel(modelId, settings);
+    return createChatModel(modelId);
   };
   provider.languageModel = createChatModel;
   provider.chat = createChatModel;
   provider.embedding = createEmbeddingModel;
   provider.textEmbedding = createEmbeddingModel;
   provider.textEmbeddingModel = createEmbeddingModel;
+  provider.imageModel = (modelId) => {
+    throw new import_provider4.NoSuchModelError({ modelId, modelType: "imageModel" });
+  };
   return provider;
 }
 var mistral = createMistral();