AI/firecrawl
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/mendableai/firecrawl synced 2025-08-12 01:59:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(scrape): bill llm on fail

Browse Source
This commit is contained in:
Gergő Móricz 2025-04-19 02:56:18 -07:00
parent b35462456e
commit 0e027fe430
1 changed files with 29 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
apps/api/src/controllers/v1/scrape.ts
View File

@ -74,6 +74,35 @@ export async function scrapeController(
scrapeId: jobId,
startTime,
});
let creditsToBeBilled = 0;
if (req.body.agent?.model?.toLowerCase() === "fire-1" || req.body.extract?.agent?.model?.toLowerCase() === "fire-1" || req.body.jsonOptions?.agent?.model?.toLowerCase() === "fire-1") {
if (process.env.USE_DB_AUTHENTICATION === "true") {
// @Nick this is a hack pushed at 2AM pls help - mogery
const job = await supabaseGetJobById(jobId);
if (!job?.cost_tracking) {
logger.warn("No cost tracking found for job", {
jobId,
});
}
creditsToBeBilled = Math.ceil((job?.cost_tracking?.totalCost ?? 1) * 1800);
} else {
creditsToBeBilled = 150;
}
}
if (creditsToBeBilled > 0) {
billTeam(req.auth.team_id, req.acuc?.sub_id, creditsToBeBilled).catch(
(error) => {
logger.error(
`Failed to bill team ${req.auth.team_id} for ${creditsToBeBilled} credits: ${error}`,
);
// Optionally, you could notify an admin or add to a retry queue here
},
);
}
if (
e instanceof Error &&
(e.message.startsWith("Job wait") || e.message === "timeout")