fix: image generation supports test mode

HeyPuter · Nov 22, 2024 · f533dca · f533dca
1 parent 2d43129
commit f533dca
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/backend/src/modules/puterai/OpenAIImageGenerationService.js b/src/backend/src/modules/puterai/OpenAIImageGenerationService.js
@@ -17,6 +17,12 @@ class OpenAIImageGenerationService extends BaseService {
     }
 
     static IMPLEMENTS = {
+        ['driver-capabilities']: {
+            supports_test_mode (iface, method_name) {
+                return iface === 'puter-image-generation' &&
+                    method_name === 'generate';
+            }
+        },
         ['puter-image-generation']: {
             async generate ({ prompt, test_mode }) {
                 if ( test_mode ) {