ai system improvements

cookbook/helper/ai_helper.py

@@ -1,3 +1,5 @@
+from decimal import Decimal
+
 from django.utils import timezone
 from django.db.models import Sum
 from litellm import CustomLogger
@@ -55,19 +57,15 @@ class AiCallbackHandler(CustomLogger):
         if self.ai_provider.log_credit_cost:
             credit_cost = kwargs.get("response_cost", 0) * 100
 
-        print(not has_monthly_token(self.space) , self.space.ai_credits_balance > 0, not has_monthly_token(self.space) and self.space.ai_credits_balance > 0)
         if (not has_monthly_token(self.space)) and self.space.ai_credits_balance > 0:
-            print('taking credits from balance')
-            self.space.ai_credits_balance = max(0, self.space.ai_credits_balance - credit_cost)
-            print('setting from balance to true')
-            credits_from_balance = True
-            print('saving space')
-            self.space.save()
-            print('done')
-        else:
-            print('not taking credits from balance')
+            remaining_balance = self.space.ai_credits_balance - Decimal(str(credit_cost))
+            if remaining_balance < 0:
+                remaining_balance = 0
+
+            self.space.ai_credits_balance = remaining_balance
+            credits_from_balance = True
+            self.space.save()
 
-        print('creating AI log with credit cost ', credit_cost , ' from balance: ', credits_from_balance)
         AiLog.objects.create(
             created_by=self.user,
             space=self.space,

cookbook/helper/permission_helper.py

@@ -333,6 +333,9 @@ class CustomRecipePermission(permissions.BasePermission):
 class CustomAiProviderPermission(permissions.BasePermission):
     """
     Custom permission class for the AiProvider api endpoint
+    users: can read all
+    admins: can read and write
+    superusers: can read and write + write providers without a space
     """
     message = _('You do not have the required permissions to view this page!')