agency-lang 0.7.1 → 0.7.3

package/dist/lib/agents/agency-agent/shared.js

@@ -184,12 +184,14 @@ __registerTool(getModelSettings);
 __registerTool(loadSettings);
 __registerTool(slotKind);
 let __staticInitPromise = null;
+let ONE_SHOT_MAX_TOOL_CALL_ROUNDS = __UNINIT_STATIC;
 let AGENCY_AGENT_DIR = __UNINIT_STATIC;
 async function __initializeStatic(__ctx) {
   if (__staticInitPromise) {
     return __staticInitPromise;
   }
   __staticInitPromise = (async () => {
+    ONE_SHOT_MAX_TOOL_CALL_ROUNDS = __deepFreeze(50);
     AGENCY_AGENT_DIR = __deepFreeze(await __call(computeAgentDir, {
       type: "positional",
       args: []
@@ -199,6 +201,7 @@ async function __initializeStatic(__ctx) {
 }
 function __getStaticVars() {
   return {
+    ONE_SHOT_MAX_TOOL_CALL_ROUNDS,
     AGENCY_AGENT_DIR
   };
 }
@@ -220,6 +223,158 @@ async function __registerTopLevelCallbacks(__ctx) {
   __ctx.topLevelCallbacks = [];
 }
 __functionRefReviver.registry = __toolRegistry;
+async function __resolveMaxToolCallRounds_impl(explicit, oneShot) {
+  const __setupData = setupFunction();
+  const __stack = __setupData.stack;
+  const __step = __setupData.step;
+  const __self = __setupData.self;
+  const __ctx = getRuntimeContext().ctx;
+  let __forked;
+  let __functionCompleted = false;
+  if (!__globals().isInitialized("dist/lib/agents/agency-agent/shared.agency")) {
+    await __initializeGlobals(__ctx);
+  }
+  let __funcStartTime = performance.now();
+  __stack.args["explicit"] = explicit;
+  __stack.args["oneShot"] = oneShot;
+  __self.__retryable = __self.__retryable ?? true;
+  const runner = new Runner(__ctx, __stack, { state: __stack, moduleId: "dist/lib/agents/agency-agent/shared.agency", scopeName: "resolveMaxToolCallRounds", threads: __setupData.threads });
+  let __resultCheckpointId = -1;
+  if (__ctx._pendingArgOverrides) {
+    const __overrides = __ctx._pendingArgOverrides;
+    __ctx._pendingArgOverrides = void 0;
+    if ("explicit" in __overrides) {
+      explicit = __overrides["explicit"];
+      __stack.args["explicit"] = explicit;
+    }
+    if ("oneShot" in __overrides) {
+      oneShot = __overrides["oneShot"];
+      __stack.args["oneShot"] = oneShot;
+    }
+  }
+  try {
+    await agencyStore.run({
+      ...getRuntimeContext(),
+      ctx: __ctx,
+      stack: __setupData.stateStack,
+      threads: __setupData.threads
+    }, async () => {
+      await runner.hook(0, async () => {
+        await callHook({
+          name: "onFunctionStart",
+          data: {
+            functionName: "resolveMaxToolCallRounds",
+            args: {
+              explicit,
+              oneShot
+            },
+            moduleId: "dist/lib/agents/agency-agent/shared.agency"
+          }
+        });
+      });
+      await runner.ifElse(1, [
+        {
+          condition: async () => !__eq(__stack.args.explicit, null),
+          body: async (runner2) => {
+            await runner2.step(0, async (runner3) => {
+              __functionCompleted = true;
+              runner3.halt(__stack.args.explicit);
+              return;
+            });
+          }
+        }
+      ]);
+      await runner.ifElse(2, [
+        {
+          condition: async () => __stack.args.oneShot,
+          body: async (runner2) => {
+            await runner2.step(0, async (runner3) => {
+              __functionCompleted = true;
+              runner3.halt(__readStatic(ONE_SHOT_MAX_TOOL_CALL_ROUNDS, "ONE_SHOT_MAX_TOOL_CALL_ROUNDS", "dist/lib/agents/agency-agent/shared.agency"));
+              return;
+            });
+          }
+        }
+      ]);
+      await runner.step(3, async (runner2) => {
+        __functionCompleted = true;
+        runner2.halt(null);
+        return;
+      });
+    });
+    if (runner.halted) {
+      if (isFailure(runner.haltResult)) {
+        runner.haltResult.retryable = runner.haltResult.retryable && __self.__retryable;
+      }
+      return runner.haltResult;
+    }
+  } catch (__error) {
+    if (__error instanceof RestoreSignal) {
+      throw __error;
+    }
+    if (__error instanceof AgencyAbort) {
+      throw __error;
+    }
+    {
+      const __errMsg = __error instanceof Error ? __error.message : String(__error);
+      const __errStack = __error instanceof Error && __error.stack ? __error.stack : "";
+      const __log = __createLogger(__ctx.logLevel);
+      __log.error("Function resolveMaxToolCallRounds threw an exception (converted to Failure): " + __errMsg);
+      if (__errStack) __log.error(__errStack);
+      __ctx.statelogClient?.error?.({
+        errorType: "runtimeError",
+        message: __errMsg,
+        functionName: "resolveMaxToolCallRounds",
+        retryable: __self.__retryable
+      });
+    }
+    return failure(
+      __error instanceof Error ? __error.message : String(__error),
+      {
+        checkpoint: getRuntimeContext().ctx.getResultCheckpoint(),
+        retryable: __self.__retryable,
+        functionName: "resolveMaxToolCallRounds",
+        args: __stack.args
+      }
+    );
+  } finally {
+    __stateStack()?.pop();
+    if (__functionCompleted) {
+      await callHook({
+        name: "onFunctionEnd",
+        data: {
+          functionName: "resolveMaxToolCallRounds",
+          timeTaken: performance.now() - __funcStartTime
+        }
+      });
+    }
+  }
+}
+const resolveMaxToolCallRounds = __AgencyFunction.create({
+  name: "resolveMaxToolCallRounds",
+  module: "dist/lib/agents/agency-agent/shared.agency",
+  fn: __resolveMaxToolCallRounds_impl,
+  params: [{
+    name: "explicit",
+    hasDefault: false,
+    defaultValue: void 0,
+    variadic: false,
+    isFunctionTyped: false
+  }, {
+    name: "oneShot",
+    hasDefault: false,
+    defaultValue: void 0,
+    variadic: false,
+    isFunctionTyped: false
+  }],
+  toolDefinition: {
+    name: "resolveMaxToolCallRounds",
+    description: "No description provided.",
+    schema: z.object({ "explicit": z.union([z.number(), z.null()]), "oneShot": z.boolean() })
+  },
+  safe: false,
+  exported: true
+}, __toolRegistry);
 async function __computeAgentDir_impl() {
   const __setupData = setupFunction();
   const __stack = __setupData.stack;
@@ -1993,9 +2148,10 @@ const configureLocalModel = __AgencyFunction.create({
   exported: true
 }, __toolRegistry);
 var stdin_default = graph;
-const __sourceMap = { "dist/lib/agents/agency-agent/shared.agency:computeAgentDir": { "1": { "line": 34, "col": 2 }, "2": { "line": 35, "col": 2 }, "3": { "line": 38, "col": 2 }, "2.0": { "line": 36, "col": 4 } }, "dist/lib/agents/agency-agent/shared.agency:enableAgentMemory": { "1": { "line": 59, "col": 2 } }, "dist/lib/agents/agency-agent/shared.agency:getSlowModel": { "1": { "line": 72, "col": 2 } }, "dist/lib/agents/agency-agent/shared.agency:getSlowProvider": { "1": { "line": 76, "col": 2 } }, "dist/lib/agents/agency-agent/shared.agency:setSlowModel": { "1": { "line": 84, "col": 2 }, "2": { "line": 85, "col": 2 } }, "dist/lib/agents/agency-agent/shared.agency:promoteProvider": { "1": { "line": 93, "col": 2 }, "2": { "line": 96, "col": 2 }, "1.0": { "line": 94, "col": 4 } }, "dist/lib/agents/agency-agent/shared.agency:promoteAll": { "1": { "line": 102, "col": 2 }, "2": { "line": 103, "col": 2 }, "3": { "line": 114, "col": 2 }, "2.0": { "line": 104, "col": 4 }, "2.1": { "line": 105, "col": 4 }, "2.2": { "line": 108, "col": 4 } }, "dist/lib/agents/agency-agent/shared.agency:applyResolved": { "1": { "line": 122, "col": 2 }, "2": { "line": 123, "col": 2 }, "2.0": { "line": 124, "col": 4 }, "2.1": { "line": 125, "col": 4 }, "2.2": { "line": 128, "col": 4 }, "2.3.0": { "line": 130, "col": 6 }, "2.3.1": { "line": 137, "col": 6 }, "2.3": { "line": 129, "col": 4 } }, "dist/lib/agents/agency-agent/shared.agency:getResolvedSlots": { "1": { "line": 143, "col": 2 } }, "dist/lib/agents/agency-agent/shared.agency:currentProvider": { "1": { "line": 147, "col": 2 }, "2": { "line": 148, "col": 2 }, "3": { "line": 151, "col": 2 }, "2.0": { "line": 149, "col": 4 } }, "dist/lib/agents/agency-agent/shared.agency:configureModels": { "1": { "line": 170, "col": 2 }, "2": { "line": 175, "col": 2 }, "3": { "line": 176, "col": 2 }, "4": { "line": 180, "col": 2 }, "5": { "line": 181, "col": 2 }, "7": { "line": 190, "col": 2 }, "8": { "line": 194, "col": 2 }, "9": { "line": 195, "col": 2 }, "10": { "line": 198, "col": 2 }, "12": { "line": 204, "col": 2 }, "13": { "line": 205, "col": 2 }, "14": { "line": 214, "col": 2 }, "15": { "line": 215, "col": 2 }, "16": { "line": 220, "col": 2 }, "3.0": { "line": 177, "col": 4 }, "5.0": { "line": 182, "col": 4 }, "5.1": { "line": 183, "col": 4 }, "5.2": { "line": 184, "col": 4 }, "9.0": { "line": 196, "col": 4 }, "10.0": { "line": 199, "col": 4 }, "15.0": { "line": 216, "col": 4 }, "15.1": { "line": 217, "col": 4 }, "15.2": { "line": 218, "col": 4 } }, "dist/lib/agents/agency-agent/shared.agency:configureLocalModel": { "2": { "line": 236, "col": 2 }, "3": { "line": 237, "col": 2 }, "4": { "line": 242, "col": 2 }, "5": { "line": 246, "col": 2 }, "6": { "line": 251, "col": 2 }, "3.0": { "line": 238, "col": 4 }, "3.1": { "line": 239, "col": 4 }, "3.2": { "line": 240, "col": 4 } } };
+const __sourceMap = { "dist/lib/agents/agency-agent/shared.agency:resolveMaxToolCallRounds": { "1": { "line": 35, "col": 2 }, "2": { "line": 38, "col": 2 }, "3": { "line": 41, "col": 2 }, "1.0": { "line": 36, "col": 4 }, "2.0": { "line": 39, "col": 4 } }, "dist/lib/agents/agency-agent/shared.agency:computeAgentDir": { "1": { "line": 56, "col": 2 }, "2": { "line": 57, "col": 2 }, "3": { "line": 60, "col": 2 }, "2.0": { "line": 58, "col": 4 } }, "dist/lib/agents/agency-agent/shared.agency:enableAgentMemory": { "1": { "line": 81, "col": 2 } }, "dist/lib/agents/agency-agent/shared.agency:getSlowModel": { "1": { "line": 94, "col": 2 } }, "dist/lib/agents/agency-agent/shared.agency:getSlowProvider": { "1": { "line": 98, "col": 2 } }, "dist/lib/agents/agency-agent/shared.agency:setSlowModel": { "1": { "line": 106, "col": 2 }, "2": { "line": 107, "col": 2 } }, "dist/lib/agents/agency-agent/shared.agency:promoteProvider": { "1": { "line": 115, "col": 2 }, "2": { "line": 118, "col": 2 }, "1.0": { "line": 116, "col": 4 } }, "dist/lib/agents/agency-agent/shared.agency:promoteAll": { "1": { "line": 124, "col": 2 }, "2": { "line": 125, "col": 2 }, "3": { "line": 136, "col": 2 }, "2.0": { "line": 126, "col": 4 }, "2.1": { "line": 127, "col": 4 }, "2.2": { "line": 130, "col": 4 } }, "dist/lib/agents/agency-agent/shared.agency:applyResolved": { "1": { "line": 144, "col": 2 }, "2": { "line": 145, "col": 2 }, "2.0": { "line": 146, "col": 4 }, "2.1": { "line": 147, "col": 4 }, "2.2": { "line": 150, "col": 4 }, "2.3.0": { "line": 152, "col": 6 }, "2.3.1": { "line": 159, "col": 6 }, "2.3": { "line": 151, "col": 4 } }, "dist/lib/agents/agency-agent/shared.agency:getResolvedSlots": { "1": { "line": 165, "col": 2 } }, "dist/lib/agents/agency-agent/shared.agency:currentProvider": { "1": { "line": 169, "col": 2 }, "2": { "line": 170, "col": 2 }, "3": { "line": 173, "col": 2 }, "2.0": { "line": 171, "col": 4 } }, "dist/lib/agents/agency-agent/shared.agency:configureModels": { "1": { "line": 192, "col": 2 }, "2": { "line": 197, "col": 2 }, "3": { "line": 198, "col": 2 }, "4": { "line": 202, "col": 2 }, "5": { "line": 203, "col": 2 }, "7": { "line": 212, "col": 2 }, "8": { "line": 216, "col": 2 }, "9": { "line": 217, "col": 2 }, "10": { "line": 220, "col": 2 }, "12": { "line": 226, "col": 2 }, "13": { "line": 227, "col": 2 }, "14": { "line": 236, "col": 2 }, "15": { "line": 237, "col": 2 }, "16": { "line": 242, "col": 2 }, "3.0": { "line": 199, "col": 4 }, "5.0": { "line": 204, "col": 4 }, "5.1": { "line": 205, "col": 4 }, "5.2": { "line": 206, "col": 4 }, "9.0": { "line": 218, "col": 4 }, "10.0": { "line": 221, "col": 4 }, "15.0": { "line": 238, "col": 4 }, "15.1": { "line": 239, "col": 4 }, "15.2": { "line": 240, "col": 4 } }, "dist/lib/agents/agency-agent/shared.agency:configureLocalModel": { "2": { "line": 258, "col": 2 }, "3": { "line": 259, "col": 2 }, "4": { "line": 264, "col": 2 }, "5": { "line": 268, "col": 2 }, "6": { "line": 273, "col": 2 }, "3.0": { "line": 260, "col": 4 }, "3.1": { "line": 261, "col": 4 }, "3.2": { "line": 262, "col": 4 } } };
 export {
   AGENCY_AGENT_DIR,
+  ONE_SHOT_MAX_TOOL_CALL_ROUNDS,
   __getCheckpoints,
   __invokeFunction,
   __setDebugger,
@@ -2021,6 +2177,7 @@ export {
   promoteAll,
   promoteProvider,
   reject,
+  resolveMaxToolCallRounds,
   respondToInterrupts,
   rewindFrom,
   setSlowModel

package/dist/lib/agents/agency-agent/subagents/code.agency

@@ -27,7 +27,7 @@ import { skillsDir } from "std::skills"
 import { systemMessage } from "std::thread"
 
 import { agentNames } from "../lib/config.agency"
-import { withWebSearch } from "../lib/search.agency"
+import { withWebSearch, getSearchBackend } from "../lib/search.agency"
 import { oracleAgent } from "./oracle.agency"
 import { review, renderFeedback, feedbackHasErrors } from "./review.agency"
 
@@ -72,30 +72,16 @@ You are the **code specialist** of a multi-thread Agency-language
 assistant. You handle anything that touches code or the file system:
 writing, editing, running, and typechecking Agency or shell code.
 
-You do NOT have access to web search, Wikipedia,
-or URL fetching. If the user asks a question that needs research,
-do one of the following:
-
-- If the research involves Agency features of syntax:
-   - call the bundled docs tools first — they're authoritative:
-     * `docSkill`      — language guide (syntax, control flow,
-                         types, error handling, etc.),
-                         built-ins, callbacks, advanced types
-     * `cliSkill`      — CLI reference (`agency run`, `agency test`,
-                         `agency compile`, ...)
-   Pick whichever matches the question rather than guessing.
-
-- If the research involves something else (a research question,
-  an external API, a Wikipedia summary, or a URL fetch), call
-  `handoff(reason)` — the only valid target for you is the research
-  specialist, so just pass a brief reason. The runtime will re-route
-  the user's original message there.
-
-**Bias toward staying in the code thread.** Most follow-up messages
-("can you make it faster?", "fix the error", "what about edge
-cases?") are continuations of the current coding task, NOT new
-categories. Only call `handoff` when the message clearly needs
-research-specialist tools.
+For Agency-language questions (syntax, control flow, types, the CLI),
+call the bundled docs tools — they are authoritative for Agency:
+  * `docSkill` — the language guide (syntax, types, error handling,
+                 built-ins, callbacks, advanced types)
+  * `cliSkill` — the CLI reference (`agency run`, `agency test`, ...)
+Pick whichever matches the question rather than guessing.
+
+**Stay focused on the current task.** Most follow-ups ("can you make it
+faster?", "fix the error", "what about edge cases?") are continuations
+of the current work, not new categories.
 
 ## Style
 
@@ -359,6 +345,40 @@ export static const codeTools: any[] = [
   oracleAgent.partial(allowHandoff: false),
 ]
 
+// Appended to the system prompt only when a web-search backend is active
+// (see `withWebSearch`, which already wires the search tool onto the llm()
+// call). The prompt just needs to tell the model it HAS the capability — on
+// the cloud path `configureSearch` defaults to hosted web search.
+static const WEB_SEARCH_NOTE = """
+You have **web search**. Use it directly whenever a task needs current or
+external information — a fact to verify, a library or API to check, a
+dataset or leaderboard to consult. Do NOT ask the user to look things up,
+and do NOT decline a task as "needs research / out of scope": search for
+what you need and keep going."""
+
+// Appended in one-shot / autonomous mode (`agency agent -p`), where there is
+// no human to answer questions and no next turn. Gated on `_oneShot` so the
+// interactive REPL keeps its normal ask-when-unsure behavior.
+static const ONE_SHOT_NOTE = """
+You are running in **one-shot, autonomous mode**: no human will answer
+questions and there is NO next turn. Therefore:
+- NEVER end your turn by asking a clarifying question or waiting for input.
+  If something is ambiguous, state the most reasonable assumption and
+  proceed.
+- NEVER defer work to "the next turn" or hand back a plan — do it now.
+- Leave the task COMPLETE on disk: write the required file(s), then verify
+  (compile, run, run any tests) and fix what fails before you stop.
+- If you are running low on tool budget, spend the remaining calls
+  producing and saving the best COMPLETE artifact you can. A working file
+  on disk beats a perfect plan that was never written."""
+
+// Set by `agent.agency` (`setCodeOneShot`) before the first dispatch when the
+// run is non-interactive. Gates `ONE_SHOT_NOTE`.
+let _oneShot = false
+export def setCodeOneShot(oneShot: boolean) {
+  _oneShot = oneShot
+}
+
 // only add system message the first time
 let _first = true
 
@@ -397,7 +417,16 @@ export def codeAgent(userMsg: string, allowHandoff: boolean) {
       firstIteration = false
       iterations += 1
       if (_first) {
-        systemMessage(codeSysPrompt)
+        // Compose the system prompt from the static base plus the
+        // capabilities that are actually active this run.
+        let sys = codeSysPrompt
+        if (getSearchBackend() != "off") {
+          sys = "${sys}\n\n${WEB_SEARCH_NOTE}"
+        }
+        if (_oneShot) {
+          sys = "${sys}\n\n${ONE_SHOT_NOTE}"
+        }
+        systemMessage(sys)
         _first = false
       }
       reply = llm(

package/dist/lib/agents/agency-agent/subagents/code.js

@@ -11,7 +11,7 @@ import { gitStatus, gitLog, gitDiff, gitShow, gitBranchList, gitRemoteList, gitB
 import { skillsDir } from "agency-lang/stdlib/skills.js";
 import { systemMessage } from "agency-lang/stdlib/thread.js";
 import { agentNames } from "../lib/config.js";
-import { withWebSearch } from "../lib/search.js";
+import { withWebSearch, getSearchBackend } from "../lib/search.js";
 import { oracleAgent } from "./oracle.js";
 import { review, renderFeedback, feedbackHasErrors } from "./review.js";
 import { fileURLToPath } from "url";
@@ -214,6 +214,7 @@ __registerTool(skillsDir);
 __registerTool(systemMessage);
 __registerTool(agentNames);
 __registerTool(withWebSearch);
+__registerTool(getSearchBackend);
 __registerTool(oracleAgent);
 __registerTool(review);
 __registerTool(renderFeedback);
@@ -224,6 +225,8 @@ let docSkill = __UNINIT_STATIC;
 let cliSkill = __UNINIT_STATIC;
 let codeSysPrompt = __UNINIT_STATIC;
 let codeTools = __UNINIT_STATIC;
+let WEB_SEARCH_NOTE = __UNINIT_STATIC;
+let ONE_SHOT_NOTE = __UNINIT_STATIC;
 async function __initializeStatic(__ctx) {
   if (__staticInitPromise) {
     return __staticInitPromise;
@@ -261,30 +264,16 @@ You are the **code specialist** of a multi-thread Agency-language
 assistant. You handle anything that touches code or the file system:
 writing, editing, running, and typechecking Agency or shell code.
 
-You do NOT have access to web search, Wikipedia,
-or URL fetching. If the user asks a question that needs research,
-do one of the following:
+For Agency-language questions (syntax, control flow, types, the CLI),
+call the bundled docs tools \u2014 they are authoritative for Agency:
+  * \`docSkill\` \u2014 the language guide (syntax, types, error handling,
+                 built-ins, callbacks, advanced types)
+  * \`cliSkill\` \u2014 the CLI reference (\`agency run\`, \`agency test\`, ...)
+Pick whichever matches the question rather than guessing.
 
-- If the research involves Agency features of syntax:
-   - call the bundled docs tools first \u2014 they're authoritative:
-     * \`docSkill\`      \u2014 language guide (syntax, control flow,
-                         types, error handling, etc.),
-                         built-ins, callbacks, advanced types
-     * \`cliSkill\`      \u2014 CLI reference (\`agency run\`, \`agency test\`,
-                         \`agency compile\`, ...)
-   Pick whichever matches the question rather than guessing.
-
-- If the research involves something else (a research question,
-  an external API, a Wikipedia summary, or a URL fetch), call
-  \`handoff(reason)\` \u2014 the only valid target for you is the research
-  specialist, so just pass a brief reason. The runtime will re-route
-  the user's original message there.
-
-**Bias toward staying in the code thread.** Most follow-up messages
-("can you make it faster?", "fix the error", "what about edge
-cases?") are continuations of the current coding task, NOT new
-categories. Only call \`handoff\` when the message clearly needs
-research-specialist tools.
+**Stay focused on the current task.** Most follow-ups ("can you make it
+faster?", "fix the error", "what about edge cases?") are continuations
+of the current work, not new categories.
 
 ## Style
 
@@ -551,6 +540,24 @@ diagram needs scrolling, prose is probably better.
         allowHandoff: false
       }
     })]);
+    WEB_SEARCH_NOTE = __deepFreeze(`
+You have **web search**. Use it directly whenever a task needs current or
+external information \u2014 a fact to verify, a library or API to check, a
+dataset or leaderboard to consult. Do NOT ask the user to look things up,
+and do NOT decline a task as "needs research / out of scope": search for
+what you need and keep going.`);
+    ONE_SHOT_NOTE = __deepFreeze(`
+You are running in **one-shot, autonomous mode**: no human will answer
+questions and there is NO next turn. Therefore:
+- NEVER end your turn by asking a clarifying question or waiting for input.
+  If something is ambiguous, state the most reasonable assumption and
+  proceed.
+- NEVER defer work to "the next turn" or hand back a plan \u2014 do it now.
+- Leave the task COMPLETE on disk: write the required file(s), then verify
+  (compile, run, run any tests) and fix what fails before you stop.
+- If you are running low on tool budget, spend the remaining calls
+  producing and saving the best COMPLETE artifact you can. A working file
+  on disk beats a perfect plan that was never written.`);
   })();
   return __staticInitPromise;
 }
@@ -560,7 +567,9 @@ function __getStaticVars() {
     docSkill,
     cliSkill,
     codeSysPrompt,
-    codeTools
+    codeTools,
+    WEB_SEARCH_NOTE,
+    ONE_SHOT_NOTE
   };
 }
 __globalCtx.getStaticVars = __getStaticVars;
@@ -572,6 +581,7 @@ async function __initializeGlobals(__ctx) {
   __ctx.globals.markInitialized("dist/lib/agents/agency-agent/subagents/code.agency");
   await __initializeStatic(__ctx);
   await __ctx.writeStaticStateToTrace(__globalCtx.getStaticVars());
+  __ctx.globals.set("dist/lib/agents/agency-agent/subagents/code.agency", "_oneShot", false);
   __ctx.globals.set("dist/lib/agents/agency-agent/subagents/code.agency", "_first", true);
 }
 __registerGlobalsInit("dist/lib/agents/agency-agent/subagents/code.agency", __initializeGlobals);
@@ -713,6 +723,120 @@ const agencyCli = __AgencyFunction.create({
   safe: false,
   exported: true
 }, __toolRegistry);
+async function __setCodeOneShot_impl(oneShot) {
+  const __setupData = setupFunction();
+  const __stack = __setupData.stack;
+  const __step = __setupData.step;
+  const __self = __setupData.self;
+  const __ctx = getRuntimeContext().ctx;
+  let __forked;
+  let __functionCompleted = false;
+  if (!__globals().isInitialized("dist/lib/agents/agency-agent/subagents/code.agency")) {
+    await __initializeGlobals(__ctx);
+  }
+  let __funcStartTime = performance.now();
+  __stack.args["oneShot"] = oneShot;
+  __self.__retryable = __self.__retryable ?? true;
+  const runner = new Runner(__ctx, __stack, { state: __stack, moduleId: "dist/lib/agents/agency-agent/subagents/code.agency", scopeName: "setCodeOneShot", threads: __setupData.threads });
+  let __resultCheckpointId = -1;
+  if (__ctx._pendingArgOverrides) {
+    const __overrides = __ctx._pendingArgOverrides;
+    __ctx._pendingArgOverrides = void 0;
+    if ("oneShot" in __overrides) {
+      oneShot = __overrides["oneShot"];
+      __stack.args["oneShot"] = oneShot;
+    }
+  }
+  try {
+    await agencyStore.run({
+      ...getRuntimeContext(),
+      ctx: __ctx,
+      stack: __setupData.stateStack,
+      threads: __setupData.threads
+    }, async () => {
+      await runner.hook(0, async () => {
+        await callHook({
+          name: "onFunctionStart",
+          data: {
+            functionName: "setCodeOneShot",
+            args: {
+              oneShot
+            },
+            moduleId: "dist/lib/agents/agency-agent/subagents/code.agency"
+          }
+        });
+      });
+      await runner.step(1, async (runner2) => {
+        __globals().set("dist/lib/agents/agency-agent/subagents/code.agency", "_oneShot", __stack.args.oneShot);
+      });
+    });
+    if (runner.halted) {
+      if (isFailure(runner.haltResult)) {
+        runner.haltResult.retryable = runner.haltResult.retryable && __self.__retryable;
+      }
+      return runner.haltResult;
+    }
+  } catch (__error) {
+    if (__error instanceof RestoreSignal) {
+      throw __error;
+    }
+    if (__error instanceof AgencyAbort) {
+      throw __error;
+    }
+    {
+      const __errMsg = __error instanceof Error ? __error.message : String(__error);
+      const __errStack = __error instanceof Error && __error.stack ? __error.stack : "";
+      const __log = __createLogger(__ctx.logLevel);
+      __log.error("Function setCodeOneShot threw an exception (converted to Failure): " + __errMsg);
+      if (__errStack) __log.error(__errStack);
+      __ctx.statelogClient?.error?.({
+        errorType: "runtimeError",
+        message: __errMsg,
+        functionName: "setCodeOneShot",
+        retryable: __self.__retryable
+      });
+    }
+    return failure(
+      __error instanceof Error ? __error.message : String(__error),
+      {
+        checkpoint: getRuntimeContext().ctx.getResultCheckpoint(),
+        retryable: __self.__retryable,
+        functionName: "setCodeOneShot",
+        args: __stack.args
+      }
+    );
+  } finally {
+    __stateStack()?.pop();
+    if (__functionCompleted) {
+      await callHook({
+        name: "onFunctionEnd",
+        data: {
+          functionName: "setCodeOneShot",
+          timeTaken: performance.now() - __funcStartTime
+        }
+      });
+    }
+  }
+}
+const setCodeOneShot = __AgencyFunction.create({
+  name: "setCodeOneShot",
+  module: "dist/lib/agents/agency-agent/subagents/code.agency",
+  fn: __setCodeOneShot_impl,
+  params: [{
+    name: "oneShot",
+    hasDefault: false,
+    defaultValue: void 0,
+    variadic: false,
+    isFunctionTyped: false
+  }],
+  toolDefinition: {
+    name: "setCodeOneShot",
+    description: "No description provided.",
+    schema: z.object({ "oneShot": z.boolean() })
+  },
+  safe: false,
+  exported: true
+}, __toolRegistry);
 async function __codeAgent_impl(userMsg, allowHandoff) {
   const __setupData = setupFunction();
   const __stack = __setupData.stack;
@@ -854,10 +978,45 @@ async function __codeAgent_impl(userMsg, allowHandoff) {
               condition: async () => __globals().get("dist/lib/agents/agency-agent/subagents/code.agency", "_first"),
               body: async (runner4) => {
                 await runner4.step(0, async (runner5) => {
+                });
+                await runner4.step(1, async (runner5) => {
+                  __stack.locals.sys = __readStatic(codeSysPrompt, "codeSysPrompt", "dist/lib/agents/agency-agent/subagents/code.agency");
+                });
+                await runner4.ifElse(2, [
+                  {
+                    condition: async () => !__eq(await __call(getSearchBackend, {
+                      type: "positional",
+                      args: []
+                    }), `off`),
+                    body: async (runner5) => {
+                      await runner5.step(0, async (runner6) => {
+                        __stack.locals.sys = `${__stack.locals.sys}
+
+${__readStatic(WEB_SEARCH_NOTE, "WEB_SEARCH_NOTE", "dist/lib/agents/agency-agent/subagents/code.agency")}`;
+                      });
+                    }
+                  }
+                ]);
+                await runner4.step(3, async (runner5) => {
+                  __self.__retryable = false;
+                });
+                await runner4.ifElse(4, [
+                  {
+                    condition: async () => __globals().get("dist/lib/agents/agency-agent/subagents/code.agency", "_oneShot"),
+                    body: async (runner5) => {
+                      await runner5.step(0, async (runner6) => {
+                        __stack.locals.sys = `${__stack.locals.sys}
+
+${__readStatic(ONE_SHOT_NOTE, "ONE_SHOT_NOTE", "dist/lib/agents/agency-agent/subagents/code.agency")}`;
+                      });
+                    }
+                  }
+                ]);
+                await runner4.step(5, async (runner5) => {
                   __self.__retryable = false;
                   const __funcResult = await __call(systemMessage, {
                     type: "positional",
-                    args: [__readStatic(codeSysPrompt, "codeSysPrompt", "dist/lib/agents/agency-agent/subagents/code.agency")]
+                    args: [__stack.locals.sys]
                   });
                   if (hasInterrupts(__funcResult)) {
                     await getRuntimeContext().ctx.pendingPromises.awaitAll();
@@ -865,7 +1024,7 @@ async function __codeAgent_impl(userMsg, allowHandoff) {
                     return;
                   }
                 });
-                await runner4.step(1, async (runner5) => {
+                await runner4.step(6, async (runner5) => {
                   __globals().set("dist/lib/agents/agency-agent/subagents/code.agency", "_first", false);
                 });
               }
@@ -1073,7 +1232,7 @@ const codeAgent = __AgencyFunction.create({
   exported: true
 }, __toolRegistry);
 var stdin_default = graph;
-const __sourceMap = { "dist/lib/agents/agency-agent/subagents/code.agency:agencyCli": { "1": { "line": 66, "col": 2 } }, "dist/lib/agents/agency-agent/subagents/code.agency:codeAgent": { "1": { "line": 374, "col": 2 }, "2": { "line": 375, "col": 2 }, "3": { "line": 376, "col": 2 }, "4": { "line": 377, "col": 2 }, "5": { "line": 381, "col": 2 }, "6": { "line": 382, "col": 2 }, "7": { "line": 385, "col": 2 }, "8": { "line": 392, "col": 2 }, "9": { "line": 393, "col": 2 }, "11": { "line": 436, "col": 2 }, "4.0": { "line": 378, "col": 4 }, "9.0": { "line": 394, "col": 4 }, "9.1.0": { "line": 396, "col": 6 }, "9.1.2.0": { "line": 399, "col": 8 }, "9.1.2.1": { "line": 400, "col": 8 }, "9.1.2": { "line": 398, "col": 6 }, "9.1.4": { "line": 402, "col": 6 }, "9.1.5": { "line": 410, "col": 6 }, "9.1.6.0": { "line": 412, "col": 8 }, "9.1.6.1": { "line": 413, "col": 8 }, "9.1.6.2": { "line": 414, "col": 8 }, "9.1.6.3": { "line": 416, "col": 8 }, "9.1.6": { "line": 411, "col": 6 }, "9.1": { "line": 395, "col": 4 }, "9.3": { "line": 420, "col": 4 }, "9.4": { "line": 434, "col": 4 } }, "dist/lib/agents/agency-agent/subagents/code.agency:__block_0": {} };
+const __sourceMap = { "dist/lib/agents/agency-agent/subagents/code.agency:agencyCli": { "1": { "line": 66, "col": 2 } }, "dist/lib/agents/agency-agent/subagents/code.agency:setCodeOneShot": { "1": { "line": 378, "col": 2 } }, "dist/lib/agents/agency-agent/subagents/code.agency:codeAgent": { "1": { "line": 394, "col": 2 }, "2": { "line": 395, "col": 2 }, "3": { "line": 396, "col": 2 }, "4": { "line": 397, "col": 2 }, "5": { "line": 401, "col": 2 }, "6": { "line": 402, "col": 2 }, "7": { "line": 405, "col": 2 }, "8": { "line": 412, "col": 2 }, "9": { "line": 413, "col": 2 }, "11": { "line": 465, "col": 2 }, "4.0": { "line": 398, "col": 4 }, "9.0": { "line": 414, "col": 4 }, "9.1.0": { "line": 416, "col": 6 }, "9.1.2.1": { "line": 421, "col": 8 }, "9.1.2.2.0": { "line": 423, "col": 10 }, "9.1.2.2": { "line": 422, "col": 8 }, "9.1.2.4.0": { "line": 426, "col": 10 }, "9.1.2.4": { "line": 425, "col": 8 }, "9.1.2.5": { "line": 428, "col": 8 }, "9.1.2.6": { "line": 429, "col": 8 }, "9.1.2": { "line": 418, "col": 6 }, "9.1.4": { "line": 431, "col": 6 }, "9.1.5": { "line": 439, "col": 6 }, "9.1.6.0": { "line": 441, "col": 8 }, "9.1.6.1": { "line": 442, "col": 8 }, "9.1.6.2": { "line": 443, "col": 8 }, "9.1.6.3": { "line": 445, "col": 8 }, "9.1.6": { "line": 440, "col": 6 }, "9.1": { "line": 415, "col": 4 }, "9.3": { "line": 449, "col": 4 }, "9.4": { "line": 463, "col": 4 } }, "dist/lib/agents/agency-agent/subagents/code.agency:__block_0": {} };
 export {
   __getCheckpoints,
   __invokeFunction,
@@ -1094,5 +1253,6 @@ export {
   isInterrupt,
   reject,
   respondToInterrupts,
-  rewindFrom
+  rewindFrom,
+  setCodeOneShot
 };

package/dist/lib/agents/agency-agent/tests/oneShotRounds.agency

@@ -0,0 +1,20 @@
+// resolveMaxToolCallRounds: an explicit --max-tool-call-rounds always wins;
+// otherwise one-shot runs get the higher default and interactive runs keep the
+// compiled default (null).
+import { resolveMaxToolCallRounds } from "../shared.agency"
+
+node explicitWinsOverOneShot(): boolean {
+  return resolveMaxToolCallRounds(7, true) == 7
+}
+
+node oneShotDefaults50(): boolean {
+  return resolveMaxToolCallRounds(null, true) == 50
+}
+
+node explicitWinsOverInteractive(): boolean {
+  return resolveMaxToolCallRounds(3, false) == 3
+}
+
+node interactiveKeepsBaked(): boolean {
+  return resolveMaxToolCallRounds(null, false) == null
+}