diff --git a/src/renderer/src/context/ThemeProvider.tsx b/src/renderer/src/context/ThemeProvider.tsx index a1281983e4..85b0dbae86 100644 --- a/src/renderer/src/context/ThemeProvider.tsx +++ b/src/renderer/src/context/ThemeProvider.tsx @@ -52,13 +52,11 @@ export const ThemeProvider: React.FC = ({ children }) => { initUserTheme() // listen for theme updates from main process - const cleanup = window.electron.ipcRenderer.on(IpcChannel.ThemeUpdated, (_, actualTheme: ThemeMode) => { + return window.electron.ipcRenderer.on(IpcChannel.ThemeUpdated, (_, actualTheme: ThemeMode) => { document.body.setAttribute('theme-mode', actualTheme) setActualTheme(actualTheme) }) - - return cleanup - }, []) + }, [actualTheme, initUserTheme, setSettedTheme, settedTheme]) useEffect(() => { window.api.setTheme(settedTheme) diff --git a/src/renderer/src/services/StreamProcessingService.ts b/src/renderer/src/services/StreamProcessingService.ts index 0209ad1194..67acc2f87d 100644 --- a/src/renderer/src/services/StreamProcessingService.ts +++ b/src/renderer/src/services/StreamProcessingService.ts @@ -59,7 +59,7 @@ export function createStreamProcessor(callbacks: StreamProcessorCallbacks = {}) callbacks.onTextComplete(data.text) } if (data.type === ChunkType.THINKING_DELTA && callbacks.onThinkingChunk) { - callbacks.onThinkingChunk(data.text) + callbacks.onThinkingChunk(data.text, data.thinking_millsec) } if (data.type === ChunkType.THINKING_COMPLETE && callbacks.onThinkingComplete) { callbacks.onThinkingComplete(data.text, data.thinking_millsec)