salman/igny8
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixed site id issue for credits in wdigets for specific site

Browse Source
This commit is contained in:
IGNY8 VPS (Salman)
2026-01-13 00:32:49 +00:00
parent 25caea5d90
commit 78c9cd38e0
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
backend/igny8_core/ai/tasks.py
View File

@@ -765,6 +765,9 @@ def process_image_generation_queue(self, image_ids: list, account_id: int = None
cost_usd = result.get('cost', 0) or result.get('cost_usd', 0) # generate_image returns 'cost' cost_usd = result.get('cost', 0) or result.get('cost_usd', 0) # generate_image returns 'cost'
if account: if account:
try: try:
# Get site_id from content if available
site_id = content.site_id if content else None
credits_deducted = CreditService.deduct_credits_for_image( credits_deducted = CreditService.deduct_credits_for_image(
account=account, account=account,
model_name=model, model_name=model,
@@ -773,6 +776,7 @@ def process_image_generation_queue(self, image_ids: list, account_id: int = None
metadata={ metadata={
'image_id': image_id, 'image_id': image_id,
'content_id': content_id, 'content_id': content_id,
'site_id': site_id,
'provider': provider, 'provider': provider,
'model': model, 'model': model,
'image_type': image.image_type if image else 'unknown', 'image_type': image.image_type if image else 'unknown',
@@ -780,9 +784,10 @@ def process_image_generation_queue(self, image_ids: list, account_id: int = None
}, },
cost_usd=cost_usd, cost_usd=cost_usd,
related_object_type='image', related_object_type='image',
related_object_id=image_id related_object_id=image_id,
site=site_id # Pass site_id for credit logging
) )
logger.info(f"[process_image_generation_queue] Credits deducted for image {image_id}: account balance now {credits_deducted}") logger.info(f"[process_image_generation_queue] Credits deducted for image {image_id}: account balance now {credits_deducted}, site_id={site_id}")
except Exception as credit_error: except Exception as credit_error:
logger.error(f"[process_image_generation_queue] Failed to deduct credits for image {image_id}: {credit_error}") logger.error(f"[process_image_generation_queue] Failed to deduct credits for image {image_id}: {credit_error}")
# Don't fail the image generation if credit deduction fails # Don't fail the image generation if credit deduction fails