mirrors/mindcraft
1
0
Fork 0
mirror of https://github.com/kolbytn/mindcraft.git synced 2025-08-11 01:35:34 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add select_num exception range judgment

Browse source
This commit is contained in:
Qu Yi 2024-11-01 13:39:13 +08:00
parent 90df61d2be
commit f264b23ccc
1 changed files with 8 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
src/agent/prompter.js
View file

@ -136,17 +136,19 @@ export class Prompter {
})) }))
.sort((a, b) => b.similarity_score - a.similarity_score); .sort((a, b) => b.similarity_score - a.similarity_score);
// select_num = -1 means select all let length = skill_doc_similarities.length;
let selected_docs = skill_doc_similarities.slice(0, select_num === -1 ? skill_doc_similarities.length : select_num); 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'; 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'); message += selected_docs.map(doc => `${doc.doc_key}`).join('\n');
console.log(message);
return message; return message;
} }
async replaceStrings(prompt, messages, examples=null, to_summarize=[], last_goals=null) { async replaceStrings(prompt, messages, examples=null, to_summarize=[], last_goals=null) {
prompt = prompt.replaceAll('$NAME', this.agent.name); prompt = prompt.replaceAll('$NAME', this.agent.name);