fix: adapt preview team checks

This commit is contained in:
Gergő Móricz 2025-01-25 19:02:32 +01:00
parent 4d8f4109b5
commit f3982c0894
3 changed files with 6 additions and 6 deletions

View File

@ -48,7 +48,7 @@ export async function supaBillTeam(
credits,
});
if (team_id === "preview") {
if (team_id === "preview" || team_id.startsWith("preview_")) {
return { success: true, message: "Preview team, no credits used" };
}
_logger.info(`Billing team ${team_id} for ${credits} credits`);
@ -109,7 +109,7 @@ export async function supaCheckTeamCredits(
credits: number,
): Promise<CheckTeamCreditsResponse> {
// WARNING: chunk will be null if team_id is preview -- do not perform operations on it under ANY circumstances - mogery
if (team_id === "preview") {
if (team_id === "preview" || team_id.startsWith("preview_")) {
return {
success: true,
message: "Preview team, no credits used",

View File

@ -50,7 +50,7 @@ export async function logJob(job: FirecrawlJob, force: boolean = false) {
num_docs: job.num_docs,
docs: cleanOfNull(job.docs),
time_taken: job.time_taken,
team_id: job.team_id === "preview" ? null : job.team_id,
team_id: (job.team_id === "preview" || job.team_id?.startsWith("preview_"))? null : job.team_id,
mode: job.mode,
url: job.url,
crawler_options: job.crawlerOptions,
@ -112,7 +112,7 @@ export async function logJob(job: FirecrawlJob, force: boolean = false) {
if (process.env.POSTHOG_API_KEY && !job.crawl_id) {
let phLog = {
distinctId: "from-api", //* To identify this on the group level, setting distinctid to a static string per posthog docs: https://posthog.com/docs/product-analytics/group-analytics#advanced-server-side-only-capturing-group-events-without-a-user
...(job.team_id !== "preview" && {
...((job.team_id !== "preview" && !job.team_id?.startsWith("preview_")) && {
groups: { team: job.team_id },
}), //* Identifying event on this team
event: "job-logged",
@ -121,7 +121,7 @@ export async function logJob(job: FirecrawlJob, force: boolean = false) {
message: job.message,
num_docs: job.num_docs,
time_taken: job.time_taken,
team_id: job.team_id === "preview" ? null : job.team_id,
team_id: (job.team_id === "preview" || job.team_id?.startsWith("preview_"))? null : job.team_id,
mode: job.mode,
url: job.url,
crawler_options: job.crawlerOptions,

View File

@ -86,7 +86,7 @@ export async function sendNotificationInternal(
chunk: AuthCreditUsageChunk,
bypassRecentChecks: boolean = false,
): Promise<{ success: boolean }> {
if (team_id === "preview") {
if (team_id === "preview" || team_id.startsWith("preview_")) {
return { success: true };
}
return await redlock.using(