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

chore: first commit

Browse Source
This commit is contained in:
Han Xiao 2025-02-02 22:41:53 +08:00
parent d38944f976
commit be5548f5da
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/agent.ts
View File

@ -9,7 +9,7 @@ import {evaluateAnswer} from "./tools/evaluator";
import {analyzeSteps} from "./tools/error-analyzer";
import {GEMINI_API_KEY, JINA_API_KEY, SEARCH_PROVIDER, STEP_SLEEP, modelConfigs} from "./config";
import {tokenTracker} from "./utils/token-tracker";
import {StepAction, SchemaProperty, ResponseSchema} from "./types";
import {StepAction, SchemaProperty, ResponseSchema, AnswerAction} from "./types";
async function sleep(ms: number) {
const seconds = Math.ceil(ms / 1000);
@ -241,7 +241,7 @@ function removeAllLineBreaks(text: string) {
return text.replace(/(\r\n|\n|\r)/gm, " ");
}
export async function getResponse(question: string, tokenBudget: number = 1_000_000, maxBadAttempts: number = 3) {
export async function getResponse(question: string, tokenBudget: number = 1_000_000, maxBadAttempts: number = 3): Promise<StepAction> {
let step = 0;
let totalStep = 0;
let badAttempts = 0;
@ -624,7 +624,9 @@ You decided to think out of the box or cut from a completely different angle.`);
tokenTracker.trackUsage('agent', usage?.totalTokenCount || 0);
await storeContext(prompt, [allContext, allKeywords, allQuestions, allKnowledge], totalStep);
return JSON.parse(response.text());
thisStep = JSON.parse(response.text());
console.log(thisStep)
return thisStep;
}
}
@ -646,7 +648,7 @@ const genAI = new GoogleGenerativeAI(GEMINI_API_KEY);
export async function main() {
const question = process.argv[2] || "";
const finalStep = await getResponse(question);
const finalStep = await getResponse(question) as AnswerAction;
console.log('Final Answer:', finalStep.answer);
tokenTracker.printSummary();
}