fix(scrape/json): move back to 4o mini

This commit is contained in:
Gergő Móricz 2025-04-17 10:48:13 -07:00
parent 8546bcacc0
commit 06770bc63f
3 changed files with 5 additions and 6 deletions

View File

@ -51,7 +51,8 @@ export async function singleAnswerCompletion({
},
markdown: `${singleAnswerDocs.map((x, i) => `[START_PAGE (ID: ${i})]` + buildDocument(x)).join("\n")} [END_PAGE]\n`,
isExtractEndpoint: true,
model: getModel("gemini-2.0-flash", "google"),
model: getModel("gemini-2.5-pro-preview-03-25", "vertex"),
retryModel: getModel("gemini-2.5-pro-preview-03-25", "google"),
costTrackingOptions: {
costTracking,
metadata: {
@ -75,7 +76,7 @@ export async function singleAnswerCompletion({
promptTokens: 0,
completionTokens: 0,
totalTokens: 0,
model: "gemini-2.0-flash",
model: "gemini-2.5-pro-preview-03-25",
},
sources: singleAnswerDocs.map(
(doc) => doc.metadata.url || doc.metadata.sourceURL || "",

View File

@ -269,8 +269,6 @@ export async function extractData({
totalUsage: t,
} = await generateCompletions({
...extractOptionsNewSchema,
model: getModel("gemini-2.5-pro-preview-03-25", "vertex"),
retryModel: getModel("gemini-2.5-pro-preview-03-25", "google"),
costTrackingOptions: {
costTracking: extractOptions.costTrackingOptions.costTracking,
metadata: {

View File

@ -658,8 +658,8 @@ export async function performLLMExtract(
// model: getModel("qwen-qwq-32b", "groq"),
// model: getModel("gemini-2.0-flash", "google"),
// model: getModel("gemini-2.5-pro-preview-03-25", "vertex"),
model: getModel("gemini-2.5-pro-preview-03-25", "vertex"),
retryModel: getModel("gemini-2.5-pro-preview-03-25", "google"),
model: getModel("gpt-4o-mini", "openai"),
retryModel: getModel("gpt-4o", "openai"),
costTrackingOptions: {
costTracking: meta.costTracking,
metadata: {