72 lines
2.9 KiB
JavaScript
72 lines
2.9 KiB
JavaScript
|
"use strict";
|
||
|
|
var __importDefault = (this && this.__importDefault) || function (mod) {
|
||
|
|
return (mod && mod.__esModule) ? mod : { "default": mod };
|
||
|
|
};
|
||
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
||
|
|
exports.HandlebarsPromptTemplate = exports.interpolateHandlebars = exports.parseHandlebars = void 0;
|
||
|
|
const handlebars_1 = __importDefault(require("handlebars"));
|
||
|
|
const custom_format_js_1 = require("./custom_format.cjs");
|
||
|
|
const parseHandlebars = (template) => {
|
||
|
|
const parsed = [];
|
||
|
|
const nodes = [...handlebars_1.default.parse(template).body];
|
||
|
|
while (nodes.length) {
|
||
|
|
// eslint-disable-next-line @typescript-eslint/no-non-null-assertion
|
||
|
|
const node = nodes.pop();
|
||
|
|
if (node.type === "ContentStatement") {
|
||
|
|
// @ts-expect-error - handlebars' hbs.AST.ContentStatement isn't exported
|
||
|
|
const text = node.value;
|
||
|
|
parsed.push({ type: "literal", text });
|
||
|
|
}
|
||
|
|
else if (node.type === "MustacheStatement") {
|
||
|
|
// @ts-expect-error - handlebars' hbs.AST.MustacheStatement isn't exported
|
||
|
|
const name = node.path.parts[0];
|
||
|
|
// @ts-expect-error - handlebars' hbs.AST.MustacheStatement isn't exported
|
||
|
|
const { original } = node.path;
|
||
|
|
if (!!name &&
|
||
|
|
!original.startsWith("this.") &&
|
||
|
|
!original.startsWith("@")) {
|
||
|
|
parsed.push({ type: "variable", name });
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else if (node.type === "PathExpression") {
|
||
|
|
// @ts-expect-error - handlebars' hbs.AST.PathExpression isn't exported
|
||
|
|
const name = node.parts[0];
|
||
|
|
// @ts-expect-error - handlebars' hbs.AST.PathExpression isn't exported
|
||
|
|
const { original } = node;
|
||
|
|
if (!!name &&
|
||
|
|
!original.startsWith("this.") &&
|
||
|
|
!original.startsWith("@")) {
|
||
|
|
parsed.push({ type: "variable", name });
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else if (node.type === "BlockStatement") {
|
||
|
|
// @ts-expect-error - handlebars' hbs.AST.BlockStatement isn't exported
|
||
|
|
nodes.push(...node.params, ...node.program.body);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return parsed;
|
||
|
|
};
|
||
|
|
exports.parseHandlebars = parseHandlebars;
|
||
|
|
const interpolateHandlebars = (template, values) => {
|
||
|
|
const compiled = handlebars_1.default.compile(template, { noEscape: true });
|
||
|
|
return compiled(values);
|
||
|
|
};
|
||
|
|
exports.interpolateHandlebars = interpolateHandlebars;
|
||
|
|
class HandlebarsPromptTemplate extends custom_format_js_1.CustomFormatPromptTemplate {
|
||
|
|
static lc_name() {
|
||
|
|
return "HandlebarsPromptTemplate";
|
||
|
|
}
|
||
|
|
/**
|
||
|
|
* Load prompt template from a template
|
||
|
|
*/
|
||
|
|
static fromTemplate(template, params) {
|
||
|
|
return super.fromTemplate(template, {
|
||
|
|
...params,
|
||
|
|
validateTemplate: false,
|
||
|
|
customParser: exports.parseHandlebars,
|
||
|
|
renderer: exports.interpolateHandlebars,
|
||
|
|
});
|
||
|
|
}
|
||
|
|
}
|
||
|
|
exports.HandlebarsPromptTemplate = HandlebarsPromptTemplate;
|