86 lines
2.9 KiB
JavaScript
86 lines
2.9 KiB
JavaScript
|
"use strict";
|
||
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
||
|
|
exports.AutoGPTOutputParser = exports.preprocessJsonInput = void 0;
|
||
|
|
const output_parsers_1 = require("@langchain/core/output_parsers");
|
||
|
|
/**
|
||
|
|
* Utility function used to preprocess a string to be parsed as JSON.
|
||
|
|
* It replaces single backslashes with double backslashes, while leaving
|
||
|
|
* already escaped ones intact.
|
||
|
|
* It also extracts the json code if it is inside a code block
|
||
|
|
*/
|
||
|
|
function preprocessJsonInput(inputStr) {
|
||
|
|
const correctedStr = inputStr.replace(/(?<!\\)\\(?!["\\/bfnrt]|u[0-9a-fA-F]{4})/g, "\\\\");
|
||
|
|
const match = correctedStr.match(/```(.*)(\r\n|\r|\n)(?<code>[\w\W\n]+)(\r\n|\r|\n)```/);
|
||
|
|
if (match?.groups?.code) {
|
||
|
|
return match.groups.code.trim();
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
return correctedStr;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
exports.preprocessJsonInput = preprocessJsonInput;
|
||
|
|
/**
|
||
|
|
* Class responsible for parsing the output of AutoGPT. It extends the
|
||
|
|
* BaseOutputParser class.
|
||
|
|
*/
|
||
|
|
class AutoGPTOutputParser extends output_parsers_1.BaseOutputParser {
|
||
|
|
constructor() {
|
||
|
|
super(...arguments);
|
||
|
|
Object.defineProperty(this, "lc_namespace", {
|
||
|
|
enumerable: true,
|
||
|
|
configurable: true,
|
||
|
|
writable: true,
|
||
|
|
value: ["langchain", "experimental", "autogpt"]
|
||
|
|
});
|
||
|
|
}
|
||
|
|
/**
|
||
|
|
* Method not implemented in the class and will throw an error if called.
|
||
|
|
* It is likely meant to be overridden in subclasses to provide specific
|
||
|
|
* format instructions.
|
||
|
|
* @returns Throws an error.
|
||
|
|
*/
|
||
|
|
getFormatInstructions() {
|
||
|
|
throw new Error("Method not implemented.");
|
||
|
|
}
|
||
|
|
/**
|
||
|
|
* Asynchronous method that takes a string as input and attempts to parse
|
||
|
|
* it into an AutoGPTAction object. If the input string cannot be parsed
|
||
|
|
* directly, the method tries to preprocess the string using the
|
||
|
|
* preprocessJsonInput function and parse it again. If parsing fails
|
||
|
|
* again, it returns an AutoGPTAction object with an error message.
|
||
|
|
* @param text The string to be parsed.
|
||
|
|
* @returns A Promise that resolves to an AutoGPTAction object.
|
||
|
|
*/
|
||
|
|
async parse(text) {
|
||
|
|
let parsed;
|
||
|
|
try {
|
||
|
|
parsed = JSON.parse(text);
|
||
|
|
}
|
||
|
|
catch (error) {
|
||
|
|
const preprocessedText = preprocessJsonInput(text);
|
||
|
|
try {
|
||
|
|
parsed = JSON.parse(preprocessedText);
|
||
|
|
}
|
||
|
|
catch (error) {
|
||
|
|
return {
|
||
|
|
name: "ERROR",
|
||
|
|
args: { error: `Could not parse invalid json: ${text}` },
|
||
|
|
};
|
||
|
|
}
|
||
|
|
}
|
||
|
|
try {
|
||
|
|
return {
|
||
|
|
name: parsed.command.name,
|
||
|
|
args: parsed.command.args,
|
||
|
|
};
|
||
|
|
}
|
||
|
|
catch (error) {
|
||
|
|
return {
|
||
|
|
name: "ERROR",
|
||
|
|
args: { error: `Incomplete command args: ${parsed}` },
|
||
|
|
};
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
exports.AutoGPTOutputParser = AutoGPTOutputParser;
|