@topce/pizx 0.5.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 {
@@ -327,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:
@@ -430,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",
@@ -463,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}
@@ -672,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);
     })
   );
@@ -688,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}
 
@@ -809,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)
@@ -885,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:
@@ -905,7 +944,7 @@ Revise the content based on the critique.`,
     }
     if (!opts.quiet) process.stderr.write(`  \u2192 Critiquing (round ${r + 1})...
 `);
-    const critique2 = await ask(currentContent, {
+    const critique2 = await executeTask(currentContent, {
       ...opts,
       model: plannerModel,
       system: mergeSystem(opts.system, CRITIQUE_SYSTEM)
@@ -1036,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.`,
@@ -1117,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("");
@@ -1128,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)
@@ -1162,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") {
@@ -1309,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)
@@ -1404,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 };
       })
     );
@@ -1418,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:
@@ -1509,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,
@@ -1541,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:
@@ -1571,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))
@@ -1739,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",
@@ -1804,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}
 
@@ -1955,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)
@@ -1977,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,
@@ -2003,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 } : {}
   });
@@ -2053,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}
@@ -2074,7 +2113,7 @@ ${plan}
 Goal: ${currentGoal}`, {
       ...opts,
       model: workerModel
-    }) : await ask(`Implement this plan:
+    }) : await executeTask(`Implement this plan:
 ${plan}
 
 Goal: ${currentGoal}`, {
@@ -2082,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,
@@ -2163,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}
@@ -2253,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}
 
@@ -2371,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,
@@ -2426,7 +2468,7 @@ async function executeRound(roles, assignments, store2, round, opts) {
       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)
@@ -2597,7 +2639,7 @@ async function execute15(pieces, args, opts) {
     for (let a = 0; a < roles.length; a++) {
       const role = roles[a] ?? `Agent ${a + 1}`;
       const prompt = buildThreadPrompt(role, thread2);
-      const response = await ask(prompt, { ...opts, model: workerModel });
+      const response = await executeTask(prompt, { ...opts, model: workerModel });
       messages.push(new ThreadMessage(role, turn, response));
       thread2 += `
 [${role}] (Turn ${turn}): ${response}
@@ -2605,7 +2647,7 @@ async function execute15(pieces, args, opts) {
     }
   }
   if (!opts.quiet) process.stderr.write("  \u2192 Synthesizing conclusion...\n");
-  const conclusion = await ask(
+  const conclusion = await executeTask(
     `Topic: ${topic}
 
 Conversation thread:
@@ -2823,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 = {
@@ -2888,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,