From 698d2f6c3f966cda62c520c84dad3b4695eefe7c Mon Sep 17 00:00:00 2001
From: M57 <41645758+Hy4ri@users.noreply.github.com>
Date: Thu, 30 Jul 2026 11:37:36 +0300
Subject: [PATCH] fix(chat): fetch context usage immediately on session
 change/connect (#746)

Clears stale context token states on session switch/creation and fetches

fresh context usage info immediately instead of waiting for the 5s poll.

Co-authored-by: M57 <hy4ri@users.noreply.github.com>
---
 .../java/com/m57/hermescontrol/ui/chat/ChatScreen.kt |  4 ++++
 .../com/m57/hermescontrol/ui/chat/ChatViewModel.kt   | 12 ++++++++++++
 2 files changed, 16 insertions(+)

diff --git a/app/src/main/java/com/m57/hermescontrol/ui/chat/ChatScreen.kt b/app/src/main/java/com/m57/hermescontrol/ui/chat/ChatScreen.kt
index b84ae955..0e42e774 100644
--- a/app/src/main/java/com/m57/hermescontrol/ui/chat/ChatScreen.kt
+++ b/app/src/main/java/com/m57/hermescontrol/ui/chat/ChatScreen.kt
@@ -134,6 +134,10 @@ fun ChatScreen(
 
     // Periodic session sync while connected.
     LaunchedEffect(state.currentSessionId, state.connectionStatus) {
+        if (state.currentSessionId != null && state.connectionStatus == ConnectionStatus.CONNECTED) {
+            viewModel.syncCurrentSession()
+            viewModel.fetchContextUsage()
+        }
         while (state.currentSessionId != null && state.connectionStatus == ConnectionStatus.CONNECTED) {
             delay(SESSION_SYNC_INTERVAL_MS)
             viewModel.syncCurrentSession()
diff --git a/app/src/main/java/com/m57/hermescontrol/ui/chat/ChatViewModel.kt b/app/src/main/java/com/m57/hermescontrol/ui/chat/ChatViewModel.kt
index 1b7101df..e1c7f02c 100644
--- a/app/src/main/java/com/m57/hermescontrol/ui/chat/ChatViewModel.kt
+++ b/app/src/main/java/com/m57/hermescontrol/ui/chat/ChatViewModel.kt
@@ -595,6 +595,9 @@ class ChatViewModel(
                         isLoading = false,
                         messages = emptyList(),
                         chatTitle = "Hermes",
+                        usedContextTokens = null,
+                        fullContextTokens = null,
+                        contextBreakdown = null,
                     )
                 }
                 // Mirror the active session id app-wide so session-scoped
@@ -604,6 +607,7 @@ class ChatViewModel(
                 _streamingState.update { StreamingState() }
                 addSystemMessage("Session created", persist = true)
                 loadSessions()
+                fetchContextUsage()
             }
 
             WsMethods.SESSION_BRANCH -> {
@@ -616,6 +620,9 @@ class ChatViewModel(
                         isLoading = false,
                         messages = emptyList(),
                         chatTitle = (resultMap["title"] as? String)?.takeIf { t -> t.isNotBlank() } ?: "Hermes",
+                        usedContextTokens = null,
+                        fullContextTokens = null,
+                        contextBreakdown = null,
                     )
                 }
                 ActiveSessionHolder.set(newId)
@@ -623,6 +630,7 @@ class ChatViewModel(
                 addSystemMessage("Session branched", persist = true)
                 loadSessionMessages(newId)
                 loadSessions()
+                fetchContextUsage()
             }
 
             WsMethods.SESSION_LIST -> {
@@ -684,6 +692,7 @@ class ChatViewModel(
                 // Mirror the active runtime session id app-wide (issue #532).
                 ActiveSessionHolder.set(runtimeSessionId ?: sessionId)
                 addSystemMessage("Session resumed")
+                fetchContextUsage()
             }
 
             WsMethods.SESSION_INTERRUPT -> {
@@ -1447,6 +1456,9 @@ class ChatViewModel(
                 chatTitle = title,
                 showSessionPicker = false,
                 isAgentTyping = false,
+                usedContextTokens = null,
+                fullContextTokens = null,
+                contextBreakdown = null,
             )
         }
         // Mirror the active session id app-wide (issue #532).