Skip to content

Commit 476acae

Browse files
committed
fix: disable unconfigured ai services
1 parent c976887 commit 476acae

File tree

1 file changed

+18
-8
lines changed

1 file changed

+18
-8
lines changed

src/backend/src/modules/puterai/PuterAIModule.js

+18-8
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
const { AdvancedBase } = require("@heyputer/puter-js-common");
2+
const config = require("../../config");
23

34
class PuterAIModule extends AdvancedBase {
45
async install (context) {
@@ -7,17 +8,26 @@ class PuterAIModule extends AdvancedBase {
78
const { AIInterfaceService } = require('./AIInterfaceService');
89
services.registerService('__ai-interfaces', AIInterfaceService);
910

10-
const { AWSTextractService } = require('./AWSTextractService');
11-
services.registerService('aws-textract', AWSTextractService);
11+
// TODO: services should govern their own availability instead of
12+
// the module deciding what to register
1213

13-
const { AWSPollyService } = require('./AWSPollyService');
14-
services.registerService('aws-polly', AWSPollyService);
14+
if ( !! config?.services?.['aws-textract']?.aws ) {
15+
const { AWSTextractService } = require('./AWSTextractService');
16+
services.registerService('aws-textract', AWSTextractService);
17+
}
1518

16-
const { OpenAICompletionService } = require('./OpenAICompletionService');
17-
services.registerService('openai-completion', OpenAICompletionService);
19+
if ( !! config?.services?.['aws-polly']?.aws ) {
20+
const { AWSPollyService } = require('./AWSPollyService');
21+
services.registerService('aws-polly', AWSPollyService);
22+
}
1823

19-
const { OpenAIImageGenerationService } = require('./OpenAIImageGenerationService');
20-
services.registerService('openai-image-generation', OpenAIImageGenerationService);
24+
if ( !! config?.openai ) {
25+
const { OpenAICompletionService } = require('./OpenAICompletionService');
26+
services.registerService('openai-completion', OpenAICompletionService);
27+
28+
const { OpenAIImageGenerationService } = require('./OpenAIImageGenerationService');
29+
services.registerService('openai-image-generation', OpenAIImageGenerationService);
30+
}
2131
}
2232
}
2333

0 commit comments

Comments
 (0)