billingconfig

ainize-team · Sep 13, 2023 · de87a41 · de87a41
1 parent 778b297
commit de87a41
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/src/modules/app.ts b/src/modules/app.ts
@@ -161,11 +161,19 @@ export default class App extends ModuleBase {
    */
     async checkCostAndBalance(appName: string, serviceName: string, value: string, requesterAddress?: string) {
       requesterAddress = requesterAddress ? requesterAddress : this.getDefaultAccount().address;
-      const billingConfig = await this.getBillingConfig(appName);
+      const billingConfig = (await this.getBillingConfig(appName)).service;
+      const serviceBillingConfig = billingConfig.default;
       // TODO(woojae): calculate cost more accurately
-      let serviceBillingConfig = billingConfig.service.default;
-      if(billingConfig.service[serviceName]) {
-        serviceBillingConfig = billingConfig.service[serviceName];
+      if(billingConfig[serviceName]) {
+        if(billingConfig[serviceName].costPerToken) {
+          serviceBillingConfig.costPerToken = billingConfig[serviceName].costPerToken;
+        }
+        if(billingConfig[serviceName].minCost) {
+          serviceBillingConfig.minCost = billingConfig[serviceName].minCost;
+        }
+        if(billingConfig[serviceName].maxCost) {
+          serviceBillingConfig.maxCost = billingConfig[serviceName].maxCost;
+        }
       }
       const token = value.split(' ').length;
       let cost = token * serviceBillingConfig.costPerToken;