From f533dca1a6d88ca7a14bd69f15d0a151e24c58e1 Mon Sep 17 00:00:00 2001
From: KernelDeimos <eric.alex.dube@gmail.com>
Date: Fri, 22 Nov 2024 09:22:24 -0500
Subject: [PATCH] fix: image generation supports test mode

---
 .../src/modules/puterai/OpenAIImageGenerationService.js     | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/backend/src/modules/puterai/OpenAIImageGenerationService.js b/src/backend/src/modules/puterai/OpenAIImageGenerationService.js
index e715788d7..b4271e124 100644
--- 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 ) {