94 lines
2.8 KiB
JavaScript
94 lines
2.8 KiB
JavaScript
|
"use strict";
|
||
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
||
|
|
exports.WriteFileTool = exports.ReadFileTool = void 0;
|
||
|
|
const zod_1 = require("zod");
|
||
|
|
const tools_1 = require("@langchain/core/tools");
|
||
|
|
/**
|
||
|
|
* Class for reading files from the disk. Extends the StructuredTool
|
||
|
|
* class.
|
||
|
|
*/
|
||
|
|
class ReadFileTool extends tools_1.StructuredTool {
|
||
|
|
static lc_name() {
|
||
|
|
return "ReadFileTool";
|
||
|
|
}
|
||
|
|
constructor({ store }) {
|
||
|
|
super(...arguments);
|
||
|
|
Object.defineProperty(this, "schema", {
|
||
|
|
enumerable: true,
|
||
|
|
configurable: true,
|
||
|
|
writable: true,
|
||
|
|
value: zod_1.z.object({
|
||
|
|
file_path: zod_1.z.string().describe("name of file"),
|
||
|
|
})
|
||
|
|
});
|
||
|
|
Object.defineProperty(this, "name", {
|
||
|
|
enumerable: true,
|
||
|
|
configurable: true,
|
||
|
|
writable: true,
|
||
|
|
value: "read_file"
|
||
|
|
});
|
||
|
|
Object.defineProperty(this, "description", {
|
||
|
|
enumerable: true,
|
||
|
|
configurable: true,
|
||
|
|
writable: true,
|
||
|
|
value: "Read file from disk"
|
||
|
|
});
|
||
|
|
Object.defineProperty(this, "store", {
|
||
|
|
enumerable: true,
|
||
|
|
configurable: true,
|
||
|
|
writable: true,
|
||
|
|
value: void 0
|
||
|
|
});
|
||
|
|
this.store = store;
|
||
|
|
}
|
||
|
|
async _call({ file_path }) {
|
||
|
|
return await this.store.readFile(file_path);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
exports.ReadFileTool = ReadFileTool;
|
||
|
|
/**
|
||
|
|
* Class for writing data to files on the disk. Extends the StructuredTool
|
||
|
|
* class.
|
||
|
|
*/
|
||
|
|
class WriteFileTool extends tools_1.StructuredTool {
|
||
|
|
static lc_name() {
|
||
|
|
return "WriteFileTool";
|
||
|
|
}
|
||
|
|
constructor({ store, ...rest }) {
|
||
|
|
super(rest);
|
||
|
|
Object.defineProperty(this, "schema", {
|
||
|
|
enumerable: true,
|
||
|
|
configurable: true,
|
||
|
|
writable: true,
|
||
|
|
value: zod_1.z.object({
|
||
|
|
file_path: zod_1.z.string().describe("name of file"),
|
||
|
|
text: zod_1.z.string().describe("text to write to file"),
|
||
|
|
})
|
||
|
|
});
|
||
|
|
Object.defineProperty(this, "name", {
|
||
|
|
enumerable: true,
|
||
|
|
configurable: true,
|
||
|
|
writable: true,
|
||
|
|
value: "write_file"
|
||
|
|
});
|
||
|
|
Object.defineProperty(this, "description", {
|
||
|
|
enumerable: true,
|
||
|
|
configurable: true,
|
||
|
|
writable: true,
|
||
|
|
value: "Write file from disk"
|
||
|
|
});
|
||
|
|
Object.defineProperty(this, "store", {
|
||
|
|
enumerable: true,
|
||
|
|
configurable: true,
|
||
|
|
writable: true,
|
||
|
|
value: void 0
|
||
|
|
});
|
||
|
|
this.store = store;
|
||
|
|
}
|
||
|
|
async _call({ file_path, text }) {
|
||
|
|
await this.store.writeFile(file_path, text);
|
||
|
|
return "File written to successfully.";
|
||
|
|
}
|
||
|
|
}
|
||
|
|
exports.WriteFileTool = WriteFileTool;
|