diff --git a/src/backend/src/modules/puterai/AIInterfaceService.js b/src/backend/src/modules/puterai/AIInterfaceService.js
index 5ec2ccb88..521c53458 100644
--- a/src/backend/src/modules/puterai/AIInterfaceService.js
+++ b/src/backend/src/modules/puterai/AIInterfaceService.js
@@ -98,6 +98,8 @@ class AIInterfaceService extends BaseService {
                         quality: { type: 'string' },
                         model: { type: 'string' },
                         ratio: { type: 'json' },
+                        input_image: { type: 'string', optional: true },
+                        input_image_mime_type: { type: 'string', optional: true },
                     },
                     result_choices: [
                         {