feat(agent): add token tracking to CodingAgentRenderer #453

Introduce updateTokenInfo to CodingAgentRenderer for direct token updates, replacing reflection-based invocation and enabling consistent token tracking across renderers.

Author: phodal

mpp-core/src/commonMain/kotlin/cc/unitmesh/agent/executor/CodingAgentExecutor.kt

@@ -53,13 +53,7 @@ class CodingAgentExecutor(
 
         // Set up token tracking callback to update renderer
         conversationManager?.onTokenUpdate = { tokenInfo ->
-            // Update renderer if it supports token tracking
-            try {
-                val updateMethod = renderer::class.members.find { it.name == "updateTokenInfo" }
-                updateMethod?.call(renderer, tokenInfo)
-            } catch (e: Exception) {
-                // Renderer doesn't support token tracking, ignore
-            }
+            renderer.updateTokenInfo(tokenInfo)
         }
 
         val initialUserMessage = buildInitialUserMessage(task)

mpp-core/src/commonMain/kotlin/cc/unitmesh/agent/render/CodingAgentRenderer.kt

@@ -1,5 +1,7 @@
 package cc.unitmesh.agent.render
 
+import cc.unitmesh.llm.compression.TokenInfo
+
 interface CodingAgentRenderer {
     fun renderIterationHeader(current: Int, max: Int)
     fun renderLLMResponseStart()
@@ -22,6 +24,8 @@ interface CodingAgentRenderer {
 
     fun renderRecoveryAdvice(recoveryAdvice: String)
 
+    fun updateTokenInfo(tokenInfo: TokenInfo) {}
+
     fun renderUserConfirmationRequest(toolName: String, params: Map<String, Any>)
 
     fun addLiveTerminal(

mpp-ui/src/commonMain/kotlin/cc/unitmesh/devins/ui/compose/agent/ComposeRenderer.kt

@@ -581,7 +581,7 @@ class ComposeRenderer : BaseRenderer() {
      * Update token information from LLM response
      * Called when StreamFrame.End is received with token metadata
      */
-    fun updateTokenInfo(tokenInfo: TokenInfo) {
+    override fun updateTokenInfo(tokenInfo: TokenInfo) {
         _lastMessageTokenInfo = tokenInfo
         // Accumulate total tokens
         _totalTokenInfo = TokenInfo(