diff --git a/app/MindWork AI Studio/Settings/ProviderExtensions.Mistral.cs b/app/MindWork AI Studio/Settings/ProviderExtensions.Mistral.cs
index 3d0150c9..597b0aae 100644
--- a/app/MindWork AI Studio/Settings/ProviderExtensions.Mistral.cs	
+++ b/app/MindWork AI Studio/Settings/ProviderExtensions.Mistral.cs	
@@ -23,9 +23,12 @@ public static partial class ProviderExtensions
         if (modelName.IndexOf("mistral-large-") is not -1)
             return
             [
-                Capability.TEXT_INPUT, Capability.MULTIPLE_IMAGE_INPUT,
+                Capability.TEXT_INPUT, 
+                Capability.SINGLE_IMAGE_INPUT, Capability.MULTIPLE_IMAGE_INPUT,
                 Capability.TEXT_OUTPUT,
                 
+                Capability.OPTIONAL_REASONING,
+                
                 Capability.FUNCTION_CALLING,
                 Capability.CHAT_COMPLETION_API,
             ];
@@ -34,9 +37,12 @@ public static partial class ProviderExtensions
         if (modelName.IndexOf("mistral-medium-") is not -1)
             return
             [
-                Capability.TEXT_INPUT, Capability.MULTIPLE_IMAGE_INPUT,
+                Capability.TEXT_INPUT, 
+                Capability.SINGLE_IMAGE_INPUT, Capability.MULTIPLE_IMAGE_INPUT,
                 Capability.TEXT_OUTPUT,
                 
+                Capability.OPTIONAL_REASONING,
+                
                 Capability.FUNCTION_CALLING,
                 Capability.CHAT_COMPLETION_API,
             ];
@@ -45,8 +51,11 @@ public static partial class ProviderExtensions
         if (modelName.IndexOf("mistral-small-") is not -1)
             return
             [
-                Capability.TEXT_INPUT, Capability.MULTIPLE_IMAGE_INPUT,
+                Capability.TEXT_INPUT, 
+                Capability.SINGLE_IMAGE_INPUT, Capability.MULTIPLE_IMAGE_INPUT,
                 Capability.TEXT_OUTPUT,
+
+                Capability.OPTIONAL_REASONING,
                 
                 Capability.FUNCTION_CALLING,
                 Capability.CHAT_COMPLETION_API,
diff --git a/app/MindWork AI Studio/Settings/ProviderExtensions.OpenSource.cs b/app/MindWork AI Studio/Settings/ProviderExtensions.OpenSource.cs
index dc30e53b..b9b31aa2 100644
--- a/app/MindWork AI Studio/Settings/ProviderExtensions.OpenSource.cs	
+++ b/app/MindWork AI Studio/Settings/ProviderExtensions.OpenSource.cs	
@@ -150,13 +150,30 @@ public static partial class ProviderExtensions
                 modelName.IndexOf("mistral-large-3") is not -1)
                 return
                 [
-                    Capability.TEXT_INPUT, Capability.MULTIPLE_IMAGE_INPUT,
+                    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-") is not -1)
+                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("voxtral-") is not -1)
                 return
                 [
@@ -305,4 +322,4 @@ public static partial class ProviderExtensions
             Capability.CHAT_COMPLETION_API,
         ];
     }
-}
\ No newline at end of file
+}