From f264b23ccc0cbe63c5d307717efdaf9ae6479f7e Mon Sep 17 00:00:00 2001 From: Qu Yi Date: Fri, 1 Nov 2024 13:39:13 +0800 Subject: [PATCH] Add select_num exception range judgment --- src/agent/prompter.js | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/src/agent/prompter.js b/src/agent/prompter.js index 3ba51dd..d44829e 100644 --- a/src/agent/prompter.js +++ b/src/agent/prompter.js @@ -136,17 +136,19 @@ export class Prompter { })) .sort((a, b) => b.similarity_score - a.similarity_score); - // select_num = -1 means select all - let selected_docs = skill_doc_similarities.slice(0, select_num === -1 ? skill_doc_similarities.length : select_num); + let length = skill_doc_similarities.length; + if (typeof select_num !== 'number' || isNaN(select_num) || select_num <= 0) { + select_num = length; + } else { + select_num = Math.min(Math.floor(select_num), length); + } + let selected_docs = skill_doc_similarities.slice(0, select_num); let message = '\nThe following recommended functions are listed in descending order of task relevance.\nSkillDocs:\n'; message += selected_docs.map(doc => `${doc.doc_key}`).join('\n'); + console.log(message); return message; } - - - - async replaceStrings(prompt, messages, examples=null, to_summarize=[], last_goals=null) { prompt = prompt.replaceAll('$NAME', this.agent.name);