Skip to content

Commit f533dca

Browse files
committed
fix: image generation supports test mode
1 parent 2d43129 commit f533dca

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

src/backend/src/modules/puterai/OpenAIImageGenerationService.js

+6
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,12 @@ class OpenAIImageGenerationService extends BaseService {
1717
}
1818

1919
static IMPLEMENTS = {
20+
['driver-capabilities']: {
21+
supports_test_mode (iface, method_name) {
22+
return iface === 'puter-image-generation' &&
23+
method_name === 'generate';
24+
}
25+
},
2026
['puter-image-generation']: {
2127
async generate ({ prompt, test_mode }) {
2228
if ( test_mode ) {

0 commit comments

Comments
 (0)