various fixes

This commit is contained in:
vabene1111
2025-09-18 18:10:58 +02:00
parent f0342d4568
commit bca1ebbf99
9 changed files with 61 additions and 22 deletions

View File

@@ -5,6 +5,7 @@ from django.db.models import Sum
from litellm import CustomLogger
from cookbook.models import AiLog
from recipes import settings
def get_monthly_token_usage(space):
@@ -61,6 +62,8 @@ class AiCallbackHandler(CustomLogger):
remaining_balance = self.space.ai_credits_balance - Decimal(str(credit_cost))
if remaining_balance < 0:
remaining_balance = 0
if settings.HOSTED:
self.space.ai_enabled = False
self.space.ai_credits_balance = remaining_balance
credits_from_balance = True