dirac-lang 0.1.94 → 0.1.96

package/dist/{agent-L6C3Z2YG.js → agent-SVKTURXE.js}

@@ -2,8 +2,8 @@ import {
   SessionServer,
   getSocketPath,
   isSessionRunning
-} from "./chunk-3MD2NFUM.js";
-import "./chunk-6CVWLZYL.js";
+} from "./chunk-IFBJYN2F.js";
+import "./chunk-NUNYBLVJ.js";
 import "./chunk-ECAW4X46.js";
 import "./chunk-SLGJRZ3P.js";
 import "./chunk-HRHAMPOB.js";

package/dist/{chunk-3MD2NFUM.js → chunk-IFBJYN2F.js}

@@ -1,6 +1,6 @@
 import {
   integrate
-} from "./chunk-6CVWLZYL.js";
+} from "./chunk-NUNYBLVJ.js";
 import {
   DiracParser
 } from "./chunk-HRHAMPOB.js";

package/dist/{chunk-LHNUYXML.js → chunk-LGFNVEHY.js}

@@ -1,6 +1,6 @@
 import {
   integrate
-} from "./chunk-6CVWLZYL.js";
+} from "./chunk-NUNYBLVJ.js";
 import {
   DiracParser
 } from "./chunk-HRHAMPOB.js";

package/dist/{chunk-6CVWLZYL.js → chunk-NUNYBLVJ.js}

