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: categorize agent tokens as reasoning tokens

Browse Source 
Co-Authored-By: sha.zhou@jina.ai <sha.zhou@jina.ai>
This commit is contained in:
Devin AI 2025-02-11 10:34:12 +00:00
parent 6fbbb5301e
commit 67347d7a40
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/agent.ts
View File

@ -12,7 +12,7 @@ import {evaluateAnswer, evaluateQuestion} from "./tools/evaluator";
import {analyzeSteps} from "./tools/error-analyzer";
import {TokenTracker} from "./utils/token-tracker";
import {ActionTracker} from "./utils/action-tracker";
import {StepAction, AnswerAction} from "./types";
import {StepAction, AnswerAction, TOKEN_CATEGORIES} from "./types";
import {TrackerContext} from "./types";
import {search} from "./tools/jina-search";
// import {grounding} from "./tools/grounding";
@ -366,7 +366,7 @@ export async function getResponse(question: string, tokenBudget: number = 1_000_
console.log(thisStep)
context.actionTracker.trackAction({totalStep, thisStep, gaps, badAttempts});
context.tokenTracker.trackUsage('agent', totalTokens);
context.tokenTracker.trackUsage('agent', totalTokens, TOKEN_CATEGORIES.REASONING);
// reset allowAnswer to true
allowAnswer = true;
@ -707,7 +707,7 @@ You decided to think out of the box or cut from a completely different angle.`);
await storeContext(prompt, schema, [allContext, allKeywords, allQuestions, allKnowledge], totalStep);
thisStep = object as StepAction;
context.actionTracker.trackAction({totalStep, thisStep, gaps, badAttempts});
context.tokenTracker.trackUsage('agent', totalTokens);
context.tokenTracker.trackUsage('agent', totalTokens, TOKEN_CATEGORIES.REASONING);
console.log(thisStep)
return {result: thisStep, context};
}