diff --git a/settings.js b/settings.js index d3899b4..f252363 100644 --- a/settings.js +++ b/settings.js @@ -36,6 +36,7 @@ export default "allow_insecure_coding": false, // allows newAction command and model can write/run code on your computer. enable at own risk "allow_vision": false, // allows vision model to interpret screenshots as inputs + "blocked_actions" : [], // commands to disable and remove from docs. Ex: ["!setMode"] "code_timeout_mins": -1, // minutes code is allowed to run. -1 for no timeout "relevant_docs_count": 5, // Parameter: -1 = all, 0 = no references, 5 = five references. If exceeding the maximum, all reference documents are returned. diff --git a/src/agent/agent.js b/src/agent/agent.js index fb123f6..370afb0 100644 --- a/src/agent/agent.js +++ b/src/agent/agent.js @@ -47,7 +47,7 @@ export class Agent { await this.prompter.initExamples(); console.log('Initializing task...'); this.task = new Task(this, task_path, task_id); - const blocked_actions = this.task.blocked_actions || []; + const blocked_actions = settings.blocked_actions.concat(this.task.blocked_actions || []); blacklistCommands(blocked_actions); serverProxy.connect(this); diff --git a/src/models/prompter.js b/src/models/prompter.js index b412ee5..efa231e 100644 --- a/src/models/prompter.js +++ b/src/models/prompter.js @@ -152,6 +152,8 @@ export class Prompter { profile.api = 'xai'; else if (profile.model.includes('deepseek')) profile.api = 'deepseek'; + else if (profile.model.includes('mistral')) + profile.api = 'mistral'; else if (profile.model.includes('llama3')) profile.api = 'ollama'; else