github_mirrors/puter
1
0
Fork 0
mirror of https://github.com/HeyPuter/puter.git synced 2025-02-02 23:28:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: image generation supports test mode

Browse Source
This commit is contained in:
KernelDeimos 2024-11-22 09:22:24 -05:00
parent 2d4312972a
commit f533dca1a6
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/backend/src/modules/puterai/OpenAIImageGenerationService.js
View File

@ -17,6 +17,12 @@ class OpenAIImageGenerationService extends BaseService {
} }
static IMPLEMENTS = { static IMPLEMENTS = {
['driver-capabilities']: {
supports_test_mode (iface, method_name) {
return iface === 'puter-image-generation' &&
method_name === 'generate';
}
},
['puter-image-generation']: { ['puter-image-generation']: {
async generate ({ prompt, test_mode }) { async generate ({ prompt, test_mode }) {
if ( test_mode ) { if ( test_mode ) {