AI/firecrawl
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/mendableai/firecrawl synced 2025-08-14 15:35:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

Nick: links-billed update (temp)

Browse Source
This commit is contained in:
Nicolas 2025-01-08 15:13:33 -03:00
parent 363021ea78
commit aa31508ccd
2 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
apps/api/src/lib/extract/config.ts
View File

@ -5,3 +5,7 @@ export const extractConfig = {
FALLBACK_SCORE_THRESHOLD: 0.5, FALLBACK_SCORE_THRESHOLD: 0.5,
MIN_REQUIRED_LINKS: 1, MIN_REQUIRED_LINKS: 1,
}; };
export const CUSTOM_U_TEAMS = [
"874d40cc-a5c0-4e93-b661-9ddfbad5e51e"
]

10
apps/api/src/lib/extract/extraction-service.ts
View File

@ -10,6 +10,7 @@ import { logJob } from "../../services/logging/log_job";
import { _addScrapeJobToBullMQ } from "../../services/queue-jobs"; import { _addScrapeJobToBullMQ } from "../../services/queue-jobs";
import { saveCrawl, StoredCrawl } from "../crawl-redis"; import { saveCrawl, StoredCrawl } from "../crawl-redis";
import { updateExtract } from "./extract-redis"; import { updateExtract } from "./extract-redis";
import { CUSTOM_U_TEAMS } from "./config";
interface ExtractServiceOptions { interface ExtractServiceOptions {
request: ExtractRequest; request: ExtractRequest;
@ -182,10 +183,15 @@ export async function performExtraction(extractId: string, options: ExtractServi
// }, {}, crypto.randomUUID(), 50); // }, {}, crypto.randomUUID(), 50);
// }); // });
let linksBilled = links.length * 5;
if(CUSTOM_U_TEAMS.includes(teamId)){
linksBilled = 1;
}
// Bill team for usage // Bill team for usage
billTeam(teamId, subId, links.length * 5).catch((error) => { billTeam(teamId, subId, linksBilled).catch((error) => {
logger.error( logger.error(
`Failed to bill team ${teamId} for ${links.length * 5} credits: ${error}`, `Failed to bill team ${teamId} for ${linksBilled} credits: ${error}`,
); );
}); });