minor change

Author: iceljc

src/Infrastructure/BotSharp.OpenAPI/Controllers/InstructModeController.cs

@@ -141,6 +141,10 @@ public async Task<ImageGenerationViewModel> ImageVariation([FromBody] IncomingMe
         try
         {
             var file = input.Files.FirstOrDefault(x => !string.IsNullOrWhiteSpace(x.FileUrl) || !string.IsNullOrWhiteSpace(x.FileData));
+            if (file == null)
+            {
+                return new ImageGenerationViewModel { Message = "Error! Cannot find an image!" };
+            }
             var message = await fileService.VarifyImage(input.Provider, input.Model, file);
             imageViewModel.Content = message.Content;
             imageViewModel.Images = message.GeneratedImages.Select(x => ImageViewModel.ToViewModel(x)).ToList();

src/Plugins/BotSharp.Plugin.AzureOpenAI/Providers/Image/ImageCompletionProvider.Generation.cs

@@ -53,18 +53,18 @@ public async Task<RoleDialogModel> GetImageGeneration(Agent agent, RoleDialogMod
         var prompt = message?.Payload ?? message?.Content ?? string.Empty;
 
         var state = _services.GetRequiredService<IConversationStateService>();
-        var size = state.GetState("image_size");
-        var quality = state.GetState("image_quality");
-        var style = state.GetState("image_style");
-        var format = state.GetState("image_format");
+        var size = GetImageSize(state.GetState("image_size"));
+        var quality = GetImageQuality(state.GetState("image_quality"));
+        var style = GetImageStyle(state.GetState("image_style"));
+        var format = GetImageFormat(state.GetState("image_format"));
         var count = GetImageCount(state.GetState("image_count", "1"));
 
         var options = new ImageGenerationOptions
         {
-            Size = GetImageSize(size),
-            Quality = GetImageQuality(quality),
-            Style = GetImageStyle(style),
-            ResponseFormat = GetImageFormat(format)
+            Size = size,
+            Quality = quality,
+            Style = style,
+            ResponseFormat = format
         };
         return (prompt, count, options);
     }

src/Plugins/BotSharp.Plugin.AzureOpenAI/Providers/Image/ImageCompletionProvider.Variation.cs

@@ -51,14 +51,14 @@ public async Task<RoleDialogModel> GetImageVariation(Agent agent, RoleDialogMode
     private (int, ImageVariationOptions) PrepareOptions()
     {
         var state = _services.GetRequiredService<IConversationStateService>();
-        var size = state.GetState("image_size");
-        var format = state.GetState("image_format");
+        var size = GetImageSize(state.GetState("image_size"));
+        var format = GetImageFormat(state.GetState("image_format"));
         var count = GetImageCount(state.GetState("image_count", "1"));
 
         var options = new ImageVariationOptions
         {
-            Size = GetImageSize(size),
-            ResponseFormat = GetImageFormat(format)
+            Size = size,
+            ResponseFormat = format
         };
         return (count, options);
     }

src/Plugins/BotSharp.Plugin.OpenAI/Providers/Image/ImageCompletionProvider.Generation.cs

@@ -53,21 +53,19 @@ public async Task<RoleDialogModel> GetImageGeneration(Agent agent, RoleDialogMod
         var prompt = message?.Payload ?? message?.Content ?? string.Empty;
 
         var state = _services.GetRequiredService<IConversationStateService>();
-        var size = state.GetState("image_size");
-        var quality = state.GetState("image_quality");
-        var style = state.GetState("image_style");
-        var format = state.GetState("image_format");
+        var size = GetImageSize(state.GetState("image_size"));
+        var quality = GetImageQuality(state.GetState("image_quality"));
+        var style = GetImageStyle(state.GetState("image_style"));
+        var format = GetImageFormat(state.GetState("image_format"));
         var count = GetImageCount(state.GetState("image_count", "1"));
 
         var options = new ImageGenerationOptions
         {
-            Size = GetImageSize(size),
-            Quality = GetImageQuality(quality),
-            Style = GetImageStyle(style),
-            ResponseFormat = GetImageFormat(format)
+            Size = size,
+            Quality = quality,
+            Style = style,
+            ResponseFormat = format
         };
         return (prompt, count, options);
     }
-
-    
 }

src/Plugins/BotSharp.Plugin.OpenAI/Providers/Image/ImageCompletionProvider.Variation.cs

@@ -10,7 +10,7 @@ public async Task<RoleDialogModel> GetImageVariation(Agent agent, RoleDialogMode
         var (imageCount, options) = PrepareOptions();
         var imageClient = client.GetImageClient(_model);
 
-        var response = imageClient.GenerateImageVariations(image, imageFileName, imageCount, options);
+        var response = await imageClient.GenerateImageVariationsAsync(image, imageFileName, imageCount, options);
         var values = response.Value;
 
         var generatedImages = new List<ImageGeneration>();
@@ -51,14 +51,14 @@ public async Task<RoleDialogModel> GetImageVariation(Agent agent, RoleDialogMode
     private (int, ImageVariationOptions) PrepareOptions()
     {
         var state = _services.GetRequiredService<IConversationStateService>();
-        var size = state.GetState("image_size");
-        var format = state.GetState("image_format");
+        var size = GetImageSize(state.GetState("image_size"));
+        var format = GetImageFormat(state.GetState("image_format"));
         var count = GetImageCount(state.GetState("image_count", "1"));
 
         var options = new ImageVariationOptions
         {
-            Size = GetImageSize(size),
-            ResponseFormat = GetImageFormat(format)
+            Size = size,
+            ResponseFormat = format
         };
         return (count, options);
     }