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

fix: remove unnecessary budget guard

Browse Source
This commit is contained in:
Han Xiao
2025-02-05 10:41:29 +08:00
parent 2d489812a3
commit d562d518ed
1 changed files with 0 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/agent.ts
View File

@@ -309,13 +309,6 @@ export async function getResponse(question: string, tokenBudget: number = 1_000_
} }
}); });
// Check if we have enough budget for this operation (estimate 50 tokens for prompt + response)
const estimatedTokens = 50;
const currentUsage = context.tokenTracker.getTotalUsage();
if (currentUsage + estimatedTokens > tokenBudget) {
throw new Error(`Token budget would be exceeded: ${currentUsage + estimatedTokens} > ${tokenBudget}`);
}
const result = await model.generateContent(prompt); const result = await model.generateContent(prompt);
const response = await result.response; const response = await result.response;
const usage = response.usageMetadata; const usage = response.usageMetadata;
@@ -635,13 +628,6 @@ You decided to think out of the box or cut from a completely different angle.`);
} }
}); });
// Check if we have enough budget for this operation (estimate 50 tokens for prompt + response)
const estimatedTokens = 50;
const currentUsage = context.tokenTracker.getTotalUsage();
if (currentUsage + estimatedTokens > tokenBudget) {
throw new Error(`Token budget would be exceeded: ${currentUsage + estimatedTokens} > ${tokenBudget}`);
}
const result = await model.generateContent(prompt); const result = await model.generateContent(prompt);
const response = await result.response; const response = await result.response;
const usage = response.usageMetadata; const usage = response.usageMetadata;