dev: migrate image gen service to use cost service

2026-08-26 16:07:13 +00:00 · 2025-04-09 14:34:32 -04:00
parent 14304143af
commit 82a4ab0884
1 changed files with 8 additions and 3 deletions
@@ -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,