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

feat: add api params

Browse Source
This commit is contained in:
Han Xiao 2025-02-26 15:34:54 +08:00
parent 142e22a3fe
commit 4cedd57fcc
3 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/agent.ts
View File

@ -261,6 +261,8 @@ export async function getResponse(question?: string,
actionTracker: existingContext?.actionTracker || new ActionTracker()
};
const generator = new ObjectGeneratorSafe(context.tokenTracker);
let schema: ZodObject<any> = SchemaGen.getAgentSchema(true, true, true, true, true)
const gaps: string[] = [question]; // All questions to be answered including the orginal question
const allQuestions = [question];
@ -280,7 +282,9 @@ export async function getResponse(question?: string,
const allURLs: Record<string, SearchResult> = {};
const visitedURLs: string[] = [];
const evaluationMetrics: Record<string, EvaluationType[]> = {};
while (context.tokenTracker.getTotalUsage().totalTokens < tokenBudget && badAttempts <= maxBadAttempts) {
// reserve the 10% final budget for the beast mode
const regularBudget = tokenBudget * 0.9;
while (context.tokenTracker.getTotalUsage().totalTokens < regularBudget && badAttempts <= maxBadAttempts) {
// add 1s delay to avoid rate limiting
step++;
totalStep++;
@ -314,7 +318,6 @@ export async function getResponse(question?: string,
false,
);
schema = SchemaGen.getAgentSchema(allowReflect, allowRead, allowAnswer, allowSearch, allowCoding)
const generator = new ObjectGeneratorSafe(context.tokenTracker);
const result = await generator.generateObject({
model: 'agent',
schema,
@ -721,7 +724,6 @@ But unfortunately, you failed to solve the issue. You need to think out of the b
);
schema = SchemaGen.getAgentSchema(false, false, true, false, false);
const generator = new ObjectGeneratorSafe(context.tokenTracker);
const result = await generator.generateObject({
model: 'agentBeastMode',
schema,

9
src/app.ts
View File

@ -392,11 +392,18 @@ app.post('/v1/chat/completions', (async (req: Request, res: Response) => {
});
console.log('messages', body.messages);
const {tokenBudget, maxBadAttempts} = getTokenBudgetAndMaxAttempts(
let {tokenBudget, maxBadAttempts} = getTokenBudgetAndMaxAttempts(
body.reasoning_effort,
body.max_completion_tokens
);
if (body.budget_tokens) {
tokenBudget = body.budget_tokens;
}
if (body.max_attempts) {
maxBadAttempts = body.max_attempts;
}
const requestId = Date.now().toString();
const created = Math.floor(Date.now() / 1000);
const context: TrackerContext = {

3
src/types.ts
View File

@ -187,6 +187,9 @@ export interface ChatCompletionRequest {
stream?: boolean;
reasoning_effort?: 'low' | 'medium' | 'high' | null;
max_completion_tokens?: number | null;
budget_tokens?: number | null;
max_attempts?: number | null;
}
export interface ChatCompletionResponse {