fix error
This commit is contained in:
parent
f3e7fb913d
commit
779f50ea87
@ -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(
|
||||||
|
@ -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
|
||||||
|
Loading…
Reference in New Issue
Block a user