Added support for images (#609)

app/MindWork AI Studio/Tools/RAG/IRetrievalContextExtensions.cs

@@ -81,7 +81,9 @@ public static class IRetrievalContextExtensions
                 sb.AppendLine();
                 sb.AppendLine("Matched image content as base64-encoded data:");
                 sb.AppendLine("````");
-                sb.AppendLine(await imageContext.AsBase64(token));
+                sb.AppendLine(await imageContext.TryAsBase64(token) is (success: true, { } base64Image)
+                        ? base64Image 
+                        : string.Empty);
                 sb.AppendLine("````");
                 break;
                 

app/MindWork AI Studio/Tools/Validation/FileExtensionValidation.cs

@@ -1,3 +1,5 @@
+using AIStudio.Provider;
+using AIStudio.Settings;
 using AIStudio.Tools.PluginSystem;
 using AIStudio.Tools.Rust;
 
@@ -10,15 +12,38 @@ public static class FileExtensionValidation
 {
     private static string TB(string fallbackEN) => I18N.I.T(fallbackEN, typeof(FileExtensionValidation).Namespace, nameof(FileExtensionValidation));
 
+    /// <summary>
+    /// Defines the use cases for file extension validation.
+    /// </summary>
+    public enum UseCase
+    {
+        /// <summary>
+        /// No specific use case; general validation.
+        /// </summary>
+        NONE,
+        
+        /// <summary>
+        /// Validating for directly loading content into the UI. In this state, there might be no provider selected yet.
+        /// </summary>
+        DIRECTLY_LOADING_CONTENT,
+        
+        /// <summary>
+        /// Validating for attaching content to a message or prompt.
+        /// </summary>
+        ATTACHING_CONTENT,
+    }
+    
     /// <summary>
     /// Validates the file extension and sends appropriate MessageBus notifications when invalid.
     /// </summary>
+    /// <param name="useCae">The validation use case.</param>
     /// <param name="filePath">The file path to validate.</param>
+    /// <param name="provider">The selected provider.</param>
     /// <returns>True if valid, false if invalid (error/warning already sent via MessageBus).</returns>
-    public static async Task<bool> IsExtensionValidWithNotifyAsync(string filePath)
+    public static async Task<bool> IsExtensionValidWithNotifyAsync(UseCase useCae, string filePath, Settings.Provider? provider = null)
     {
-        var ext = Path.GetExtension(filePath).TrimStart('.');
-        if (Array.Exists(FileTypeFilter.Executables.FilterExtensions, x => x.Equals(ext, StringComparison.OrdinalIgnoreCase)))
+        var ext = Path.GetExtension(filePath).TrimStart('.').ToLowerInvariant();
+        if(FileTypeFilter.Executables.FilterExtensions.Contains(ext))
         {
             await MessageBus.INSTANCE.SendError(new(
                 Icons.Material.Filled.AppBlocking,
@@ -26,15 +51,39 @@ public static class FileExtensionValidation
             return false;
         }
 
-        if (Array.Exists(FileTypeFilter.AllImages.FilterExtensions, x => x.Equals(ext, StringComparison.OrdinalIgnoreCase)))
+        var capabilities = provider?.GetModelCapabilities() ?? new();
+        if (FileTypeFilter.AllImages.FilterExtensions.Contains(ext))
         {
-            await MessageBus.INSTANCE.SendWarning(new(
-                Icons.Material.Filled.ImageNotSupported,
-                TB("Images are not supported yet")));
-            return false;
+            switch (useCae)
+            {
+                // In this use case, we cannot guarantee that a provider is selected yet:
+                case UseCase.DIRECTLY_LOADING_CONTENT:
+                    await MessageBus.INSTANCE.SendWarning(new(
+                        Icons.Material.Filled.ImageNotSupported,
+                        TB("Images are not supported at this place")));
+                    return false;
+                
+                // In this use case, we can check the provider capabilities:
+                case UseCase.ATTACHING_CONTENT when capabilities.Contains(Capability.SINGLE_IMAGE_INPUT) ||
+                                                    capabilities.Contains(Capability.MULTIPLE_IMAGE_INPUT):
+                    return true;
+                
+                // We know that images are not supported:
+                case UseCase.ATTACHING_CONTENT:
+                    await MessageBus.INSTANCE.SendWarning(new(
+                        Icons.Material.Filled.ImageNotSupported,
+                        TB("Images are not supported by the selected provider and model")));
+                    return false;
+                
+                default:
+                    await MessageBus.INSTANCE.SendWarning(new(
+                        Icons.Material.Filled.ImageNotSupported,
+                        TB("Images are not supported yet")));
+                    return false;
+            }
         }
 
-        if (Array.Exists(FileTypeFilter.AllVideos.FilterExtensions, x => x.Equals(ext, StringComparison.OrdinalIgnoreCase)))
+        if(FileTypeFilter.AllVideos.FilterExtensions.Contains(ext))
         {
             await MessageBus.INSTANCE.SendWarning(new(
                 Icons.Material.Filled.FeaturedVideo,
@@ -42,7 +91,7 @@ public static class FileExtensionValidation
             return false;
         }
 
-        if (Array.Exists(FileTypeFilter.AllAudio.FilterExtensions, x => x.Equals(ext, StringComparison.OrdinalIgnoreCase)))
+        if(FileTypeFilter.AllAudio.FilterExtensions.Contains(ext))
         {
             await MessageBus.INSTANCE.SendWarning(new(
                 Icons.Material.Filled.AudioFile,