@librechat/agents 3.1.90 → 3.1.92

package/dist/cjs/tools/local/resolveLocalExecutionTools.cjs

@@ -4,13 +4,53 @@ var _enum = require('../../common/enum.cjs');
 var LocalExecutionTools = require('./LocalExecutionTools.cjs');
 var LocalCodingTools = require('./LocalCodingTools.cjs');
 var LocalProgrammaticToolCalling = require('./LocalProgrammaticToolCalling.cjs');
+require('path');
+require('../cloudflare/CloudflareProgrammaticToolCalling.cjs');
+require('events');
+require('stream');
+require('./LocalExecutionEngine.cjs');
+var CloudflareSandboxTools = require('../cloudflare/CloudflareSandboxTools.cjs');
 
 function shouldUseLocalExecution(config) {
     return config?.engine === 'local';
 }
+function shouldUseCloudflareSandboxExecution(config) {
+    return config?.engine === 'cloudflare-sandbox';
+}
 function shouldIncludeCodingTools(config) {
-    return (shouldUseLocalExecution(config) &&
-        config?.local?.includeCodingTools !== false);
+    return ((shouldUseLocalExecution(config) &&
+        config?.local?.includeCodingTools !== false) ||
+        (shouldUseCloudflareSandboxExecution(config) &&
+            config?.cloudflare?.includeCodingTools !== false));
+}
+function getCloudflareConfig(config) {
+    if (config?.cloudflare == null) {
+        throw new Error('toolExecution.cloudflare is required when engine is "cloudflare-sandbox".');
+    }
+    return config.cloudflare;
+}
+function getSelectedCloudflareCodingToolNames(config) {
+    return new Set(config.codingToolNames ?? _enum.LOCAL_CODING_BUNDLE_NAMES);
+}
+function filterCloudflareCodingToolAllowlist(tools, selectedNames) {
+    const existingTools = tools ?? [];
+    if (existingTools.length === 0) {
+        return tools;
+    }
+    return existingTools.filter((existingTool) => {
+        if (!('name' in existingTool) || typeof existingTool.name !== 'string') {
+            return true;
+        }
+        return (!_enum.LOCAL_CODING_BUNDLE_NAMES.includes(existingTool.name) ||
+            selectedNames.has(existingTool.name));
+    });
+}
+function pruneCloudflareCodingToolAllowlist(toolMap, selectedNames) {
+    for (const name of _enum.LOCAL_CODING_BUNDLE_NAMES) {
+        if (!selectedNames.has(name)) {
+            toolMap.delete(name);
+        }
+    }
 }
 function createLocalExecutionTool(name, config) {
     switch (name) {
@@ -47,9 +87,26 @@ function mergeToolsByName(baseTools, localTools) {
     return orderedTools;
 }
 function resolveLocalToolsForBinding(args) {
-    if (!shouldUseLocalExecution(args.toolExecution)) {
+    if (!shouldUseLocalExecution(args.toolExecution) &&
+        !shouldUseCloudflareSandboxExecution(args.toolExecution)) {
         return args.tools;
     }
+    if (shouldUseCloudflareSandboxExecution(args.toolExecution)) {
+        const cloudflareConfig = getCloudflareConfig(args.toolExecution);
+        if (shouldIncludeCodingTools(args.toolExecution)) {
+            const selectedNames = getSelectedCloudflareCodingToolNames(cloudflareConfig);
+            return mergeToolsByName(filterCloudflareCodingToolAllowlist(args.tools, selectedNames), CloudflareSandboxTools.createCloudflareCodingTools(cloudflareConfig));
+        }
+        const replacements = (args.tools ?? [])
+            .filter((existingTool) => 'name' in existingTool &&
+            typeof existingTool.name === 'string' &&
+            _enum.CODE_EXECUTION_TOOLS.has(existingTool.name))
+            .map((existingTool) => CloudflareSandboxTools.createCloudflareExecutionTool(existingTool.name, cloudflareConfig))
+            .filter((cloudflareTool) => cloudflareTool != null);
+        return replacements.length === 0
+            ? args.tools
+            : mergeToolsByName(args.tools, replacements);
+    }
     const localConfig = args.toolExecution?.local ?? {};
     if (shouldIncludeCodingTools(args.toolExecution)) {
         return mergeToolsByName(args.tools, LocalCodingTools.createLocalCodingTools(localConfig));
@@ -69,22 +126,68 @@ function resolveLocalToolRegistry(args) {
         return args.toolRegistry;
     }
     const registry = new Map(args.toolRegistry ?? []);
+    const selectedNames = shouldUseCloudflareSandboxExecution(args.toolExecution)
+        ? getSelectedCloudflareCodingToolNames(getCloudflareConfig(args.toolExecution))
+        : undefined;
     for (const definition of LocalCodingTools.createLocalCodingToolDefinitions()) {
+        if (selectedNames != null && !selectedNames.has(definition.name)) {
+            registry.delete(definition.name);
+            continue;
+        }
         registry.set(definition.name, definition);
     }
     return registry;
 }
 function resolveLocalExecutionTools(args) {
     const directToolNames = new Set();
-    if (!shouldUseLocalExecution(args.toolExecution)) {
+    if (!shouldUseLocalExecution(args.toolExecution) &&
+        !shouldUseCloudflareSandboxExecution(args.toolExecution)) {
         return {
             toolMap: args.toolMap,
             directToolNames,
         };
     }
-    const localConfig = args.toolExecution?.local ?? {};
     const toolMap = new Map(args.toolMap);
     let fileCheckpointer;
+    if (shouldUseCloudflareSandboxExecution(args.toolExecution)) {
+        const cloudflareConfig = getCloudflareConfig(args.toolExecution);
+        if (shouldIncludeCodingTools(args.toolExecution)) {
+            const selectedNames = getSelectedCloudflareCodingToolNames(cloudflareConfig);
+            pruneCloudflareCodingToolAllowlist(toolMap, selectedNames);
+            if (cloudflareConfig.fileCheckpointing === true ||
+                args.fileCheckpointer != null) {
+                const bundle = CloudflareSandboxTools.createCloudflareCodingToolBundle(cloudflareConfig, {
+                    checkpointer: args.fileCheckpointer,
+                });
+                fileCheckpointer = bundle.checkpointer;
+                for (const cloudflareTool of bundle.tools) {
+                    toolMap.set(cloudflareTool.name, cloudflareTool);
+                    directToolNames.add(cloudflareTool.name);
+                }
+            }
+            else {
+                for (const cloudflareTool of CloudflareSandboxTools.createCloudflareCodingTools(cloudflareConfig)) {
+                    toolMap.set(cloudflareTool.name, cloudflareTool);
+                    directToolNames.add(cloudflareTool.name);
+                }
+            }
+        }
+        const includeCodingTools = shouldIncludeCodingTools(args.toolExecution);
+        for (const name of _enum.CODE_EXECUTION_TOOLS) {
+            if (includeCodingTools)
+                continue;
+            if (!toolMap.has(name))
+                continue;
+            const cloudflareTool = CloudflareSandboxTools.createCloudflareExecutionTool(name, cloudflareConfig);
+            if (cloudflareTool == null) {
+                continue;
+            }
+            toolMap.set(name, cloudflareTool);
+            directToolNames.add(name);
+        }
+        return { toolMap, directToolNames, fileCheckpointer };
+    }
+    const localConfig = args.toolExecution?.local ?? {};
     if (shouldIncludeCodingTools(args.toolExecution)) {
         // Use the bundle factory when fileCheckpointing is on so we can
         // surface the checkpointer back to the caller — without this, the
@@ -92,7 +195,8 @@ function resolveLocalExecutionTools(args) {
         // immediately discarded, making the public `fileCheckpointing`
         // config flag a silent no-op outside of direct
         // `createLocalCodingToolBundle()` use.
-        if (localConfig.fileCheckpointing === true || args.fileCheckpointer != null) {
+        if (localConfig.fileCheckpointing === true ||
+            args.fileCheckpointer != null) {
             const bundle = LocalCodingTools.createLocalCodingToolBundle(localConfig, {
                 checkpointer: args.fileCheckpointer,
             });

package/dist/cjs/tools/local/resolveLocalExecutionTools.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"resolveLocalExecutionTools.cjs","sources":["../../../../src/tools/local/resolveLocalExecutionTools.ts"],"sourcesContent":["import { Constants, CODE_EXECUTION_TOOLS } from '@/common';\nimport {\n  createLocalBashExecutionTool,\n  createLocalCodeExecutionTool,\n} from './LocalExecutionTools';\nimport {\n  createLocalCodingToolBundle,\n  createLocalCodingToolDefinitions,\n  createLocalCodingTools,\n} from './LocalCodingTools';\nimport {\n  createLocalBashProgrammaticToolCallingTool,\n  createLocalProgrammaticToolCallingTool,\n} from './LocalProgrammaticToolCalling';\nimport type * as t from '@/types';\n\ntype ResolveLocalToolsResult = {\n  toolMap: t.ToolMap;\n  directToolNames: Set<string>;\n  /**\n   * Set when `local.fileCheckpointing === true` AND the auto-bind\n   * coding suite is in use. ToolNode stashes this on the node and\n   * exposes it via `getFileCheckpointer()` so the host can call\n   * `rewind()` after a failed batch. Manual review (finding E)\n   * flagged that the config flag was previously a no-op in the\n   * Run/ToolNode auto-bind path — only direct\n   * `createLocalCodingToolBundle()` callers could access the\n   * checkpointer.\n   */\n  fileCheckpointer?: t.LocalFileCheckpointer;\n};\n\nfunction shouldUseLocalExecution(config?: t.ToolExecutionConfig): boolean {\n  return config?.engine === 'local';\n}\n\nfunction shouldIncludeCodingTools(config?: t.ToolExecutionConfig): boolean {\n  return (\n    shouldUseLocalExecution(config) &&\n    config?.local?.includeCodingTools !== false\n  );\n}\n\nfunction createLocalExecutionTool(\n  name: string,\n  config: t.LocalExecutionConfig\n): t.GenericTool | undefined {\n  switch (name) {\n  case Constants.EXECUTE_CODE:\n    return createLocalCodeExecutionTool(config);\n  case Constants.BASH_TOOL:\n    return createLocalBashExecutionTool({ config });\n  case Constants.PROGRAMMATIC_TOOL_CALLING:\n    return createLocalProgrammaticToolCallingTool(config);\n  case Constants.BASH_PROGRAMMATIC_TOOL_CALLING:\n    return createLocalBashProgrammaticToolCallingTool(config);\n  default:\n    return undefined;\n  }\n}\n\nfunction mergeToolsByName(\n  baseTools: t.GraphTools | undefined,\n  localTools: t.GenericTool[]\n): t.GraphTools {\n  const orderedTools: t.GenericTool[] = [];\n  const indexByName = new Map<string, number>();\n\n  for (const tool of (baseTools as t.GenericTool[] | undefined) ?? []) {\n    if ('name' in tool && typeof tool.name === 'string') {\n      indexByName.set(tool.name, orderedTools.length);\n    }\n    orderedTools.push(tool);\n  }\n\n  for (const tool of localTools) {\n    const existingIndex = indexByName.get(tool.name);\n    if (existingIndex == null) {\n      indexByName.set(tool.name, orderedTools.length);\n      orderedTools.push(tool);\n      continue;\n    }\n    orderedTools[existingIndex] = tool;\n  }\n\n  return orderedTools;\n}\n\nexport function resolveLocalToolsForBinding(args: {\n  tools?: t.GraphTools;\n  toolExecution?: t.ToolExecutionConfig;\n}): t.GraphTools | undefined {\n  if (!shouldUseLocalExecution(args.toolExecution)) {\n    return args.tools;\n  }\n\n  const localConfig = args.toolExecution?.local ?? {};\n  if (shouldIncludeCodingTools(args.toolExecution)) {\n    return mergeToolsByName(args.tools, createLocalCodingTools(localConfig));\n  }\n\n  const replacements = ((args.tools as t.GenericTool[] | undefined) ?? [])\n    .filter(\n      (existingTool): existingTool is t.GenericTool & { name: string } =>\n        'name' in existingTool &&\n        typeof existingTool.name === 'string' &&\n        CODE_EXECUTION_TOOLS.has(existingTool.name)\n    )\n    .map((existingTool) =>\n      createLocalExecutionTool(existingTool.name, localConfig)\n    )\n    .filter((localTool): localTool is t.GenericTool => localTool != null);\n\n  return replacements.length === 0\n    ? args.tools\n    : mergeToolsByName(args.tools, replacements);\n}\n\nexport function resolveLocalToolRegistry(args: {\n  toolRegistry?: t.LCToolRegistry;\n  toolExecution?: t.ToolExecutionConfig;\n}): t.LCToolRegistry | undefined {\n  if (!shouldIncludeCodingTools(args.toolExecution)) {\n    return args.toolRegistry;\n  }\n\n  const registry = new Map(args.toolRegistry ?? []);\n  for (const definition of createLocalCodingToolDefinitions()) {\n    registry.set(definition.name, definition);\n  }\n  return registry;\n}\n\nexport function resolveLocalExecutionTools(args: {\n  toolMap: t.ToolMap;\n  toolExecution?: t.ToolExecutionConfig;\n  /**\n   * Caller-provided checkpointer that overrides the bundle's\n   * auto-created one. The Graph layer threads a single per-Run\n   * instance so every ToolNode it compiles shares one snapshot\n   * store — without that, a multi-agent graph would each get a\n   * private checkpointer and `Run.rewindFiles()` couldn't reach\n   * any of them.\n   */\n  fileCheckpointer?: t.LocalFileCheckpointer;\n}): ResolveLocalToolsResult {\n  const directToolNames = new Set<string>();\n  if (!shouldUseLocalExecution(args.toolExecution)) {\n    return {\n      toolMap: args.toolMap,\n      directToolNames,\n    };\n  }\n\n  const localConfig = args.toolExecution?.local ?? {};\n  const toolMap = new Map(args.toolMap);\n  let fileCheckpointer: t.LocalFileCheckpointer | undefined;\n\n  if (shouldIncludeCodingTools(args.toolExecution)) {\n    // Use the bundle factory when fileCheckpointing is on so we can\n    // surface the checkpointer back to the caller — without this, the\n    // execution-path tools each captured into a checkpointer that was\n    // immediately discarded, making the public `fileCheckpointing`\n    // config flag a silent no-op outside of direct\n    // `createLocalCodingToolBundle()` use.\n    if (localConfig.fileCheckpointing === true || args.fileCheckpointer != null) {\n      const bundle = createLocalCodingToolBundle(localConfig, {\n        checkpointer: args.fileCheckpointer,\n      });\n      fileCheckpointer = bundle.checkpointer;\n      for (const localTool of bundle.tools) {\n        toolMap.set(localTool.name, localTool);\n        directToolNames.add(localTool.name);\n      }\n    } else {\n      for (const localTool of createLocalCodingTools(localConfig)) {\n        toolMap.set(localTool.name, localTool);\n        directToolNames.add(localTool.name);\n      }\n    }\n  }\n\n  // When the coding-tool bundle was already installed above, it\n  // already created `bash_tool` / `execute_code` / programmatic-tool\n  // variants. Skip re-creating them here — the audit-of-audit (manual\n  // finding #4) flagged that the original loop overwrote those bundle\n  // instances with fresh ones via `createLocalExecutionTool`, wasting\n  // work and (more importantly) replacing tools the bundle had\n  // already wired up with shared state. The CODE_EXECUTION_TOOLS\n  // loop is now only relevant when the host pre-bound a tool with\n  // one of these names (the `toolMap.has(name)` branch) and coding\n  // tools are off.\n  const includeCodingTools = shouldIncludeCodingTools(args.toolExecution);\n  for (const name of CODE_EXECUTION_TOOLS) {\n    if (includeCodingTools) continue;\n    if (!toolMap.has(name)) continue;\n\n    const localTool = createLocalExecutionTool(name, localConfig);\n    if (localTool == null) {\n      continue;\n    }\n\n    toolMap.set(name, localTool);\n    directToolNames.add(name);\n  }\n\n  return { toolMap, directToolNames, fileCheckpointer };\n}\n"],"names":["Constants","createLocalCodeExecutionTool","createLocalBashExecutionTool","createLocalProgrammaticToolCallingTool","createLocalBashProgrammaticToolCallingTool","createLocalCodingTools","CODE_EXECUTION_TOOLS","createLocalCodingToolDefinitions","createLocalCodingToolBundle"],"mappings":";;;;;;;AAgCA,SAAS,uBAAuB,CAAC,MAA8B,EAAA;AAC7D,IAAA,OAAO,MAAM,EAAE,MAAM,KAAK,OAAO;AACnC;AAEA,SAAS,wBAAwB,CAAC,MAA8B,EAAA;AAC9D,IAAA,QACE,uBAAuB,CAAC,MAAM,CAAC;AAC/B,QAAA,MAAM,EAAE,KAAK,EAAE,kBAAkB,KAAK,KAAK;AAE/C;AAEA,SAAS,wBAAwB,CAC/B,IAAY,EACZ,MAA8B,EAAA;IAE9B,QAAQ,IAAI;QACZ,KAAKA,eAAS,CAAC,YAAY;AACzB,YAAA,OAAOC,gDAA4B,CAAC,MAAM,CAAC;QAC7C,KAAKD,eAAS,CAAC,SAAS;AACtB,YAAA,OAAOE,gDAA4B,CAAC,EAAE,MAAM,EAAE,CAAC;QACjD,KAAKF,eAAS,CAAC,yBAAyB;AACtC,YAAA,OAAOG,mEAAsC,CAAC,MAAM,CAAC;QACvD,KAAKH,eAAS,CAAC,8BAA8B;AAC3C,YAAA,OAAOI,uEAA0C,CAAC,MAAM,CAAC;AAC3D,QAAA;AACE,YAAA,OAAO,SAAS;;AAEpB;AAEA,SAAS,gBAAgB,CACvB,SAAmC,EACnC,UAA2B,EAAA;IAE3B,MAAM,YAAY,GAAoB,EAAE;AACxC,IAAA,MAAM,WAAW,GAAG,IAAI,GAAG,EAAkB;AAE7C,IAAA,KAAK,MAAM,IAAI,IAAK,SAAyC,IAAI,EAAE,EAAE;QACnE,IAAI,MAAM,IAAI,IAAI,IAAI,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE;YACnD,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,YAAY,CAAC,MAAM,CAAC;QACjD;AACA,QAAA,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC;IACzB;AAEA,IAAA,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE;QAC7B,MAAM,aAAa,GAAG,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC;AAChD,QAAA,IAAI,aAAa,IAAI,IAAI,EAAE;YACzB,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,YAAY,CAAC,MAAM,CAAC;AAC/C,YAAA,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC;YACvB;QACF;AACA,QAAA,YAAY,CAAC,aAAa,CAAC,GAAG,IAAI;IACpC;AAEA,IAAA,OAAO,YAAY;AACrB;AAEM,SAAU,2BAA2B,CAAC,IAG3C,EAAA;IACC,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE;QAChD,OAAO,IAAI,CAAC,KAAK;IACnB;IAEA,MAAM,WAAW,GAAG,IAAI,CAAC,aAAa,EAAE,KAAK,IAAI,EAAE;AACnD,IAAA,IAAI,wBAAwB,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE;QAChD,OAAO,gBAAgB,CAAC,IAAI,CAAC,KAAK,EAAEC,uCAAsB,CAAC,WAAW,CAAC,CAAC;IAC1E;IAEA,MAAM,YAAY,GAAG,CAAE,IAAI,CAAC,KAAqC,IAAI,EAAE;SACpE,MAAM,CACL,CAAC,YAAY,KACX,MAAM,IAAI,YAAY;AACtB,QAAA,OAAO,YAAY,CAAC,IAAI,KAAK,QAAQ;AACrC,QAAAC,0BAAoB,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,CAAC;AAE9C,SAAA,GAAG,CAAC,CAAC,YAAY,KAChB,wBAAwB,CAAC,YAAY,CAAC,IAAI,EAAE,WAAW,CAAC;SAEzD,MAAM,CAAC,CAAC,SAAS,KAAiC,SAAS,IAAI,IAAI,CAAC;AAEvE,IAAA,OAAO,YAAY,CAAC,MAAM,KAAK;UAC3B,IAAI,CAAC;UACL,gBAAgB,CAAC,IAAI,CAAC,KAAK,EAAE,YAAY,CAAC;AAChD;AAEM,SAAU,wBAAwB,CAAC,IAGxC,EAAA;IACC,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE;QACjD,OAAO,IAAI,CAAC,YAAY;IAC1B;IAEA,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,YAAY,IAAI,EAAE,CAAC;AACjD,IAAA,KAAK,MAAM,UAAU,IAAIC,iDAAgC,EAAE,EAAE;QAC3D,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC;IAC3C;AACA,IAAA,OAAO,QAAQ;AACjB;AAEM,SAAU,0BAA0B,CAAC,IAY1C,EAAA;AACC,IAAA,MAAM,eAAe,GAAG,IAAI,GAAG,EAAU;IACzC,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE;QAChD,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,eAAe;SAChB;IACH;IAEA,MAAM,WAAW,GAAG,IAAI,CAAC,aAAa,EAAE,KAAK,IAAI,EAAE;IACnD,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC;AACrC,IAAA,IAAI,gBAAqD;AAEzD,IAAA,IAAI,wBAAwB,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE;;;;;;;AAOhD,QAAA,IAAI,WAAW,CAAC,iBAAiB,KAAK,IAAI,IAAI,IAAI,CAAC,gBAAgB,IAAI,IAAI,EAAE;AAC3E,YAAA,MAAM,MAAM,GAAGC,4CAA2B,CAAC,WAAW,EAAE;gBACtD,YAAY,EAAE,IAAI,CAAC,gBAAgB;AACpC,aAAA,CAAC;AACF,YAAA,gBAAgB,GAAG,MAAM,CAAC,YAAY;AACtC,YAAA,KAAK,MAAM,SAAS,IAAI,MAAM,CAAC,KAAK,EAAE;gBACpC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,SAAS,CAAC;AACtC,gBAAA,eAAe,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC;YACrC;QACF;aAAO;YACL,KAAK,MAAM,SAAS,IAAIH,uCAAsB,CAAC,WAAW,CAAC,EAAE;gBAC3D,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,SAAS,CAAC;AACtC,gBAAA,eAAe,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC;YACrC;QACF;IACF;;;;;;;;;;;IAYA,MAAM,kBAAkB,GAAG,wBAAwB,CAAC,IAAI,CAAC,aAAa,CAAC;AACvE,IAAA,KAAK,MAAM,IAAI,IAAIC,0BAAoB,EAAE;AACvC,QAAA,IAAI,kBAAkB;YAAE;AACxB,QAAA,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC;YAAE;QAExB,MAAM,SAAS,GAAG,wBAAwB,CAAC,IAAI,EAAE,WAAW,CAAC;AAC7D,QAAA,IAAI,SAAS,IAAI,IAAI,EAAE;YACrB;QACF;AAEA,QAAA,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,SAAS,CAAC;AAC5B,QAAA,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC;IAC3B;AAEA,IAAA,OAAO,EAAE,OAAO,EAAE,eAAe,EAAE,gBAAgB,EAAE;AACvD;;;;;;"}
+{"version":3,"file":"resolveLocalExecutionTools.cjs","sources":["../../../../src/tools/local/resolveLocalExecutionTools.ts"],"sourcesContent":["import {\n  Constants,\n  CODE_EXECUTION_TOOLS,\n  LOCAL_CODING_BUNDLE_NAMES,\n} from '@/common';\nimport {\n  createLocalBashExecutionTool,\n  createLocalCodeExecutionTool,\n} from './LocalExecutionTools';\nimport {\n  createLocalCodingToolBundle,\n  createLocalCodingToolDefinitions,\n  createLocalCodingTools,\n} from './LocalCodingTools';\nimport {\n  createLocalBashProgrammaticToolCallingTool,\n  createLocalProgrammaticToolCallingTool,\n} from './LocalProgrammaticToolCalling';\nimport {\n  createCloudflareCodingToolBundle,\n  createCloudflareCodingTools,\n  createCloudflareExecutionTool,\n} from '@/tools/cloudflare';\nimport type * as t from '@/types';\n\ntype ResolveLocalToolsResult = {\n  toolMap: t.ToolMap;\n  directToolNames: Set<string>;\n  /**\n   * Set when `local.fileCheckpointing === true` AND the auto-bind\n   * coding suite is in use. ToolNode stashes this on the node and\n   * exposes it via `getFileCheckpointer()` so the host can call\n   * `rewind()` after a failed batch. Manual review (finding E)\n   * flagged that the config flag was previously a no-op in the\n   * Run/ToolNode auto-bind path — only direct\n   * `createLocalCodingToolBundle()` callers could access the\n   * checkpointer.\n   */\n  fileCheckpointer?: t.LocalFileCheckpointer;\n};\n\nfunction shouldUseLocalExecution(config?: t.ToolExecutionConfig): boolean {\n  return config?.engine === 'local';\n}\n\nfunction shouldUseCloudflareSandboxExecution(\n  config?: t.ToolExecutionConfig\n): boolean {\n  return config?.engine === 'cloudflare-sandbox';\n}\n\nfunction shouldIncludeCodingTools(config?: t.ToolExecutionConfig): boolean {\n  return (\n    (shouldUseLocalExecution(config) &&\n      config?.local?.includeCodingTools !== false) ||\n    (shouldUseCloudflareSandboxExecution(config) &&\n      config?.cloudflare?.includeCodingTools !== false)\n  );\n}\n\nfunction getCloudflareConfig(\n  config?: t.ToolExecutionConfig\n): t.CloudflareSandboxExecutionConfig {\n  if (config?.cloudflare == null) {\n    throw new Error(\n      'toolExecution.cloudflare is required when engine is \"cloudflare-sandbox\".'\n    );\n  }\n  return config.cloudflare;\n}\n\nfunction getSelectedCloudflareCodingToolNames(\n  config: t.CloudflareSandboxExecutionConfig\n): Set<string> {\n  return new Set(config.codingToolNames ?? LOCAL_CODING_BUNDLE_NAMES);\n}\n\nfunction filterCloudflareCodingToolAllowlist(\n  tools: t.GraphTools | undefined,\n  selectedNames: Set<string>\n): t.GraphTools | undefined {\n  const existingTools = (tools as t.GenericTool[] | undefined) ?? [];\n  if (existingTools.length === 0) {\n    return tools;\n  }\n  return existingTools.filter((existingTool) => {\n    if (!('name' in existingTool) || typeof existingTool.name !== 'string') {\n      return true;\n    }\n    return (\n      !LOCAL_CODING_BUNDLE_NAMES.includes(existingTool.name) ||\n      selectedNames.has(existingTool.name)\n    );\n  });\n}\n\nfunction pruneCloudflareCodingToolAllowlist(\n  toolMap: t.ToolMap,\n  selectedNames: Set<string>\n): void {\n  for (const name of LOCAL_CODING_BUNDLE_NAMES) {\n    if (!selectedNames.has(name)) {\n      toolMap.delete(name);\n    }\n  }\n}\n\nfunction createLocalExecutionTool(\n  name: string,\n  config: t.LocalExecutionConfig\n): t.GenericTool | undefined {\n  switch (name) {\n  case Constants.EXECUTE_CODE:\n    return createLocalCodeExecutionTool(config);\n  case Constants.BASH_TOOL:\n    return createLocalBashExecutionTool({ config });\n  case Constants.PROGRAMMATIC_TOOL_CALLING:\n    return createLocalProgrammaticToolCallingTool(config);\n  case Constants.BASH_PROGRAMMATIC_TOOL_CALLING:\n    return createLocalBashProgrammaticToolCallingTool(config);\n  default:\n    return undefined;\n  }\n}\n\nfunction mergeToolsByName(\n  baseTools: t.GraphTools | undefined,\n  localTools: t.GenericTool[]\n): t.GraphTools {\n  const orderedTools: t.GenericTool[] = [];\n  const indexByName = new Map<string, number>();\n\n  for (const tool of (baseTools as t.GenericTool[] | undefined) ?? []) {\n    if ('name' in tool && typeof tool.name === 'string') {\n      indexByName.set(tool.name, orderedTools.length);\n    }\n    orderedTools.push(tool);\n  }\n\n  for (const tool of localTools) {\n    const existingIndex = indexByName.get(tool.name);\n    if (existingIndex == null) {\n      indexByName.set(tool.name, orderedTools.length);\n      orderedTools.push(tool);\n      continue;\n    }\n    orderedTools[existingIndex] = tool;\n  }\n\n  return orderedTools;\n}\n\nexport function resolveLocalToolsForBinding(args: {\n  tools?: t.GraphTools;\n  toolExecution?: t.ToolExecutionConfig;\n}): t.GraphTools | undefined {\n  if (\n    !shouldUseLocalExecution(args.toolExecution) &&\n    !shouldUseCloudflareSandboxExecution(args.toolExecution)\n  ) {\n    return args.tools;\n  }\n\n  if (shouldUseCloudflareSandboxExecution(args.toolExecution)) {\n    const cloudflareConfig = getCloudflareConfig(args.toolExecution);\n    if (shouldIncludeCodingTools(args.toolExecution)) {\n      const selectedNames =\n        getSelectedCloudflareCodingToolNames(cloudflareConfig);\n      return mergeToolsByName(\n        filterCloudflareCodingToolAllowlist(args.tools, selectedNames),\n        createCloudflareCodingTools(cloudflareConfig)\n      );\n    }\n\n    const replacements = ((args.tools as t.GenericTool[] | undefined) ?? [])\n      .filter(\n        (existingTool): existingTool is t.GenericTool & { name: string } =>\n          'name' in existingTool &&\n          typeof existingTool.name === 'string' &&\n          CODE_EXECUTION_TOOLS.has(existingTool.name)\n      )\n      .map((existingTool) =>\n        createCloudflareExecutionTool(existingTool.name, cloudflareConfig)\n      )\n      .filter(\n        (cloudflareTool): cloudflareTool is t.GenericTool =>\n          cloudflareTool != null\n      );\n\n    return replacements.length === 0\n      ? args.tools\n      : mergeToolsByName(args.tools, replacements);\n  }\n\n  const localConfig = args.toolExecution?.local ?? {};\n  if (shouldIncludeCodingTools(args.toolExecution)) {\n    return mergeToolsByName(args.tools, createLocalCodingTools(localConfig));\n  }\n\n  const replacements = ((args.tools as t.GenericTool[] | undefined) ?? [])\n    .filter(\n      (existingTool): existingTool is t.GenericTool & { name: string } =>\n        'name' in existingTool &&\n        typeof existingTool.name === 'string' &&\n        CODE_EXECUTION_TOOLS.has(existingTool.name)\n    )\n    .map((existingTool) =>\n      createLocalExecutionTool(existingTool.name, localConfig)\n    )\n    .filter((localTool): localTool is t.GenericTool => localTool != null);\n\n  return replacements.length === 0\n    ? args.tools\n    : mergeToolsByName(args.tools, replacements);\n}\n\nexport function resolveLocalToolRegistry(args: {\n  toolRegistry?: t.LCToolRegistry;\n  toolExecution?: t.ToolExecutionConfig;\n}): t.LCToolRegistry | undefined {\n  if (!shouldIncludeCodingTools(args.toolExecution)) {\n    return args.toolRegistry;\n  }\n\n  const registry = new Map(args.toolRegistry ?? []);\n  const selectedNames = shouldUseCloudflareSandboxExecution(args.toolExecution)\n    ? getSelectedCloudflareCodingToolNames(\n      getCloudflareConfig(args.toolExecution)\n    )\n    : undefined;\n  for (const definition of createLocalCodingToolDefinitions()) {\n    if (selectedNames != null && !selectedNames.has(definition.name)) {\n      registry.delete(definition.name);\n      continue;\n    }\n    registry.set(definition.name, definition);\n  }\n  return registry;\n}\n\nexport function resolveLocalExecutionTools(args: {\n  toolMap: t.ToolMap;\n  toolExecution?: t.ToolExecutionConfig;\n  /**\n   * Caller-provided checkpointer that overrides the bundle's\n   * auto-created one. The Graph layer threads a single per-Run\n   * instance so every ToolNode it compiles shares one snapshot\n   * store — without that, a multi-agent graph would each get a\n   * private checkpointer and `Run.rewindFiles()` couldn't reach\n   * any of them.\n   */\n  fileCheckpointer?: t.LocalFileCheckpointer;\n}): ResolveLocalToolsResult {\n  const directToolNames = new Set<string>();\n  if (\n    !shouldUseLocalExecution(args.toolExecution) &&\n    !shouldUseCloudflareSandboxExecution(args.toolExecution)\n  ) {\n    return {\n      toolMap: args.toolMap,\n      directToolNames,\n    };\n  }\n\n  const toolMap = new Map(args.toolMap);\n  let fileCheckpointer: t.LocalFileCheckpointer | undefined;\n\n  if (shouldUseCloudflareSandboxExecution(args.toolExecution)) {\n    const cloudflareConfig = getCloudflareConfig(args.toolExecution);\n    if (shouldIncludeCodingTools(args.toolExecution)) {\n      const selectedNames =\n        getSelectedCloudflareCodingToolNames(cloudflareConfig);\n      pruneCloudflareCodingToolAllowlist(toolMap, selectedNames);\n      if (\n        cloudflareConfig.fileCheckpointing === true ||\n        args.fileCheckpointer != null\n      ) {\n        const bundle = createCloudflareCodingToolBundle(cloudflareConfig, {\n          checkpointer: args.fileCheckpointer,\n        });\n        fileCheckpointer = bundle.checkpointer;\n        for (const cloudflareTool of bundle.tools) {\n          toolMap.set(cloudflareTool.name, cloudflareTool);\n          directToolNames.add(cloudflareTool.name);\n        }\n      } else {\n        for (const cloudflareTool of createCloudflareCodingTools(\n          cloudflareConfig\n        )) {\n          toolMap.set(cloudflareTool.name, cloudflareTool);\n          directToolNames.add(cloudflareTool.name);\n        }\n      }\n    }\n\n    const includeCodingTools = shouldIncludeCodingTools(args.toolExecution);\n    for (const name of CODE_EXECUTION_TOOLS) {\n      if (includeCodingTools) continue;\n      if (!toolMap.has(name)) continue;\n\n      const cloudflareTool = createCloudflareExecutionTool(\n        name,\n        cloudflareConfig\n      );\n      if (cloudflareTool == null) {\n        continue;\n      }\n\n      toolMap.set(name, cloudflareTool);\n      directToolNames.add(name);\n    }\n\n    return { toolMap, directToolNames, fileCheckpointer };\n  }\n\n  const localConfig = args.toolExecution?.local ?? {};\n\n  if (shouldIncludeCodingTools(args.toolExecution)) {\n    // Use the bundle factory when fileCheckpointing is on so we can\n    // surface the checkpointer back to the caller — without this, the\n    // execution-path tools each captured into a checkpointer that was\n    // immediately discarded, making the public `fileCheckpointing`\n    // config flag a silent no-op outside of direct\n    // `createLocalCodingToolBundle()` use.\n    if (\n      localConfig.fileCheckpointing === true ||\n      args.fileCheckpointer != null\n    ) {\n      const bundle = createLocalCodingToolBundle(localConfig, {\n        checkpointer: args.fileCheckpointer,\n      });\n      fileCheckpointer = bundle.checkpointer;\n      for (const localTool of bundle.tools) {\n        toolMap.set(localTool.name, localTool);\n        directToolNames.add(localTool.name);\n      }\n    } else {\n      for (const localTool of createLocalCodingTools(localConfig)) {\n        toolMap.set(localTool.name, localTool);\n        directToolNames.add(localTool.name);\n      }\n    }\n  }\n\n  // When the coding-tool bundle was already installed above, it\n  // already created `bash_tool` / `execute_code` / programmatic-tool\n  // variants. Skip re-creating them here — the audit-of-audit (manual\n  // finding #4) flagged that the original loop overwrote those bundle\n  // instances with fresh ones via `createLocalExecutionTool`, wasting\n  // work and (more importantly) replacing tools the bundle had\n  // already wired up with shared state. The CODE_EXECUTION_TOOLS\n  // loop is now only relevant when the host pre-bound a tool with\n  // one of these names (the `toolMap.has(name)` branch) and coding\n  // tools are off.\n  const includeCodingTools = shouldIncludeCodingTools(args.toolExecution);\n  for (const name of CODE_EXECUTION_TOOLS) {\n    if (includeCodingTools) continue;\n    if (!toolMap.has(name)) continue;\n\n    const localTool = createLocalExecutionTool(name, localConfig);\n    if (localTool == null) {\n      continue;\n    }\n\n    toolMap.set(name, localTool);\n    directToolNames.add(name);\n  }\n\n  return { toolMap, directToolNames, fileCheckpointer };\n}\n"],"names":["LOCAL_CODING_BUNDLE_NAMES","Constants","createLocalCodeExecutionTool","createLocalBashExecutionTool","createLocalProgrammaticToolCallingTool","createLocalBashProgrammaticToolCallingTool","createCloudflareCodingTools","CODE_EXECUTION_TOOLS","createCloudflareExecutionTool","createLocalCodingTools","createLocalCodingToolDefinitions","createCloudflareCodingToolBundle","createLocalCodingToolBundle"],"mappings":";;;;;;;;;;;;;AAyCA,SAAS,uBAAuB,CAAC,MAA8B,EAAA;AAC7D,IAAA,OAAO,MAAM,EAAE,MAAM,KAAK,OAAO;AACnC;AAEA,SAAS,mCAAmC,CAC1C,MAA8B,EAAA;AAE9B,IAAA,OAAO,MAAM,EAAE,MAAM,KAAK,oBAAoB;AAChD;AAEA,SAAS,wBAAwB,CAAC,MAA8B,EAAA;AAC9D,IAAA,QACE,CAAC,uBAAuB,CAAC,MAAM,CAAC;AAC9B,QAAA,MAAM,EAAE,KAAK,EAAE,kBAAkB,KAAK,KAAK;SAC5C,mCAAmC,CAAC,MAAM,CAAC;YAC1C,MAAM,EAAE,UAAU,EAAE,kBAAkB,KAAK,KAAK,CAAC;AAEvD;AAEA,SAAS,mBAAmB,CAC1B,MAA8B,EAAA;AAE9B,IAAA,IAAI,MAAM,EAAE,UAAU,IAAI,IAAI,EAAE;AAC9B,QAAA,MAAM,IAAI,KAAK,CACb,2EAA2E,CAC5E;IACH;IACA,OAAO,MAAM,CAAC,UAAU;AAC1B;AAEA,SAAS,oCAAoC,CAC3C,MAA0C,EAAA;IAE1C,OAAO,IAAI,GAAG,CAAC,MAAM,CAAC,eAAe,IAAIA,+BAAyB,CAAC;AACrE;AAEA,SAAS,mCAAmC,CAC1C,KAA+B,EAC/B,aAA0B,EAAA;AAE1B,IAAA,MAAM,aAAa,GAAI,KAAqC,IAAI,EAAE;AAClE,IAAA,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE;AAC9B,QAAA,OAAO,KAAK;IACd;AACA,IAAA,OAAO,aAAa,CAAC,MAAM,CAAC,CAAC,YAAY,KAAI;AAC3C,QAAA,IAAI,EAAE,MAAM,IAAI,YAAY,CAAC,IAAI,OAAO,YAAY,CAAC,IAAI,KAAK,QAAQ,EAAE;AACtE,YAAA,OAAO,IAAI;QACb;QACA,QACE,CAACA,+BAAyB,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC;YACtD,aAAa,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,CAAC;AAExC,IAAA,CAAC,CAAC;AACJ;AAEA,SAAS,kCAAkC,CACzC,OAAkB,EAClB,aAA0B,EAAA;AAE1B,IAAA,KAAK,MAAM,IAAI,IAAIA,+BAAyB,EAAE;QAC5C,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;AAC5B,YAAA,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC;QACtB;IACF;AACF;AAEA,SAAS,wBAAwB,CAC/B,IAAY,EACZ,MAA8B,EAAA;IAE9B,QAAQ,IAAI;QACZ,KAAKC,eAAS,CAAC,YAAY;AACzB,YAAA,OAAOC,gDAA4B,CAAC,MAAM,CAAC;QAC7C,KAAKD,eAAS,CAAC,SAAS;AACtB,YAAA,OAAOE,gDAA4B,CAAC,EAAE,MAAM,EAAE,CAAC;QACjD,KAAKF,eAAS,CAAC,yBAAyB;AACtC,YAAA,OAAOG,mEAAsC,CAAC,MAAM,CAAC;QACvD,KAAKH,eAAS,CAAC,8BAA8B;AAC3C,YAAA,OAAOI,uEAA0C,CAAC,MAAM,CAAC;AAC3D,QAAA;AACE,YAAA,OAAO,SAAS;;AAEpB;AAEA,SAAS,gBAAgB,CACvB,SAAmC,EACnC,UAA2B,EAAA;IAE3B,MAAM,YAAY,GAAoB,EAAE;AACxC,IAAA,MAAM,WAAW,GAAG,IAAI,GAAG,EAAkB;AAE7C,IAAA,KAAK,MAAM,IAAI,IAAK,SAAyC,IAAI,EAAE,EAAE;QACnE,IAAI,MAAM,IAAI,IAAI,IAAI,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE;YACnD,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,YAAY,CAAC,MAAM,CAAC;QACjD;AACA,QAAA,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC;IACzB;AAEA,IAAA,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE;QAC7B,MAAM,aAAa,GAAG,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC;AAChD,QAAA,IAAI,aAAa,IAAI,IAAI,EAAE;YACzB,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,YAAY,CAAC,MAAM,CAAC;AAC/C,YAAA,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC;YACvB;QACF;AACA,QAAA,YAAY,CAAC,aAAa,CAAC,GAAG,IAAI;IACpC;AAEA,IAAA,OAAO,YAAY;AACrB;AAEM,SAAU,2BAA2B,CAAC,IAG3C,EAAA;AACC,IAAA,IACE,CAAC,uBAAuB,CAAC,IAAI,CAAC,aAAa,CAAC;AAC5C,QAAA,CAAC,mCAAmC,CAAC,IAAI,CAAC,aAAa,CAAC,EACxD;QACA,OAAO,IAAI,CAAC,KAAK;IACnB;AAEA,IAAA,IAAI,mCAAmC,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE;QAC3D,MAAM,gBAAgB,GAAG,mBAAmB,CAAC,IAAI,CAAC,aAAa,CAAC;AAChE,QAAA,IAAI,wBAAwB,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE;AAChD,YAAA,MAAM,aAAa,GACjB,oCAAoC,CAAC,gBAAgB,CAAC;AACxD,YAAA,OAAO,gBAAgB,CACrB,mCAAmC,CAAC,IAAI,CAAC,KAAK,EAAE,aAAa,CAAC,EAC9DC,kDAA2B,CAAC,gBAAgB,CAAC,CAC9C;QACH;QAEA,MAAM,YAAY,GAAG,CAAE,IAAI,CAAC,KAAqC,IAAI,EAAE;aACpE,MAAM,CACL,CAAC,YAAY,KACX,MAAM,IAAI,YAAY;AACtB,YAAA,OAAO,YAAY,CAAC,IAAI,KAAK,QAAQ;AACrC,YAAAC,0BAAoB,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,CAAC;AAE9C,aAAA,GAAG,CAAC,CAAC,YAAY,KAChBC,oDAA6B,CAAC,YAAY,CAAC,IAAI,EAAE,gBAAgB,CAAC;aAEnE,MAAM,CACL,CAAC,cAAc,KACb,cAAc,IAAI,IAAI,CACzB;AAEH,QAAA,OAAO,YAAY,CAAC,MAAM,KAAK;cAC3B,IAAI,CAAC;cACL,gBAAgB,CAAC,IAAI,CAAC,KAAK,EAAE,YAAY,CAAC;IAChD;IAEA,MAAM,WAAW,GAAG,IAAI,CAAC,aAAa,EAAE,KAAK,IAAI,EAAE;AACnD,IAAA,IAAI,wBAAwB,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE;QAChD,OAAO,gBAAgB,CAAC,IAAI,CAAC,KAAK,EAAEC,uCAAsB,CAAC,WAAW,CAAC,CAAC;IAC1E;IAEA,MAAM,YAAY,GAAG,CAAE,IAAI,CAAC,KAAqC,IAAI,EAAE;SACpE,MAAM,CACL,CAAC,YAAY,KACX,MAAM,IAAI,YAAY;AACtB,QAAA,OAAO,YAAY,CAAC,IAAI,KAAK,QAAQ;AACrC,QAAAF,0BAAoB,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,CAAC;AAE9C,SAAA,GAAG,CAAC,CAAC,YAAY,KAChB,wBAAwB,CAAC,YAAY,CAAC,IAAI,EAAE,WAAW,CAAC;SAEzD,MAAM,CAAC,CAAC,SAAS,KAAiC,SAAS,IAAI,IAAI,CAAC;AAEvE,IAAA,OAAO,YAAY,CAAC,MAAM,KAAK;UAC3B,IAAI,CAAC;UACL,gBAAgB,CAAC,IAAI,CAAC,KAAK,EAAE,YAAY,CAAC;AAChD;AAEM,SAAU,wBAAwB,CAAC,IAGxC,EAAA;IACC,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE;QACjD,OAAO,IAAI,CAAC,YAAY;IAC1B;IAEA,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,YAAY,IAAI,EAAE,CAAC;AACjD,IAAA,MAAM,aAAa,GAAG,mCAAmC,CAAC,IAAI,CAAC,aAAa;UACxE,oCAAoC,CACpC,mBAAmB,CAAC,IAAI,CAAC,aAAa,CAAC;UAEvC,SAAS;AACb,IAAA,KAAK,MAAM,UAAU,IAAIG,iDAAgC,EAAE,EAAE;AAC3D,QAAA,IAAI,aAAa,IAAI,IAAI,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE;AAChE,YAAA,QAAQ,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC;YAChC;QACF;QACA,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC;IAC3C;AACA,IAAA,OAAO,QAAQ;AACjB;AAEM,SAAU,0BAA0B,CAAC,IAY1C,EAAA;AACC,IAAA,MAAM,eAAe,GAAG,IAAI,GAAG,EAAU;AACzC,IAAA,IACE,CAAC,uBAAuB,CAAC,IAAI,CAAC,aAAa,CAAC;AAC5C,QAAA,CAAC,mCAAmC,CAAC,IAAI,CAAC,aAAa,CAAC,EACxD;QACA,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,eAAe;SAChB;IACH;IAEA,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC;AACrC,IAAA,IAAI,gBAAqD;AAEzD,IAAA,IAAI,mCAAmC,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE;QAC3D,MAAM,gBAAgB,GAAG,mBAAmB,CAAC,IAAI,CAAC,aAAa,CAAC;AAChE,QAAA,IAAI,wBAAwB,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE;AAChD,YAAA,MAAM,aAAa,GACjB,oCAAoC,CAAC,gBAAgB,CAAC;AACxD,YAAA,kCAAkC,CAAC,OAAO,EAAE,aAAa,CAAC;AAC1D,YAAA,IACE,gBAAgB,CAAC,iBAAiB,KAAK,IAAI;AAC3C,gBAAA,IAAI,CAAC,gBAAgB,IAAI,IAAI,EAC7B;AACA,gBAAA,MAAM,MAAM,GAAGC,uDAAgC,CAAC,gBAAgB,EAAE;oBAChE,YAAY,EAAE,IAAI,CAAC,gBAAgB;AACpC,iBAAA,CAAC;AACF,gBAAA,gBAAgB,GAAG,MAAM,CAAC,YAAY;AACtC,gBAAA,KAAK,MAAM,cAAc,IAAI,MAAM,CAAC,KAAK,EAAE;oBACzC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,cAAc,CAAC;AAChD,oBAAA,eAAe,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,CAAC;gBAC1C;YACF;iBAAO;gBACL,KAAK,MAAM,cAAc,IAAIL,kDAA2B,CACtD,gBAAgB,CACjB,EAAE;oBACD,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,cAAc,CAAC;AAChD,oBAAA,eAAe,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,CAAC;gBAC1C;YACF;QACF;QAEA,MAAM,kBAAkB,GAAG,wBAAwB,CAAC,IAAI,CAAC,aAAa,CAAC;AACvE,QAAA,KAAK,MAAM,IAAI,IAAIC,0BAAoB,EAAE;AACvC,YAAA,IAAI,kBAAkB;gBAAE;AACxB,YAAA,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC;gBAAE;YAExB,MAAM,cAAc,GAAGC,oDAA6B,CAClD,IAAI,EACJ,gBAAgB,CACjB;AACD,YAAA,IAAI,cAAc,IAAI,IAAI,EAAE;gBAC1B;YACF;AAEA,YAAA,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,cAAc,CAAC;AACjC,YAAA,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC;QAC3B;AAEA,QAAA,OAAO,EAAE,OAAO,EAAE,eAAe,EAAE,gBAAgB,EAAE;IACvD;IAEA,MAAM,WAAW,GAAG,IAAI,CAAC,aAAa,EAAE,KAAK,IAAI,EAAE;AAEnD,IAAA,IAAI,wBAAwB,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE;;;;;;;AAOhD,QAAA,IACE,WAAW,CAAC,iBAAiB,KAAK,IAAI;AACtC,YAAA,IAAI,CAAC,gBAAgB,IAAI,IAAI,EAC7B;AACA,YAAA,MAAM,MAAM,GAAGI,4CAA2B,CAAC,WAAW,EAAE;gBACtD,YAAY,EAAE,IAAI,CAAC,gBAAgB;AACpC,aAAA,CAAC;AACF,YAAA,gBAAgB,GAAG,MAAM,CAAC,YAAY;AACtC,YAAA,KAAK,MAAM,SAAS,IAAI,MAAM,CAAC,KAAK,EAAE;gBACpC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,SAAS,CAAC;AACtC,gBAAA,eAAe,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC;YACrC;QACF;aAAO;YACL,KAAK,MAAM,SAAS,IAAIH,uCAAsB,CAAC,WAAW,CAAC,EAAE;gBAC3D,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,SAAS,CAAC;AACtC,gBAAA,eAAe,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC;YACrC;QACF;IACF;;;;;;;;;;;IAYA,MAAM,kBAAkB,GAAG,wBAAwB,CAAC,IAAI,CAAC,aAAa,CAAC;AACvE,IAAA,KAAK,MAAM,IAAI,IAAIF,0BAAoB,EAAE;AACvC,QAAA,IAAI,kBAAkB;YAAE;AACxB,QAAA,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC;YAAE;QAExB,MAAM,SAAS,GAAG,wBAAwB,CAAC,IAAI,EAAE,WAAW,CAAC;AAC7D,QAAA,IAAI,SAAS,IAAI,IAAI,EAAE;YACrB;QACF;AAEA,QAAA,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,SAAS,CAAC;AAC5B,QAAA,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC;IAC3B;AAEA,IAAA,OAAO,EAAE,OAAO,EAAE,eAAe,EAAE,gBAAgB,EAAE;AACvD;;;;;;"}

package/dist/cjs/utils/callbacks.cjs

@@ -0,0 +1,27 @@
+'use strict';
+
+var manager = require('@langchain/core/callbacks/manager');
+
+function appendCallbacks(callbacks, additions) {
+    if (additions.length === 0) {
+        return callbacks ?? [];
+    }
+    if (callbacks == null) {
+        return [...additions];
+    }
+    if (Array.isArray(callbacks)) {
+        return callbacks.concat(additions);
+    }
+    return callbacks.copy(additions.map(manager.ensureHandler));
+}
+function findCallback(callbacks, predicate) {
+    if (callbacks == null) {
+        return undefined;
+    }
+    const handlers = Array.isArray(callbacks) ? callbacks : callbacks.handlers;
+    return handlers.find(predicate);
+}
+
+exports.appendCallbacks = appendCallbacks;
+exports.findCallback = findCallback;
+//# sourceMappingURL=callbacks.cjs.map

package/dist/cjs/utils/callbacks.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"callbacks.cjs","sources":["../../../src/utils/callbacks.ts"],"sourcesContent":["import { ensureHandler } from '@langchain/core/callbacks/manager';\nimport type {\n  BaseCallbackHandler,\n  CallbackHandlerMethods,\n} from '@langchain/core/callbacks/base';\nimport type { Callbacks } from '@langchain/core/callbacks/manager';\n\nexport type CallbackEntry = BaseCallbackHandler | CallbackHandlerMethods;\n\nexport function appendCallbacks(\n  callbacks: Callbacks | undefined,\n  additions: readonly CallbackEntry[]\n): Callbacks {\n  if (additions.length === 0) {\n    return callbacks ?? [];\n  }\n\n  if (callbacks == null) {\n    return [...additions];\n  }\n\n  if (Array.isArray(callbacks)) {\n    return callbacks.concat(additions);\n  }\n\n  return callbacks.copy(additions.map(ensureHandler));\n}\n\nexport function findCallback(\n  callbacks: Callbacks | undefined,\n  predicate: (callback: CallbackEntry) => boolean\n): CallbackEntry | undefined {\n  if (callbacks == null) {\n    return undefined;\n  }\n\n  const handlers = Array.isArray(callbacks) ? callbacks : callbacks.handlers;\n  return handlers.find(predicate);\n}\n"],"names":["ensureHandler"],"mappings":";;;;AASM,SAAU,eAAe,CAC7B,SAAgC,EAChC,SAAmC,EAAA;AAEnC,IAAA,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE;QAC1B,OAAO,SAAS,IAAI,EAAE;IACxB;AAEA,IAAA,IAAI,SAAS,IAAI,IAAI,EAAE;AACrB,QAAA,OAAO,CAAC,GAAG,SAAS,CAAC;IACvB;AAEA,IAAA,IAAI,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE;AAC5B,QAAA,OAAO,SAAS,CAAC,MAAM,CAAC,SAAS,CAAC;IACpC;IAEA,OAAO,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAACA,qBAAa,CAAC,CAAC;AACrD;AAEM,SAAU,YAAY,CAC1B,SAAgC,EAChC,SAA+C,EAAA;AAE/C,IAAA,IAAI,SAAS,IAAI,IAAI,EAAE;AACrB,QAAA,OAAO,SAAS;IAClB;AAEA,IAAA,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,GAAG,SAAS,GAAG,SAAS,CAAC,QAAQ;AAC1E,IAAA,OAAO,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC;AACjC;;;;;"}

package/dist/esm/agents/AgentContext.mjs

@@ -20,12 +20,13 @@ class AgentContext {
      * Create an AgentContext from configuration with token accounting initialization
      */
     static fromConfig(agentConfig, tokenCounter, indexTokenCountMap) {
-        const { agentId, name, provider, clientOptions, tools, toolMap, toolEnd, toolRegistry, toolDefinitions, instructions, additional_instructions, streamBuffer, maxContextTokens, reasoningKey, useLegacyContent, discoveredTools, summarizationEnabled, summarizationConfig, initialSummary, contextPruningConfig, maxToolResultChars, toolSchemaTokens, subagentConfigs, maxSubagentDepth, } = agentConfig;
+        const { agentId, name, provider, clientOptions, langfuse, tools, toolMap, toolEnd, toolRegistry, toolDefinitions, instructions, additional_instructions, streamBuffer, maxContextTokens, reasoningKey, useLegacyContent, discoveredTools, summarizationEnabled, summarizationConfig, initialSummary, contextPruningConfig, maxToolResultChars, toolSchemaTokens, subagentConfigs, maxSubagentDepth, } = agentConfig;
         const agentContext = new AgentContext({
             agentId,
             name: name ?? agentId,
             provider,
             clientOptions,
+            langfuse,
             maxContextTokens,
             streamBuffer,
             tools,
@@ -87,6 +88,8 @@ class AgentContext {
     provider;
     /** Client options for this agent */
     clientOptions;
+    /** Per-agent Langfuse tracing configuration. */
+    langfuse;
     /** Token count map indexed by message position */
     indexTokenCountMap = {};
     /** Canonical pre-run token map used to restore token accounting on reset */
@@ -221,11 +224,12 @@ class AgentContext {
      * Contains source and parallel execution info for system message context.
      */
     handoffContext;
-    constructor({ agentId, name, provider, clientOptions, maxContextTokens, streamBuffer, tokenCounter, tools, toolMap, toolRegistry, toolDefinitions, instructions, additionalInstructions, reasoningKey, toolEnd, instructionTokens, useLegacyContent, discoveredTools, summarizationEnabled, summarizationConfig, contextPruningConfig, maxToolResultChars, }) {
+    constructor({ agentId, name, provider, clientOptions, langfuse, maxContextTokens, streamBuffer, tokenCounter, tools, toolMap, toolRegistry, toolDefinitions, instructions, additionalInstructions, reasoningKey, toolEnd, instructionTokens, useLegacyContent, discoveredTools, summarizationEnabled, summarizationConfig, contextPruningConfig, maxToolResultChars, }) {
         this.agentId = agentId;
         this.name = name;
         this.provider = provider;
         this.clientOptions = clientOptions;
+        this.langfuse = langfuse;
         this.maxContextTokens = maxContextTokens;
         this.streamBuffer = streamBuffer;
         this.tokenCounter = tokenCounter;
@@ -313,12 +317,12 @@ class AgentContext {
         return { name: Constants.BASH_PROGRAMMATIC_TOOL_CALLING, language: 'bash' };
     }
     hasAvailableTool(name) {
-        if (this.toolDefinitions?.some((tool) => tool.name === name))
+        if (this.toolDefinitions?.some((tool) => tool.name === name) === true)
             return true;
-        if (this.tools?.some((tool) => 'name' in tool && tool.name === name)) {
+        if (this.tools?.some((tool) => 'name' in tool && tool.name === name) === true) {
             return true;
         }
-        if (this.toolMap?.has(name))
+        if (this.toolMap?.has(name) === true)
             return true;
         return this.toolRegistry?.has(name) === true;
     }