tkmind/tkmind_go
1
0
Fork 0
Code Issues Pull Requests Actions 4 Packages Projects Releases Wiki Activity

Fix tokenState loading on new sessions (#6129)

Browse Source
This commit is contained in:
David Katz
2025-12-16 11:35:19 -05:00
committed by GitHub
parent 0a5b7d7ce6
commit 15aa8187f1
1 changed files with 16 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ui/desktop/src/hooks/useChatStream.ts
+16
View File
@@ -212,6 +212,14 @@ export function useChatStream({
if (cached) { if (cached) {
setSession(cached.session); setSession(cached.session);
updateMessages(cached.messages); updateMessages(cached.messages);
setTokenState({
inputTokens: cached.session?.input_tokens ?? 0,
outputTokens: cached.session?.output_tokens ?? 0,
totalTokens: cached.session?.total_tokens ?? 0,
accumulatedInputTokens: cached.session?.accumulated_input_tokens ?? 0,
accumulatedOutputTokens: cached.session?.accumulated_output_tokens ?? 0,
accumulatedTotalTokens: cached.session?.accumulated_total_tokens ?? 0,
});
setChatState(ChatState.Idle); setChatState(ChatState.Idle);
return; return;
} }
@@ -241,6 +249,14 @@ export function useChatStream({
const session = response.data; const session = response.data;
setSession(session); setSession(session);
updateMessages(session?.conversation || []); updateMessages(session?.conversation || []);
setTokenState({
inputTokens: session?.input_tokens ?? 0,
outputTokens: session?.output_tokens ?? 0,
totalTokens: session?.total_tokens ?? 0,
accumulatedInputTokens: session?.accumulated_input_tokens ?? 0,
accumulatedOutputTokens: session?.accumulated_output_tokens ?? 0,
accumulatedTotalTokens: session?.accumulated_total_tokens ?? 0,
});
setChatState(ChatState.Idle); setChatState(ChatState.Idle);
onSessionLoaded?.(); onSessionLoaded?.();
} catch (error) { } catch (error) {