image strugles

This commit is contained in:
IGNY8 VPS (Salman)
2026-01-10 11:54:18 +00:00
parent 622e66b0fb
commit 1246f8ac5d
3 changed files with 8 additions and 6 deletions

View File

@@ -118,9 +118,9 @@ class ModelRegistry:
# Handle AIModelConfig instance
if rate_type == 'input':
return model.input_cost_per_1m or Decimal('0')
return model.cost_per_1k_input or Decimal('0')
elif rate_type == 'output':
return model.output_cost_per_1m or Decimal('0')
return model.cost_per_1k_output or Decimal('0')
elif rate_type == 'image':
return model.cost_per_image or Decimal('0')