Update analyzeSchemaAndPrompt.ts (#1108)

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

View File

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