agsamantha/node_modules/langchain/dist/experimental/autogpt/output_parser.js
2024-10-02 15:15:21 -05:00

80 lines
2.7 KiB
JavaScript

import { BaseOutputParser } from "@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
*/
export 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;
}
}
/**
* Class responsible for parsing the output of AutoGPT. It extends the
* BaseOutputParser class.
*/
export class AutoGPTOutputParser extends 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}` },
};
}
}
}