image strugles
This commit is contained in:
@@ -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')
|
||||
|
||||
|
||||
Reference in New Issue
Block a user