github/node-DeepResearch
1
0
Fork 0
mirror of https://github.com/jina-ai/node-DeepResearch.git synced 2026-03-22 07:29:35 +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
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 {analyzeSteps} from "./tools/error-analyzer";
import {GEMINI_API_KEY, JINA_API_KEY, SEARCH_PROVIDER, STEP_SLEEP, modelConfigs} from "./config"; import {GEMINI_API_KEY, JINA_API_KEY, SEARCH_PROVIDER, STEP_SLEEP, modelConfigs} from "./config";
import {tokenTracker} from "./utils/token-tracker"; import {tokenTracker} from "./utils/token-tracker";
import {StepAction, SchemaProperty, ResponseSchema} from "./types"; import {StepAction, SchemaProperty, ResponseSchema, AnswerAction} from "./types";
async function sleep(ms: number) { async function sleep(ms: number) {
const seconds = Math.ceil(ms / 1000); const seconds = Math.ceil(ms / 1000);
@@ -241,7 +241,7 @@ function removeAllLineBreaks(text: string) {
return text.replace(/(\r\n|\n|\r)/gm, " "); 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 step = 0;
let totalStep = 0; let totalStep = 0;
let badAttempts = 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); tokenTracker.trackUsage('agent', usage?.totalTokenCount || 0);
await storeContext(prompt, [allContext, allKeywords, allQuestions, allKnowledge], totalStep); 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() { export async function main() {
const question = process.argv[2] || ""; const question = process.argv[2] || "";
const finalStep = await getResponse(question); const finalStep = await getResponse(question) as AnswerAction;
console.log('Final Answer:', finalStep.answer); console.log('Final Answer:', finalStep.answer);
tokenTracker.printSummary(); tokenTracker.printSummary();
} }