mozempk/trueref
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(FEEDBACK-0001): complete iteration 0 - harden context search

Browse Source
This commit is contained in:
Giancarmine Salucci
2026-03-27 01:25:46 +01:00
parent e7a2a83cdb
commit 16436bfab2
15 changed files with 1469 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/lib/server/api/token-budget.ts
View File

@@ -11,8 +11,8 @@ import type { Snippet } from '$lib/types';
* Select snippets from a ranked list up to a maximum token budget.
*
* Snippets are evaluated in order. A snippet is included when its token count
* does not push the running total past `maxTokens`. The loop halts at the
* first snippet that would exceed the budget.
* does not push the running total past `maxTokens`. Oversized snippets are
* skipped so lower-ranked results can still be considered.
*
* @param snippets - Ranked list of snippets (best first).
* @param maxTokens - Inclusive upper bound on total token count.
@@ -24,7 +24,7 @@ export function selectSnippetsWithinBudget(snippets: Snippet[], maxTokens: numbe
let usedTokens = 0;
for (const snippet of snippets) {
if (usedTokens + (snippet.tokenCount ?? 0) > maxTokens) break;
if (usedTokens + (snippet.tokenCount ?? 0) > maxTokens) continue;
selected.push(snippet);
usedTokens += snippet.tokenCount ?? 0;
}