From be5548f5dab94494aedcc7975510645976917e34 Mon Sep 17 00:00:00 2001 From: Han Xiao Date: Sun, 2 Feb 2025 22:41:53 +0800 Subject: [PATCH] chore: first commit --- src/agent.ts | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/src/agent.ts b/src/agent.ts index e95c127..d36f3b1 100644 --- a/src/agent.ts +++ b/src/agent.ts @@ -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 { 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(); }