admin-valentin/agent-ecosystem
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: correct context badge count to sum actual items instead of injection objects (#2)

Browse source
This commit is contained in:
Psypeal Gwai 2026-02-21 01:15:55 -08:00
parent 4ec272758c
commit 6e9c6219b2
2 changed files with 15 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
src/renderer/components/chat/ContextBadge.tsx
View file

@ -215,6 +215,17 @@ export const ContextBadge = ({
[newTaskCoordinationInjections]
);
// Compute actual item counts (not injection-object counts) for accurate badge display
const toolOutputCount = useMemo(
() => newToolOutputInjections.reduce((sum, inj) => sum + inj.toolCount, 0),
[newToolOutputInjections]
);
const taskCoordinationCount = useMemo(
() => newTaskCoordinationInjections.reduce((sum, inj) => sum + inj.breakdown.length, 0),
[newTaskCoordinationInjections]
);
const userMessageTokens = useMemo(
() => newUserMessageInjections.reduce((sum, inj) => sum + inj.estimatedTokens, 0),
[newUserMessageInjections]
@ -478,7 +489,7 @@ export const ContextBadge = ({
{newToolOutputInjections.length > 0 && (
<PopoverSection
title="Tool Outputs"
count={newToolOutputInjections.length}
count={toolOutputCount}
tokenCount={toolOutputTokens}
>
{newToolOutputInjections.map((injection) =>
@ -501,7 +512,7 @@ export const ContextBadge = ({
{newTaskCoordinationInjections.length > 0 && (
<PopoverSection
title="Task Coordination"
count={newTaskCoordinationInjections.length}
count={taskCoordinationCount}
tokenCount={taskCoordinationTokens}
>
{newTaskCoordinationInjections.map((injection) =>

4
src/renderer/utils/contextTracker.ts
View file

@ -836,13 +836,13 @@ function computeContextStats(params: ComputeContextStatsParams): ContextStats {
newCounts.mentionedFiles++;
break;
case 'tool-output':
newCounts.toolOutputs++;
newCounts.toolOutputs += (injection).toolCount;
break;
case 'thinking-text':
newCounts.thinkingText++;
break;
case 'task-coordination':
newCounts.taskCoordination++;
newCounts.taskCoordination += (injection).breakdown.length;
break;
case 'user-message':
newCounts.userMessages++;