@mistralys/persona-builder 2.3.0 → 2.4.0

package/dist/cli.js

@@ -123,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;
@@ -132,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) {
@@ -359,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
@@ -416,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);
@@ -463,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) };
   }
@@ -473,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) {