mirror of
https://git.mirrors.martin98.com/https://github.com/mendableai/firecrawl
synced 2025-08-13 18:46:04 +08:00
Update analyzeSchemaAndPrompt.ts (#1108)
This commit is contained in:
parent
d09e0603f8
commit
cf8f7d0ce3
@ -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,
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user