@@ -487,12 +487,12 @@ async function executeIf(session, element) {
   const condition = await evaluatePredicate(session, conditionElement);
   if (condition) {
     if (thenElement) {
-      const { integrateChildren: integrateChildren2 } = await import("./interpreter-QR2YWK6R.js");
+      const { integrateChildren: integrateChildren2 } = await import("./interpreter-WNHDVVDD.js");
       await integrateChildren2(session, thenElement);
     }
   } else {
     if (elseElement) {
-      const { integrateChildren: integrateChildren2 } = await import("./interpreter-QR2YWK6R.js");
+      const { integrateChildren: integrateChildren2 } = await import("./interpreter-WNHDVVDD.js");
       await integrateChildren2(session, elseElement);
     }
   }
@@ -505,7 +505,7 @@ async function evaluatePredicate(session, predicateElement) {
     return await evaluateCondition(session, predicateElement);
   }
   const outputLengthBefore = session.output.length;
-  const { integrate: integrate2 } = await import("./interpreter-QR2YWK6R.js");
+  const { integrate: integrate2 } = await import("./interpreter-WNHDVVDD.js");
   await integrate2(session, predicateElement);
   const newOutputChunks = session.output.slice(outputLengthBefore);
   const result = newOutputChunks.join("").trim();
@@ -528,11 +528,11 @@ async function evaluateCondition(session, condElement) {
   }
   const outputLengthBefore = session.output.length;
   const args = [];
-  const { integrate: integrate2 } = await import("./interpreter-QR2YWK6R.js");
+  const { integrate: integrate2 } = await import("./interpreter-WNHDVVDD.js");
   for (const child of condElement.children) {
     if (child.tag.toLowerCase() === "arg") {
       const argOutputStart = session.output.length;
-      const { integrateChildren: integrateChildren2 } = await import("./interpreter-QR2YWK6R.js");
+      const { integrateChildren: integrateChildren2 } = await import("./interpreter-WNHDVVDD.js");
       await integrateChildren2(session, child);
       const newChunks = session.output.slice(argOutputStart);
       const argValue = newChunks.join("");
@@ -1052,7 +1052,8 @@ ${result}
       const autocorrect = element.attributes["autocorrect"] === "true";
       const maxRetries = parseInt(element.attributes["max-retries"] || "0", 10);
       const feedbackMode = element.attributes["feedback"] === "true";
-      console.error(`[LLM] Execute mode - validate: ${validateTags}, autocorrect: ${autocorrect}, feedback: ${feedbackMode}, debug: ${session.debug}`);
+      const confirmCorrections = element.attributes["confirm-corrections"] === "true";
+      console.error(`[LLM] Execute mode - validate: ${validateTags}, autocorrect: ${autocorrect}, feedback: ${feedbackMode}, confirm-corrections: ${confirmCorrections}, debug: ${session.debug}`);
       const maxIterationsAttr = substituteAttribute(session, element.attributes["max-iterations"] || "3");
       const maxIterations = parseInt(maxIterationsAttr, 10);
       const replaceTick = element.attributes["replace-tick"] === "true";
@@ -1201,18 +1202,73 @@ ${validation.errorMessages.join("\n")}`);
                 }
               }
               const correctedCode = correctedCodeLines.join("\n");
-              const correctionFeedback = `System: Auto-corrections applied:
+              if (confirmCorrections) {
+                const correctionFeedback = `System: Your submitted code had errors and was auto-corrected:
+${correctionMessages.join("\n")}
+
+Corrected code:
+\`\`\`xml
+${correctedCode}
+\`\`\`
+
+If you resubmit this corrected code, I will execute it for you. Please review and resubmit the corrected code, or provide a different solution.`;
+                dialogHistory.push({ role: "user", content: correctionFeedback });
+                if (contextVar) {
+                  setVariable(session, contextVar, JSON.stringify(dialogHistory), true);
+                } else if (saveDialog) {
+                  setVariable(session, "__llm_dialog__", JSON.stringify(dialogHistory), true);
+                }
+                console.error("[LLM] Corrections made - waiting for LLM confirmation (not executing yet)");
+                if (isOpenAI) {
+                  const response = await llmClient.chat.completions.create({
+                    model,
+                    max_tokens: maxTokens,
+                    temperature,
+                    messages: dialogHistory
+                  });
+                  result = response.choices[0]?.message?.content || "";
+                } else if (isOllama) {
+                  const ollamaPrompt = dialogHistory.map((m) => `${m.role.charAt(0).toUpperCase() + m.role.slice(1)}: ${m.content}`).join("\n");
+                  result = await llmClient.complete(ollamaPrompt, {
+                    model,
+                    temperature,
+                    max_tokens: maxTokens
+                  });
+                } else if (isCustom) {
+                  const customPrompt = dialogHistory.map((m) => `${m.role}: ${m.content}`).join("\n");
+                  result = await llmClient.complete(customPrompt, {
+                    model,
+                    temperature,
+                    max_tokens: maxTokens,
+                    messages: dialogHistory
+                  });
+                } else {
+                  result = await callAnthropic(llmClient, model, maxTokens, temperature, dialogHistory);
+                }
+                dialogHistory.push({ role: "assistant", content: result });
+                if (contextVar) {
+                  setVariable(session, contextVar, JSON.stringify(dialogHistory), true);
+                } else if (saveDialog) {
+                  setVariable(session, "__llm_dialog__", JSON.stringify(dialogHistory), true);
+                }
+                console.error(`[LLM] LLM confirmation response:
+${result}
+`);
+                continue;
+              } else {
+                const correctionFeedback = `System: Auto-corrections applied:
 ${correctionMessages.join("\n")}
 
 Actual code executed:
 \`\`\`xml
 ${correctedCode}
 \`\`\``;
-              dialogHistory.push({ role: "user", content: correctionFeedback });
-              if (contextVar) {
-                setVariable(session, contextVar, JSON.stringify(dialogHistory), true);
-              } else if (saveDialog) {
-                setVariable(session, "__llm_dialog__", JSON.stringify(dialogHistory), true);
+                dialogHistory.push({ role: "user", content: correctionFeedback });
+                if (contextVar) {
+                  setVariable(session, contextVar, JSON.stringify(dialogHistory), true);
+                } else if (saveDialog) {
+                  setVariable(session, "__llm_dialog__", JSON.stringify(dialogHistory), true);
+                }
               }
             }
           }
@@ -1871,7 +1927,7 @@ async function executeTagCheck(session, element) {
       const executeTag = correctedTag || tagName;
       console.error(`[tag-check] Executing <${executeTag}/> as all checks passed and execute=true.`);
       const elementToExecute = correctedTag ? { ...child, tag: correctedTag } : child;
-      const { integrate: integrate2 } = await import("./interpreter-QR2YWK6R.js");
+      const { integrate: integrate2 } = await import("./interpreter-WNHDVVDD.js");
       await integrate2(session, elementToExecute);
     }
   }
@@ -1880,7 +1936,7 @@ async function executeTagCheck(session, element) {
 // src/tags/throw.ts
 async function executeThrow(session, element) {
   const exceptionName = element.attributes?.name || "exception";
-  const { integrateChildren: integrateChildren2 } = await import("./interpreter-QR2YWK6R.js");
+  const { integrateChildren: integrateChildren2 } = await import("./interpreter-WNHDVVDD.js");
   const exceptionDom = {
     tag: "exception-content",
     attributes: { name: exceptionName },
@@ -1893,7 +1949,7 @@ async function executeThrow(session, element) {
 // src/tags/try.ts
 async function executeTry(session, element) {
   setExceptionBoundary(session);
-  const { integrateChildren: integrateChildren2 } = await import("./interpreter-QR2YWK6R.js");
+  const { integrateChildren: integrateChildren2 } = await import("./interpreter-WNHDVVDD.js");
   await integrateChildren2(session, element);
   unsetExceptionBoundary(session);
 }
@@ -1903,7 +1959,7 @@ async function executeCatch(session, element) {
   const exceptionName = element.attributes?.name || "exception";
   const caughtCount = lookupException(session, exceptionName);
   if (caughtCount > 0) {
-    const { integrateChildren: integrateChildren2 } = await import("./interpreter-QR2YWK6R.js");
+    const { integrateChildren: integrateChildren2 } = await import("./interpreter-WNHDVVDD.js");
     await integrateChildren2(session, element);
   }
   flushCurrentException(session);
@@ -1912,7 +1968,7 @@ async function executeCatch(session, element) {
 // src/tags/exception.ts
 async function executeException(session, element) {
   const exceptions = getCurrentExceptions(session);
-  const { integrateChildren: integrateChildren2 } = await import("./interpreter-QR2YWK6R.js");
+  const { integrateChildren: integrateChildren2 } = await import("./interpreter-WNHDVVDD.js");
   for (const exceptionDom of exceptions) {
     await integrateChildren2(session, exceptionDom);
   }
@@ -2113,7 +2169,7 @@ async function executeLoadContext(session, element) {
     query = element.text.trim();
   }
   if (!query && element.children.length > 0) {
-    const { integrate: integrate2 } = await import("./interpreter-QR2YWK6R.js");
+    const { integrate: integrate2 } = await import("./interpreter-WNHDVVDD.js");
     const beforeOutput = session.output.length;
     for (const child of element.children) {
       await integrate2(session, child);
@@ -2957,7 +3013,7 @@ async function executeForeach(session, element) {
     const parser2 = new DiracParser2();
     try {
       const fromElement = parser2.parse(fromAttr);
-      const { integrate: integrate2 } = await import("./interpreter-QR2YWK6R.js");
+      const { integrate: integrate2 } = await import("./interpreter-WNHDVVDD.js");
       await integrate2(session, fromElement);
     } catch (e) {
       session.output = savedOutput;

package/dist/cli.js

@@ -1,10 +1,10 @@
 #!/usr/bin/env node
 import {
   execute
-} from "./chunk-LHNUYXML.js";
+} from "./chunk-LGFNVEHY.js";
 import {
   BraKetParser
-} from "./chunk-6CVWLZYL.js";
+} from "./chunk-NUNYBLVJ.js";
 import "./chunk-ECAW4X46.js";
 import "./chunk-SLGJRZ3P.js";
 import "./chunk-HRHAMPOB.js";
@@ -16,7 +16,7 @@ import "dotenv/config";
 // package.json
 var package_default = {
   name: "dirac-lang",
-  version: "0.1.94",
+  version: "0.1.95",
   description: "LLM-Augmented Declarative Execution",
   type: "module",
   main: "dist/index.js",
@@ -150,7 +150,7 @@ async function main() {
   const args = process.argv.slice(2);
   const calledAs = process.argv[1];
   if (calledAs && calledAs.endsWith("/dish")) {
-    const { DiracShell } = await import("./shell-3FJRBNSJ.js");
+    const { DiracShell } = await import("./shell-N3TM23ZB.js");
     const shellConfig = loadShellConfig(args);
     const shell = new DiracShell(shellConfig);
     await shell.start();
@@ -199,11 +199,11 @@ async function main() {
   if (args[0] === "agent") {
     const subcommand = args[1];
     if (subcommand === "daemon") {
-      const { runAgentDaemon } = await import("./agent-L6C3Z2YG.js");
+      const { runAgentDaemon } = await import("./agent-SVKTURXE.js");
       await runAgentDaemon();
       return;
     }
-    const { AgentCLI } = await import("./agent-L6C3Z2YG.js");
+    const { AgentCLI } = await import("./agent-SVKTURXE.js");
     const agent = new AgentCLI();
     switch (subcommand) {
       case "start":
@@ -230,8 +230,8 @@ async function main() {
     return;
   }
   if (args[0] === "shell") {
-    const { DiracShell } = await import("./shell-3FJRBNSJ.js");
-    const { SessionServer, isSessionRunning, getSocketPath } = await import("./session-server-H6P4IXUF.js");
+    const { DiracShell } = await import("./shell-N3TM23ZB.js");
+    const { SessionServer, isSessionRunning, getSocketPath } = await import("./session-server-GPITH5QY.js");
     const { SessionClient } = await import("./session-client-3VTC5MLO.js");
     const daemonMode = args.includes("--daemon") || args.includes("-d");
     const agentMode = args.includes("--agent") || args.includes("-a");
@@ -277,7 +277,7 @@ async function main() {
     return;
   }
   if (args[0] === "daemon") {
-    const { SessionServer } = await import("./session-server-H6P4IXUF.js");
+    const { SessionServer } = await import("./session-server-GPITH5QY.js");
     const server = new SessionServer();
     await server.start();
     console.log("Session daemon started");

package/dist/{cron-CI4QNHTO.js → cron-YIBSPTLT.js}

@@ -3,7 +3,7 @@ import {
   listCronJobs,
   stopAllCronJobs,
   stopCronJob
-} from "./chunk-6CVWLZYL.js";
+} from "./chunk-NUNYBLVJ.js";
 import "./chunk-ECAW4X46.js";
 import "./chunk-SLGJRZ3P.js";
 import "./chunk-HRHAMPOB.js";

package/dist/index.js

@@ -2,10 +2,10 @@ import {
   createLLMAdapter,
   execute,
   executeUserCommand
-} from "./chunk-LHNUYXML.js";
+} from "./chunk-LGFNVEHY.js";
 import {
   integrate
-} from "./chunk-6CVWLZYL.js";
+} from "./chunk-NUNYBLVJ.js";
 import "./chunk-ECAW4X46.js";
 import "./chunk-SLGJRZ3P.js";
 import {

package/dist/{interpreter-QR2YWK6R.js → interpreter-WNHDVVDD.js}

@@ -1,7 +1,7 @@
 import {
   integrate,
   integrateChildren
-} from "./chunk-6CVWLZYL.js";
+} from "./chunk-NUNYBLVJ.js";
 import "./chunk-ECAW4X46.js";
 import "./chunk-SLGJRZ3P.js";
 import "./chunk-HRHAMPOB.js";

package/dist/{run-at-IDE7KGHV.js → run-at-H7RN6DOD.js}

@@ -3,7 +3,7 @@ import {
   cancelScheduledRun,
   executeRunAt,
   listScheduledRuns
-} from "./chunk-6CVWLZYL.js";
+} from "./chunk-NUNYBLVJ.js";
 import "./chunk-ECAW4X46.js";
 import "./chunk-SLGJRZ3P.js";
 import "./chunk-HRHAMPOB.js";

package/dist/{schedule-5SRRRO6K.js → schedule-JY54UROJ.js}

@@ -3,7 +3,7 @@ import {
   listScheduledTasks,
   stopAllScheduledTasks,
   stopScheduledTask
-} from "./chunk-6CVWLZYL.js";
+} from "./chunk-NUNYBLVJ.js";
 import "./chunk-ECAW4X46.js";
 import "./chunk-SLGJRZ3P.js";
 import "./chunk-HRHAMPOB.js";

package/dist/{session-server-H6P4IXUF.js → session-server-GPITH5QY.js}

@@ -2,8 +2,8 @@ import {
   SessionServer,
   getSocketPath,
   isSessionRunning
-} from "./chunk-3MD2NFUM.js";
-import "./chunk-6CVWLZYL.js";
+} from "./chunk-IFBJYN2F.js";
+import "./chunk-NUNYBLVJ.js";
 import "./chunk-ECAW4X46.js";
 import "./chunk-SLGJRZ3P.js";
 import "./chunk-HRHAMPOB.js";

package/dist/{shell-3FJRBNSJ.js → shell-N3TM23ZB.js}

@@ -2,7 +2,7 @@
 import {
   BraKetParser,
   integrate
-} from "./chunk-6CVWLZYL.js";
+} from "./chunk-NUNYBLVJ.js";
 import "./chunk-ECAW4X46.js";
 import "./chunk-SLGJRZ3P.js";
 import {
@@ -358,7 +358,7 @@ var DiracShell = class {
     if (this.client) {
       this.client.disconnect();
     }
-    import("./schedule-5SRRRO6K.js").then(({ stopAllScheduledTasks }) => {
+    import("./schedule-JY54UROJ.js").then(({ stopAllScheduledTasks }) => {
       stopAllScheduledTasks();
       console.log("\nGoodbye!");
       process.exit(0);
@@ -542,7 +542,7 @@ Commands:
   :scheduled      List all scheduled runs (run-at)
   :cancel <name>  Cancel a scheduled run
   :cancelall      Cancel all scheduled runs
-  :save-training  Save LLM dialog as training data (opens in editor)
+  :save-training [mode=full|pruned|both]  Save LLM dialog as training data (opens in editor)
   :exit           Exit shell
 
 Syntax:
@@ -763,7 +763,7 @@ Examples:
         break;
       case "tasks":
         try {
-          const { listScheduledTasks } = await import("./schedule-5SRRRO6K.js");
+          const { listScheduledTasks } = await import("./schedule-JY54UROJ.js");
           const tasks = listScheduledTasks();
           if (tasks.length === 0) {
             console.log("No scheduled tasks running.");
@@ -782,7 +782,7 @@ Examples:
           console.log("Usage: :stop <task-name>");
         } else {
           try {
-            const { stopScheduledTask } = await import("./schedule-5SRRRO6K.js");
+            const { stopScheduledTask } = await import("./schedule-JY54UROJ.js");
             const taskName = args[0];
             const stopped = stopScheduledTask(taskName);
             if (stopped) {
@@ -797,7 +797,7 @@ Examples:
         break;
       case "stopall":
         try {
-          const { stopAllScheduledTasks } = await import("./schedule-5SRRRO6K.js");
+          const { stopAllScheduledTasks } = await import("./schedule-JY54UROJ.js");
           stopAllScheduledTasks();
           console.log("All scheduled tasks stopped.");
         } catch (error) {
@@ -806,7 +806,7 @@ Examples:
         break;
       case "crons":
         try {
-          const { listCronJobs } = await import("./cron-CI4QNHTO.js");
+          const { listCronJobs } = await import("./cron-YIBSPTLT.js");
           const jobs = listCronJobs();
           if (jobs.length === 0) {
             console.log("No cron jobs running.");
@@ -826,7 +826,7 @@ Examples:
           console.log("Usage: :stopcron <job-name>");
         } else {
           try {
-            const { stopCronJob } = await import("./cron-CI4QNHTO.js");
+            const { stopCronJob } = await import("./cron-YIBSPTLT.js");
             const jobName = args[0];
             const stopped = stopCronJob(jobName);
             if (stopped) {
@@ -841,7 +841,7 @@ Examples:
         break;
       case "stopallcrons":
         try {
-          const { stopAllCronJobs } = await import("./cron-CI4QNHTO.js");
+          const { stopAllCronJobs } = await import("./cron-YIBSPTLT.js");
           stopAllCronJobs();
           console.log("All cron jobs stopped.");
         } catch (error) {
@@ -850,7 +850,7 @@ Examples:
         break;
       case "scheduled":
         try {
-          const { listScheduledRuns } = await import("./run-at-IDE7KGHV.js");
+          const { listScheduledRuns } = await import("./run-at-H7RN6DOD.js");
           const runs = listScheduledRuns();
           if (runs.length === 0) {
             console.log("No scheduled runs pending.");
@@ -870,7 +870,7 @@ Examples:
           console.log("Usage: :cancel <run-name>");
         } else {
           try {
-            const { cancelScheduledRun } = await import("./run-at-IDE7KGHV.js");
+            const { cancelScheduledRun } = await import("./run-at-H7RN6DOD.js");
             const runName = args[0];
             const cancelled = cancelScheduledRun(runName);
             if (cancelled) {
@@ -885,7 +885,7 @@ Examples:
         break;
       case "cancelall":
         try {
-          const { cancelAllScheduledRuns } = await import("./run-at-IDE7KGHV.js");
+          const { cancelAllScheduledRuns } = await import("./run-at-H7RN6DOD.js");
           cancelAllScheduledRuns();
           console.log("All scheduled runs cancelled.");
         } catch (error) {
@@ -894,6 +894,21 @@ Examples:
         break;
       case "save-training":
         try {
+          const fullCommand = cmd.slice(1);
+          const saveArgs = fullCommand.split(/\s+/).slice(1);
+          let saveMode = "full";
+          console.error(`[DEBUG] Full command: "${fullCommand}"`);
+          console.error(`[DEBUG] Args:`, saveArgs);
+          for (const arg of saveArgs) {
+            if (arg.startsWith("mode=")) {
+              saveMode = arg.split("=")[1];
+              console.error(`[DEBUG] Set mode to: ${saveMode}`);
+            } else if (arg === "prune=true") {
+              saveMode = "pruned";
+              console.error(`[DEBUG] Set mode to pruned`);
+            }
+          }
+          console.error(`[DEBUG] Final saveMode: ${saveMode}`);
           const dialogVar = this.client ? (await this.client.getState()).variables.find((v) => v.name === "__llm_dialog__") : this.session.variables.find((v) => v.name === "__llm_dialog__");
           if (!dialogVar || !dialogVar.value) {
             console.log("No LLM dialog to save");
@@ -904,10 +919,64 @@ Examples:
             console.log("Dialog is empty");
             break;
           }
-          const trainingExample = { messages: dialog };
+          const pruneCorrections = (msgs) => {
+            const pruned = [];
+            let skipNext = false;
+            let correctionCount = 0;
+            for (let i = 0; i < msgs.length; i++) {
+              const msg = msgs[i];
+              if (msg.role === "system") {
+                pruned.push(msg);
+                continue;
+              }
+              if (skipNext) {
+                skipNext = false;
+                continue;
+              }
+              if (msg.role === "user" && (msg.content.includes("System: Your submitted code had errors") || msg.content.includes("System: Auto-corrections applied"))) {
+                correctionCount++;
+                console.error(`[PRUNE] Found correction message at index ${i}`);
+                if (pruned.length > 0 && pruned[pruned.length - 1].role === "assistant") {
+                  console.error(`[PRUNE] Removing wrong assistant response`);
+                  pruned.pop();
+                }
+                if (i + 1 < msgs.length && msgs[i + 1].role === "assistant") {
+                  console.error(`[PRUNE] Keeping corrected assistant response`);
+                  pruned.push(msgs[i + 1]);
+                  skipNext = true;
+                }
+                continue;
+              }
+              pruned.push(msg);
+            }
+            console.error(`[PRUNE] Processed ${msgs.length} messages, found ${correctionCount} corrections, result has ${pruned.length} messages`);
+            return pruned;
+          };
+          const fullExample = { messages: dialog };
+          const prunedExample = { messages: pruneCorrections(dialog) };
+          console.log(`
+Mode: ${saveMode}`);
+          if (saveMode === "full" || saveMode === "both") {
+            console.log(`Full dialog: ${dialog.length} messages`);
+          }
+          if (saveMode === "pruned" || saveMode === "both") {
+            console.log(`Pruned dialog: ${prunedExample.messages.length} messages (removed ${dialog.length - prunedExample.messages.length} correction messages)`);
+          }
           const timestamp = (/* @__PURE__ */ new Date()).toISOString().replace(/[:.]/g, "-").split("T")[0] + "-" + Date.now();
           const tempFile = path.join(os.tmpdir(), `dirac-training-${timestamp}.jsonl`);
-          fs.writeFileSync(tempFile, JSON.stringify(trainingExample, null, 2), "utf-8");
+          let tempContent;
+          if (saveMode === "both") {
+            tempContent = `// FULL VERSION (with corrections)
+${JSON.stringify(fullExample, null, 2)}
+
+// PRUNED VERSION (mistakes removed)
+${JSON.stringify(prunedExample, null, 2)}`;
+          } else if (saveMode === "pruned") {
+            tempContent = JSON.stringify(prunedExample, null, 2);
+          } else {
+            tempContent = JSON.stringify(fullExample, null, 2);
+          }
+          fs.writeFileSync(tempFile, tempContent, "utf-8");
           console.log("Opening in editor...");
           const editor = process.env.EDITOR || process.env.VISUAL || "vi";
           const { spawnSync } = await import("child_process");
@@ -944,10 +1013,40 @@ Examples:
             savePath = path.join(trainingDir, answer.endsWith(".jsonl") ? answer : `${answer}.jsonl`);
           }
           const editedContent = fs.readFileSync(tempFile, "utf-8");
-          const editedData = JSON.parse(editedContent);
-          fs.appendFileSync(savePath, JSON.stringify(editedData) + "\n");
+          if (saveMode === "both") {
+            const lines = editedContent.split("\n");
+            let fullJson = "";
+            let prunedJson = "";
+            let inFull = false;
+            let inPruned = false;
+            let braceCount = 0;
+            for (const line of lines) {
+              if (line.includes("// FULL VERSION")) {
+                inFull = true;
+                continue;
+              }
+              if (line.includes("// PRUNED VERSION")) {
+                inFull = false;
+                inPruned = true;
+                continue;
+              }
+              if (inFull) {
+                fullJson += line + "\n";
+              } else if (inPruned) {
+                prunedJson += line + "\n";
+              }
+            }
+            const fullData = JSON.parse(fullJson.trim());
+            const prunedData = JSON.parse(prunedJson.trim());
+            fs.appendFileSync(savePath, JSON.stringify(fullData) + "\n");
+            fs.appendFileSync(savePath, JSON.stringify(prunedData) + "\n");
+            console.log(`\u2713 Saved 2 training examples (full + pruned) to ${savePath}`);
+          } else {
+            const editedData = JSON.parse(editedContent);
+            fs.appendFileSync(savePath, JSON.stringify(editedData) + "\n");
+            console.log(`\u2713 Saved training example (${saveMode}) to ${savePath}`);
+          }
           fs.unlinkSync(tempFile);
-          console.log(`\u2713 Saved training example to ${savePath}`);
         } catch (error) {
           console.error("Error saving training data:", error instanceof Error ? error.message : String(error));
         }

package/dist/test-runner.js

@@ -1,6 +1,6 @@
 import {
   integrate
-} from "./chunk-6CVWLZYL.js";
+} from "./chunk-NUNYBLVJ.js";
 import "./chunk-ECAW4X46.js";
 import "./chunk-SLGJRZ3P.js";
 import {

package/lib/ai.di

@@ -32,7 +32,7 @@
   
   
   
-  <llm execute="true" save-dialog="true" show="boundary" feedback="true" validate="true" autocorrect="true">
+  <llm execute="true" save-dialog="true" show="boundary" feedback="true" validate="true" autocorrect="true" confirm-corrections="true">
     
     
     

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dirac-lang",
-  "version": "0.1.94",
+  "version": "0.1.96",
   "description": "LLM-Augmented Declarative Execution",
   "type": "module",
   "main": "dist/index.js",