92752a2a15
* Improve MCP functionality * apply ANSI strip path * attempt ARM build * improve dockerfile IO build time and migrate to ARM build * fix comment * add ability to disable MCP cooldown feature * update devbuild name * move chromium arm build patch to CDN
99 lines
3.5 KiB
JavaScript
99 lines
3.5 KiB
JavaScript
const { Deduplicator } = require("../utils/dedupe");
|
|
|
|
const saveFileInBrowser = {
|
|
name: "save-file-to-browser",
|
|
startupConfig: {
|
|
params: {},
|
|
},
|
|
plugin: function () {
|
|
return {
|
|
name: this.name,
|
|
setup(aibitat) {
|
|
// List and summarize the contents of files that are embedded in the workspace
|
|
aibitat.function({
|
|
super: aibitat,
|
|
tracker: new Deduplicator(),
|
|
name: this.name,
|
|
description:
|
|
"Save content to a file when the user explicitly asks for a download of the file.",
|
|
examples: [
|
|
{
|
|
prompt: "Save me that to a file named 'output'",
|
|
call: JSON.stringify({
|
|
file_content:
|
|
"<content of the file we will write previous conversation>",
|
|
filename: "output.txt",
|
|
}),
|
|
},
|
|
{
|
|
prompt: "Save me that to my desktop",
|
|
call: JSON.stringify({
|
|
file_content:
|
|
"<content of the file we will write previous conversation>",
|
|
filename: "<relevant filename>.txt",
|
|
}),
|
|
},
|
|
{
|
|
prompt: "Save me that to a file",
|
|
call: JSON.stringify({
|
|
file_content:
|
|
"<content of the file we will write from previous conversation>",
|
|
filename: "<descriptive filename>.txt",
|
|
}),
|
|
},
|
|
],
|
|
parameters: {
|
|
$schema: "http://json-schema.org/draft-07/schema#",
|
|
type: "object",
|
|
properties: {
|
|
file_content: {
|
|
type: "string",
|
|
description: "The content of the file that will be saved.",
|
|
},
|
|
filename: {
|
|
type: "string",
|
|
description:
|
|
"filename to save the file as with extension. Extension should be plaintext file extension.",
|
|
},
|
|
},
|
|
additionalProperties: false,
|
|
},
|
|
handler: async function ({ file_content = "", filename }) {
|
|
try {
|
|
const { isDuplicate, reason } = this.tracker.isDuplicate(
|
|
this.name,
|
|
{ file_content, filename }
|
|
);
|
|
if (isDuplicate) {
|
|
this.super.handlerProps.log(
|
|
`${this.name} was called, but exited early because ${reason}.`
|
|
);
|
|
return `${filename} file has been saved successfully!`;
|
|
}
|
|
|
|
this.super.socket.send("fileDownload", {
|
|
filename,
|
|
b64Content:
|
|
"data:text/plain;base64," +
|
|
Buffer.from(file_content, "utf8").toString("base64"),
|
|
});
|
|
this.super.introspect(`${this.caller}: Saving file ${filename}.`);
|
|
this.tracker.trackRun(this.name, { file_content, filename });
|
|
return `${filename} file has been saved successfully and will be downloaded automatically to the users browser.`;
|
|
} catch (error) {
|
|
this.super.handlerProps.log(
|
|
`save-file-to-browser raised an error. ${error.message}`
|
|
);
|
|
return `Let the user know this action was not successful. An error was raised while saving a file to the browser. ${error.message}`;
|
|
}
|
|
},
|
|
});
|
|
},
|
|
};
|
|
},
|
|
};
|
|
|
|
module.exports = {
|
|
saveFileInBrowser,
|
|
};
|