@arizeai/phoenix-evals/dist/src/template/getTemplateVariables.js

A library for running evaluations for AI use cases

"use strict";
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.getTemplateVariables = getTemplateVariables;
const mustache_1 = __importDefault(require("mustache"));
/**
 * Parse out the template variables of a prompt
 * @param {GetTemplateVariableArgs} args
 * @returns {string[]} a list of prompt template variables
 */
function getTemplateVariables(args) {
    const { template } = args;
    if (typeof template === "string") {
        return getTemplateVariablesFromString(template);
    }
    return template.reduce((acc, message) => {
        if (message.content !== undefined && typeof message.content === "string") {
            return [...acc, ...getTemplateVariablesFromString(message.content)];
        }
        return acc;
    }, []);
}
/**
 * Parse out the template variables of a string template
 * @param template - The template to get the variables from
 * @returns {string[]} a list of prompt template variables
 */
function getTemplateVariablesFromString(template) {
    const templateSpans = mustache_1.default.parse(template);
    return templateSpans.reduce((acc, templateSpan) => {
        const [spanType, value] = templateSpan;
        if (spanType === "name" && typeof value === "string") {
            acc = [...acc, value];
        }
        return acc;
    }, []);
}
//# sourceMappingURL=getTemplateVariables.js.map