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

feat(goose-cli): theme persistence & selection (#1693)

Browse Source 
Co-authored-by: Bradley Axen <baxen@squareup.com>
This commit is contained in:
Shodipo Ayomide
2025-07-04 01:23:34 +01:00
committed by GitHub
parent f59077587f
commit 5e8e5fb7c2
5 changed files with 70 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ui/desktop/src/hooks/useMessageStream.ts
+10 -6
View File
@@ -317,15 +317,16 @@ export function useMessageStream({
const lastMessage = currentMessages[currentMessages.length - 1];
onFinish(lastMessage, parsedEvent.reason);
}
// Fetch updated session metadata with token counts
const sessionId = (extraMetadataRef.current.body as Record<string, unknown>)?.session_id as string;
const sessionId = (extraMetadataRef.current.body as Record<string, unknown>)
?.session_id as string;
if (sessionId) {
try {
const sessionResponse = await getSessionHistory({
path: { session_id: sessionId },
});
if (sessionResponse.data?.metadata) {
setSessionMetadata({
workingDir: sessionResponse.data.metadata.working_dir,
@@ -335,9 +336,12 @@ export function useMessageStream({
totalTokens: sessionResponse.data.metadata.total_tokens || null,
inputTokens: sessionResponse.data.metadata.input_tokens || null,
outputTokens: sessionResponse.data.metadata.output_tokens || null,
accumulatedTotalTokens: sessionResponse.data.metadata.accumulated_total_tokens || null,
accumulatedInputTokens: sessionResponse.data.metadata.accumulated_input_tokens || null,
accumulatedOutputTokens: sessionResponse.data.metadata.accumulated_output_tokens || null,
accumulatedTotalTokens:
sessionResponse.data.metadata.accumulated_total_tokens || null,
accumulatedInputTokens:
sessionResponse.data.metadata.accumulated_input_tokens || null,
accumulatedOutputTokens:
sessionResponse.data.metadata.accumulated_output_tokens || null,
});
}
} catch (error) {