AI/firecrawl
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/mendableai/firecrawl synced 2025-08-13 18:08:58 +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,
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 { saveCrawl, StoredCrawl } from "../crawl-redis";
import { updateExtract } from "./extract-redis";
import { CUSTOM_U_TEAMS } from "./config";
interface ExtractServiceOptions {
request: ExtractRequest;
@ -182,10 +183,15 @@ export async function performExtraction(extractId: string, options: ExtractServi
// }, {}, crypto.randomUUID(), 50);
// });
let linksBilled = links.length * 5;
if(CUSTOM_U_TEAMS.includes(teamId)){
linksBilled = 1;
}
// Bill team for usage
billTeam(teamId, subId, links.length * 5).catch((error) => {
billTeam(teamId, subId, linksBilled).catch((error) => {
logger.error(
`Failed to bill team ${teamId} for ${links.length * 5} credits: ${error}`,
`Failed to bill team ${teamId} for ${linksBilled} credits: ${error}`,
);
});