From 82a4ab08846dc381e3680e52809a80035f2f0248 Mon Sep 17 00:00:00 2001 From: KernelDeimos Date: Thu, 3 Apr 2025 17:59:34 -0400 Subject: [PATCH] dev: migrate image gen service to use cost service --- .../modules/puterai/OpenAIImageGenerationService.js | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/src/backend/src/modules/puterai/OpenAIImageGenerationService.js b/src/backend/src/modules/puterai/OpenAIImageGenerationService.js index 6d36a705f..e42605707 100644 --- a/src/backend/src/modules/puterai/OpenAIImageGenerationService.js +++ b/src/backend/src/modules/puterai/OpenAIImageGenerationService.js @@ -173,16 +173,21 @@ class OpenAIImageGenerationService extends BaseService { }); } + const exact_cost = this.models_[model][price_key] + * 100 // $ USD to cents USD + * Math.pow(10,6) // cents to microcents + const svc_cost = this.services.get('cost'); const usageAllowed = await svc_cost.get_funding_allowed({ - minimum: this.models_[model][price_key] - * 100 // $ USD to cents USD - * Math.pow(10,6) // cents to microcents + minimum: exact_cost, }); if ( ! usageAllowed ) { throw APIError.create('insufficient_funds'); } + + // We can charge immediately + await svc_cost.record_cost({ cost: exact_cost }); const result = await this.openai.images.generate({ user: user_private_uid,