);
}
diff --git a/lib/i18n/messages/en.ts b/lib/i18n/messages/en.ts
index 2c3aa09d42..846e1b9906 100644
--- a/lib/i18n/messages/en.ts
+++ b/lib/i18n/messages/en.ts
@@ -20,6 +20,9 @@ export const enLocale: LocalePlugin = {
"settings.themeLight": "Light",
"settings.themeDark": "Dark",
"settings.themeSystem": "System",
+ "settings.thinkingDisplay": "Thinking display",
+ "settings.thinkingDisplayDescription": "Choose whether model reasoning blocks are expanded when messages load.",
+ "settings.thinkingExpandedDefault": "Expand thinking blocks by default",
"settings.languageDescription": "Choose the language used throughout the interface.",
"settings.shellTool": "Shell tool",
"settings.shellToolDescription": "Choose which shell the model uses for commands. Direct ! and !! commands still use Bash.",
diff --git a/lib/i18n/messages/zh-CN.ts b/lib/i18n/messages/zh-CN.ts
index d977b50606..b4ab258771 100644
--- a/lib/i18n/messages/zh-CN.ts
+++ b/lib/i18n/messages/zh-CN.ts
@@ -20,6 +20,9 @@ export const zhCNLocale: LocalePlugin = {
"settings.themeLight": "浅色",
"settings.themeDark": "深色",
"settings.themeSystem": "跟随系统",
+ "settings.thinkingDisplay": "思考过程显示",
+ "settings.thinkingDisplayDescription": "选择消息加载时模型思考块是否默认展开。",
+ "settings.thinkingExpandedDefault": "默认展开思考块",
"settings.languageDescription": "选择整个界面使用的语言。",
"settings.shellTool": "Shell 工具",
"settings.shellToolDescription": "选择模型执行命令时使用的 Shell。直接输入的 ! 和 !! 命令仍使用 Bash。",
diff --git a/lib/i18n/messages/zh-TW.ts b/lib/i18n/messages/zh-TW.ts
index 204f381f0c..9357648e60 100644
--- a/lib/i18n/messages/zh-TW.ts
+++ b/lib/i18n/messages/zh-TW.ts
@@ -20,6 +20,9 @@ export const zhTWLocale: LocalePlugin = {
"settings.themeLight": "淺色",
"settings.themeDark": "深色",
"settings.themeSystem": "跟隨系統",
+ "settings.thinkingDisplay": "思考過程顯示",
+ "settings.thinkingDisplayDescription": "選擇訊息載入時模型思考區塊是否預設展開。",
+ "settings.thinkingExpandedDefault": "預設展開思考區塊",
"settings.languageDescription": "選擇整個介面使用的語言。",
"settings.shellTool": "Shell 工具",
"settings.shellToolDescription": "選擇模型執行命令時使用的 Shell。直接輸入的 ! 和 !! 命令仍使用 Bash。",
diff --git a/lib/message-display.test.mjs b/lib/message-display.test.mjs
index 6165549d6d..b518ff4061 100644
--- a/lib/message-display.test.mjs
+++ b/lib/message-display.test.mjs
@@ -101,6 +101,39 @@ test("keeps deferred historical thinking placeholders", async () => {
);
});
+test("keeps thinking outside contiguous process segments in original order", async () => {
+ const { getDisplayableAssistantBlocks, splitThinkingBlocks, splitFinalAssistantBlocks } = await loadSubject();
+ const message = assistant([
+ { type: "thinking", thinking: "" },
+ { type: "text", text: "Inspecting" },
+ { type: "thinking", thinking: "", deferred: true },
+ { type: "thinking", thinking: "Check the result" },
+ { type: "toolCall", toolCallId: "call-1", toolName: "bash", input: {} },
+ { type: "text", text: "Tool commentary" },
+ { type: "thinking", thinking: "Conclude" },
+ { type: "text", text: "Final answer" },
+ ]);
+ const { processBlocks, answerBlocks } = splitFinalAssistantBlocks(message);
+ const groups = splitThinkingBlocks(processBlocks);
+ assert.deepEqual(groups.map(({ thinking, blocks }) => ({
+ thinking,
+ indices: blocks.map((block) => message.content.indexOf(block)),
+ })), [
+ { thinking: false, indices: [1] },
+ { thinking: true, indices: [2, 3] },
+ { thinking: false, indices: [4, 5] },
+ { thinking: true, indices: [6] },
+ ]);
+ assert.deepEqual([...groups.flatMap(({ blocks }) => blocks), ...answerBlocks], getDisplayableAssistantBlocks(message));
+
+ const simple = splitFinalAssistantBlocks(assistant([
+ { type: "thinking", thinking: "Reasoning" },
+ { type: "text", text: "Answer" },
+ ]));
+ assert.equal(splitThinkingBlocks(simple.processBlocks).some(({ thinking }) => !thinking), false);
+ assert.deepEqual(splitThinkingBlocks([]), []);
+});
+
test("returns completed provider errors even when the message has no content", async () => {
const { getAssistantErrorMessage } = await loadSubject();
const message = {
diff --git a/lib/message-display.ts b/lib/message-display.ts
index b64282c1a4..001e371fc7 100644
--- a/lib/message-display.ts
+++ b/lib/message-display.ts
@@ -50,3 +50,14 @@ export function splitFinalAssistantBlocks(
export function countToolCallBlocks(blocks: AssistantContentBlock[]): number {
return blocks.filter((block): block is ToolCallContent => block.type === "toolCall").length;
}
+
+export function splitThinkingBlocks(blocks: AssistantContentBlock[]): { thinking: boolean; blocks: AssistantContentBlock[] }[] {
+ const groups: { thinking: boolean; blocks: AssistantContentBlock[] }[] = [];
+ for (const block of blocks) {
+ const thinking = block.type === "thinking";
+ const previous = groups.at(-1);
+ if (previous?.thinking === thinking) previous.blocks.push(block);
+ else groups.push({ thinking, blocks: [block] });
+ }
+ return groups;
+}
diff --git a/lib/thinking-expansion-preference.test.mjs b/lib/thinking-expansion-preference.test.mjs
new file mode 100644
index 0000000000..bd49250135
--- /dev/null
+++ b/lib/thinking-expansion-preference.test.mjs
@@ -0,0 +1,57 @@
+import assert from "node:assert/strict";
+import test from "node:test";
+
+import {
+ isThinkingExpandedByDefault,
+ setThinkingExpandedByDefault,
+ THINKING_EXPANDED_EVENT,
+} from "./thinking-expansion-preference.ts";
+
+function installWindow() {
+ const store = new Map();
+ const listeners = new Set();
+ const fakeWindow = {
+ localStorage: {
+ getItem: (key) => (store.has(key) ? store.get(key) : null),
+ setItem: (key, value) => store.set(key, String(value)),
+ removeItem: (key) => store.delete(key),
+ },
+ addEventListener: (_type, handler) => listeners.add(handler),
+ removeEventListener: (_type, handler) => listeners.delete(handler),
+ dispatchEvent: (event) => {
+ for (const handler of listeners) handler(event);
+ return true;
+ },
+ };
+ globalThis.window = fakeWindow;
+ return { fakeWindow, listeners };
+}
+
+test("defaults to collapsed when no preference is stored", () => {
+ installWindow();
+ assert.equal(isThinkingExpandedByDefault(), false);
+});
+
+test("persists the preference and notifies listeners on change", () => {
+ installWindow();
+ let notified = 0;
+ globalThis.window.addEventListener(THINKING_EXPANDED_EVENT, () => notified++);
+
+ setThinkingExpandedByDefault(true);
+ assert.equal(isThinkingExpandedByDefault(), true);
+ assert.equal(notified, 1);
+
+ setThinkingExpandedByDefault(false);
+ assert.equal(isThinkingExpandedByDefault(), false);
+ assert.equal(notified, 2);
+});
+
+test("returns false when window is unavailable (SSR)", () => {
+ const original = globalThis.window;
+ delete globalThis.window;
+ try {
+ assert.equal(isThinkingExpandedByDefault(), false);
+ } finally {
+ if (original !== undefined) globalThis.window = original;
+ }
+});
diff --git a/lib/thinking-expansion-preference.ts b/lib/thinking-expansion-preference.ts
new file mode 100644
index 0000000000..dc1c522d58
--- /dev/null
+++ b/lib/thinking-expansion-preference.ts
@@ -0,0 +1,15 @@
+const STORAGE_KEY = "pi-thinking-expanded";
+
+// Broadcast so already-mounted ThinkingBlock instances update when the
+// preference changes in the settings panel.
+export const THINKING_EXPANDED_EVENT = "pi-thinking-expanded-changed";
+
+export function isThinkingExpandedByDefault(): boolean {
+ if (typeof window === "undefined") return false;
+ return window.localStorage.getItem(STORAGE_KEY) === "true";
+}
+
+export function setThinkingExpandedByDefault(expanded: boolean): void {
+ window.localStorage.setItem(STORAGE_KEY, String(expanded));
+ window.dispatchEvent(new Event(THINKING_EXPANDED_EVENT));
+}