kunthawat/ALwrity
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix preflight model mapping when skipping invalid providers

Browse Source
This commit is contained in:
ي
2026-03-05 11:10:54 +05:30
parent 2318fd8a48
commit 81052d06b4
1 changed files with 14 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
backend/api/subscription/routes/preflight.py
View File

@@ -75,7 +75,10 @@ async def preflight_check(
'provider': provider_enum,
'tokens_requested': op.tokens_requested or 0,
'actual_provider_name': op.actual_provider_name or op.provider,
'operation_type': op.operation_type
'operation_type': op.operation_type,
# Keep the originating request fields together so model lookup
# cannot drift when invalid providers are skipped.
'model': op.model
})
except Exception as e:
logger.warning(f"Error processing operation {op.operation_type}: {e}")
@@ -94,7 +97,7 @@ async def preflight_check(
operation_results = []
total_cost = 0.0
for i, op in enumerate(operations_to_validate):
for op in operations_to_validate:
op_result = {
'provider': op['actual_provider_name'],
'operation_type': op['operation_type'],
@@ -105,7 +108,7 @@ async def preflight_check(
}
# Get pricing for this operation
model_name = request.operations[i].model
model_name = op.get('model')
if model_name:
pricing_info = pricing_service.get_pricing_for_provider_model(
op['provider'],
@@ -124,11 +127,15 @@ async def preflight_check(
chars = max(0, int(op.get('tokens_requested') or 0))
cost = max(0.005, 0.005 * (chars / 100.0))
else:
# Audio pricing is per character (every character is 1 token)
cost = (pricing_info.get('cost_per_input_token', 0.0) or 0.0) * (op['tokens_requested'] / 1000.0)
# Audio pricing uses per-token/per-character unit pricing from DB.
# Do not divide by 1000 here: pricing values are already normalized
# as per-unit costs in APIProviderPricing.
cost = (pricing_info.get('cost_per_input_token', 0.0) or 0.0) * op['tokens_requested']
elif op['tokens_requested'] > 0:
# Token-based cost estimation (rough estimate)
cost = (pricing_info.get('cost_per_input_token', 0.0) or 0.0) * (op['tokens_requested'] / 1000)
# Token-based cost estimation (rough estimate).
# IMPORTANT: cost_per_input_token is stored as cost-per-token.
# Multiplying by tokens_requested gives the correct estimate.
cost = (pricing_info.get('cost_per_input_token', 0.0) or 0.0) * op['tokens_requested']
else:
cost = pricing_info.get('cost_per_request', 0.0) or 0.0