From a3c69f13576a747077e542cb42ca7cc9159ac7c6 Mon Sep 17 00:00:00 2001
From: hart_s3 <sabrina.hartmann@dlr.de>
Date: Thu, 2 Apr 2026 11:38:22 +0200
Subject: [PATCH] Modify the capability of the Mistral models

---
 .../Settings/ProviderExtensions.Mistral.cs    | 40 +++++++++++++++++--
 .../Settings/ProviderExtensions.OpenSource.cs |  4 +-
 .../wwwroot/changelog/v26.3.1.md              |  1 +
 3 files changed, 40 insertions(+), 5 deletions(-)

diff --git a/app/MindWork AI Studio/Settings/ProviderExtensions.Mistral.cs b/app/MindWork AI Studio/Settings/ProviderExtensions.Mistral.cs
index 597b0aae..9868505d 100644
--- a/app/MindWork AI Studio/Settings/ProviderExtensions.Mistral.cs	
+++ b/app/MindWork AI Studio/Settings/ProviderExtensions.Mistral.cs	
@@ -24,7 +24,17 @@ public static partial class ProviderExtensions
             return
             [
                 Capability.TEXT_INPUT, 
-                Capability.SINGLE_IMAGE_INPUT, Capability.MULTIPLE_IMAGE_INPUT,
+                Capability.TEXT_OUTPUT,
+                
+                Capability.FUNCTION_CALLING,
+                Capability.CHAT_COMPLETION_API,
+            ];
+        
+        if (modelName.IndexOf("mistral-large-latest") is not -1)                  // The latest mistral-Large-3 model
+            return
+            [
+                Capability.TEXT_INPUT, 
+                Capability.MULTIPLE_IMAGE_INPUT,
                 Capability.TEXT_OUTPUT,
                 
                 Capability.OPTIONAL_REASONING,
@@ -38,7 +48,19 @@ public static partial class ProviderExtensions
             return
             [
                 Capability.TEXT_INPUT, 
-                Capability.SINGLE_IMAGE_INPUT, Capability.MULTIPLE_IMAGE_INPUT,
+                Capability.TEXT_OUTPUT,
+                
+                Capability.OPTIONAL_REASONING,
+                
+                Capability.FUNCTION_CALLING,
+                Capability.CHAT_COMPLETION_API,
+            ];
+        
+        if (modelName.IndexOf("mistral-medium-latest") is not -1)           // The latest mistral-medium-3.1 model
+            return
+            [
+                Capability.TEXT_INPUT, 
+                Capability.MULTIPLE_IMAGE_INPUT,
                 Capability.TEXT_OUTPUT,
                 
                 Capability.OPTIONAL_REASONING,
@@ -52,7 +74,19 @@ public static partial class ProviderExtensions
             return
             [
                 Capability.TEXT_INPUT, 
-                Capability.SINGLE_IMAGE_INPUT, Capability.MULTIPLE_IMAGE_INPUT,
+                Capability.TEXT_OUTPUT,
+
+                Capability.OPTIONAL_REASONING,
+                
+                Capability.FUNCTION_CALLING,
+                Capability.CHAT_COMPLETION_API,
+            ];
+        
+        if (modelName.IndexOf("mistral-small-latest") is not -1)        // The latest mistral-medium-3.1 model
+            return
+            [
+                Capability.TEXT_INPUT, 
+                Capability.MULTIPLE_IMAGE_INPUT,
                 Capability.TEXT_OUTPUT,
 
                 Capability.OPTIONAL_REASONING,
diff --git a/app/MindWork AI Studio/Settings/ProviderExtensions.OpenSource.cs b/app/MindWork AI Studio/Settings/ProviderExtensions.OpenSource.cs
index b9b31aa2..106acd7b 100644
--- a/app/MindWork AI Studio/Settings/ProviderExtensions.OpenSource.cs	
+++ b/app/MindWork AI Studio/Settings/ProviderExtensions.OpenSource.cs	
@@ -151,7 +151,7 @@ public static partial class ProviderExtensions
                 return
                 [
                     Capability.TEXT_INPUT, 
-                    Capability.SINGLE_IMAGE_INPUT, Capability.MULTIPLE_IMAGE_INPUT,
+                    Capability.MULTIPLE_IMAGE_INPUT,
                     Capability.TEXT_OUTPUT,
                 
                     Capability.OPTIONAL_REASONING,
@@ -164,7 +164,7 @@ public static partial class ProviderExtensions
                 return
                 [
                     Capability.TEXT_INPUT, 
-                    Capability.SINGLE_IMAGE_INPUT, Capability.MULTIPLE_IMAGE_INPUT,
+                    Capability.MULTIPLE_IMAGE_INPUT,
                     Capability.TEXT_OUTPUT,
 
                     Capability.OPTIONAL_REASONING,
diff --git a/app/MindWork AI Studio/wwwroot/changelog/v26.3.1.md b/app/MindWork AI Studio/wwwroot/changelog/v26.3.1.md
index 05ee5fce..7e5e8817 100644
--- a/app/MindWork AI Studio/wwwroot/changelog/v26.3.1.md	
+++ b/app/MindWork AI Studio/wwwroot/changelog/v26.3.1.md	
@@ -6,6 +6,7 @@
 - Added the ability to load a system prompt from a file when creating or editing chat templates.
 - Added a start-page setting, so AI Studio can now open directly on your preferred page when the app starts. Configuration plugins can also provide and optionally lock this default for organizations.
 - Added math rendering in chats for LaTeX display formulas, including block formats such as `$$ ... $$` and `\[ ... \]`.
+- Added the latest Mistral models.
 - Released the document analysis assistant after an intense testing phase.
 - Improved enterprise deployment for organizations: administrators can now provide up to 10 centrally managed enterprise configuration slots, use policy files on Linux and macOS, and continue using older configuration formats as a fallback during migration.
 - Improved the profile selection for assistants and the chat. You can now explicitly choose between the app default profile, no profile, or a specific profile.