titan-agent 6.4.1 → 6.4.4

package/dist/agent/agent.js

@@ -1278,66 +1278,81 @@ CRITICAL: Widgets are NOT files. Do NOT use write_file, edit_file, or any file t
   }
   modelUsed = activeModel;
   let activeTools = tools;
-  if (earlyPersona) {
-    const { applyPersonaToolFilter } = await import("./personaProfiles.js");
+  let allToolsBackup = activeTools;
+  let toolSearchEnabled = true;
+  const explicitAllowList = overrides?.allowedTools;
+  if (explicitAllowList && explicitAllowList.length > 0) {
+    const allow = new Set(explicitAllowList);
     const before = activeTools.length;
-    activeTools = applyPersonaToolFilter(activeTools, earlyPersona);
-    if (activeTools.length !== before) {
-      logger.info(COMPONENT, `[Persona:${earlyPersona.id}] tools ${before} \u2192 ${activeTools.length}`);
+    activeTools = tools.filter((t) => allow.has(t.function.name));
+    allToolsBackup = activeTools;
+    logger.info(COMPONENT, `[AllowedTools] override bound ${activeTools.length}/${explicitAllowList.length} requested tools (filter chain bypassed): [${activeTools.map((t) => t.function.name).join(", ")}] from pool of ${before}`);
+    const missing = explicitAllowList.filter((name) => !activeTools.some((t) => t.function.name === name));
+    if (missing.length > 0) {
+      logger.warn(COMPONENT, `[AllowedTools] caller asked for tools that aren't registered: [${missing.join(", ")}]`);
     }
-  }
-  const SMALL_MODEL_PATTERNS = ["llama3.2", "llama3.1:8b", "phi", "gemma:2b", "qwen3.5:4b", "tinyllama", "dolphin3"];
-  const isSmallModel = SMALL_MODEL_PATTERNS.some((p) => activeModel.toLowerCase().includes(p));
-  if (isSmallModel) {
-    const CORE_TOOL_NAMES = ["shell", "read_file", "write_file", "edit_file", "list_dir", "memory"];
-    const coreTools = activeTools.filter((t) => CORE_TOOL_NAMES.includes(t.function.name));
-    logger.info(COMPONENT, `[SmallModel] Reducing tools from ${activeTools.length} to ${coreTools.length} for ${activeModel}`);
-    activeTools = coreTools;
-  }
-  if (!voiceFastPath && !isSmallModel && isBrainAvailable()) {
-    const brainFiltered = await brainSelectTools(message, activeTools);
-    if (brainFiltered.length > 0 && brainFiltered.length < activeTools.length) {
-      logger.info(COMPONENT, `[Brain] Filtered: ${activeTools.length} \u2192 ${brainFiltered.length} tools`);
-      activeTools = brainFiltered;
+  } else {
+    if (earlyPersona) {
+      const { applyPersonaToolFilter } = await import("./personaProfiles.js");
+      const before = activeTools.length;
+      activeTools = applyPersonaToolFilter(activeTools, earlyPersona);
+      if (activeTools.length !== before) {
+        logger.info(COMPONENT, `[Persona:${earlyPersona.id}] tools ${before} \u2192 ${activeTools.length}`);
+      }
     }
-  }
-  const toolSearchConfig = config.toolSearch;
-  const toolSearchEnabled = toolSearchConfig?.enabled ?? true;
-  const allToolsBackup = activeTools;
-  if (pipelineEnsureTools.length > 0 && !(toolSearchEnabled && !isSmallModel && activeTools.length > 12)) {
-    const activeNames = new Set(activeTools.map((t) => t.function.name));
-    const missing = pipelineEnsureTools.filter((name) => !activeNames.has(name));
-    if (missing.length > 0) {
-      const rescued = allToolsBackup.filter((t) => missing.includes(t.function.name));
-      activeTools.push(...rescued);
-      if (rescued.length > 0) {
-        logger.info(COMPONENT, `[Pipeline:${pipelineType}] Ensured ${rescued.length} tools (no-compact): [${rescued.map((t) => t.function.name).join(", ")}]`);
+    const SMALL_MODEL_PATTERNS = ["llama3.2", "llama3.1:8b", "phi", "gemma:2b", "qwen3.5:4b", "tinyllama", "dolphin3"];
+    const isSmallModel = SMALL_MODEL_PATTERNS.some((p) => activeModel.toLowerCase().includes(p));
+    if (isSmallModel) {
+      const CORE_TOOL_NAMES = ["shell", "read_file", "write_file", "edit_file", "list_dir", "memory"];
+      const coreTools = activeTools.filter((t) => CORE_TOOL_NAMES.includes(t.function.name));
+      logger.info(COMPONENT, `[SmallModel] Reducing tools from ${activeTools.length} to ${coreTools.length} for ${activeModel}`);
+      activeTools = coreTools;
+    }
+    if (!voiceFastPath && !isSmallModel && isBrainAvailable()) {
+      const brainFiltered = await brainSelectTools(message, activeTools);
+      if (brainFiltered.length > 0 && brainFiltered.length < activeTools.length) {
+        logger.info(COMPONENT, `[Brain] Filtered: ${activeTools.length} \u2192 ${brainFiltered.length} tools`);
+        activeTools = brainFiltered;
       }
     }
-  }
-  if (toolSearchEnabled && !isSmallModel && activeTools.length > 12) {
-    const VOICE_CORE_TOOLS = ["shell", "web_search", "weather", "memory", "ha_control", "ha_devices", "ha_status", "ha_setup", "tool_search"];
-    const configCoreTools = toolSearchConfig?.coreTools;
-    const effectiveCoreTools = configCoreTools && configCoreTools.length > 0 ? configCoreTools : DEFAULT_CORE_TOOLS;
-    const pipelineMerged = pipelineEnsureTools.length > 0 ? [.../* @__PURE__ */ new Set([...effectiveCoreTools, ...pipelineEnsureTools])] : effectiveCoreTools;
-    const coreNames = new Set(voiceFastPath ? VOICE_CORE_TOOLS : pipelineMerged);
-    activeTools = activeTools.filter((t) => coreNames.has(t.function.name));
-    if (pipelineEnsureTools.length > 0) {
+    const toolSearchConfig = config.toolSearch;
+    toolSearchEnabled = toolSearchConfig?.enabled ?? true;
+    allToolsBackup = activeTools;
+    if (pipelineEnsureTools.length > 0 && !(toolSearchEnabled && !isSmallModel && activeTools.length > 12)) {
       const activeNames = new Set(activeTools.map((t) => t.function.name));
       const missing = pipelineEnsureTools.filter((name) => !activeNames.has(name));
       if (missing.length > 0) {
         const rescued = allToolsBackup.filter((t) => missing.includes(t.function.name));
         activeTools.push(...rescued);
         if (rescued.length > 0) {
-          logger.info(COMPONENT, `[Pipeline:${pipelineType}] Rescued ${rescued.length} tools: [${rescued.map((t) => t.function.name).join(", ")}]`);
+          logger.info(COMPONENT, `[Pipeline:${pipelineType}] Ensured ${rescued.length} tools (no-compact): [${rescued.map((t) => t.function.name).join(", ")}]`);
         }
       }
     }
-    logger.info(COMPONENT, `[ToolSearch] Compact mode: ${allToolsBackup.length} \u2192 ${activeTools.length} tools (${allToolsBackup.length - activeTools.length} discoverable via tool_search)`);
-  }
-  if (dangerous) {
-    activeTools = [];
-    logger.info(COMPONENT, "[Safety] Stripped all tools \u2014 dangerous command detected");
+    if (toolSearchEnabled && !isSmallModel && activeTools.length > 12) {
+      const VOICE_CORE_TOOLS = ["shell", "web_search", "weather", "memory", "ha_control", "ha_devices", "ha_status", "ha_setup", "tool_search"];
+      const configCoreTools = toolSearchConfig?.coreTools;
+      const effectiveCoreTools = configCoreTools && configCoreTools.length > 0 ? configCoreTools : DEFAULT_CORE_TOOLS;
+      const pipelineMerged = pipelineEnsureTools.length > 0 ? [.../* @__PURE__ */ new Set([...effectiveCoreTools, ...pipelineEnsureTools])] : effectiveCoreTools;
+      const coreNames = new Set(voiceFastPath ? VOICE_CORE_TOOLS : pipelineMerged);
+      activeTools = activeTools.filter((t) => coreNames.has(t.function.name));
+      if (pipelineEnsureTools.length > 0) {
+        const activeNames = new Set(activeTools.map((t) => t.function.name));
+        const missing = pipelineEnsureTools.filter((name) => !activeNames.has(name));
+        if (missing.length > 0) {
+          const rescued = allToolsBackup.filter((t) => missing.includes(t.function.name));
+          activeTools.push(...rescued);
+          if (rescued.length > 0) {
+            logger.info(COMPONENT, `[Pipeline:${pipelineType}] Rescued ${rescued.length} tools: [${rescued.map((t) => t.function.name).join(", ")}]`);
+          }
+        }
+      }
+      logger.info(COMPONENT, `[ToolSearch] Compact mode: ${allToolsBackup.length} \u2192 ${activeTools.length} tools (${allToolsBackup.length - activeTools.length} discoverable via tool_search)`);
+    }
+    if (dangerous) {
+      activeTools = [];
+      logger.info(COMPONENT, "[Safety] Stripped all tools \u2014 dangerous command detected");
+    }
   }
   setAutonomousMode(isAutonomous);
   heartbeat(session.id);