github/node-DeepResearch
1
0
Fork 0
mirror of https://github.com/jina-ai/node-DeepResearch.git synced 2025-12-25 22:16:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: refine subproblem handling and aggregation logic in agent

Browse Source
This commit is contained in:
Han Xiao 2025-06-12 19:57:50 -07:00
parent e8b99a5889
commit a664e4d851
1 changed files with 36 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

66
src/agent.ts
View File

@ -796,39 +796,45 @@ But then you realized you have asked them before. You decided to to think out of
if (teamSize > 1) {
const subproblems = await researchPlan(question, teamSize, soundBites, context, SchemaGen);
// parallel call getResponse for each subproblem with exact same parameters from the current step, but their teamSize is 1
const subproblemResponses = await Promise.all(subproblems.map(subproblem => getResponse(subproblem,
tokenBudget,
maxBadAttempts,
context,
messages,
numReturnedURLs,
noDirectAnswer,
boostHostnames,
badHostnames,
onlyHostnames,
maxRef,
minRelScore, languageCode, searchLanguageCode, searchProvider, withImages, 1)));
// convert current step to AnswerAction
thisStep = {
action: 'answer',
think: thisStep.think,
answer: subproblemResponses.map(r => (r.result as AnswerAction).answer).join('\n\n'),
mdAnswer: subproblemResponses.map(r => (r.result as AnswerAction).mdAnswer).join('\n\n'),
references: subproblemResponses.map(r => (r.result as AnswerAction).references).flat(),
isFinal: true,
isAggregated: true
} as AnswerAction;
candidateAnswers = subproblemResponses.map(r => (r.result as AnswerAction).mdAnswer).filter(a => a) as string[];
if (subproblems.length > 1) {
// aggregate urls
visitedURLs.push(...subproblemResponses.map(r => r.readURLs).flat());
weightedURLs = subproblemResponses.map(r => r.allURLs.map(url => ({ url, title: '' } as BoostedSearchSnippet))).flat();
// parallel call getResponse for each subproblem with exact same parameters from the current step, but their teamSize is 1
const subproblemResponses = await Promise.all(subproblems.map(subproblem => getResponse(subproblem,
tokenBudget,
maxBadAttempts,
context,
messages,
numReturnedURLs,
noDirectAnswer,
boostHostnames,
badHostnames,
onlyHostnames,
maxRef,
minRelScore, languageCode, searchLanguageCode, searchProvider, withImages, 1)));
// convert current step to AnswerAction
thisStep = {
action: 'answer',
think: thisStep.think,
answer: subproblemResponses.map(r => (r.result as AnswerAction).answer).join('\n\n'),
mdAnswer: subproblemResponses.map(r => (r.result as AnswerAction).mdAnswer).join('\n\n'),
references: subproblemResponses.map(r => (r.result as AnswerAction).references).flat(),
isFinal: true,
isAggregated: true
} as AnswerAction;
candidateAnswers = subproblemResponses.map(r => (r.result as AnswerAction).mdAnswer).filter(a => a) as string[];
// TODO aggregate images @shazhou2015
// aggregate urls
visitedURLs.push(...subproblemResponses.map(r => r.readURLs).flat());
weightedURLs = subproblemResponses.map(r => r.allURLs.map(url => ({ url, title: '' } as BoostedSearchSnippet))).flat();
// break the loop, jump directly final boxing
break;
// TODO aggregate images @shazhou2015
// break the loop, jump directly final boxing
break;
} else {
// if there is only one subproblem, then we skip the recurrsion
gaps.push(subproblems[0]);
}
}
// rewrite queries with initial soundbites