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

Update analyzeSchemaAndPrompt.ts (#1108)

Browse Source
This commit is contained in:
Nicolas 2025-01-29 12:36:13 -03:00 committed by GitHub
parent d09e0603f8
commit cf8f7d0ce3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 22 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
apps/api/src/lib/extract/completions/analyzeSchemaAndPrompt.ts
View File

@ -6,6 +6,7 @@ import {
buildAnalyzeSchemaUserPrompt,
} from "../build-prompts";
import OpenAI from "openai";
import { logger } from "../../../lib/logger";
const openai = new OpenAI();
export async function analyzeSchemaAndPrompt(
@ -75,20 +76,34 @@ export async function analyzeSchemaAndPrompt(
},
});
const { isMultiEntity, multiEntityKeys, reasoning, keyIndicators } =
checkSchema.parse(result.choices[0].message.parsed);
const tokenUsage: TokenUsage = {
promptTokens: result.usage?.prompt_tokens ?? 0,
completionTokens: result.usage?.completion_tokens ?? 0,
totalTokens: result.usage?.total_tokens ?? 0,
model: model,
};
try {
const { isMultiEntity, multiEntityKeys, reasoning, keyIndicators } =
checkSchema.parse(result.choices[0].message.parsed);
return {
isMultiEntity,
multiEntityKeys,
reasoning,
keyIndicators,
tokenUsage,
};
} catch (e) {
logger.warn("(analyzeSchemaAndPrompt) Error parsing schema analysis", {
error: e,
});
}
return {
isMultiEntity,
multiEntityKeys,
reasoning,
keyIndicators,
isMultiEntity: false,
multiEntityKeys: [],
reasoning: "",
keyIndicators: [],
tokenUsage,
};
}