fix(extract/oldExtract): do if for old/new extract

This commit is contained in:
Gergő Móricz 2025-04-22 00:53:17 +02:00
parent 1afc6258bd
commit f47d8779f5

View File

@ -10,6 +10,7 @@ import * as Sentry from "@sentry/node";
import { saveExtract } from "../../lib/extract/extract-redis"; import { saveExtract } from "../../lib/extract/extract-redis";
import { getTeamIdSyncB } from "../../lib/extract/team-id-sync"; import { getTeamIdSyncB } from "../../lib/extract/team-id-sync";
import { performExtraction } from "../../lib/extract/extraction-service"; import { performExtraction } from "../../lib/extract/extraction-service";
import { performExtraction_F0 } from "../../lib/extract/fire-0/extraction-service-f0";
export async function oldExtract( export async function oldExtract(
req: RequestWithAuth<{}, ExtractResponse, ExtractRequest>, req: RequestWithAuth<{}, ExtractResponse, ExtractRequest>,
@ -19,11 +20,21 @@ export async function oldExtract(
// Means that are in the non-queue system // Means that are in the non-queue system
// TODO: Remove this once all teams have transitioned to the new system // TODO: Remove this once all teams have transitioned to the new system
try { try {
const result = await performExtraction(extractId, { let result;
request: req.body, const model = req.body.agent?.model
teamId: req.auth.team_id, if (req.body.agent && model && model.toLowerCase().includes("fire-1")) {
subId: req.acuc?.sub_id ?? undefined, result = await performExtraction(extractId, {
}); request: req.body,
teamId: req.auth.team_id,
subId: req.acuc?.sub_id ?? undefined,
});
} else {
result = await performExtraction_F0(extractId, {
request: req.body,
teamId: req.auth.team_id,
subId: req.acuc?.sub_id ?? undefined,
});
}
return res.status(200).json(result); return res.status(200).json(result);
} catch (error) { } catch (error) {