Fix: Preserve images with file attachments across providers

Detect when the latest message contains images and, instead of clearing them, run a memory/kernel search to build contextual text for the LLM. GeminiService and LLMService now: perform SearchAsync, delete the temporary index, aggregate citation text into a context block, inject that context into the message content (nulling Files), call Send, then restore the original message content and return the result. LLMService also performs additional model/resource cleanup when disableCache is true (dispose models/generator and embedder weights, remove model from loader). DeepSeekService and OllamaService no longer null out Images on the message so image data is preserved for the new image-handling flow.

Author: Madzionator

src/MaIN.Services/Services/LLMService/DeepSeekService.cs

@@ -67,7 +67,6 @@ protected override void ValidateApiKey()
 
         chat.Messages.Last().Content = message.Content;
         chat.Messages.Last().Files = [];
-        chat.Messages.Last().Images = null;
         var result = await Send(chat, requestOptions, cancellationToken);
         chat.Messages.Last().Content = lastMsg.Content;
         return result;

src/MaIN.Services/Services/LLMService/GeminiService.cs

@@ -98,6 +98,27 @@ protected override void ValidateApiKey()
                 $"{userQuery} | For your next response only, please respond using exactly the following JSON format: \n{jsonGrammar}\n. Do not include any explanations, code blocks, or additional content. After this single JSON response, resume your normal conversational style.";
         }
 
+        var lastMessage = chat.Messages.Last();
+        if (lastMessage.Images?.Count > 0)
+        {
+            var searchResult = await kernel.SearchAsync(userQuery, cancellationToken: cancellationToken);
+            await kernel.DeleteIndexAsync(cancellationToken: cancellationToken);
+
+            var ctxBuilder = new StringBuilder();
+            foreach (var citation in searchResult.Results.SelectMany(r => r.Partitions))
+                ctxBuilder.AppendLine(citation.Text);
+
+            var originalContent = lastMessage.Content;
+            if (ctxBuilder.Length > 0)
+                lastMessage.Content =
+                    $"Use the following context to answer the question:\n\n{ctxBuilder}\n\nQuestion: {originalContent}";
+            lastMessage.Files = null;
+
+            var result = await Send(chat, requestOptions, cancellationToken);
+            lastMessage.Content = originalContent;
+            return result;
+        }
+
         MemoryAnswer retrievedContext;
 
         if (requestOptions.InteractiveUpdates || requestOptions.TokenCallback != null)

src/MaIN.Services/Services/LLMService/LLMService.cs

@@ -129,7 +129,37 @@ public Task CleanSessionCache(string? id)
 
         await memoryService.ImportDataToMemory((km, generator), memoryOptions, cancellationToken);
         var userMessage = chat.Messages.Last();
-        
+
+        if (userMessage.Images?.Count > 0)
+        {
+            var searchResult = await km.SearchAsync(userMessage.Content, cancellationToken: cancellationToken);
+            await km.DeleteIndexAsync(cancellationToken: cancellationToken);
+
+            if (disableCache)
+            {
+                llmModel.Dispose();
+                ModelLoader.RemoveModel(model.FileName);
+                textGenerator.Dispose();
+            }
+            generator._embedder.Dispose();
+            generator._embedder._weights.Dispose();
+            generator.Dispose();
+
+            var ctxBuilder = new StringBuilder();
+            foreach (var citation in searchResult.Results.SelectMany(r => r.Partitions))
+                ctxBuilder.AppendLine(citation.Text);
+
+            var originalContent = userMessage.Content;
+            if (ctxBuilder.Length > 0)
+                userMessage.Content =
+                    $"Use the following context to answer the question:\n\n{ctxBuilder}\n\nQuestion: {originalContent}";
+            userMessage.Files = null;
+
+            var chatResult = await Send(chat, requestOptions, cancellationToken);
+            userMessage.Content = originalContent;
+            return chatResult;
+        }
+
         MemoryAnswer result;
 
         var tokens = new List<LLMTokenValue>();

src/MaIN.Services/Services/LLMService/OllamaService.cs

@@ -58,7 +58,6 @@ protected override void ValidateApiKey()
 
         chat.Messages.Last().Content = message.Content;
         chat.Messages.Last().Files = [];
-        chat.Messages.Last().Images = null;
         var result = await Send(chat, requestOptions, cancellationToken);
         chat.Messages.Last().Content = lastMsg.Content;
         return result;