@myagentroam/agent 0.9.136 → 0.9.138

package/dist/tools/registry.js

@@ -1,6 +1,6 @@
 import { applyPatchToolDefinition } from './apply-patch.js';
 import { execToolDefinition } from './exec.js';
-import { codeModeToolDefinition } from './code-mode.js';
+import { codeModeToolDefinition, codeExecWaitToolDefinition } from './code-mode.js';
 import { executeQuestion, questionToolDefinition } from './question.js';
 import { readToolDefinition } from './read.js';
 import { viewImageToolDefinition } from './view-image.js';
@@ -21,6 +21,7 @@ export const BUILTIN_TOOL_DEFINITIONS = [
     viewImageToolDefinition,
     execToolDefinition,
     codeModeToolDefinition,
+    codeExecWaitToolDefinition,
     applyPatchToolDefinition,
     questionToolDefinition,
     todoReadToolDefinition,

package/dist/tools/scheduler.d.ts

@@ -1,4 +1,6 @@
 import type { ClientToolDefinition } from '../model/contracts.js';
+/** One barrier domain for foreground tools and all retained cells in a Session. */
+export declare function createSessionToolScheduler(): <T>(definition: ClientToolDefinition | undefined, run: () => Promise<T>) => Promise<T>;
 /** Schedules an incremental stream while preserving the same safe/serial barriers as batches. */
 export declare function createToolCallScheduler<TCall extends {
     readonly name: string;

package/dist/tools/scheduler.js

@@ -1,3 +1,14 @@
+/** One barrier domain for foreground tools and all retained cells in a Session. */
+export function createSessionToolScheduler() {
+    const schedule = createToolCallScheduler([
+        { name: 'safe', parallelSafety: 'safe' },
+        { name: 'serial', parallelSafety: 'serial' }
+    ], (job) => job.run());
+    return (definition, run) => schedule({
+        name: definition?.parallelSafety === 'safe' ? 'safe' : 'serial',
+        run
+    });
+}
 /** Schedules an incremental stream while preserving the same safe/serial barriers as batches. */
 export function createToolCallScheduler(definitions, execute) {
     const safety = new Map(definitions.map((definition) => [definition.name, definition.parallelSafety]));

package/dist/tools/tool-search.js

@@ -1,7 +1,7 @@
 import { MarAgentError } from '../error.js';
 import { assertMcpToolCatalog } from '../sdk/mcp-catalog.js';
 import { countModelTokens } from '../runtime/token-budget.js';
-import { projectCodeModeTools } from './code-mode.js';
+import { projectCodeModeSearchResult, projectCodeModeTools } from './code-mode.js';
 export const TOOL_SEARCH_LIMITS = Object.freeze({
     defaultLimit: 5,
     maximumLimit: 20,
@@ -13,7 +13,7 @@ export const TOOL_SEARCH_LIMITS = Object.freeze({
 });
 export const toolSearchToolDefinition = {
     name: 'tool_search',
-    description: 'Search deferred MCP metadata and expose matching schemas for the next model call. Use the full callable name for exact selection, or keywords for BM25 search. Returns tool names, never executes them. Already exposed tools need not be searched again. The selected definitions must fit the exposure budget or the search fails atomically.',
+    description: 'Search deferred MCP metadata and return matching tool definitions for the next model call. Use the full callable name for exact selection, or keywords for BM25 search. Never executes the discovered tools. With Code Mode Only, programmatic tools must be called through code_exec, not directly. Already exposed tools need not be searched again. The selected definitions must fit the exposure budget or the search fails atomically.',
     inputSchema: {
         type: 'object',
         properties: {
@@ -201,10 +201,16 @@ export class DeferredMcpRegistry {
         };
     }
     assertBudget(base, window, system, selection = this.#selected, executionSelection = this.#executionSelected) {
-        const tools = projectCodeModeTools(this.project(base, selection, new Set(executionSelection)));
-        const direct = projectCodeModeTools(this.projectDirect(base));
-        const full = countModelTokens(JSON.stringify(tools));
-        const estimated = Math.max(0, full - countModelTokens(JSON.stringify(direct)));
+        const resident = this.projectDirect(base);
+        const active = this.project(base, selection, new Set(executionSelection));
+        const tools = projectCodeModeTools(active, resident);
+        const direct = projectCodeModeTools(resident);
+        const residentNames = new Set(resident.map((tool) => tool.name));
+        const definitions = active.filter((tool) => !residentNames.has(tool.name));
+        const searchResult = projectCodeModeSearchResult({ role: 'tool', content: '', toolSearchTools: definitions }, base.some((tool) => tool.name === 'code_exec'));
+        // Search-result definitions still consume context even when Only mode hides direct tools.
+        const estimated = countModelTokens(searchResult.content || JSON.stringify(searchResult.toolSearchTools));
+        const full = Math.max(countModelTokens(JSON.stringify(tools)), countModelTokens(JSON.stringify(direct)) + estimated);
         const maximum = Math.min(TOOL_SEARCH_LIMITS.maximumDefinitionTokens, Math.floor(window * TOOL_SEARCH_LIMITS.definitionWindowFraction));
         if (estimated > maximum ||
             full + countModelTokens(system) + TOOL_SEARCH_LIMITS.minimumContinuationTokens > window)

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@myagentroam/agent",
-  "version": "0.9.136",
+  "version": "0.9.138",
   "description": "Embeddable MAR coding agent SDK and CLI.",
   "license": "Apache-2.0",
   "type": "module",