fix error

This commit is contained in:
songtianlun 2025-09-02 00:18:05 +08:00
parent f3e7fb913d
commit 779f50ea87
2 changed files with 2 additions and 2 deletions

View File

@ -130,7 +130,7 @@ export async function POST(
// Check user's credit balance before execution // Check user's credit balance before execution
const userBalance = await getUserBalance(user.id); const userBalance = await getUserBalance(user.id);
const costMultiplier = (run.user.subscriptionPlan as { costMultiplier?: number })?.costMultiplier || 1.0; const costMultiplier = (run.user.subscriptionPlan as { costMultiplier?: number })?.costMultiplier || 1.0;
const estimatedCost = calculateCost(0, 100, run.model, costMultiplier); // Rough estimate const estimatedCost = calculateCost(50, 100, run.model, costMultiplier); // Rough estimate
if (userBalance < estimatedCost) { if (userBalance < estimatedCost) {
return NextResponse.json( return NextResponse.json(

View File

@ -53,7 +53,7 @@ export function calculateCost(
inputCostPer1k?: number | null inputCostPer1k?: number | null
outputCostPer1k?: number | null outputCostPer1k?: number | null
}, },
// costMultiplier?: number costMultiplier?: number
): number { ): number {
const inputCostPer1k = model.inputCostPer1k || 0 const inputCostPer1k = model.inputCostPer1k || 0
const outputCostPer1k = model.outputCostPer1k || 0 const outputCostPer1k = model.outputCostPer1k || 0