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,20 +76,34 @@ 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 {
isMultiEntity,
multiEntityKeys,
reasoning,
keyIndicators,
tokenUsage,
};
} catch (e) {
logger.warn("(analyzeSchemaAndPrompt) Error parsing schema analysis", {
error: e,
});
}
return { return {
isMultiEntity, isMultiEntity: false,
multiEntityKeys, multiEntityKeys: [],
reasoning, reasoning: "",
keyIndicators, keyIndicators: [],
tokenUsage, tokenUsage,
}; };
} }