Browse Source

fix: vector_size convert bytes to MB. (#1684)

Garfield Dai 1 year ago
parent
commit
1263b7de75
1 changed files with 3 additions and 2 deletions
  1. 3 2
      api/services/billing_service.py

+ 3 - 2
api/services/billing_service.py

@@ -14,8 +14,9 @@ class BillingService:
 
         billing_info = cls._send_request('GET', '/info', params=params)
 
-        vector_size = DatasetService.get_tenant_datasets_usage(tenant_id) / 1024
-        billing_info['vector_space']['size'] = int(vector_size)
+        vector_size = DatasetService.get_tenant_datasets_usage(tenant_id)
+        # Convert bytes to MB
+        billing_info['vector_space']['size'] = int(vector_size / 1024 / 1024)
 
         return billing_info