@topce/pizx 0.4.0 → 0.6.0

package/dist/index.js

@@ -4,6 +4,7 @@ export * from "zx";
 // src/patterns/types.ts
 import { createInterface } from "node:readline";
 import { completeSimple } from "@earendil-works/pi-ai";
+import { createAgentSession } from "@earendil-works/pi-coding-agent";
 
 // src/model-picker.ts
 import {
@@ -301,7 +302,8 @@ ${skillContext}` : skillContext;
       reasoning: opts.thinkingLevel ?? "medium",
       thinkingBudgets: opts.thinkingBudgets,
       timeoutMs: opts.timeoutMs,
-      maxRetries: opts.maxRetries
+      maxRetries: opts.maxRetries,
+      apiKey: opts.apiKey
     }
   );
   const durationMs = Date.now() - t0;
@@ -326,6 +328,40 @@ ${skillContext}` : skillContext;
   }
   return text.trim();
 }
+async function executeTask(prompt, opts = {}) {
+  if (opts.mode === "agent") {
+    return runAgentTask(prompt, opts);
+  }
+  return ask(prompt, opts);
+}
+async function runAgentTask(prompt, opts) {
+  const model = pickModel(opts.model);
+  if (!model) throw new Error("pizx/patterns: No AI models configured. Run `pi auth login` first.");
+  const tools = ["read", "bash", "edit", "write", "grep", "ls"];
+  const { session } = await createAgentSession({
+    tools,
+    ...model ? { model } : {}
+  });
+  try {
+    await session.sendUserMessage(prompt);
+    const messages = session.messages;
+    for (let i = messages.length - 1; i >= 0; i--) {
+      const msg = messages[i];
+      if (msg?.role !== "assistant") continue;
+      const c = "content" in msg ? msg.content : void 0;
+      if (typeof c === "string") return c.trim();
+      if (Array.isArray(c)) {
+        const texts = c.filter(
+          (block) => typeof block === "object" && block !== null && "type" in block && "text" in block
+        ).map((block) => block.text);
+        if (texts.length > 0) return texts.join("\n").trim();
+      }
+    }
+    return "";
+  } finally {
+    session.dispose();
+  }
+}
 var QUALITY_REVIEW_SYSTEM = `You are a quality assurance reviewer. Evaluate the final deliverable against the original request.
 
 Output format:
@@ -429,7 +465,7 @@ async function execute(pieces, args, opts) {
 `);
   }
   if (!opts.quiet) process.stderr.write("  \u2192 Planning...\n");
