Skip to content

Commit 6e95f9e

Browse files
Refactor pruneFileTokenScores to accept a single options object to improve readability and reduce parameter coupling; update call sites accordingly.
🤖 Generated with Codebuff Co-Authored-By: Codebuff <noreply@codebuff.com>
1 parent dbba78a commit 6e95f9e

File tree

1 file changed

+9
-8
lines changed

1 file changed

+9
-8
lines changed

backend/src/system-prompt/truncate-file-tree.ts

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -65,11 +65,11 @@ export const truncateFileTreeBasedOnTokenBudget = (
6565
truncationLevel: 'unimportant-files',
6666
}
6767
}
68-
const { printedTree, tokenCount } = pruneFileTokenScores(
69-
filteredTree,
68+
const { printedTree, tokenCount } = pruneFileTokenScores({
69+
fileTree: filteredTree,
7070
fileTokenScores,
7171
tokenBudget,
72-
)
72+
})
7373

7474
if (tokenCount <= tokenBudget) {
7575
if (DEBUG) {
@@ -216,11 +216,12 @@ export const truncateFileTreeBasedOnTokenBudget = (
216216
}
217217
}
218218

219-
function pruneFileTokenScores(
220-
fileTree: FileTreeNode[],
221-
fileTokenScores: Record<string, Record<string, number>>,
222-
tokenBudget: number,
223-
) {
219+
function pruneFileTokenScores(params: {
220+
fileTree: FileTreeNode[]
221+
fileTokenScores: Record<string, Record<string, number>>
222+
tokenBudget: number
223+
}) {
224+
const { fileTree, fileTokenScores, tokenBudget } = params
224225
const startTime = performance.now()
225226

226227
// Create sorted array of tokens by score

0 commit comments

Comments
 (0)