@mistralys/persona-builder 2.2.0 → 2.4.0

package/dist/cli.js

@@ -19,10 +19,30 @@ function resolvePartials(text, partialsMap, depth = 0) {
 }
 
 // src/engine/conditionals.ts
+var NO_NESTED_IF = String.raw`(?:(?!\{\{#if\b)[\s\S])*?`;
+var ELSE_IF_PATTERN = new RegExp(
+  String.raw`\{\{else if (\w+)\}\}(${NO_NESTED_IF})\{\{\/if\}\}`,
+  "g"
+);
+function resolveElseIf(text) {
+  if (!text.includes("{{else if ")) {
+    return text;
+  }
+  let result = text;
+  let prev;
+  do {
+    prev = result;
+    result = result.replace(
+      ELSE_IF_PATTERN,
+      (_match, flag, content) => `{{else}}{{#if ${flag}}}${content}{{/if}}{{/if}}`
+    );
+  } while (result !== prev);
+  return result;
+}
 function resolveConditionals(text, context) {
-  const noNestedIf = String.raw`(?:(?!\{\{#if\b)[\s\S])*?`;
+  const normalized = resolveElseIf(text);
   const pattern = new RegExp(
-    String.raw`\n*\{\{#if (\w+)\}\}(${noNestedIf})` + String.raw`(?:\{\{else\}\}(${noNestedIf}))?\{\{\/if\}\}\n*`,
+    String.raw`\n*\{\{#if (\w+)\}\}(${NO_NESTED_IF})` + String.raw`(?:\{\{else\}\}(${NO_NESTED_IF}))?\{\{\/if\}\}\n*`,
     "g"
   );
   const resolve = (_match, flag, inner, elseInner) => {
@@ -34,7 +54,7 @@ function resolveConditionals(text, context) {
     }
     return "\n";
   };
-  let result = text;
+  let result = normalized;
   let prev;
   do {
     prev = result;
@@ -103,7 +123,7 @@ function runBuildContext(plugins, ctx, persona, suite, target) {
   let accumulated = ctx;
   for (const plugin of plugins) {
     if (typeof plugin.onBuildContext === "function") {
-      accumulated = plugin.onBuildContext(accumulated, persona, suite, target);
+      accumulated = plugin.onBuildContext(accumulated, persona, suite, target) ?? accumulated;
     }
   }
   return accumulated;
@@ -112,11 +132,29 @@ function runPostRender(plugins, rendered, persona, target) {
   let output = rendered;
   for (const plugin of plugins) {
     if (typeof plugin.onPostRender === "function") {
-      output = plugin.onPostRender(output, persona, target);
+      output = plugin.onPostRender(output, persona, target) ?? output;
     }
   }
   return output;
 }
+function runPartials(plugins, partialsMap, suiteName, suite) {
+  let accumulated = partialsMap;
+  for (const plugin of plugins) {
+    if (typeof plugin.onPartials === "function") {
+      accumulated = plugin.onPartials(accumulated, suiteName, suite) ?? accumulated;
+    }
+  }
+  return accumulated;
+}
+function runPersonaPartials(plugins, partialsMap, persona, context, suite, target) {
+  let accumulated = partialsMap;
+  for (const plugin of plugins) {
+    if (typeof plugin.onPersonaPartials === "function") {
+      accumulated = plugin.onPersonaPartials(accumulated, persona, context, suite, target) ?? accumulated;
+    }
+  }
+  return accumulated;
+}
 function runValidate(plugins, persona, suite, target) {
   const results = [];
   for (const plugin of plugins) {
@@ -339,9 +377,20 @@ async function buildAgentNameMap(config) {
   }
   return agentMap;
 }
-function buildContext(personaMeta, sharedMeta, agentMap = {}, target, registry) {
+function buildContext(options) {
+  const {
+    personaMeta,
+    sharedMeta,
+    agentMap = {},
+    target,
+    registry,
+    configVariables,
+    suiteVariables
+  } = options;
   const version = typeof personaMeta["version"] === "string" ? personaMeta["version"] : typeof sharedMeta["default_version"] === "string" ? sharedMeta["default_version"] : "0.0.0";
   const merged = {
+    ...configVariables ?? {},
+    ...suiteVariables ?? {},
     ...sharedMeta,
     ...personaMeta,
     version
@@ -396,16 +445,32 @@ function buildContext(personaMeta, sharedMeta, agentMap = {}, target, registry)
 }
 async function buildPersona(personaYamlPath, suiteName, suiteConfig, sharedMeta, partialsMap, config, plugins, target, agentMap = {}, registry = defaultRegistry) {
   const personaMeta = await loadPersonaYaml(personaYamlPath);
-  let context = buildContext(personaMeta, sharedMeta, agentMap, target, registry);
+  let context = buildContext({
+    personaMeta,
+    sharedMeta,
+    agentMap,
+    target,
+    registry,
+    configVariables: config.variables,
+    suiteVariables: suiteConfig.variables
+  });
   const personaMetaTyped = personaMeta;
   context = runBuildContext(plugins, context, personaMetaTyped, suiteConfig, target);
+  const personaPartialsMap = runPersonaPartials(
+    plugins,
+    { ...partialsMap },
+    personaMetaTyped,
+    context,
+    suiteConfig,
+    target
+  );
   const fmTemplate = resolveFrontmatterTemplate(target, plugins, config.frontmatter, registry);
   const contentBasename = path2.basename(personaYamlPath, ".yaml") + ".md";
   const frontmatter = renderFrontmatter(fmTemplate, context, contentBasename);
   const contentSubdir = suiteConfig.contentSubdir ?? "content";
   const contentPath = path2.join(suiteConfig.srcDir, contentSubdir, contentBasename);
   const bodyTemplate = normalizeNewlines(await readFile(contentPath, "utf8"));
-  let body = resolvePartials(bodyTemplate, partialsMap);
+  let body = resolvePartials(bodyTemplate, personaPartialsMap);
   body = resolveConditionals(body, context);
   body = resolveVariables(body, context, contentBasename);
   body = collapseBlankLines(body);
@@ -443,7 +508,7 @@ async function buildSuite(suiteName, suiteConfig, config, plugins, target, agent
   const metaSubdir = suiteConfig.metaSubdir ?? "meta";
   const sharedYamlPath = path2.join(suiteConfig.srcDir, metaSubdir, "_shared.yaml");
   const sharedMeta = await loadRawYaml(sharedYamlPath);
-  let partialsMap = {};
+  let partialsMap = { ...config.partials ?? {} };
   if (config.sharedPartialsDir && existsSync(config.sharedPartialsDir)) {
     partialsMap = { ...partialsMap, ...await loadPartials(config.sharedPartialsDir) };
   }
@@ -453,6 +518,7 @@ async function buildSuite(suiteName, suiteConfig, config, plugins, target, agent
     partialsMap = { ...partialsMap, ...await loadPartials(suitePartialsDir) };
   }
   runSuiteInit(plugins, suiteConfig, sharedMeta);
+  partialsMap = runPartials(plugins, partialsMap, suiteName, suiteConfig);
   const personaYamlPaths = await discoverSuitePersonaYamls(suiteConfig);
   const results = [];
   for (const yamlPath of personaYamlPaths) {