Something went wrong. Try again.
source dump of claude code forked from oppi.li/claude-code
Something went wrong. Try again.
5.0 kB · 153 lines
TypeScript
at main
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154import { FILE_EDIT_TOOL_NAME } from 'src/tools/FileEditTool/constants.js'import { FILE_READ_TOOL_NAME } from 'src/tools/FileReadTool/prompt.js'import { FILE_WRITE_TOOL_NAME } from 'src/tools/FileWriteTool/prompt.js'import { GLOB_TOOL_NAME } from 'src/tools/GlobTool/prompt.js'import { GREP_TOOL_NAME } from 'src/tools/GrepTool/prompt.js'import { NOTEBOOK_EDIT_TOOL_NAME } from 'src/tools/NotebookEditTool/constants.js'import { WEB_FETCH_TOOL_NAME } from 'src/tools/WebFetchTool/prompt.js'import { WEB_SEARCH_TOOL_NAME } from 'src/tools/WebSearchTool/prompt.js'import { SHELL_TOOL_NAMES } from 'src/utils/shell/shellToolUtils.js'import { isEnvTruthy } from '../../utils/envUtils.js'
// docs: https://docs.google.com/document/d/1oCT4evvWTh3P6z-kcfNQwWTCxAhkoFndSaNS9Gm40uw/edit?tab=t.0
// Default values for context management strategies// Match client-side microcompact token valuesconst DEFAULT_MAX_INPUT_TOKENS = 180_000 // Typical warning thresholdconst DEFAULT_TARGET_INPUT_TOKENS = 40_000 // Keep last 40k tokens like client-side
const TOOLS_CLEARABLE_RESULTS = [ ...SHELL_TOOL_NAMES, GLOB_TOOL_NAME, GREP_TOOL_NAME, FILE_READ_TOOL_NAME, WEB_FETCH_TOOL_NAME, WEB_SEARCH_TOOL_NAME,]
const TOOLS_CLEARABLE_USES = [ FILE_EDIT_TOOL_NAME, FILE_WRITE_TOOL_NAME, NOTEBOOK_EDIT_TOOL_NAME,]
// Context management strategy types matching API documentationexport type ContextEditStrategy = | { type: 'clear_tool_uses_20250919' trigger?: { type: 'input_tokens' value: number } keep?: { type: 'tool_uses' value: number } clear_tool_inputs?: boolean | string[] exclude_tools?: string[] clear_at_least?: { type: 'input_tokens' value: number } } | { type: 'clear_thinking_20251015' keep: { type: 'thinking_turns'; value: number } | 'all' }
// Context management configuration wrapperexport type ContextManagementConfig = { edits: ContextEditStrategy[]}
// API-based microcompact implementation that uses native context managementexport function getAPIContextManagement(options?: { hasThinking?: boolean isRedactThinkingActive?: boolean clearAllThinking?: boolean}): ContextManagementConfig | undefined { const { hasThinking = false, isRedactThinkingActive = false, clearAllThinking = false, } = options ?? {}
const strategies: ContextEditStrategy[] = []
// Preserve thinking blocks in previous assistant turns. Skip when // redact-thinking is active — redacted blocks have no model-visible content. // When clearAllThinking is set (>1h idle = cache miss), keep only the last // thinking turn — the API schema requires value >= 1, and omitting the edit // falls back to the model-policy default (often "all"), which wouldn't clear. if (hasThinking && !isRedactThinkingActive) { strategies.push({ type: 'clear_thinking_20251015', keep: clearAllThinking ? { type: 'thinking_turns', value: 1 } : 'all', }) }
// Tool clearing strategies are ant-only if (process.env.USER_TYPE !== 'ant') { return strategies.length > 0 ? { edits: strategies } : undefined }
const useClearToolResults = isEnvTruthy( process.env.USE_API_CLEAR_TOOL_RESULTS, ) const useClearToolUses = isEnvTruthy(process.env.USE_API_CLEAR_TOOL_USES)
// If no tool clearing strategy is enabled, return early if (!useClearToolResults && !useClearToolUses) { return strategies.length > 0 ? { edits: strategies } : undefined }
if (useClearToolResults) { const triggerThreshold = process.env.API_MAX_INPUT_TOKENS ? parseInt(process.env.API_MAX_INPUT_TOKENS) : DEFAULT_MAX_INPUT_TOKENS const keepTarget = process.env.API_TARGET_INPUT_TOKENS ? parseInt(process.env.API_TARGET_INPUT_TOKENS) : DEFAULT_TARGET_INPUT_TOKENS
const strategy: ContextEditStrategy = { type: 'clear_tool_uses_20250919', trigger: { type: 'input_tokens', value: triggerThreshold, }, clear_at_least: { type: 'input_tokens', value: triggerThreshold - keepTarget, }, clear_tool_inputs: TOOLS_CLEARABLE_RESULTS, }
strategies.push(strategy) }
if (useClearToolUses) { const triggerThreshold = process.env.API_MAX_INPUT_TOKENS ? parseInt(process.env.API_MAX_INPUT_TOKENS) : DEFAULT_MAX_INPUT_TOKENS const keepTarget = process.env.API_TARGET_INPUT_TOKENS ? parseInt(process.env.API_TARGET_INPUT_TOKENS) : DEFAULT_TARGET_INPUT_TOKENS
const strategy: ContextEditStrategy = { type: 'clear_tool_uses_20250919', trigger: { type: 'input_tokens', value: triggerThreshold, }, clear_at_least: { type: 'input_tokens', value: triggerThreshold - keepTarget, }, exclude_tools: TOOLS_CLEARABLE_USES, }
strategies.push(strategy) }
return strategies.length > 0 ? { edits: strategies } : undefined}