From 4cdfeacce6bffb50fdabdc835d7a33e7ebfd9c8f Mon Sep 17 00:00:00 2001 From: Jicheng Lu Date: Tue, 2 Jul 2024 18:39:06 -0500 Subject: [PATCH] minor change --- .../Files/Functions/GenerateImageFn.cs | 23 ++++--------------- 1 file changed, 4 insertions(+), 19 deletions(-) diff --git a/src/Infrastructure/BotSharp.Core/Files/Functions/GenerateImageFn.cs b/src/Infrastructure/BotSharp.Core/Files/Functions/GenerateImageFn.cs index eb9a49e3..18c34c12 100644 --- a/src/Infrastructure/BotSharp.Core/Files/Functions/GenerateImageFn.cs +++ b/src/Infrastructure/BotSharp.Core/Files/Functions/GenerateImageFn.cs @@ -81,26 +81,11 @@ public class GenerateImageFn : IFunctionCallback { if (images.IsNullOrEmpty()) return; - var files = new List(); - foreach (var image in images) + var files = images.Where(x => !string.IsNullOrEmpty(x?.ImageData)).Select(x => new BotSharpFile { - if (string.IsNullOrEmpty(image?.ImageData)) - { - continue; - } - - try - { - var name = $"{Guid.NewGuid()}.png"; - var data = $"data:image/png;base64,{image.ImageData}"; - files.Add(new BotSharpFile { FileName = name, FileData = data }); - } - catch (Exception ex) - { - _logger.LogWarning($"Error when saving generated image: {image.ImageData}\r\n{ex.Message}"); - continue; - } - } + FileName = $"{Guid.NewGuid()}.png", + FileData = $"data:image/png;base64,{x.ImageData}" + }).ToList(); var fileService = _services.GetRequiredService(); fileService.SaveMessageFiles(_conversationId, _messageId, FileSourceType.Bot, files);