agency-lang 0.2.0 → 0.3.0

package/dist/lib/agents/review/agent.js

@@ -1,11 +1,11 @@
-import { print, printJSON, parseJSON, input, sleep, round, fetch, fetchJSON, read, write, readImage, notify, range, mostCommon, keys, values, entries, emit } from "agency-lang/stdlib/index.js";
+import { print, printJSON, parseJSON, input, sleep, round, read, write, readImage, notify, range, mostCommon, keys, values, entries, emit, callback } from "agency-lang/stdlib/index.js";
 import { exec } from "agency-lang/stdlib/shell.js";
 import { args } from "agency-lang/stdlib/system.js";
+import { systemMessage } from "agency-lang/stdlib/thread.js";
 import { fileURLToPath } from "url";
 import __process from "process";
 import { z } from "agency-lang/zod";
 import { goToNode, nanoid } from "agency-lang";
-import { smoltalk } from "agency-lang";
 import path from "path";
 import {
   RuntimeContext,
@@ -26,12 +26,14 @@ import {
   respondToInterrupts as _respondToInterrupts,
   rewindFrom as _rewindFrom,
   RestoreSignal,
+  GuardExceededError,
   failure,
-  readSkill as _readSkillRaw,
-  readSkillTool as __readSkillTool,
-  readSkillToolParams as __readSkillToolParams,
   AgencyFunction as __AgencyFunction,
   __call,
+  __threads,
+  __stateStack,
+  getRuntimeContext,
+  agencyStore,
   functionRefReviver as __functionRefReviver,
   DeterministicClient as __DeterministicClient
 } from "agency-lang/runtime";
@@ -65,15 +67,9 @@ const __globalCtx = new RuntimeContext({
   traceConfig: {
     program: "dist/lib/agents/review/agent.agency",
     traceDir: "traces"
-  },
-  memory: {
-    dir: ".agency-memory"
   }
 });
 const graph = __globalCtx.graph;
-function readSkill({ filepath }) {
-  return _readSkillRaw({ filepath, dirname: __dirname });
-}
 function approve(value) {
   return { type: "approve", value };
 }
@@ -83,8 +79,8 @@ function reject(value) {
 function propagate() {
   return { type: "propagate" };
 }
-const respondToInterrupts = (interrupts, responses, opts) => _respondToInterrupts({ ctx: __globalCtx, interrupts, responses, overrides: opts?.overrides, metadata: opts?.metadata });
-const rewindFrom = (checkpoint2, overrides, opts) => _rewindFrom({ ctx: __globalCtx, checkpoint: checkpoint2, overrides, metadata: opts?.metadata });
+const respondToInterrupts = (interrupts, responses, opts) => _respondToInterrupts({ ctx: __globalCtx, interrupts, responses, overrides: opts?.overrides, metadata: opts?.metadata, registerTopLevelCallbacks: __registerTopLevelCallbacks, moduleDir: __dirname });
+const rewindFrom = (checkpoint2, overrides, opts) => _rewindFrom({ ctx: __globalCtx, checkpoint: checkpoint2, overrides, metadata: opts?.metadata, registerTopLevelCallbacks: __registerTopLevelCallbacks, moduleDir: __dirname });
 const __setDebugger = (dbg) => {
   __globalCtx.debuggerState = dbg;
 };
@@ -126,8 +122,6 @@ __registerTool(parseJSON);
 __registerTool(input);
 __registerTool(sleep);
 __registerTool(round);
-__registerTool(fetch);
-__registerTool(fetchJSON);
 __registerTool(read);
 __registerTool(write);
 __registerTool(readImage);
@@ -138,24 +132,20 @@ __registerTool(keys);
 __registerTool(values);
 __registerTool(entries);
 __registerTool(emit);
+__registerTool(callback);
 __registerTool(exec);
 __registerTool(args);
+__registerTool(systemMessage);
 async function __initializeGlobals(__ctx) {
   __ctx.globals.markInitialized("dist/lib/agents/review/agent.agency");
   __ctx.globals.set("dist/lib/agents/review/agent.agency", "systemPrompt", await __call(read, {
     type: "positional",
     args: [`./prompts/system.md`]
-  }, {
-    ctx: __ctx
   }));
 }
-__toolRegistry["readSkill"] = __AgencyFunction.create({
-  name: "readSkill",
-  module: "dist/lib/agents/review/agent.agency",
-  fn: readSkill,
-  params: __readSkillToolParams.map((p) => ({ name: p, hasDefault: false, defaultValue: void 0, variadic: false })),
-  toolDefinition: __readSkillTool
-}, __toolRegistry);
+async function __registerTopLevelCallbacks(__ctx) {
+  __ctx.topLevelCallbacks = [];
+}
 __functionRefReviver.registry = __toolRegistry;
 const Severity = z.union([z.literal("critical"), z.literal("warning"), z.literal("suggestion")]);
 const Issue = z.object({ "severity": Severity, "line": z.union([z.number(), z.null()]), "message": z.string(), "suggestion": z.union([z.string(), z.null()]) });
@@ -164,242 +154,217 @@ graph.node("main", async (__state) => {
   const __setupData = setupNode({
     state: __state
   });
-  const __stateStack = __state.ctx.stateStack;
   const __stack = __setupData.stack;
   const __step = __setupData.step;
   const __self = __setupData.self;
-  const __threads = __setupData.threads;
-  const __ctx = __state.ctx;
-  const statelogClient = __ctx.statelogClient;
-  const __graph = __ctx.graph;
+  const __ctx = getRuntimeContext().ctx;
   let __forked;
   let __functionCompleted = false;
-  await callHook({
-    callbacks: __ctx.callbacks,
-    name: "onNodeStart",
-    data: {
-      nodeName: "main"
-    }
-  });
-  const runner = new Runner(__ctx, __stack, { nodeContext: true, state: __stack, moduleId: "dist/lib/agents/review/agent.agency", scopeName: "main" });
+  const runner = new Runner(__ctx, __stack, { nodeContext: true, state: __stack, moduleId: "dist/lib/agents/review/agent.agency", scopeName: "main", threads: __setupData.threads });
   try {
-    await runner.step(0, async (runner2) => {
-      __stack.locals.cliArgs = await __call(args, {
-        type: "positional",
-        args: []
-      }, {
-        ctx: __ctx,
-        threads: __threads,
-        stateStack: __stateStack
+    await agencyStore.run({
+      ...getRuntimeContext(),
+      ctx: __ctx,
+      stack: __ctx.stateStack,
+      threads: __setupData.threads
+    }, async () => {
+      await runner.hook(0, async () => {
+        await callHook({
+          name: "onNodeStart",
+          data: {
+            nodeName: "main"
+          }
+        });
       });
-      if (hasInterrupts(__stack.locals.cliArgs)) {
-        await __ctx.pendingPromises.awaitAll();
-        runner2.halt({
-          ...__state,
-          data: __stack.locals.cliArgs
+      await runner.step(1, async (runner2) => {
+        __stack.locals.cliArgs = await __call(args, {
+          type: "positional",
+          args: []
         });
-        return;
-      }
-    });
-    await runner.ifElse(1, [
-      {
-        condition: async () => __stack.locals.cliArgs.length < 2,
-        body: async (runner2) => {
-          await runner2.step(0, async (runner3) => {
-            const __funcResult = await __call(print, {
-              type: "positional",
-              args: [`Usage: agency review <file.agency>`]
-            }, {
-              ctx: __ctx,
-              threads: __threads,
-              stateStack: __stateStack
-            });
-            if (hasInterrupts(__funcResult)) {
-              await __ctx.pendingPromises.awaitAll();
-              runner3.halt({
-                ...__state,
-                data: __funcResult
-              });
-              return;
-            }
-          });
-          await runner2.step(1, async (runner3) => {
-            __stateStack.pop();
-            __functionCompleted = true;
-            runner3.halt(goToNode("end", {
-              messages: __threads,
-              ctx: __ctx,
-              data: {}
-            }));
-            return;
+        if (hasInterrupts(__stack.locals.cliArgs)) {
+          await getRuntimeContext().ctx.pendingPromises.awaitAll();
+          runner2.halt({
+            ...__state,
+            data: __stack.locals.cliArgs
           });
+          return;
         }
-      }
-    ]);
-    await runner.step(2, async (runner2) => {
-      __stack.locals.agencyScript = __stack.locals.cliArgs[0];
-    });
-    await runner.step(3, async (runner2) => {
-      __stack.locals.filePath = __stack.locals.cliArgs[1];
-    });
-    await runner.step(4, async (runner2) => {
-      __self.__retryable = false;
-      __stack.locals.code = await __call(read, {
-        type: "positional",
-        args: [__stack.locals.filePath]
-      }, {
-        ctx: __ctx,
-        threads: __threads,
-        stateStack: __stateStack
       });
-      if (hasInterrupts(__stack.locals.code)) {
-        await __ctx.pendingPromises.awaitAll();
-        runner2.halt({
-          ...__state,
-          data: __stack.locals.code
-        });
-        return;
-      }
-    });
-    await runner.step(5, async (runner2) => {
-      const __funcResult = await __call(print, {
-        type: "positional",
-        args: [`Reviewing: ${__stack.locals.filePath}
-`]
-      }, {
-        ctx: __ctx,
-        threads: __threads,
-        stateStack: __stateStack
-      });
-      if (hasInterrupts(__funcResult)) {
-        await __ctx.pendingPromises.awaitAll();
-        runner2.halt({
-          ...__state,
-          data: __funcResult
-        });
-        return;
-      }
-    });
-    await runner.step(6, async (runner2) => {
-      const __funcResult = await __call(print, {
-        type: "positional",
-        args: [`Running type checker...`]
-      }, {
-        ctx: __ctx,
-        threads: __threads,
-        stateStack: __stateStack
+      await runner.ifElse(2, [
+        {
+          condition: async () => __stack.locals.cliArgs.length < 2,
+          body: async (runner2) => {
+            await runner2.step(0, async (runner3) => {
+              const __funcResult = await __call(print, {
+                type: "positional",
+                args: [`Usage: agency review <file.agency>`]
+              });
+              if (hasInterrupts(__funcResult)) {
+                await getRuntimeContext().ctx.pendingPromises.awaitAll();
+                runner3.halt({
+                  ...__state,
+                  data: __funcResult
+                });
+                return;
+              }
+            });
+            await runner2.step(1, async (runner3) => {
+              __stateStack()?.pop();
+              __functionCompleted = true;
+              runner3.halt(goToNode("end", {
+                messages: __threads(),
+                ctx: getRuntimeContext().ctx,
+                data: {}
+              }));
+              return;
+            });
+          }
+        }
+      ]);
+      await runner.step(3, async (runner2) => {
+        __stack.locals.agencyScript = __stack.locals.cliArgs[0];
       });
-      if (hasInterrupts(__funcResult)) {
-        await __ctx.pendingPromises.awaitAll();
-        runner2.halt({
-          ...__state,
-          data: __funcResult
-        });
-        return;
-      }
-    });
-    await runner.step(7, async (runner2) => {
-      __self.__retryable = false;
-      __stack.locals.typeCheckResult = await __call(exec, {
-        type: "positional",
-        args: [`node`, [__stack.locals.agencyScript, `typecheck`, __stack.locals.filePath]]
-      }, {
-        ctx: __ctx,
-        threads: __threads,
-        stateStack: __stateStack
+      await runner.step(4, async (runner2) => {
+        __stack.locals.filePath = __stack.locals.cliArgs[1];
       });
-      if (hasInterrupts(__stack.locals.typeCheckResult)) {
-        await __ctx.pendingPromises.awaitAll();
-        runner2.halt({
-          ...__state,
-          data: __stack.locals.typeCheckResult
+      await runner.step(5, async (runner2) => {
+        __self.__retryable = false;
+        __stack.locals.code = await __call(read, {
+          type: "positional",
+          args: [__stack.locals.filePath]
         });
-        return;
-      }
-    });
-    await runner.step(8, async (runner2) => {
-      __stack.locals.typeErrors = ``;
-    });
-    await runner.ifElse(9, [
-      {
-        condition: async () => __stack.locals.typeCheckResult.exitCode !== 0,
-        body: async (runner2) => {
-          await runner2.step(0, async (runner3) => {
-            __stack.locals.typeErrors = __stack.locals.typeCheckResult.stderr + __stack.locals.typeCheckResult.stdout;
-          });
-          await runner2.step(1, async (runner3) => {
-            const __funcResult = await __call(print, {
-              type: "positional",
-              args: [`Type checker found issues:
-${__stack.locals.typeErrors}
-`]
-            }, {
-              ctx: __ctx,
-              threads: __threads,
-              stateStack: __stateStack
-            });
-            if (hasInterrupts(__funcResult)) {
-              await __ctx.pendingPromises.awaitAll();
-              runner3.halt({
-                ...__state,
-                data: __funcResult
-              });
-              return;
-            }
+        if (hasInterrupts(__stack.locals.code)) {
+          await getRuntimeContext().ctx.pendingPromises.awaitAll();
+          runner2.halt({
+            ...__state,
+            data: __stack.locals.code
           });
+          return;
         }
-      }
-    ], async (runner2) => {
-      await runner2.step(2, async (runner3) => {
+      });
+      await runner.step(6, async (runner2) => {
         const __funcResult = await __call(print, {
           type: "positional",
-          args: [`Type checker passed.
+          args: [`Reviewing: ${__stack.locals.filePath}
 `]
-        }, {
-          ctx: __ctx,
-          threads: __threads,
-          stateStack: __stateStack
         });
         if (hasInterrupts(__funcResult)) {
-          await __ctx.pendingPromises.awaitAll();
-          runner3.halt({
+          await getRuntimeContext().ctx.pendingPromises.awaitAll();
+          runner2.halt({
             ...__state,
             data: __funcResult
           });
           return;
         }
       });
-    });
-    await runner.step(10, async (runner2) => {
-    });
-    await runner.thread(11, __threads, "create", async (runner2) => {
-      await runner2.step(0, async (runner3) => {
-        const __funcResult = __threads.active().push(smoltalk.systemMessage(__ctx.globals.get("dist/lib/agents/review/agent.agency", "systemPrompt")));
+      await runner.step(7, async (runner2) => {
+        const __funcResult = await __call(print, {
+          type: "positional",
+          args: [`Running type checker...`]
+        });
         if (hasInterrupts(__funcResult)) {
-          await __ctx.pendingPromises.awaitAll();
-          runner3.halt({
+          await getRuntimeContext().ctx.pendingPromises.awaitAll();
+          runner2.halt({
             ...__state,
             data: __funcResult
           });
           return;
         }
       });
-      await runner2.step(1, async (runner3) => {
-        __stack.locals.typeErrorMessage = `The type checker found no errors.`;
+      await runner.step(8, async (runner2) => {
+        __self.__retryable = false;
+        __stack.locals.typeCheckResult = await __call(exec, {
+          type: "positional",
+          args: [`node`, [__stack.locals.agencyScript, `typecheck`, __stack.locals.filePath]]
+        });
+        if (hasInterrupts(__stack.locals.typeCheckResult)) {
+          await getRuntimeContext().ctx.pendingPromises.awaitAll();
+          runner2.halt({
+            ...__state,
+            data: __stack.locals.typeCheckResult
+          });
+          return;
+        }
       });
-      await runner2.ifElse(2, [
+      await runner.step(9, async (runner2) => {
+        __stack.locals.typeErrors = ``;
+      });
+      await runner.ifElse(10, [
         {
-          condition: async () => __stack.locals.typeErrors !== ``,
-          body: async (runner3) => {
-            await runner3.step(0, async (runner4) => {
-              __stack.locals.typeErrorMessage = `The type checker reported the following errors:
-${__stack.locals.typeErrors}`;
+          condition: async () => __stack.locals.typeCheckResult.exitCode !== 0,
+          body: async (runner2) => {
+            await runner2.step(0, async (runner3) => {
+              __stack.locals.typeErrors = __stack.locals.typeCheckResult.stderr + __stack.locals.typeCheckResult.stdout;
+            });
+            await runner2.step(1, async (runner3) => {
+              const __funcResult = await __call(print, {
+                type: "positional",
+                args: [`Type checker found issues:
+${__stack.locals.typeErrors}
+`]
+              });
+              if (hasInterrupts(__funcResult)) {
+                await getRuntimeContext().ctx.pendingPromises.awaitAll();
+                runner3.halt({
+                  ...__state,
+                  data: __funcResult
+                });
+                return;
+              }
             });
           }
         }
-      ]);
-      await runner2.step(3, async (runner3) => {
-        __stack.locals.reviewPrompt = `
+      ], async (runner2) => {
+        await runner2.step(2, async (runner3) => {
+          const __funcResult = await __call(print, {
+            type: "positional",
+            args: [`Type checker passed.
+`]
+          });
+          if (hasInterrupts(__funcResult)) {
+            await getRuntimeContext().ctx.pendingPromises.awaitAll();
+            runner3.halt({
+              ...__state,
+              data: __funcResult
+            });
+            return;
+          }
+        });
+      });
+      await runner.step(11, async (runner2) => {
+      });
+      await runner.thread(12, "create", async (runner2) => {
+        await runner2.step(0, async (runner3) => {
+          __self.__retryable = false;
+          const __funcResult = await __call(systemMessage, {
+            type: "positional",
+            args: [getRuntimeContext().ctx.globals.get("dist/lib/agents/review/agent.agency", "systemPrompt")]
+          });
+          if (hasInterrupts(__funcResult)) {
+            await getRuntimeContext().ctx.pendingPromises.awaitAll();
+            runner3.halt({
+              ...__state,
+              data: __funcResult
+            });
+            return;
+          }
+        });
+        await runner2.step(1, async (runner3) => {
+          __stack.locals.typeErrorMessage = `The type checker found no errors.`;
+        });
+        await runner2.ifElse(2, [
+          {
+            condition: async () => __stack.locals.typeErrors !== ``,
+            body: async (runner3) => {
+              await runner3.step(0, async (runner4) => {
+                __stack.locals.typeErrorMessage = `The type checker reported the following errors:
+${__stack.locals.typeErrors}`;
+              });
+            }
+          }
+        ]);
+        await runner2.step(3, async (runner3) => {
+          __stack.locals.reviewPrompt = `
               Please review the following Agency code.
 
               File: ${__stack.locals.filePath}
@@ -412,256 +377,228 @@ ${__stack.locals.typeErrors}`;
 
               Provide a structured review with a summary, list of issues (with severity, line number if applicable, message, and suggested fix), and a list of strengths.
             `;
-      });
-      await runner2.step(4, async (runner3) => {
-        __self.__removedTools = __self.__removedTools || [];
-        __stack.locals.review = await runPrompt({
-          ctx: __ctx,
-          prompt: __stack.locals.reviewPrompt,
-          messages: __threads.getOrCreateActive(),
-          responseFormat: z.object({
-            response: ReviewResult
-          }),
-          clientConfig: {},
-          maxToolCallRounds: 10,
-          stateStack: __stateStack,
-          removedTools: __self.__removedTools,
-          checkpointInfo: runner3.getCheckpointInfo()
         });
-        if (hasInterrupts(__stack.locals.review)) {
-          await __ctx.pendingPromises.awaitAll();
-          runner3.halt({
-            messages: __threads,
-            data: __stack.locals.review
+        await runner2.step(4, async (runner3) => {
+          __self.__removedTools = __self.__removedTools || [];
+          __stack.locals.review = await runPrompt({
+            prompt: __stack.locals.reviewPrompt,
+            messages: __threads().getOrCreateActive(),
+            responseFormat: z.object({
+              response: ReviewResult
+            }),
+            clientConfig: {},
+            maxToolCallRounds: 10,
+            removedTools: __self.__removedTools,
+            checkpointInfo: runner3.getCheckpointInfo()
           });
-          return;
-        }
+          if (hasInterrupts(__stack.locals.review)) {
+            await getRuntimeContext().ctx.pendingPromises.awaitAll();
+            runner3.halt({
+              messages: __threads(),
+              data: __stack.locals.review
+            });
+            return;
+          }
+        });
       });
-    });
-    await runner.step(12, async (runner2) => {
-    });
-    await runner.step(13, async (runner2) => {
-      const __funcResult = await __call(print, {
-        type: "positional",
-        args: [`
---- Review Summary ---`]
-      }, {
-        ctx: __ctx,
-        threads: __threads,
-        stateStack: __stateStack
+      await runner.step(13, async (runner2) => {
+        __self.__retryable = false;
       });
-      if (hasInterrupts(__funcResult)) {
-        await __ctx.pendingPromises.awaitAll();
-        runner2.halt({
-          ...__state,
-          data: __funcResult
+      await runner.step(14, async (runner2) => {
+        const __funcResult = await __call(print, {
+          type: "positional",
+          args: [`
+--- Review Summary ---`]
         });
-        return;
-      }
-    });
-    await runner.step(14, async (runner2) => {
-      const __funcResult = await __call(print, {
-        type: "positional",
-        args: [__stack.locals.review.summary]
-      }, {
-        ctx: __ctx,
-        threads: __threads,
-        stateStack: __stateStack
+        if (hasInterrupts(__funcResult)) {
+          await getRuntimeContext().ctx.pendingPromises.awaitAll();
+          runner2.halt({
+            ...__state,
+            data: __funcResult
+          });
+          return;
+        }
       });
-      if (hasInterrupts(__funcResult)) {
-        await __ctx.pendingPromises.awaitAll();
-        runner2.halt({
-          ...__state,
-          data: __funcResult
+      await runner.step(15, async (runner2) => {
+        const __funcResult = await __call(print, {
+          type: "positional",
+          args: [__stack.locals.review.summary]
         });
-        return;
-      }
-    });
-    await runner.ifElse(15, [
-      {
-        condition: async () => __stack.locals.review.strengths.length > 0,
-        body: async (runner2) => {
-          await runner2.step(0, async (runner3) => {
-            const __funcResult = await __call(print, {
-              type: "positional",
-              args: [`
---- Strengths ---`]
-            }, {
-              ctx: __ctx,
-              threads: __threads,
-              stateStack: __stateStack
-            });
-            if (hasInterrupts(__funcResult)) {
-              await __ctx.pendingPromises.awaitAll();
-              runner3.halt({
-                ...__state,
-                data: __funcResult
-              });
-              return;
-            }
+        if (hasInterrupts(__funcResult)) {
+          await getRuntimeContext().ctx.pendingPromises.awaitAll();
+          runner2.halt({
+            ...__state,
+            data: __funcResult
           });
-          await runner2.loop(1, __stack.locals.review.strengths, async (strength, _, runner3) => {
-            await runner3.step(0, async (runner4) => {
+          return;
+        }
+      });
+      await runner.ifElse(16, [
+        {
+          condition: async () => __stack.locals.review.strengths.length > 0,
+          body: async (runner2) => {
+            await runner2.step(0, async (runner3) => {
               const __funcResult = await __call(print, {
                 type: "positional",
-                args: [`  + ${strength}`]
-              }, {
-                ctx: __ctx,
-                threads: __threads,
-                stateStack: __stateStack
+                args: [`
+--- Strengths ---`]
               });
               if (hasInterrupts(__funcResult)) {
-                await __ctx.pendingPromises.awaitAll();
-                runner4.halt({
+                await getRuntimeContext().ctx.pendingPromises.awaitAll();
+                runner3.halt({
                   ...__state,
                   data: __funcResult
                 });
                 return;
               }
             });
-          });
-        }
-      }
-    ]);
-    await runner.ifElse(16, [
-      {
-        condition: async () => __stack.locals.review.issues.length > 0,
-        body: async (runner2) => {
-          await runner2.step(0, async (runner3) => {
-            const __funcResult = await __call(print, {
-              type: "positional",
-              args: [`
---- Issues ---`]
-            }, {
-              ctx: __ctx,
-              threads: __threads,
-              stateStack: __stateStack
-            });
-            if (hasInterrupts(__funcResult)) {
-              await __ctx.pendingPromises.awaitAll();
-              runner3.halt({
-                ...__state,
-                data: __funcResult
-              });
-              return;
-            }
-          });
-          await runner2.loop(1, __stack.locals.review.issues, async (issue, _, runner3) => {
-            await runner3.step(0, async (runner4) => {
-              __stack.locals.prefix = `[${issue.severity}]`;
-            });
-            await runner3.step(1, async (runner4) => {
-              __stack.locals.lineInfo = ``;
-            });
-            await runner3.ifElse(2, [
-              {
-                condition: async () => issue.line !== null,
-                body: async (runner4) => {
-                  await runner4.step(0, async (runner5) => {
-                    __stack.locals.lineInfo = ` (line ${issue.line})`;
+            await runner2.loop(1, __stack.locals.review.strengths, async (strength, _, runner3) => {
+              await runner3.step(0, async (runner4) => {
+                const __funcResult = await __call(print, {
+                  type: "positional",
+                  args: [`  + ${strength}`]
+                });
+                if (hasInterrupts(__funcResult)) {
+                  await getRuntimeContext().ctx.pendingPromises.awaitAll();
+                  runner4.halt({
+                    ...__state,
+                    data: __funcResult
                   });
+                  return;
                 }
-              }
-            ]);
-            await runner3.step(3, async (runner4) => {
+              });
+            });
+          }
+        }
+      ]);
+      await runner.ifElse(17, [
+        {
+          condition: async () => __stack.locals.review.issues.length > 0,
+          body: async (runner2) => {
+            await runner2.step(0, async (runner3) => {
               const __funcResult = await __call(print, {
                 type: "positional",
-                args: [`  ${__stack.locals.prefix}${__stack.locals.lineInfo} ${issue.message}`]
-              }, {
-                ctx: __ctx,
-                threads: __threads,
-                stateStack: __stateStack
+                args: [`
+--- Issues ---`]
               });
               if (hasInterrupts(__funcResult)) {
-                await __ctx.pendingPromises.awaitAll();
-                runner4.halt({
+                await getRuntimeContext().ctx.pendingPromises.awaitAll();
+                runner3.halt({
                   ...__state,
                   data: __funcResult
                 });
                 return;
               }
             });
-            await runner3.ifElse(4, [
-              {
-                condition: async () => issue.suggestion !== null,
-                body: async (runner4) => {
-                  await runner4.step(0, async (runner5) => {
-                    const __funcResult = await __call(print, {
-                      type: "positional",
-                      args: [`    Fix: ${issue.suggestion}`]
-                    }, {
-                      ctx: __ctx,
-                      threads: __threads,
-                      stateStack: __stateStack
+            await runner2.loop(1, __stack.locals.review.issues, async (issue, _, runner3) => {
+              await runner3.step(0, async (runner4) => {
+                __stack.locals.prefix = `[${issue.severity}]`;
+              });
+              await runner3.step(1, async (runner4) => {
+                __stack.locals.lineInfo = ``;
+              });
+              await runner3.ifElse(2, [
+                {
+                  condition: async () => issue.line !== null,
+                  body: async (runner4) => {
+                    await runner4.step(0, async (runner5) => {
+                      __stack.locals.lineInfo = ` (line ${issue.line})`;
                     });
-                    if (hasInterrupts(__funcResult)) {
-                      await __ctx.pendingPromises.awaitAll();
-                      runner5.halt({
-                        ...__state,
-                        data: __funcResult
-                      });
-                      return;
-                    }
+                  }
+                }
+              ]);
+              await runner3.step(3, async (runner4) => {
+                const __funcResult = await __call(print, {
+                  type: "positional",
+                  args: [`  ${__stack.locals.prefix}${__stack.locals.lineInfo} ${issue.message}`]
+                });
+                if (hasInterrupts(__funcResult)) {
+                  await getRuntimeContext().ctx.pendingPromises.awaitAll();
+                  runner4.halt({
+                    ...__state,
+                    data: __funcResult
                   });
+                  return;
                 }
-              }
-            ]);
-          });
+              });
+              await runner3.ifElse(4, [
+                {
+                  condition: async () => issue.suggestion !== null,
+                  body: async (runner4) => {
+                    await runner4.step(0, async (runner5) => {
+                      const __funcResult = await __call(print, {
+                        type: "positional",
+                        args: [`    Fix: ${issue.suggestion}`]
+                      });
+                      if (hasInterrupts(__funcResult)) {
+                        await getRuntimeContext().ctx.pendingPromises.awaitAll();
+                        runner5.halt({
+                          ...__state,
+                          data: __funcResult
+                        });
+                        return;
+                      }
+                    });
+                  }
+                }
+              ]);
+            });
+          }
         }
-      }
-    ], async (runner2) => {
-      await runner2.step(2, async (runner3) => {
-        const __funcResult = await __call(print, {
-          type: "positional",
-          args: [`
+      ], async (runner2) => {
+        await runner2.step(2, async (runner3) => {
+          const __funcResult = await __call(print, {
+            type: "positional",
+            args: [`
 No issues found. Code looks good!`]
-        }, {
-          ctx: __ctx,
-          threads: __threads,
-          stateStack: __stateStack
-        });
-        if (hasInterrupts(__funcResult)) {
-          await __ctx.pendingPromises.awaitAll();
-          runner3.halt({
-            ...__state,
-            data: __funcResult
           });
-          return;
-        }
+          if (hasInterrupts(__funcResult)) {
+            await getRuntimeContext().ctx.pendingPromises.awaitAll();
+            runner3.halt({
+              ...__state,
+              data: __funcResult
+            });
+            return;
+          }
+        });
+      });
+      await runner.step(18, async (runner2) => {
+        __stateStack()?.pop();
+        __functionCompleted = true;
+        runner2.halt(goToNode("end", {
+          messages: __threads(),
+          ctx: getRuntimeContext().ctx,
+          data: {}
+        }));
+        return;
       });
-    });
-    await runner.step(17, async (runner2) => {
-      __stateStack.pop();
-      __functionCompleted = true;
-      runner2.halt(goToNode("end", {
-        messages: __threads,
-        ctx: __ctx,
-        data: {}
-      }));
-      return;
     });
     if (runner.halted) return runner.haltResult;
-    await callHook({
-      callbacks: __ctx.callbacks,
-      name: "onNodeEnd",
-      data: {
-        nodeName: "main",
-        data: void 0
-      }
+    await runner.hook(19, async () => {
+      await callHook({
+        name: "onNodeEnd",
+        data: {
+          nodeName: "main",
+          data: void 0
+        }
+      });
     });
     return {
-      messages: __threads,
+      messages: __threads(),
       data: void 0
     };
   } catch (__error) {
     if (__error instanceof RestoreSignal) {
       throw __error;
     }
+    if (__error instanceof GuardExceededError) {
+      throw __error;
+    }
     console.error(`
 Agent crashed: ${__error.message}`);
     console.error(__error.stack);
     return {
-      messages: __threads,
+      messages: __threads(),
       data: failure(__error instanceof Error ? __error.message : String(__error), { functionName: "main" })
     };
   }
@@ -670,47 +607,55 @@ graph.node("end", async (__state) => {
   const __setupData = setupNode({
     state: __state
   });
-  const __stateStack = __state.ctx.stateStack;
   const __stack = __setupData.stack;
   const __step = __setupData.step;
   const __self = __setupData.self;
-  const __threads = __setupData.threads;
-  const __ctx = __state.ctx;
-  const statelogClient = __ctx.statelogClient;
-  const __graph = __ctx.graph;
+  const __ctx = getRuntimeContext().ctx;
   let __forked;
   let __functionCompleted = false;
-  await callHook({
-    callbacks: __ctx.callbacks,
-    name: "onNodeStart",
-    data: {
-      nodeName: "end"
-    }
-  });
-  const runner = new Runner(__ctx, __stack, { nodeContext: true, state: __stack, moduleId: "dist/lib/agents/review/agent.agency", scopeName: "end" });
+  const runner = new Runner(__ctx, __stack, { nodeContext: true, state: __stack, moduleId: "dist/lib/agents/review/agent.agency", scopeName: "end", threads: __setupData.threads });
   try {
+    await agencyStore.run({
+      ...getRuntimeContext(),
+      ctx: __ctx,
+      stack: __ctx.stateStack,
+      threads: __setupData.threads
+    }, async () => {
+      await runner.hook(0, async () => {
+        await callHook({
+          name: "onNodeStart",
+          data: {
+            nodeName: "end"
+          }
+        });
+      });
+    });
     if (runner.halted) return runner.haltResult;
-    await callHook({
-      callbacks: __ctx.callbacks,
-      name: "onNodeEnd",
-      data: {
-        nodeName: "end",
-        data: void 0
-      }
+    await runner.hook(1, async () => {
+      await callHook({
+        name: "onNodeEnd",
+        data: {
+          nodeName: "end",
+          data: void 0
+        }
+      });
     });
     return {
-      messages: __threads,
+      messages: __threads(),
       data: void 0
     };
   } catch (__error) {
     if (__error instanceof RestoreSignal) {
       throw __error;
     }
+    if (__error instanceof GuardExceededError) {
+      throw __error;
+    }
     console.error(`
 Agent crashed: ${__error.message}`);
     console.error(__error.stack);
     return {
-      messages: __threads,
+      messages: __threads(),
       data: failure(__error instanceof Error ? __error.message : String(__error), { functionName: "end" })
     };
   }
@@ -723,7 +668,9 @@ async function main({ messages, callbacks } = {}) {
     data: {},
     messages,
     callbacks,
-    initializeGlobals: __initializeGlobals
+    initializeGlobals: __initializeGlobals,
+    registerTopLevelCallbacks: __registerTopLevelCallbacks,
+    moduleDir: __dirname
   });
 }
 const __mainNodeParams = [];
@@ -734,7 +681,9 @@ async function end({ messages, callbacks } = {}) {
     data: {},
     messages,
     callbacks,
-    initializeGlobals: __initializeGlobals
+    initializeGlobals: __initializeGlobals,
+    registerTopLevelCallbacks: __registerTopLevelCallbacks,
+    moduleDir: __dirname
   });
 }
 const __endNodeParams = [];
@@ -752,7 +701,7 @@ Agent crashed: ${__error.message}`);
   }
 }
 var stdin_default = graph;
-const __sourceMap = { "dist/lib/agents/review/agent.agency:main": { "0": { "line": 20, "col": 2 }, "1": { "line": 21, "col": 2 }, "2": { "line": 25, "col": 2 }, "3": { "line": 26, "col": 2 }, "4": { "line": 27, "col": 2 }, "5": { "line": 28, "col": 2 }, "6": { "line": 29, "col": 2 }, "7": { "line": 30, "col": 2 }, "8": { "line": 31, "col": 2 }, "9": { "line": 32, "col": 2 }, "11": { "line": 39, "col": 2 }, "13": { "line": 61, "col": 2 }, "14": { "line": 62, "col": 2 }, "15": { "line": 63, "col": 2 }, "16": { "line": 69, "col": 2 }, "17": { "line": 85, "col": 2 }, "1.0": { "line": 22, "col": 4 }, "1.1": { "line": 23, "col": 4 }, "9.0": { "line": 33, "col": 4 }, "9.1": { "line": 34, "col": 4 }, "9.2": { "line": 36, "col": 4 }, "11.0": { "line": 40, "col": 4 }, "11.1": { "line": 41, "col": 4 }, "11.2.0": { "line": 43, "col": 6 }, "11.2": { "line": 42, "col": 4 }, "11.3": { "line": 45, "col": 4 }, "11.4": { "line": 58, "col": 4 }, "15.0": { "line": 64, "col": 4 }, "15.1.0": { "line": 66, "col": 6 }, "15.1": { "line": 65, "col": 4 }, "16.0": { "line": 70, "col": 4 }, "16.1.0": { "line": 72, "col": 6 }, "16.1.1": { "line": 73, "col": 6 }, "16.1.2.0": { "line": 75, "col": 8 }, "16.1.2": { "line": 74, "col": 6 }, "16.1.3": { "line": 77, "col": 6 }, "16.1.4.0": { "line": 79, "col": 8 }, "16.1.4": { "line": 78, "col": 6 }, "16.1": { "line": 71, "col": 4 }, "16.2": { "line": 83, "col": 4 } }, "dist/lib/agents/review/agent.agency:end": {} };
+const __sourceMap = { "dist/lib/agents/review/agent.agency:main": { "1": { "line": 21, "col": 2 }, "2": { "line": 22, "col": 2 }, "3": { "line": 26, "col": 2 }, "4": { "line": 27, "col": 2 }, "5": { "line": 28, "col": 2 }, "6": { "line": 29, "col": 2 }, "7": { "line": 30, "col": 2 }, "8": { "line": 31, "col": 2 }, "9": { "line": 32, "col": 2 }, "10": { "line": 33, "col": 2 }, "12": { "line": 40, "col": 2 }, "14": { "line": 62, "col": 2 }, "15": { "line": 63, "col": 2 }, "16": { "line": 64, "col": 2 }, "17": { "line": 70, "col": 2 }, "18": { "line": 86, "col": 2 }, "2.0": { "line": 23, "col": 4 }, "2.1": { "line": 24, "col": 4 }, "10.0": { "line": 34, "col": 4 }, "10.1": { "line": 35, "col": 4 }, "10.2": { "line": 37, "col": 4 }, "12.0": { "line": 41, "col": 4 }, "12.1": { "line": 42, "col": 4 }, "12.2.0": { "line": 44, "col": 6 }, "12.2": { "line": 43, "col": 4 }, "12.3": { "line": 46, "col": 4 }, "12.4": { "line": 59, "col": 4 }, "16.0": { "line": 65, "col": 4 }, "16.1.0": { "line": 67, "col": 6 }, "16.1": { "line": 66, "col": 4 }, "17.0": { "line": 71, "col": 4 }, "17.1.0": { "line": 73, "col": 6 }, "17.1.1": { "line": 74, "col": 6 }, "17.1.2.0": { "line": 76, "col": 8 }, "17.1.2": { "line": 75, "col": 6 }, "17.1.3": { "line": 78, "col": 6 }, "17.1.4.0": { "line": 80, "col": 8 }, "17.1.4": { "line": 79, "col": 6 }, "17.1": { "line": 72, "col": 4 }, "17.2": { "line": 84, "col": 4 } }, "dist/lib/agents/review/agent.agency:end": {} };
 export {
   __endNodeParams,
   __getCheckpoints,
@@ -770,7 +719,6 @@ export {
   isDebugger,
   isInterrupt,
   main,
-  readSkill,
   reject,
   respondToInterrupts,
   rewindFrom