fix: image generation supports test mode

2025-02-02 14:18:43 +08:00 · 2024-11-22 09:22:24 -05:00 · 2024-11-22 09:22:24 -05:00 · f533dca1a6
commit f533dca1a6
parent 2d4312972a
1 changed files with 6 additions and 0 deletions
--- 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 ) {