github/node-DeepResearch
1
0
Fork 0
mirror of https://github.com/jina-ai/node-DeepResearch.git synced 2025-12-26 06:28:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: eval logics

Browse Source
This commit is contained in:
Han Xiao 2025-03-20 10:54:14 +08:00
parent 4523f10283
commit 6983ead24a
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/utils/safe-generator.ts
View File

@ -179,7 +179,6 @@ export class ObjectGeneratorSafe {
});
} else {
// Second fallback: Try with fallback model if provided
const fallbackModel = getModel('fallback');
console.error(`${model} failed on object generation -> manual parsing failed -> trying fallback with distilled schema`);
try {
let failedOutput = '';
@ -194,21 +193,21 @@ export class ObjectGeneratorSafe {
const distilledSchema = this.createDistilledSchema(schema);
const fallbackResult = await generateObject({
model: fallbackModel,
model: getModel('fallback'),
schema: distilledSchema,
prompt: `Following the given JSON schema, extract the field from below: \n\n ${failedOutput}`,
maxTokens: getToolConfig('fallback').maxTokens,
temperature: getToolConfig('fallback').temperature,
});
this.tokenTracker.trackUsage(fallbackModel, fallbackResult.usage); // Track against fallback model
this.tokenTracker.trackUsage('fallback', fallbackResult.usage); // Track against fallback model
console.log('Distilled schema parse success!');
return fallbackResult;
} catch (fallbackError) {
// If fallback model also fails, try parsing its error response
try {
const lastChanceResult = await this.handleGenerateObjectError<T>(fallbackError);
this.tokenTracker.trackUsage(fallbackModel, lastChanceResult.usage);
this.tokenTracker.trackUsage('fallback', lastChanceResult.usage);
return lastChanceResult;
} catch (finalError) {
console.error(`All recovery mechanisms failed`);