Added support for images (#609)

app/MindWork AI Studio/Settings/ProviderExtensions.cs

@@ -4,26 +4,39 @@ namespace AIStudio.Settings;
 
 public static partial class ProviderExtensions
 {
-    public static List<Capability> GetModelCapabilities(this Provider provider) => provider.UsedLLMProvider switch
+    /// <summary>
+    /// Get the capabilities of the model used by the configured provider.
+    /// </summary>
+    /// <param name="provider">The configured provider.</param>
+    /// <returns>The capabilities of the configured model.</returns>
+    public static List<Capability> GetModelCapabilities(this Provider provider) => provider.UsedLLMProvider.GetModelCapabilities(provider.Model);
+    
+    /// <summary>
+    /// Get the capabilities of a model for a specific provider.
+    /// </summary>
+    /// <param name="provider">The LLM provider.</param>
+    /// <param name="model">The model to get the capabilities for.</param>
+    /// <returns>>The capabilities of the model.</returns>
+    public static List<Capability> GetModelCapabilities(this LLMProviders provider, Model model) => provider switch
     {
-        LLMProviders.OPEN_AI => GetModelCapabilitiesOpenAI(provider.Model),
-        LLMProviders.MISTRAL => GetModelCapabilitiesMistral(provider.Model),
-        LLMProviders.ANTHROPIC => GetModelCapabilitiesAnthropic(provider.Model),
-        LLMProviders.GOOGLE => GetModelCapabilitiesGoogle(provider.Model),
-        LLMProviders.X => GetModelCapabilitiesOpenSource(provider.Model),
-        LLMProviders.DEEP_SEEK => GetModelCapabilitiesDeepSeek(provider.Model),
-        LLMProviders.ALIBABA_CLOUD => GetModelCapabilitiesAlibaba(provider.Model),
-        LLMProviders.PERPLEXITY => GetModelCapabilitiesPerplexity(provider.Model),
-        LLMProviders.OPEN_ROUTER => GetModelCapabilitiesOpenRouter(provider.Model),
+        LLMProviders.OPEN_AI => GetModelCapabilitiesOpenAI(model),
+        LLMProviders.MISTRAL => GetModelCapabilitiesMistral(model),
+        LLMProviders.ANTHROPIC => GetModelCapabilitiesAnthropic(model),
+        LLMProviders.GOOGLE => GetModelCapabilitiesGoogle(model),
+        LLMProviders.X => GetModelCapabilitiesOpenSource(model),
+        LLMProviders.DEEP_SEEK => GetModelCapabilitiesDeepSeek(model),
+        LLMProviders.ALIBABA_CLOUD => GetModelCapabilitiesAlibaba(model),
+        LLMProviders.PERPLEXITY => GetModelCapabilitiesPerplexity(model),
+        LLMProviders.OPEN_ROUTER => GetModelCapabilitiesOpenRouter(model),
 
-        LLMProviders.GROQ => GetModelCapabilitiesOpenSource(provider.Model),
-        LLMProviders.FIREWORKS => GetModelCapabilitiesOpenSource(provider.Model),
-        LLMProviders.HUGGINGFACE => GetModelCapabilitiesOpenSource(provider.Model),
+        LLMProviders.GROQ => GetModelCapabilitiesOpenSource(model),
+        LLMProviders.FIREWORKS => GetModelCapabilitiesOpenSource(model),
+        LLMProviders.HUGGINGFACE => GetModelCapabilitiesOpenSource(model),
         
-        LLMProviders.HELMHOLTZ => GetModelCapabilitiesOpenSource(provider.Model),
-        LLMProviders.GWDG => GetModelCapabilitiesOpenSource(provider.Model),
+        LLMProviders.HELMHOLTZ => GetModelCapabilitiesOpenSource(model),
+        LLMProviders.GWDG => GetModelCapabilitiesOpenSource(model),
         
-        LLMProviders.SELF_HOSTED => GetModelCapabilitiesOpenSource(provider.Model),
+        LLMProviders.SELF_HOSTED => GetModelCapabilitiesOpenSource(model),
         
         _ => []
     };