Function getContextTree

src/tools/context-tree.ts:107–128

(options: ContextTreeOptions)

}

export async function getContextTree(options: ContextTreeOptions): Promise<string> {
  const entries = await walkDirectory({
    rootDir: options.rootDir,
    targetPath: options.targetPath,
    depthLimit: options.depthLimit,
  });

  const includeSymbols = options.includeSymbols !== false;
  const tree = await buildTree(entries, options.rootDir, includeSymbols);
  const maxTokens = options.maxTokens ?? 20000;

  let rendered = renderTree(tree);
  if (estimateTokens(rendered) <= maxTokens) return rendered;

  pruneSymbols(tree);
  rendered = renderTree(tree);
  if (estimateTokens(rendered) <= maxTokens) return `[Level 1: Headers only, symbols pruned to fit ${maxTokens} tokens]\n\n${rendered}`;

  pruneHeaders(tree);
  rendered = renderTree(tree);
  return `[Level 0: File names only, project too large for ${maxTokens} tokens]\n\n${rendered}`;
}