diff --git a/src/Infrastructure/BotSharp.OpenAPI/Controllers/InstructModeController.cs b/src/Infrastructure/BotSharp.OpenAPI/Controllers/InstructModeController.cs
index d1538ee2..9259bb8d 100644
--- a/src/Infrastructure/BotSharp.OpenAPI/Controllers/InstructModeController.cs
+++ b/src/Infrastructure/BotSharp.OpenAPI/Controllers/InstructModeController.cs
@@ -286,7 +286,7 @@ public class InstructModeController : ControllerBase
     }
 
     [HttpPost("/instruct/image-edit")]
-    public async Task<ImageGenerationViewModel> ImageEdit([FromBody] ImageEditRequest request)
+    public async Task<ImageGenerationViewModel> ImageEdit([FromBody] ImageEditFileRequest request)
     {
         var fileInstruct = _services.GetRequiredService<IFileInstructService>();
         var state = _services.GetRequiredService<IConversationStateService>();
diff --git a/src/Infrastructure/BotSharp.OpenAPI/ViewModels/Instructs/Request/InstructBaseRequest.cs b/src/Infrastructure/BotSharp.OpenAPI/ViewModels/Instructs/Request/InstructBaseRequest.cs
index 1071d58b..6344540f 100644
--- a/src/Infrastructure/BotSharp.OpenAPI/ViewModels/Instructs/Request/InstructBaseRequest.cs
+++ b/src/Infrastructure/BotSharp.OpenAPI/ViewModels/Instructs/Request/InstructBaseRequest.cs
@@ -50,13 +50,17 @@ public class ImageEditRequest : InstructBaseRequest
     [JsonPropertyName("text")]
     public string Text { get; set; } = string.Empty;
 
-    [JsonPropertyName("file")]
-    public InstructFileModel? File { get; set; }
-
     [JsonPropertyName("image_converter_provider")]
     public string? ImageConverterProvider { get; set; } = "file-handler";
 }
 
+public class ImageEditFileRequest : ImageEditRequest
+{
+    [JsonPropertyName("file")]
+    public InstructFileModel File { get; set; }
+}
+
+
 public class ImageMaskEditRequest : InstructBaseRequest
 {
     [JsonPropertyName("text")]