minor change

SciSharp · Jul 19, 2024 · 4311eaf · 4311eaf
1 parent bb9feb2
commit 4311eaf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/...Plugins/BotSharp.Plugin.AzureOpenAI/Providers/Image/ImageCompletionProvider.Generation.cs b/...Plugins/BotSharp.Plugin.AzureOpenAI/Providers/Image/ImageCompletionProvider.Generation.cs
@@ -45,7 +45,7 @@ public async Task<RoleDialogModel> GetImageGeneration(Agent agent, RoleDialogMod
             GeneratedImages = generatedImages
         };
 
-        return responseMessage;
+        return await Task.FromResult(responseMessage);
     }
 
     private (string, int, ImageGenerationOptions) PrepareOptions(RoleDialogModel message)

diff --git a/src/Plugins/BotSharp.Plugin.OpenAI/Providers/Image/ImageCompletionProvider.Variation.cs b/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 = await imageClient.GenerateImageVariationsAsync(image, imageFileName, imageCount, options);
+        var response = imageClient.GenerateImageVariations(image, imageFileName, imageCount, options);
         var values = response.Value;
 
         var generatedImages = new List<ImageGeneration>();