mirrors/puter
1
0
Fork 0
mirror of https://github.com/HeyPuter/puter synced 2024-11-14 22:06:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: disable unconfigured ai services

Browse Source
This commit is contained in:
KernelDeimos 2024-08-05 16:04:21 -04:00
parent c97688706e
commit 476acae0e0
1 changed files with 18 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
src/backend/src/modules/puterai/PuterAIModule.js
View File

@ -1,4 +1,5 @@
const { AdvancedBase } = require("@heyputer/puter-js-common");
const config = require("../../config");
class PuterAIModule extends AdvancedBase {
async install (context) {
@ -7,17 +8,26 @@ class PuterAIModule extends AdvancedBase {
const { AIInterfaceService } = require('./AIInterfaceService');
services.registerService('__ai-interfaces', AIInterfaceService);
const { AWSTextractService } = require('./AWSTextractService');
services.registerService('aws-textract', AWSTextractService);
// TODO: services should govern their own availability instead of
// the module deciding what to register
const { AWSPollyService } = require('./AWSPollyService');
services.registerService('aws-polly', AWSPollyService);
if ( !! config?.services?.['aws-textract']?.aws ) {
const { AWSTextractService } = require('./AWSTextractService');
services.registerService('aws-textract', AWSTextractService);
}
const { OpenAICompletionService } = require('./OpenAICompletionService');
services.registerService('openai-completion', OpenAICompletionService);
if ( !! config?.services?.['aws-polly']?.aws ) {
const { AWSPollyService } = require('./AWSPollyService');
services.registerService('aws-polly', AWSPollyService);
}
const { OpenAIImageGenerationService } = require('./OpenAIImageGenerationService');
services.registerService('openai-image-generation', OpenAIImageGenerationService);
if ( !! config?.openai ) {
const { OpenAICompletionService } = require('./OpenAICompletionService');
services.registerService('openai-completion', OpenAICompletionService);
const { OpenAIImageGenerationService } = require('./OpenAIImageGenerationService');
services.registerService('openai-image-generation', OpenAIImageGenerationService);
}
}
}