@genkit-ai/dotprompt
Version:
Genkit AI framework `.prompt` file format and management library.
183 lines • 6.69 kB
JavaScript
;
var __create = Object.create;
var __defProp = Object.defineProperty;
var __defProps = Object.defineProperties;
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
var __getOwnPropDescs = Object.getOwnPropertyDescriptors;
var __getOwnPropNames = Object.getOwnPropertyNames;
var __getOwnPropSymbols = Object.getOwnPropertySymbols;
var __getProtoOf = Object.getPrototypeOf;
var __hasOwnProp = Object.prototype.hasOwnProperty;
var __propIsEnum = Object.prototype.propertyIsEnumerable;
var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
var __spreadValues = (a, b) => {
for (var prop in b || (b = {}))
if (__hasOwnProp.call(b, prop))
__defNormalProp(a, prop, b[prop]);
if (__getOwnPropSymbols)
for (var prop of __getOwnPropSymbols(b)) {
if (__propIsEnum.call(b, prop))
__defNormalProp(a, prop, b[prop]);
}
return a;
};
var __spreadProps = (a, b) => __defProps(a, __getOwnPropDescs(b));
var __export = (target, all) => {
for (var name in all)
__defProp(target, name, { get: all[name], enumerable: true });
};
var __copyProps = (to, from, except, desc) => {
if (from && typeof from === "object" || typeof from === "function") {
for (let key of __getOwnPropNames(from))
if (!__hasOwnProp.call(to, key) && key !== except)
__defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
}
return to;
};
var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
// If the importer is in node compatibility mode or this is not an ESM
// file that has been converted to a CommonJS file using a Babel-
// compatible transform (i.e. "__esModule" has not been set), then set
// "default" to the CommonJS "module.exports" for node compatibility.
isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
mod
));
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
var template_exports = {};
__export(template_exports, {
compile: () => compile,
defineHelper: () => defineHelper,
definePartial: () => definePartial
});
module.exports = __toCommonJS(template_exports);
var import_handlebars = __toESM(require("handlebars"));
const Promptbars = global["dotprompt.handlebars"] || import_handlebars.default.create();
global["dotprompt.handlebars"] = Promptbars;
function jsonHelper(serializable, options) {
return new Promptbars.SafeString(
JSON.stringify(serializable, null, options.hash.indent || 0)
);
}
Promptbars.registerHelper("json", jsonHelper);
function roleHelper(role) {
return new Promptbars.SafeString(`<<<dotprompt:role:${role}>>>`);
}
Promptbars.registerHelper("role", roleHelper);
function historyHelper() {
return new Promptbars.SafeString("<<<dotprompt:history>>>");
}
Promptbars.registerHelper("history", historyHelper);
function sectionHelper(name) {
return new Promptbars.SafeString(`<<<dotprompt:section ${name}>>>`);
}
Promptbars.registerHelper("section", sectionHelper);
function mediaHelper(options) {
return new Promptbars.SafeString(
`<<<dotprompt:media:url ${options.hash.url}${options.hash.contentType ? ` ${options.hash.contentType}` : ""}>>>`
);
}
Promptbars.registerHelper("media", mediaHelper);
const ROLE_REGEX = /(<<<dotprompt:(?:role:[a-z]+|history))>>>/g;
function toMessages(renderedString, options) {
var _a;
let currentMessage = {
role: "user",
source: ""
};
const messageSources = [currentMessage];
for (const piece of renderedString.split(ROLE_REGEX).filter((s) => s.trim() !== "")) {
if (piece.startsWith("<<<dotprompt:role:")) {
const role = piece.substring(18);
if (currentMessage.source.trim()) {
currentMessage = { role, source: "" };
messageSources.push(currentMessage);
} else {
currentMessage.role = role;
}
} else if (piece.startsWith("<<<dotprompt:history")) {
messageSources.push(
...((_a = options == null ? void 0 : options.history) == null ? void 0 : _a.map((m) => {
return __spreadProps(__spreadValues({}, m), {
metadata: __spreadProps(__spreadValues({}, m.metadata || {}), { purpose: "history" })
});
})) || []
);
currentMessage = { role: "model", source: "" };
messageSources.push(currentMessage);
} else {
currentMessage.source += piece;
}
}
const messages = messageSources.filter((ms) => ms.content || ms.source).map((m) => {
const out = {
role: m.role,
content: m.content || toParts(m.source)
};
if (m.metadata)
out.metadata = m.metadata;
return out;
});
if (!(options == null ? void 0 : options.history) || messages.find((m) => {
var _a2;
return ((_a2 = m.metadata) == null ? void 0 : _a2.purpose) === "history";
}))
return messages;
return [
...messages.slice(0, -1),
...options.history,
messages.at(-1)
];
}
const PART_REGEX = /(<<<dotprompt:(?:media:url|section).*?)>>>/g;
function toParts(source) {
const parts = [];
const pieces = source.split(PART_REGEX).filter((s) => s.trim() !== "");
for (let i = 0; i < pieces.length; i++) {
const piece = pieces[i];
if (piece.startsWith("<<<dotprompt:media:")) {
const [_, url, contentType] = piece.split(" ");
const part = { media: { url } };
if (contentType)
part.media.contentType = contentType;
parts.push(part);
} else if (piece.startsWith("<<<dotprompt:section")) {
const [_, sectionType] = piece.split(" ");
parts.push({ metadata: { purpose: sectionType, pending: true } });
} else {
parts.push({ text: piece });
}
}
return parts;
}
function compile(source, metadata) {
const renderString = Promptbars.compile(source, {
knownHelpers: {
json: true,
section: true,
media: true,
role: true,
history: true
}
});
return (input, options) => {
const renderedString = renderString(input, {
data: {
metadata: { prompt: metadata, context: (options == null ? void 0 : options.context) || null }
}
});
return toMessages(renderedString, options);
};
}
function defineHelper(name, fn) {
Promptbars.registerHelper(name, fn);
}
function definePartial(name, source) {
Promptbars.registerPartial(name, source);
}
// Annotate the CommonJS export names for ESM import in node:
0 && (module.exports = {
compile,
defineHelper,
definePartial
});
//# sourceMappingURL=template.js.map