@ax-llm/ax 12.0.3 → 12.0.4

package/index.cjs

@@ -4164,7 +4164,8 @@ var axModelInfoMistral = [
 // ai/mistral/api.ts
 var axAIMistralDefaultConfig = () => structuredClone({
   model: "mistral-small-latest" /* MistralSmall */,
-  ...axBaseAIDefaultConfig()
+  ...axBaseAIDefaultConfig(),
+  topP: 1
 });
 var axAIMistralBestConfig = () => structuredClone({
   ...axAIMistralDefaultConfig(),
@@ -4192,6 +4193,15 @@ var AxAIMistral = class extends AxAIOpenAIBase {
       hasThinkingBudget: false,
       hasShowThoughts: false
     };
+    const chatReqUpdater = (req) => {
+      const { max_completion_tokens, stream_options, messages, ...result } = req;
+      return {
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        ...result,
+        messages: this.updateMessages(messages),
+        max_tokens: max_completion_tokens
+      };
+    };
     super({
       apiKey,
       config: _config,
@@ -4199,10 +4209,32 @@ var AxAIMistral = class extends AxAIOpenAIBase {
       apiURL: "https://api.mistral.ai/v1",
       modelInfo,
       models,
-      supportFor
+      supportFor,
+      // eslint-disable-next-line @typescript-eslint/no-explicit-any
+      chatReqUpdater
     });
     super.setName("Mistral");
   }
+  updateMessages(messages) {
+    const messagesUpdated = [];
+    if (!Array.isArray(messages)) {
+      return messages;
+    }
+    for (const message of messages) {
+      if (message.role === "user" && Array.isArray(message.content)) {
+        const contentUpdated = message.content.map((item) => {
+          if (typeof item === "object" && item !== null && item.type === "image_url") {
+            return { type: "image_url", image_url: item.image_url?.url };
+          }
+          return item;
+        });
+        messagesUpdated.push({ ...message, content: contentUpdated });
+      } else {
+        messagesUpdated.push(message);
+      }
+    }
+    return messagesUpdated;
+  }
 };
 
 // ai/ollama/api.ts