mirrors/mindcraft
1
0
Fork 0
mirror of https://github.com/kolbytn/mindcraft.git synced 2025-07-03 23:25:20 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Improve the relevance of docs to !newAction("task")

Browse source
This commit is contained in:
Qu Yi 2024-11-09 01:29:24 +08:00
parent e15c516475
commit c8302c27ac
2 changed files with 16 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/agent/action_manager.js
View file

@ -133,7 +133,7 @@ export class ActionManager {
First outputs:\n${output.substring(0, MAX_OUT / 2)}\n...skipping many lines.\nFinal outputs:\n ${output.substring(output.length - MAX_OUT / 2)}`;
}
else {
output = 'Code output:\n' + output;
output = 'Code output:\n' + output.toString();
}
return output;
}

19
src/agent/prompter.js
View file

@ -134,7 +134,7 @@ export class Prompter {
this.coding_examples.load(this.profile.coding_examples),
...getSkillDocs().map(async (doc) => {
let func_name_desc = doc.split('\n').slice(0, 2).join('');
this.skill_docs_embeddings[doc] = await this.embedding_model.embed([func_name_desc]);
this.skill_docs_embeddings[doc] = await this.embedding_model.embed(func_name_desc);
})
]);
@ -199,9 +199,20 @@ export class Prompter {
}
if (prompt.includes('$COMMAND_DOCS'))
prompt = prompt.replaceAll('$COMMAND_DOCS', getCommandDocs());
if (prompt.includes('$CODE_DOCS')){
let latest_message_content = messages.slice().reverse().find(msg => msg.role !== 'system')?.content || '';
prompt = prompt.replaceAll('$CODE_DOCS', await this.getRelevantSkillDocs(latest_message_content, 5));
if (prompt.includes('$CODE_DOCS')) {
// Find the most recent non-system message containing '!newAction('
let code_task_content = messages.slice().reverse().find(msg =>
msg.role !== 'system' && msg.content.includes('!newAction(')
)?.content || '';
// Extract content between '!newAction(' and ')'
const match = code_task_content.match(/!newAction\((.*?)\)/);
code_task_content = match ? match[1] : '';
prompt = prompt.replaceAll(
'$CODE_DOCS',
await this.getRelevantSkillDocs(code_task_content, 5)
);
}
if (prompt.includes('$EXAMPLES') && examples !== null)
prompt = prompt.replaceAll('$EXAMPLES', await examples.createExampleMessage(messages));