mirror of
https://github.com/claude-code-best/claude-code.git
synced 2026-06-15 12:55:51 +00:00
- 新增 docs/design/tool-search-design-guide.md,涵盖架构、搜索算法、执行管道、演进历史 - 禁用 getTurnZeroSearchExtraToolsPrefetch,消除用户输入时的频繁弹窗 - inter-turn 发现机制保持不变 Co-Authored-By: glm-5-turbo <zai-org@claude-code-best.win>
721 lines
24 KiB
TypeScript
721 lines
24 KiB
TypeScript
/**
|
|
* Tool Search utilities for dynamically discovering deferred tools.
|
|
*
|
|
* When enabled, deferred tools (all non-core tools) are sent with
|
|
* defer_loading: true and discovered via SearchExtraToolsTool rather than being
|
|
* loaded upfront. Core tools are defined in CORE_TOOLS (src/constants/tools.ts).
|
|
*/
|
|
|
|
import memoize from 'lodash-es/memoize.js'
|
|
import { getFeatureValue_CACHED_MAY_BE_STALE } from '../services/analytics/growthbook.js'
|
|
import {
|
|
type AnalyticsMetadata_I_VERIFIED_THIS_IS_NOT_CODE_OR_FILEPATHS,
|
|
logEvent,
|
|
} from '../services/analytics/index.js'
|
|
import type { Tool } from '../Tool.js'
|
|
import {
|
|
type ToolPermissionContext,
|
|
type Tools,
|
|
toolMatchesName,
|
|
} from '../Tool.js'
|
|
import type { AgentDefinition } from '@claude-code-best/builtin-tools/tools/AgentTool/loadAgentsDir.js'
|
|
import {
|
|
formatDeferredToolLine,
|
|
isDeferredTool,
|
|
SEARCH_EXTRA_TOOLS_TOOL_NAME,
|
|
} from '@claude-code-best/builtin-tools/tools/SearchExtraToolsTool/prompt.js'
|
|
import type { Message } from '../types/message.js'
|
|
import {
|
|
countToolDefinitionTokens,
|
|
TOOL_TOKEN_COUNT_OVERHEAD,
|
|
} from './analyzeContext.js'
|
|
import { count } from './array.js'
|
|
import { getMergedBetas } from './betas.js'
|
|
import { getContextWindowForModel } from './context.js'
|
|
import { logForDebugging } from './debug.js'
|
|
import { isEnvDefinedFalsy, isEnvTruthy } from './envUtils.js'
|
|
import { jsonStringify } from './slowOperations.js'
|
|
import { zodToJsonSchema } from './zodToJsonSchema.js'
|
|
|
|
/**
|
|
* Default percentage of context window at which to auto-enable tool search.
|
|
* When MCP tool descriptions exceed this percentage (in tokens), tool search is enabled.
|
|
* Can be overridden via ENABLE_SEARCH_EXTRA_TOOLS=auto:N where N is 0-100.
|
|
*/
|
|
const DEFAULT_AUTO_SEARCH_EXTRA_TOOLS_PERCENTAGE = 10 // 10%
|
|
|
|
/**
|
|
* Parse auto:N syntax from ENABLE_SEARCH_EXTRA_TOOLS env var.
|
|
* Returns the percentage clamped to 0-100, or null if not auto:N format or not a number.
|
|
*/
|
|
function parseAutoPercentage(value: string): number | null {
|
|
if (!value.startsWith('auto:')) return null
|
|
|
|
const percentStr = value.slice(5)
|
|
const percent = parseInt(percentStr, 10)
|
|
|
|
if (isNaN(percent)) {
|
|
logForDebugging(
|
|
`Invalid ENABLE_SEARCH_EXTRA_TOOLS value "${value}": expected auto:N where N is a number.`,
|
|
)
|
|
return null
|
|
}
|
|
|
|
// Clamp to valid range
|
|
return Math.max(0, Math.min(100, percent))
|
|
}
|
|
|
|
/**
|
|
* Check if ENABLE_SEARCH_EXTRA_TOOLS is set to auto mode (auto or auto:N).
|
|
*/
|
|
function isAutoSearchExtraToolsMode(value: string | undefined): boolean {
|
|
if (!value) return false
|
|
return value === 'auto' || value.startsWith('auto:')
|
|
}
|
|
|
|
/**
|
|
* Get the auto-enable percentage from env var or default.
|
|
*/
|
|
function getAutoSearchExtraToolsPercentage(): number {
|
|
const value = process.env.ENABLE_SEARCH_EXTRA_TOOLS
|
|
if (!value) return DEFAULT_AUTO_SEARCH_EXTRA_TOOLS_PERCENTAGE
|
|
|
|
if (value === 'auto') return DEFAULT_AUTO_SEARCH_EXTRA_TOOLS_PERCENTAGE
|
|
|
|
const parsed = parseAutoPercentage(value)
|
|
if (parsed !== null) return parsed
|
|
|
|
return DEFAULT_AUTO_SEARCH_EXTRA_TOOLS_PERCENTAGE
|
|
}
|
|
|
|
/**
|
|
* Approximate chars per token for MCP tool definitions (name + description + input schema).
|
|
* Used as fallback when the token counting API is unavailable.
|
|
*/
|
|
const CHARS_PER_TOKEN = 2.5
|
|
|
|
/**
|
|
* Get the token threshold for auto-enabling tool search for a given model.
|
|
*/
|
|
function getAutoSearchExtraToolsTokenThreshold(model: string): number {
|
|
const betas = getMergedBetas(model)
|
|
const contextWindow = getContextWindowForModel(model, betas)
|
|
const percentage = getAutoSearchExtraToolsPercentage() / 100
|
|
return Math.floor(contextWindow * percentage)
|
|
}
|
|
|
|
/**
|
|
* Get the character threshold for auto-enabling tool search for a given model.
|
|
* Used as fallback when the token counting API is unavailable.
|
|
*/
|
|
export function getAutoSearchExtraToolsCharThreshold(model: string): number {
|
|
return Math.floor(
|
|
getAutoSearchExtraToolsTokenThreshold(model) * CHARS_PER_TOKEN,
|
|
)
|
|
}
|
|
|
|
/**
|
|
* Get the total token count for all deferred tools using the token counting API.
|
|
* Memoized by deferred tool names — cache is invalidated when MCP servers connect/disconnect.
|
|
* Returns null if the API is unavailable (caller should fall back to char heuristic).
|
|
*/
|
|
const getDeferredToolTokenCount = memoize(
|
|
async (
|
|
tools: Tools,
|
|
getToolPermissionContext: () => Promise<ToolPermissionContext>,
|
|
agents: AgentDefinition[],
|
|
model: string,
|
|
): Promise<number | null> => {
|
|
const deferredTools = tools.filter(t => isDeferredTool(t))
|
|
if (deferredTools.length === 0) return 0
|
|
|
|
try {
|
|
const total = await countToolDefinitionTokens(
|
|
deferredTools,
|
|
getToolPermissionContext,
|
|
{ activeAgents: agents, allAgents: agents },
|
|
model,
|
|
)
|
|
if (total === 0) return null // API unavailable
|
|
return Math.max(0, total - TOOL_TOKEN_COUNT_OVERHEAD)
|
|
} catch {
|
|
return null // Fall back to char heuristic
|
|
}
|
|
},
|
|
(tools: Tools) =>
|
|
tools
|
|
.filter(t => isDeferredTool(t))
|
|
.map(t => t.name)
|
|
.join(','),
|
|
)
|
|
|
|
/**
|
|
* Tool search mode. Determines how deferred tools (all non-core tools)
|
|
* are surfaced:
|
|
* - 'tst': Tool Search Tool — deferred tools discovered via SearchExtraToolsTool (always enabled)
|
|
* - 'tst-auto': auto — tools deferred only when they exceed threshold
|
|
* - 'standard': tool search disabled — all tools exposed inline
|
|
*/
|
|
export type SearchExtraToolsMode = 'tst' | 'tst-auto' | 'standard'
|
|
|
|
/**
|
|
* Determines the tool search mode from ENABLE_SEARCH_EXTRA_TOOLS.
|
|
*
|
|
* ENABLE_SEARCH_EXTRA_TOOLS Mode
|
|
* auto / auto:1-99 tst-auto
|
|
* true / auto:0 tst
|
|
* false / auto:100 standard
|
|
* (unset) tst (default: always defer non-core tools)
|
|
*/
|
|
export function getSearchExtraToolsMode(): SearchExtraToolsMode {
|
|
// CLAUDE_CODE_DISABLE_EXPERIMENTAL_BETAS still acts as a kill switch
|
|
// for tool search, even though we no longer send beta headers.
|
|
// Users who set this flag explicitly opt out of tool search.
|
|
if (isEnvTruthy(process.env.CLAUDE_CODE_DISABLE_EXPERIMENTAL_BETAS)) {
|
|
return 'standard'
|
|
}
|
|
|
|
const value = process.env.ENABLE_SEARCH_EXTRA_TOOLS
|
|
|
|
// Handle auto:N syntax - check edge cases first
|
|
const autoPercent = value ? parseAutoPercentage(value) : null
|
|
if (autoPercent === 0) return 'tst' // auto:0 = always enabled
|
|
if (autoPercent === 100) return 'standard'
|
|
if (isAutoSearchExtraToolsMode(value)) {
|
|
return 'tst-auto' // auto or auto:1-99
|
|
}
|
|
|
|
if (isEnvTruthy(value)) return 'tst'
|
|
if (isEnvDefinedFalsy(process.env.ENABLE_SEARCH_EXTRA_TOOLS))
|
|
return 'standard'
|
|
return 'tst' // default: always defer non-core tools
|
|
}
|
|
|
|
/**
|
|
* Check if tool search *might* be enabled (optimistic check).
|
|
*
|
|
* Returns true if tool search could potentially be enabled, without checking
|
|
* dynamic factors like threshold. Use this for:
|
|
* - Including SearchExtraToolsTool in base tools (so it's available if needed)
|
|
* - Checking if SearchExtraToolsTool should report itself as enabled
|
|
*
|
|
* Returns false only when tool search is definitively disabled (standard mode).
|
|
*
|
|
* For the definitive check that includes threshold, use isSearchExtraToolsEnabled().
|
|
*/
|
|
let loggedOptimistic = false
|
|
|
|
export function isSearchExtraToolsEnabledOptimistic(): boolean {
|
|
const mode = getSearchExtraToolsMode()
|
|
if (mode === 'standard') {
|
|
if (!loggedOptimistic) {
|
|
loggedOptimistic = true
|
|
logForDebugging(
|
|
`[SearchExtraTools:optimistic] mode=${mode}, ENABLE_SEARCH_EXTRA_TOOLS=${process.env.ENABLE_SEARCH_EXTRA_TOOLS}, result=false`,
|
|
)
|
|
}
|
|
return false
|
|
}
|
|
|
|
// All providers use the unified self-built tool search (TF-IDF + keyword).
|
|
// No first-party / tool_reference / defer_loading distinction.
|
|
// Users can still disable via ENABLE_SEARCH_EXTRA_TOOLS=false.
|
|
|
|
if (!loggedOptimistic) {
|
|
loggedOptimistic = true
|
|
logForDebugging(
|
|
`[SearchExtraTools:optimistic] mode=${mode}, ENABLE_SEARCH_EXTRA_TOOLS=${process.env.ENABLE_SEARCH_EXTRA_TOOLS}, result=true`,
|
|
)
|
|
}
|
|
return true
|
|
}
|
|
|
|
/**
|
|
* Check if SearchExtraToolsTool is available in the provided tools list.
|
|
* If SearchExtraToolsTool is not available (e.g., disallowed via disallowedTools),
|
|
* tool search cannot function and should be disabled.
|
|
*
|
|
* @param tools Array of tools with a 'name' property
|
|
* @returns true if SearchExtraToolsTool is in the tools list, false otherwise
|
|
*/
|
|
export function isSearchExtraToolsToolAvailable(
|
|
tools: readonly { name: string }[],
|
|
): boolean {
|
|
return tools.some(tool => toolMatchesName(tool, SEARCH_EXTRA_TOOLS_TOOL_NAME))
|
|
}
|
|
|
|
/**
|
|
* Calculate total deferred tool description size in characters.
|
|
* Includes name, description text, and input schema to match what's actually sent to the API.
|
|
*/
|
|
async function calculateDeferredToolDescriptionChars(
|
|
tools: Tools,
|
|
getToolPermissionContext: () => Promise<ToolPermissionContext>,
|
|
agents: AgentDefinition[],
|
|
): Promise<number> {
|
|
const deferredTools = tools.filter(t => isDeferredTool(t))
|
|
if (deferredTools.length === 0) return 0
|
|
|
|
const sizes = await Promise.all(
|
|
deferredTools.map(async tool => {
|
|
const description = await tool.prompt({
|
|
getToolPermissionContext,
|
|
tools,
|
|
agents,
|
|
})
|
|
const inputSchema = tool.inputJSONSchema
|
|
? jsonStringify(tool.inputJSONSchema)
|
|
: tool.inputSchema
|
|
? jsonStringify(zodToJsonSchema(tool.inputSchema))
|
|
: ''
|
|
return tool.name.length + description.length + inputSchema.length
|
|
}),
|
|
)
|
|
|
|
return sizes.reduce((total, size) => total + size, 0)
|
|
}
|
|
|
|
/**
|
|
* Check if tool search (MCP tool deferral with tool_reference) is enabled for a specific request.
|
|
*
|
|
* This is the definitive check that includes:
|
|
* - MCP mode (Tst, TstAuto, McpCli, Standard)
|
|
* - Model compatibility (haiku doesn't support tool_reference)
|
|
* - SearchExtraToolsTool availability (must be in tools list)
|
|
* - Threshold check for TstAuto mode
|
|
*
|
|
* Use this when making actual API calls where all context is available.
|
|
*
|
|
* @param model The model being used (kept for API compatibility)
|
|
* @param tools Array of available tools (including MCP tools)
|
|
* @param getToolPermissionContext Function to get tool permission context
|
|
* @param agents Array of agent definitions
|
|
* @param source Optional identifier for the caller (for debugging)
|
|
* @returns true if tool search should be enabled for this request
|
|
*/
|
|
export async function isSearchExtraToolsEnabled(
|
|
model: string,
|
|
tools: Tools,
|
|
getToolPermissionContext: () => Promise<ToolPermissionContext>,
|
|
agents: AgentDefinition[],
|
|
source?: string,
|
|
): Promise<boolean> {
|
|
const mcpToolCount = count(tools, t => t.isMcp)
|
|
|
|
// Helper to log the mode decision event
|
|
function logModeDecision(
|
|
enabled: boolean,
|
|
mode: SearchExtraToolsMode,
|
|
reason: string,
|
|
extraProps?: Record<string, number>,
|
|
): void {
|
|
logEvent('tengu_search_extra_tools_mode_decision', {
|
|
enabled,
|
|
mode: mode as AnalyticsMetadata_I_VERIFIED_THIS_IS_NOT_CODE_OR_FILEPATHS,
|
|
reason:
|
|
reason as AnalyticsMetadata_I_VERIFIED_THIS_IS_NOT_CODE_OR_FILEPATHS,
|
|
// Log the actual model being checked, not the session's main model.
|
|
// This is important for debugging subagent tool search decisions where
|
|
// the subagent model (e.g., haiku) differs from the session model (e.g., opus).
|
|
checkedModel:
|
|
model as AnalyticsMetadata_I_VERIFIED_THIS_IS_NOT_CODE_OR_FILEPATHS,
|
|
mcpToolCount,
|
|
userType: (process.env.USER_TYPE ??
|
|
'external') as AnalyticsMetadata_I_VERIFIED_THIS_IS_NOT_CODE_OR_FILEPATHS,
|
|
...extraProps,
|
|
})
|
|
}
|
|
|
|
// Tool search is enabled uniformly regardless of provider or model.
|
|
// All providers use self-built TF-IDF + keyword search via SearchExtraToolsTool + ExecuteExtraTool.
|
|
|
|
// Check if SearchExtraToolsTool is available (respects disallowedTools)
|
|
if (!isSearchExtraToolsToolAvailable(tools)) {
|
|
logForDebugging(
|
|
`Tool search disabled: SearchExtraToolsTool is not available (may have been disallowed via disallowedTools).`,
|
|
)
|
|
logModeDecision(false, 'standard', 'mcp_search_unavailable')
|
|
return false
|
|
}
|
|
|
|
const mode = getSearchExtraToolsMode()
|
|
|
|
switch (mode) {
|
|
case 'tst':
|
|
logModeDecision(true, mode, 'tst_enabled')
|
|
return true
|
|
|
|
case 'tst-auto': {
|
|
const { enabled, debugDescription, metrics } = await checkAutoThreshold(
|
|
tools,
|
|
getToolPermissionContext,
|
|
agents,
|
|
model,
|
|
)
|
|
|
|
if (enabled) {
|
|
logForDebugging(
|
|
`Auto tool search enabled: ${debugDescription}` +
|
|
(source ? ` [source: ${source}]` : ''),
|
|
)
|
|
logModeDecision(true, mode, 'auto_above_threshold', metrics)
|
|
return true
|
|
}
|
|
|
|
logForDebugging(
|
|
`Auto tool search disabled: ${debugDescription}` +
|
|
(source ? ` [source: ${source}]` : ''),
|
|
)
|
|
logModeDecision(false, mode, 'auto_below_threshold', metrics)
|
|
return false
|
|
}
|
|
|
|
case 'standard':
|
|
logModeDecision(false, mode, 'standard_mode')
|
|
return false
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Check if an object is a tool_reference block.
|
|
* tool_reference is a beta feature not in the SDK types, so we need runtime checks.
|
|
*/
|
|
export function isToolReferenceBlock(obj: unknown): boolean {
|
|
return (
|
|
typeof obj === 'object' &&
|
|
obj !== null &&
|
|
'type' in obj &&
|
|
(obj as { type: unknown }).type === 'tool_reference'
|
|
)
|
|
}
|
|
|
|
/**
|
|
* Type guard for tool_reference block with tool_name.
|
|
*/
|
|
function isToolReferenceWithName(
|
|
obj: unknown,
|
|
): obj is { type: 'tool_reference'; tool_name: string } {
|
|
return (
|
|
isToolReferenceBlock(obj) &&
|
|
'tool_name' in (obj as object) &&
|
|
typeof (obj as { tool_name: unknown }).tool_name === 'string'
|
|
)
|
|
}
|
|
|
|
/**
|
|
* Type representing a tool_result block with array content.
|
|
* Used for extracting tool_reference blocks from SearchExtraToolsTool results.
|
|
*/
|
|
type ToolResultBlock = {
|
|
type: 'tool_result'
|
|
content: unknown[]
|
|
}
|
|
|
|
/**
|
|
* Type representing a tool_result block with string content.
|
|
* Used for extracting tool names from SearchExtraToolsTool text output.
|
|
*/
|
|
type ToolResultBlockWithStringContent = {
|
|
type: 'tool_result'
|
|
content: string
|
|
}
|
|
|
|
/**
|
|
* Type guard for tool_result blocks with array content.
|
|
*/
|
|
function isToolResultBlockWithContent(obj: unknown): obj is ToolResultBlock {
|
|
return (
|
|
typeof obj === 'object' &&
|
|
obj !== null &&
|
|
'type' in obj &&
|
|
(obj as { type: unknown }).type === 'tool_result' &&
|
|
'content' in obj &&
|
|
Array.isArray((obj as { content: unknown }).content)
|
|
)
|
|
}
|
|
|
|
/**
|
|
* Type guard for tool_result blocks with string content.
|
|
*/
|
|
function isToolResultBlockWithStringContent(
|
|
obj: unknown,
|
|
): obj is ToolResultBlockWithStringContent {
|
|
return (
|
|
typeof obj === 'object' &&
|
|
obj !== null &&
|
|
'type' in obj &&
|
|
(obj as { type: unknown }).type === 'tool_result' &&
|
|
'content' in obj &&
|
|
typeof (obj as { content: unknown }).content === 'string'
|
|
)
|
|
}
|
|
|
|
/**
|
|
* Regex to extract tool names from SearchExtraToolsTool text output.
|
|
* Matches: "Found N deferred tool(s): ToolA, mcp.server.ToolB."
|
|
* Uses multiline + end-of-line anchor so dots inside tool names (e.g. mcp__s__t) don't break parsing.
|
|
*/
|
|
const DISCOVERED_TOOLS_PATTERN = /^Found \d+ deferred tool\(s\): (.+)\.$/m
|
|
|
|
/**
|
|
* Extract tool names from SearchExtraToolsTool text output.
|
|
* Format: "Found N deferred tool(s): ToolA, ToolB.\n..."
|
|
*/
|
|
function extractToolNamesFromText(text: string): string[] {
|
|
const match = DISCOVERED_TOOLS_PATTERN.exec(text)
|
|
if (!match?.[1]) return []
|
|
return match[1]
|
|
.split(',')
|
|
.map(name => name.trim())
|
|
.filter(Boolean)
|
|
}
|
|
|
|
/**
|
|
* Extract tool names from SearchExtraToolsTool results in message history.
|
|
*
|
|
* Supports two formats:
|
|
* 1. Legacy tool_reference blocks (backward compat with old sessions)
|
|
* 2. Text output from unified self-built tool search
|
|
*
|
|
* Discovered tool names are used to include deferred tools in subsequent
|
|
* API requests so the model can call them directly.
|
|
*
|
|
* Compaction snapshots the discovered set onto
|
|
* compactMetadata.preCompactDiscoveredTools on the boundary marker.
|
|
*
|
|
* @param messages Array of messages that may contain tool_result blocks
|
|
* @returns Set of tool names that have been discovered
|
|
*/
|
|
export function extractDiscoveredToolNames(messages: Message[]): Set<string> {
|
|
const discoveredTools = new Set<string>()
|
|
let carriedFromBoundary = 0
|
|
|
|
for (const msg of messages) {
|
|
// Compact boundary carries the pre-compact discovered set. Inline type
|
|
// check rather than isCompactBoundaryMessage — utils/messages.ts imports
|
|
// from this file, so importing back would be circular.
|
|
if (msg.type === 'system' && msg.subtype === 'compact_boundary') {
|
|
const carried = (msg as any).compactMetadata?.preCompactDiscoveredTools as
|
|
| string[]
|
|
| undefined
|
|
if (carried) {
|
|
for (const name of carried) discoveredTools.add(name)
|
|
carriedFromBoundary += carried.length
|
|
}
|
|
continue
|
|
}
|
|
|
|
// Deferred-tools-delta attachments announce tools that the model should
|
|
// see as available. Include their addedNames so the filter in claude.ts
|
|
// keeps the corresponding tool schemas in the API request.
|
|
if (
|
|
msg.type === 'attachment' &&
|
|
(msg as any).attachment?.type === 'deferred_tools_delta'
|
|
) {
|
|
const added: string[] = (msg as any).attachment.addedNames ?? []
|
|
for (const name of added) discoveredTools.add(name)
|
|
continue
|
|
}
|
|
|
|
// Only user messages contain tool_result blocks (responses to tool_use)
|
|
if (msg.type !== 'user') continue
|
|
|
|
const content = msg.message?.content
|
|
if (!Array.isArray(content)) continue
|
|
|
|
for (const block of content) {
|
|
// Legacy: tool_reference blocks from old sessions (backward compat)
|
|
if (isToolResultBlockWithContent(block)) {
|
|
for (const item of block.content) {
|
|
if (isToolReferenceWithName(item)) {
|
|
discoveredTools.add(item.tool_name)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Unified self-built search: text output from SearchExtraToolsTool
|
|
if (isToolResultBlockWithStringContent(block)) {
|
|
const names = extractToolNamesFromText(block.content)
|
|
for (const name of names) {
|
|
discoveredTools.add(name)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (discoveredTools.size > 0) {
|
|
logForDebugging(
|
|
`Dynamic tool loading: found ${discoveredTools.size} discovered tools in message history` +
|
|
(carriedFromBoundary > 0
|
|
? ` (${carriedFromBoundary} carried from compact boundary)`
|
|
: ''),
|
|
)
|
|
}
|
|
|
|
return discoveredTools
|
|
}
|
|
|
|
export type DeferredToolsDelta = {
|
|
addedNames: string[]
|
|
/** Rendered lines for addedNames; the scan reconstructs from names. */
|
|
addedLines: string[]
|
|
removedNames: string[]
|
|
}
|
|
|
|
/**
|
|
* Call-site discriminator for the tengu_deferred_tools_pool_change event.
|
|
* The scan runs from several sites with different expected-prior semantics
|
|
* (inc-4747):
|
|
* - attachments_main: main-thread getAttachments → prior=0 is a BUG on fire-2+
|
|
* - attachments_subagent: subagent getAttachments → prior=0 is EXPECTED
|
|
* (fresh conversation, initialMessages has no DTD)
|
|
* - compact_full: compact.ts passes [] → prior=0 is EXPECTED
|
|
* - compact_partial: compact.ts passes messagesToKeep → depends on what survived
|
|
* - reactive_compact: reactiveCompact.ts passes preservedMessages → same
|
|
* Without this the 96%-prior=0 stat is dominated by EXPECTED buckets and
|
|
* the real main-thread cross-turn bug (if any) is invisible in BQ.
|
|
*/
|
|
export type DeferredToolsDeltaScanContext = {
|
|
callSite:
|
|
| 'attachments_main'
|
|
| 'attachments_subagent'
|
|
| 'compact_full'
|
|
| 'compact_partial'
|
|
| 'reactive_compact'
|
|
querySource?: string
|
|
}
|
|
|
|
/**
|
|
* True → announce deferred tools via persisted delta attachments.
|
|
* False → claude.ts keeps its per-call <available-deferred-tools>
|
|
* header prepend (the attachment does not fire).
|
|
*/
|
|
export function isDeferredToolsDeltaEnabled(): boolean {
|
|
return (
|
|
process.env.USER_TYPE === 'ant' ||
|
|
getFeatureValue_CACHED_MAY_BE_STALE('tengu_glacier_2xr', false)
|
|
)
|
|
}
|
|
|
|
/**
|
|
* Diff the current deferred-tool pool against what's already been
|
|
* announced in this conversation (reconstructed by scanning for prior
|
|
* deferred_tools_delta attachments). Returns null if nothing changed.
|
|
*
|
|
* A name that was announced but has since stopped being deferred — yet
|
|
* is still in the base pool — is NOT reported as removed. It's now
|
|
* loaded directly, so telling the model "no longer available" would be
|
|
* wrong.
|
|
*/
|
|
export function getDeferredToolsDelta(
|
|
tools: Tools,
|
|
messages: Message[],
|
|
scanContext?: DeferredToolsDeltaScanContext,
|
|
): DeferredToolsDelta | null {
|
|
const announced = new Set<string>()
|
|
let attachmentCount = 0
|
|
let dtdCount = 0
|
|
const attachmentTypesSeen = new Set<string>()
|
|
for (const msg of messages) {
|
|
if (msg.type !== 'attachment') continue
|
|
attachmentCount++
|
|
attachmentTypesSeen.add(msg.attachment!.type)
|
|
if (msg.attachment!.type !== 'deferred_tools_delta') continue
|
|
dtdCount++
|
|
for (const n of msg.attachment!.addedNames) announced.add(n)
|
|
for (const n of msg.attachment!.removedNames) announced.delete(n)
|
|
}
|
|
|
|
const deferred: Tool[] = tools.filter(isDeferredTool)
|
|
const deferredNames = new Set(deferred.map(t => t.name))
|
|
const poolNames = new Set(tools.map(t => t.name))
|
|
|
|
const added = deferred.filter(t => !announced.has(t.name))
|
|
const removed: string[] = []
|
|
for (const n of announced) {
|
|
if (deferredNames.has(n)) continue
|
|
if (!poolNames.has(n)) removed.push(n)
|
|
// else: undeferred — silent
|
|
}
|
|
|
|
if (added.length === 0 && removed.length === 0) return null
|
|
|
|
// Diagnostic for the inc-4747 scan-finds-nothing bug. Round-1 fields
|
|
// (messagesLength/attachmentCount/dtdCount from #23167) showed 45.6% of
|
|
// events have attachments-but-no-DTD, but those numbers are confounded:
|
|
// subagent first-fires and compact-path scans have EXPECTED prior=0 and
|
|
// dominate the stat. callSite/querySource/attachmentTypesSeen split the
|
|
// buckets so the real main-thread cross-turn failure is isolable in BQ.
|
|
logEvent('tengu_deferred_tools_pool_change', {
|
|
addedCount: added.length,
|
|
removedCount: removed.length,
|
|
priorAnnouncedCount: announced.size,
|
|
messagesLength: messages.length,
|
|
attachmentCount,
|
|
dtdCount,
|
|
callSite: (scanContext?.callSite ??
|
|
'unknown') as AnalyticsMetadata_I_VERIFIED_THIS_IS_NOT_CODE_OR_FILEPATHS,
|
|
querySource: (scanContext?.querySource ??
|
|
'unknown') as AnalyticsMetadata_I_VERIFIED_THIS_IS_NOT_CODE_OR_FILEPATHS,
|
|
attachmentTypesSeen: [...attachmentTypesSeen]
|
|
.sort()
|
|
.join(',') as AnalyticsMetadata_I_VERIFIED_THIS_IS_NOT_CODE_OR_FILEPATHS,
|
|
})
|
|
|
|
return {
|
|
addedNames: added.map(t => t.name).sort(),
|
|
addedLines: added.map(formatDeferredToolLine).sort(),
|
|
removedNames: removed.sort(),
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Check whether deferred tools exceed the auto-threshold for enabling TST.
|
|
* Tries exact token count first; falls back to character-based heuristic.
|
|
*/
|
|
async function checkAutoThreshold(
|
|
tools: Tools,
|
|
getToolPermissionContext: () => Promise<ToolPermissionContext>,
|
|
agents: AgentDefinition[],
|
|
model: string,
|
|
): Promise<{
|
|
enabled: boolean
|
|
debugDescription: string
|
|
metrics: Record<string, number>
|
|
}> {
|
|
// Try exact token count first (cached, one API call per toolset change)
|
|
const deferredToolTokens = await getDeferredToolTokenCount(
|
|
tools,
|
|
getToolPermissionContext,
|
|
agents,
|
|
model,
|
|
)
|
|
|
|
if (deferredToolTokens !== null) {
|
|
const threshold = getAutoSearchExtraToolsTokenThreshold(model)
|
|
return {
|
|
enabled: deferredToolTokens >= threshold,
|
|
debugDescription:
|
|
`${deferredToolTokens} tokens (threshold: ${threshold}, ` +
|
|
`${getAutoSearchExtraToolsPercentage()}% of context)`,
|
|
metrics: { deferredToolTokens, threshold },
|
|
}
|
|
}
|
|
|
|
// Fallback: character-based heuristic when token API is unavailable
|
|
const deferredToolDescriptionChars =
|
|
await calculateDeferredToolDescriptionChars(
|
|
tools,
|
|
getToolPermissionContext,
|
|
agents,
|
|
)
|
|
const charThreshold = getAutoSearchExtraToolsCharThreshold(model)
|
|
return {
|
|
enabled: deferredToolDescriptionChars >= charThreshold,
|
|
debugDescription:
|
|
`${deferredToolDescriptionChars} chars (threshold: ${charThreshold}, ` +
|
|
`${getAutoSearchExtraToolsPercentage()}% of context) (char fallback)`,
|
|
metrics: { deferredToolDescriptionChars, charThreshold },
|
|
}
|
|
}
|