Fix TPS calculation to use current message instead of other messages

Author: ariane-emory

packages/opencode/src/cli/cmd/tui/routes/session/index.tsx

@@ -1200,10 +1200,9 @@ function AssistantMessage(props: { message: AssistantMessage; parts: Part[]; las
     if (!final()) return 0
     if (!props.message.time.completed) return 0
     if (!(sync.data.config.tui as any)?.display_message_tps) return 0
-  
-    const assistantMessages : AssistantMessage[] = messages().filter((msg) => msg.role === "assistant" && msg.id !== props.message.id) as AssistantMessage[]
 
-    const allParts = assistantMessages.flatMap((msg) => getParts(msg.id))
+    // Get parts for the current message only
+    const allParts = getParts(props.message.id)
 
     const INVALID_REASONING_TEXTS = ["[REDACTED]", "", null, undefined] as const
 
@@ -1240,17 +1239,11 @@ function AssistantMessage(props: { message: AssistantMessage; parts: Part[]; las
     }
 
     if (totalStreamingTimeMs === 0) return 0
-  
-    const totals = assistantMessages.reduce(
-      (acc, m) => {
-        acc.output += m.tokens.output
-       if (hasValidReasoning) acc.reasoning += m.tokens.reasoning // Only count reasoning tokens if valid reasoning parts exists
-        return acc
-      },
-      { output: 0, reasoning: 0 },
-    )
 
-    const totalTokens = totals.reasoning + totals.output
+    // Use token counts from the current message
+    const outputTokens = props.message.tokens.output
+    const reasoningTokens = hasValidReasoning ? props.message.tokens.reasoning : 0
+    const totalTokens = outputTokens + reasoningTokens
 
     if (totalTokens === 0) return 0