-  const planText = await ask(goal, {
+  const planText = await executeTask(goal, {
     ...opts,
     model: plannerModel,
     thinkingLevel: "high",
@@ -462,12 +498,12 @@ async function execute(pieces, args, opts) {
     if (!opts.quiet)
       process.stderr.write(`  \u2192 Step ${executionStep}: ${currentStep.slice(0, 60)}...
 `);
-    const result = await ask(currentStep, {
+    const result = await executeTask(currentStep, {
       ...opts,
       model: workerModel,
       system: mergeSystem(opts.system, EXECUTE_SYSTEM)
     });
-    const evaluation = await ask(
+    const evaluation = await executeTask(
       `Goal: ${goal}
 Step executed: ${currentStep}
 Result: ${result}
@@ -671,7 +707,7 @@ async function execute2(pieces, args, opts) {
   const broadcastResults = await Promise.allSettled(
     roles.map(async (role) => {
       const prompt = WORKER_PROMPT.replace("{role}", role).replace("{question}", question);
-      const text = await ask(prompt, { ...opts, model: workerModel });
+      const text = await executeTask(prompt, { ...opts, model: workerModel });
       return new BroadcastResponse(role, text, true);
     })
   );
@@ -687,7 +723,7 @@ async function execute2(pieces, args, opts) {
   if (!opts.quiet) process.stderr.write("  \u2192 Synthesizing responses...\n");
   const responsesText = responses.map((wr) => `--- ${wr.role} ---
 ${wr.response}`).join("\n\n");
-  const synthesis = await ask(
+  const synthesis = await executeTask(
     `Original question:
 ${question}
 
@@ -808,7 +844,7 @@ async function execute3(pieces, args, opts) {
     process.stderr.write(`\u03A7: Cross-Agent Learning \u2014 analyzing ${label}
 `);
   }
-  const response = await ask(input, {
+  const response = await executeTask(input, {
     ...opts,
     model: plannerModel,
     system: mergeSystem(opts.system, ANALYSIS_SYSTEM)
@@ -838,9 +874,9 @@ var defaults4 = {
   rounds: 1
 };
 var CritiqueRound = class {
-  constructor(content, critique, round) {
+  constructor(content, critique2, round) {
     this.content = content;
-    this.critique = critique;
+    this.critique = critique2;
     this.round = round;
   }
   content;
@@ -884,12 +920,16 @@ async function execute4(pieces, args, opts) {
   for (let r = 0; r < rounds; r++) {
     if (r === 0) {
       if (!opts.quiet) process.stderr.write("  \u2192 Generating initial content...\n");
-      currentContent = await ask(prompt, { ...opts, model: workerModel, system: opts.system });
+      currentContent = await executeTask(prompt, {
+        ...opts,
+        model: workerModel,
+        system: opts.system
+      });
     } else {
       if (!opts.quiet) process.stderr.write(`  \u2192 Improving (round ${r + 1})...
 `);
       const prevCritique = critiqueRounds[r - 1]?.critique ?? "";
-      currentContent = await ask(
+      currentContent = await executeTask(
         `Original request: ${prompt}
 
 Critique:
@@ -904,12 +944,12 @@ Revise the content based on the critique.`,
     }
     if (!opts.quiet) process.stderr.write(`  \u2192 Critiquing (round ${r + 1})...
 `);
-    const critique = await ask(currentContent, {
+    const critique2 = await executeTask(currentContent, {
       ...opts,
       model: plannerModel,
       system: mergeSystem(opts.system, CRITIQUE_SYSTEM)
     });
-    critiqueRounds.push(new CritiqueRound(currentContent, critique, r));
+    critiqueRounds.push(new CritiqueRound(currentContent, critique2, r));
   }
   const t1 = Date.now();
   const finalContent = currentContent;
@@ -1035,7 +1075,7 @@ Refine your position. Address the counter-arguments directly. Strengthen your ar
 `;
   }
   if (!opts.quiet) process.stderr.write("  \u2192 Synthesizing perspectives...\n");
-  const conclusion = await ask(
+  const conclusion = await executeTask(
     `${debateHistory}
 
 Synthesize a balanced conclusion from the full debate above. Weigh the evidence from all rounds.`,
@@ -1116,7 +1156,7 @@ function describeTask(task) {
   return task;
 }
 var FLEET_SYSTEM = `You are a focused task specialist. Complete the assigned task concisely and accurately. Output only the result \u2014 no commentary about being an AI.`;
-async function executeTask(task, opts, workerModel) {
+async function executeFleetTask(task, opts, workerModel) {
   if (typeof task === "function") {
     try {
       const text = await task("");
@@ -1127,7 +1167,7 @@ async function executeTask(task, opts, workerModel) {
   }
   const model = workerModel ?? opts.model;
   try {
-    const text = await ask(task, {
+    const text = await executeTask(task, {
       ...opts,
       model,
       system: mergeSystem(opts.system, FLEET_SYSTEM)
@@ -1161,7 +1201,7 @@ async function execute6(pieces, args, opts) {
   for (let i = 0; i < tasks.length; i += concurrency) {
     const batch = tasks.slice(i, i + concurrency);
     const batchResults = await Promise.allSettled(
-      batch.map((task) => executeTask(task, opts, workerModel))
+      batch.map((task) => executeFleetTask(task, opts, workerModel))
     );
     batchResults.forEach((r, idx) => {
       if (r.status === "fulfilled") {
@@ -1308,7 +1348,7 @@ ${depResults}
 Your task: ${node.task}`;
           }
         }
-        const text = await ask(context, {
+        const text = await executeTask(context, {
           ...opts,
           model: workerModel,
           system: mergeSystem(opts.system, NODE_SYSTEM)
@@ -1403,7 +1443,7 @@ async function execute8(pieces, args, opts) {
     const roundResults = await Promise.allSettled(
       roles.map(async (role) => {
         const prompt = buildWriterPrompt(role, topic, blackboard);
-        const text = await ask(prompt, { ...opts, model: workerModel });
+        const text = await executeTask(prompt, { ...opts, model: workerModel });
         return { role, text };
       })
     );
@@ -1417,7 +1457,7 @@ async function execute8(pieces, args, opts) {
     }
   }
   if (!opts.quiet) process.stderr.write("  \u2192 Consolidating findings...\n");
-  const synthesis = await ask(
+  const synthesis = await executeTask(
     `Topic: ${topic}
 
 Blackboard findings:
@@ -1508,7 +1548,7 @@ async function negotiateRoles(task, opts) {
   const min = opts.minAgents ?? 2;
   const max = opts.maxAgents ?? 5;
   const prompt = NEGOTIATE_SYSTEM.replace("{min}", String(min)).replace("{max}", String(max));
-  const response = await ask(`Task: ${task}
+  const response = await executeTask(`Task: ${task}
 
 ${prompt}`, {
     ...opts,
@@ -1540,7 +1580,7 @@ async function decideWorkflow(roles, task, opts) {
   if (roles.length <= 1)
     return { workflow: "parallel", reasoning: "Single role \u2014 no dependencies." };
   const rolesText = roles.map((r, i) => `${i + 1}. ${r.name}: ${r.goal}`).join("\n");
-  const response = await ask(
+  const response = await executeTask(
     `Task: ${task}
 
 Roles:
@@ -1570,7 +1610,7 @@ async function executeRoles(roles, task, workflow, opts) {
   if (workflow === "sequential") {
     let context = task;
     for (const role of roles) {
-      const output = await ask(context, {
+      const output = await executeTask(context, {
         ...opts,
         model: workerModel,
         system: mergeSystem(opts.system, EXECUTE_SYSTEM2(role))
@@ -1738,7 +1778,7 @@ async function execute10(pieces, args, opts) {
   }
   if (!opts.quiet) process.stderr.write("  \u2192 Planning...\n");
   const planStart = Date.now();
-  const planText = await ask(request, {
+  const planText = await executeTask(request, {
     ...opts,
     model: plannerModel,
     thinkingLevel: "high",
@@ -1803,7 +1843,7 @@ async function execute10(pieces, args, opts) {
   const workerTexts = workerResults.map((wr, i) => `Task ${i + 1}: ${wr.task}
 Result: ${wr.output}`).join("\n\n");
   const synthStart = Date.now();
-  const synthesis = await ask(
+  const synthesis = await executeTask(
     `Original request:
 ${request}
 
@@ -1954,7 +1994,7 @@ async function execute11(pieces, args, opts) {
     } else {
       const prompt = customPrompt ?? generateStagePrompt(stage, currentInput, i === 0);
       const systemMessage = i === 0 ? `You are a specialist executing stage ${i + 1}: ${stage}. Focus only on this stage's output.` : `You are a specialist executing stage ${i + 1}: ${stage}. Process the previous stage's output according to your instructions. Maintain all important information from previous stages.`;
-      output = await ask(prompt, {
+      output = await executeTask(prompt, {
         ...opts,
         model: workerModel,
         system: mergeSystem(opts.system, systemMessage)
@@ -1976,7 +2016,7 @@ ${sr.output.slice(0, 200)}${sr.output.length > 200 ? "..." : ""}`
 var \u039B = createPatternTag(defaults11, execute11);
 
 // src/patterns/ralph.ts
-import { createAgentSession } from "@earendil-works/pi-coding-agent";
+import { createAgentSession as createAgentSession2 } from "@earendil-works/pi-coding-agent";
 var defaults12 = {
   maxIterations: 5,
   useTools: true,
@@ -2002,7 +2042,7 @@ async function executeWithTools(goal, opts) {
       `pizx/\u03A1: model not found: "${opts.model}". Run \`pi models\` to see available models.`
     );
   }
-  const { session } = await createAgentSession({
+  const { session } = await createAgentSession2({
     tools: ["read", "bash", "edit", "write", "grep", "ls"],
     ...agentModel ? { model: agentModel } : {}
   });
@@ -2052,13 +2092,13 @@ async function execute12(pieces, args, opts) {
 `);
     }
     if (!opts.quiet) process.stderr.write("  \u2192 Analyzing...\n");
-    const analysis = await ask(currentGoal, {
+    const analysis = await executeTask(currentGoal, {
       ...opts,
       model: plannerModel,
       system: mergeSystem(opts.system, ANALYSIS_SYSTEM2)
     });
     if (!opts.quiet) process.stderr.write("  \u2192 Planning...\n");
-    const plan = await ask(
+    const plan = await executeTask(
       `Goal: ${currentGoal}
 
 Analysis: ${analysis}
@@ -2073,7 +2113,7 @@ ${plan}
 Goal: ${currentGoal}`, {
       ...opts,
       model: workerModel
-    }) : await ask(`Implement this plan:
+    }) : await executeTask(`Implement this plan:
 ${plan}
 
 Goal: ${currentGoal}`, {
@@ -2081,19 +2121,22 @@ Goal: ${currentGoal}`, {
       model: workerModel
     });
     if (!opts.quiet) process.stderr.write("  \u2192 Reviewing...\n");
-    const review = await ask(`Plan:
+    const review = await executeTask(
+      `Plan:
 ${plan}
 
 Result:
 ${result}
 
-Review the implementation.`, {
-      ...opts,
-      model: plannerModel,
-      maxTokens: 1024,
-      thinkingLevel: "high",
-      system: mergeSystem(opts.system, REVIEW_SYSTEM)
-    });
+Review the implementation.`,
+      {
+        ...opts,
+        model: plannerModel,
+        maxTokens: 1024,
+        thinkingLevel: "high",
+        system: mergeSystem(opts.system, REVIEW_SYSTEM)
+      }
+    );
     const shouldContinue = review.includes("ITERATE") && !review.includes("DONE");
     iterations.push({
       iteration,
@@ -2162,7 +2205,7 @@ var DECOMPOSE_SYSTEM = `You are a task decomposition specialist. Break down comp
 var SYNTHESIS_SYSTEM5 = `You are a synthesis specialist. Combine the results from multiple sub-agent analyses into a coherent, comprehensive answer. Identify patterns, conflicts, and gaps.`;
 async function decomposeTask(task, opts) {
   if (opts.subdomains && opts.subdomains.length > 0) return opts.subdomains;
-  const result = await ask(
+  const result = await executeTask(
     `Decompose this task into ${opts.maxSubTasks ?? 4} independent sub-tasks that can be worked on in parallel:
 
 ${task}
@@ -2252,7 +2295,7 @@ async function execute13(pieces, args, opts) {
   const subResultsText = subResults.map((sr, i) => `Sub-task ${i + 1}: ${sr.subTask}
 Result: ${sr.text}`).join("\n\n");
   const synthStart = Date.now();
-  const synthesis = await ask(
+  const synthesis = await executeTask(
     `Original task:
 ${task}
 
@@ -2370,7 +2413,7 @@ var CONSOLIDATE_SYSTEM = `You are a research director. Consolidate the structure
 async function defineSchema(task, opts) {
   const agentCount = opts.agents ?? 3;
   const prompt = SCHEMA_SYSTEM.replace("{agentCount}", String(agentCount));
-  const response = await ask(`Task: ${task}
+  const response = await executeTask(`Task: ${task}
 
 ${prompt}`, {
     ...opts,
@@ -2400,21 +2443,21 @@ ${prompt}`, {
   }
   return { keys, roles: roles.slice(0, agentCount), assignments };
 }
-function formatStore(store) {
-  const entries = Object.entries(store).filter(([, v]) => v);
+function formatStore(store2) {
+  const entries = Object.entries(store2).filter(([, v]) => v);
   if (entries.length === 0) return "(empty \u2014 you are the first contributor)";
   return entries.map(([k, v]) => `[${k}]: ${v}`).join("\n\n");
 }
-function mergeEntry(store, key, content) {
-  if (store[key]) {
-    store[key] += `
+function mergeEntry(store2, key, content) {
+  if (store2[key]) {
+    store2[key] += `
 
 ${content}`;
   } else {
-    store[key] = content;
+    store2[key] = content;
   }
 }
-async function executeRound(roles, assignments, store, round, opts) {
+async function executeRound(roles, assignments, store2, round, opts) {
   const workerModel = opts.workerModel ?? opts.model;
   const isWrite = round === 1;
   const operation = isWrite ? "write" : "update";
@@ -2422,10 +2465,10 @@ async function executeRound(roles, assignments, store, round, opts) {
     roles.map(async (role) => {
       const assignedKeys = assignments.get(role) ?? ["General"];
       const keysStr = assignedKeys.join(", ");
-      const storeText = formatStore(store);
+      const storeText = formatStore(store2);
       const systemPrompt = isWrite ? WRITE_SYSTEM(role, keysStr).replace("{store}", storeText) : UPDATE_SYSTEM(role, keysStr).replace("{store}", storeText);
       const task = isWrite ? `Write your initial findings to your assigned keys: ${keysStr}` : `Review the shared context and update your entries for keys: ${keysStr}`;
-      const response = await ask(task, {
+      const response = await executeTask(task, {
         ...opts,
         model: workerModel,
         system: mergeSystem(opts.system, systemPrompt)
@@ -2434,7 +2477,7 @@ async function executeRound(roles, assignments, store, round, opts) {
     })
   );
   const entries = [];
-  const newStore = { ...store };
+  const newStore = { ...store2 };
   for (const r of roundResults) {
     if (r.status !== "fulfilled") continue;
     const { role, response } = r.value;
@@ -2457,8 +2500,8 @@ async function executeRound(roles, assignments, store, round, opts) {
   }
   return { entries, store: newStore };
 }
-async function consolidateStore(task, store, opts) {
-  const storeText = formatStore(store);
+async function consolidateStore(task, store2, opts) {
+  const storeText = formatStore(store2);
   return ask(
     `Original task: ${task}
 
@@ -2499,7 +2542,7 @@ async function execute14(pieces, args, opts) {
     }
   }
   const allEntries = [];
-  let store = {};
+  let store2 = {};
   for (let round = 1; round <= totalRounds; round++) {
     const label = round === 1 ? "Writing" : "Updating";
     if (!opts.quiet) process.stderr.write(`  \u2192 Round ${round}/${totalRounds}: ${label}...
@@ -2507,15 +2550,15 @@ async function execute14(pieces, args, opts) {
     const { entries, store: updatedStore } = await executeRound(
       roles,
       assignments,
-      store,
+      store2,
       round,
       opts
     );
     allEntries.push(...entries);
-    store = updatedStore;
+    store2 = updatedStore;
   }
   if (!opts.quiet) process.stderr.write("  \u2192 Consolidating store...\n");
-  const synthesis = await consolidateStore(task, store, { ...opts, plannerModel });
+  const synthesis = await consolidateStore(task, store2, { ...opts, plannerModel });
   if (!opts.quiet && opts.qualityCheck) process.stderr.write("  \u2192 Quality review...\n");
   const qualityReview = await runQualityReview(task, synthesis, opts);
   const t1 = Date.now();
@@ -2526,7 +2569,7 @@ async function execute14(pieces, args, opts) {
     `Entries: ${allEntries.length}`,
     `Synthesis: ${synthesis}`
   ].join("\n\n");
-  return new TauOutput(summary, allEntries, store, synthesis, t0, t1, qualityReview);
+  return new TauOutput(summary, allEntries, store2, synthesis, t0, t1, qualityReview);
 }
 var \u03A4 = createPatternTag(defaults14, execute14);
 
@@ -2567,10 +2610,10 @@ The conversation so far:
 Respond as your role. Be specific, build on or challenge what others have said.
 Keep your response under 200 words.`;
 var SYNTHESIS_SYSTEM6 = `You are a neutral facilitator. Synthesize the multi-agent conversation thread into a clear, actionable conclusion. Weigh the evidence, resolve conflicts, and present the best path forward.`;
-function buildThreadPrompt(role, thread) {
+function buildThreadPrompt(role, thread2) {
   return THREAD_PROMPT.replace("{role}", role).replace(
     "{thread}",
-    thread || "(This is the first message in the thread.)"
+    thread2 || "(This is the first message in the thread.)"
   );
 }
 async function execute15(pieces, args, opts) {
@@ -2588,27 +2631,27 @@ async function execute15(pieces, args, opts) {
 `);
   }
   const messages = [];
-  let thread = `Topic: ${topic}
+  let thread2 = `Topic: ${topic}
 `;
   for (let turn = 1; turn <= maxTurns; turn++) {
     if (!opts.quiet) process.stderr.write(`  \u2192 Turn ${turn}/${maxTurns}
 `);
     for (let a = 0; a < roles.length; a++) {
       const role = roles[a] ?? `Agent ${a + 1}`;
-      const prompt = buildThreadPrompt(role, thread);
-      const response = await ask(prompt, { ...opts, model: workerModel });
+      const prompt = buildThreadPrompt(role, thread2);
+      const response = await executeTask(prompt, { ...opts, model: workerModel });
       messages.push(new ThreadMessage(role, turn, response));
-      thread += `
+      thread2 += `
 [${role}] (Turn ${turn}): ${response}
 `;
     }
   }
   if (!opts.quiet) process.stderr.write("  \u2192 Synthesizing conclusion...\n");
-  const conclusion = await ask(
+  const conclusion = await executeTask(
     `Topic: ${topic}
 
 Conversation thread:
-${thread}
+${thread2}
 
 Synthesize a clear, actionable conclusion.`,
     {
@@ -2628,6 +2671,23 @@ Synthesize a clear, actionable conclusion.`,
 }
 var \u0398 = createPatternTag(defaults15, execute15);
 
+// src/patterns/index.ts
+var ralph = \u03A1;
+var fleet = \u03A6;
+var subagent = \u03A3;
+var debate = \u0394;
+var pipeline = \u039B;
+var critique = \u03A8;
+var orchestrator = \u03A9;
+var thread = \u0398;
+var memory = \u039C;
+var broadcast = \u0392;
+var adaptive = \u0391;
+var graph = \u0393;
+var team = \u039D;
+var learn = \u03A7;
+var store = \u03A4;
+
 // src/pi.ts
 import {
   streamSimple
@@ -2715,7 +2775,8 @@ function makeOpts(opts) {
     reasoning: opts.thinkingLevel,
     thinkingBudgets: opts.thinkingBudgets,
     timeoutMs: opts.timeoutMs,
-    maxRetries: opts.maxRetries
+    maxRetries: opts.maxRetries,
+    apiKey: opts.apiKey
   };
 }
 async function run(pieces, args, opts) {
@@ -2804,7 +2865,7 @@ function configurePi(opts) {
 
 // src/pi-agent.ts
 import {
-  createAgentSession as createAgentSession2,
+  createAgentSession as createAgentSession3,
   DefaultResourceLoader
 } from "@earendil-works/pi-coding-agent";
 var _agentDefaults = {
@@ -2869,7 +2930,7 @@ async function getSession(opts) {
         loader.extendResources({ skillPaths });
       }
     }
-    const result = await createAgentSession2({
+    const result = await createAgentSession3({
       cwd: opts.cwd,
       thinkingLevel: opts.thinkingLevel,
       tools: opts.tools,
@@ -2973,8 +3034,8 @@ export {
   OrchestratorWorkerResult,
   PatternOutput,
   PatternPromise,
+  \u03A0 as Pi,
   PiOutput,
-  PiPromise,
   PipelineOutput,
   PipelineStageResult,
   RalphOutput,
@@ -2983,12 +3044,28 @@ export {
   SubagentResult,
   ThreadMessage,
   ThreadOutput,
+  adaptive,
+  broadcast,
   closeAgent,
   configureAgent,
   configurePi,
   createPatternTag,
+  critique,
+  debate,
+  fleet,
+  graph,
+  learn,
   loadSkillContent,
   loadSkillContents,
+  memory,
+  orchestrator,
+  \u03C0 as pi,
+  pipeline,
+  ralph,
+  store,
+  subagent,
+  team,
+  thread,
   \u0391,
   \u0392,
   \u0393,