@zhushanwen/pi-smart-context 0.3.9 → 0.4.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zhushanwen/pi-smart-context",
-  "version": "0.3.9",
+  "version": "0.4.0",
   "type": "module",
   "main": "index.ts",
   "taiji": {
@@ -36,6 +36,7 @@
     "pi-package"
   ],
   "dependencies": {
+    "@zhushanwen/extension-protocol": "0.21.0",
     "@zhushanwen/pi-ext-guards": "0.5.0",
     "@zhushanwen/pi-extension-logger": "0.6.2",
     "@zhushanwen/pi-llm-shared": "0.11.3"
@@ -51,15 +52,19 @@
     "vitest.config.ts"
   ],
   "peerDependencies": {
-    "@earendil-works/pi-coding-agent": "^1.0.0",
     "@earendil-works/pi-ai": "^1.0.0",
+    "@earendil-works/pi-coding-agent": "^1.0.0",
+    "@earendil-works/pi-tui": "^1.0.0",
     "typebox": "1.3.27"
   },
   "peerDependenciesMeta": {
+    "@earendil-works/pi-ai": {
+      "optional": true
+    },
     "@earendil-works/pi-coding-agent": {
       "optional": true
     },
-    "@earendil-works/pi-ai": {
+    "@earendil-works/pi-tui": {
       "optional": true
     },
     "typebox": {

package/src/__tests__/summary.test.ts

@@ -0,0 +1,123 @@
+// compact_context 收起态摘要测试：buildCompactContextSummary 文案 + execute 产出的
+// details.summary + renderResult 折叠/展开形态。mock 形态沿用 tool.test.ts。
+
+import type { ExtensionAPI, ExtensionContext, Theme } from "@earendil-works/pi-coding-agent";
+import { beforeEach, describe, expect, it, vi } from "vitest";
+
+// mock 配置加载（不落盘，形态对齐 tool.test.ts）
+vi.mock("../pure.js", async (importOriginal) => {
+	const actual = await importOriginal<typeof import("../pure.js")>();
+	return {
+		...actual,
+		loadSmartContextConfig: vi.fn(),
+	};
+});
+
+import { DEFAULT_SMART_CONTEXT_CONFIG, loadSmartContextConfig } from "../pure.js";
+import { buildCompactContextSummary, registerCompactContextTool } from "../tool.js";
+import { renderCompactContextResult } from "../render-result.js";
+
+const mockedLoad = vi.mocked(loadSmartContextConfig);
+
+beforeEach(() => {
+	mockedLoad.mockReturnValue({ ...DEFAULT_SMART_CONTEXT_CONFIG });
+});
+
+interface RegisteredTool { // oe-exempt:20261017:test:测试 harness 契约——断言面引用其字段类型，非抽象接口
+	name: string;
+	execute: (
+		id: string,
+		params: Record<string, unknown>,
+		signal: undefined,
+		onUpdate: undefined,
+		ctx: ExtensionContext,
+	) => Promise<{ content: Array<{ type: string; text: string }>; details: Record<string, unknown> }>;
+	renderResult: (result: unknown, options: { expanded: boolean }, theme: Theme) => { content: unknown };
+}
+
+function makePi() {
+	const tools: RegisteredTool[] = [];
+	const pi = {
+		registerTool: (t: RegisteredTool) => tools.push(t),
+		sendMessage: vi.fn(),
+		sendUserMessage: vi.fn(),
+	} as unknown as ExtensionAPI;
+	return { pi, tools };
+}
+
+function makeCtx(): ExtensionContext {
+	return {
+		model: { provider: "zai", id: "glm" },
+		getContextUsage: () => ({ tokens: 450_000, contextWindow: 1_000_000 }),
+		sessionManager: { getEntries: () => [] },
+		compact: () => {},
+	} as unknown as ExtensionContext;
+}
+
+// ── buildCompactContextSummary ───────────────────────
+
+describe("buildCompactContextSummary", () => {
+	it("恒为 Context compacted（R2 降级态：返回时压缩未完成，无比例数据）", () => {
+		expect(buildCompactContextSummary()).toBe("Context compacted");
+	});
+});
+
+// ── execute 产出的 details.summary ───────────────────
+
+describe("compact_context execute details.summary", () => {
+	it("details 带现有字段 + summary", async () => {
+		const { pi, tools } = makePi();
+		registerCompactContextTool(pi);
+		const res = await tools[0]!.execute("t1", {}, undefined, undefined, makeCtx());
+		expect(res.details).toMatchObject({
+			mode: "same-model",
+			launched: true,
+			fellBack: false,
+			summary: "Context compacted",
+		});
+	});
+});
+
+// ── renderCompactContextResult（TUI renderResult）────
+
+const mockTheme = {
+	fg: (_color: string, text: string) => text,
+	bg: (_color: string, text: string) => text,
+	bold: (text: string) => text,
+	italic: (text: string) => text,
+	underline: (text: string) => text,
+	inverse: (text: string) => text,
+	strikethrough: (text: string) => text,
+	getFgAnsi: (_color: string) => "",
+	getBgAnsi: (_color: string) => "",
+	getColorMode: () => "truecolor" as const,
+	getThinkingBorderColor: () => (text: string) => text,
+	getBashModeBorderColor: () => (text: string) => text,
+} as unknown as Theme;
+
+function renderedText(result: unknown, expanded: boolean): string {
+	return renderCompactContextResult(result, { expanded }, mockTheme)
+		.render(400)
+		.map((line) => line.trimEnd())
+		.join("\n");
+}
+
+describe("renderCompactContextResult", () => {
+	const result = {
+		content: [{ type: "text", text: "压缩已启动（same-model 模式）..." }],
+		details: { mode: "same-model", launched: true, summary: "Context compacted" },
+	};
+
+	it("折叠态 → details.summary 一行", () => {
+		expect(renderedText(result, false)).toBe("Context compacted");
+	});
+
+	it("展开态 → content 纯文本", () => {
+		expect(renderedText(result, true)).toContain("压缩已启动");
+	});
+
+	it("无 summary（旧 entry）→ 折叠态回退 content 首行", () => {
+		const legacy = { content: [{ type: "text", text: "压缩已启动（same-model 模式）" }], details: {} };
+		expect(renderedText(legacy, false)).toBe("压缩已启动（same-model 模式）");
+	});
+});

package/src/render-result.ts

@@ -0,0 +1,43 @@
+/**
+ * compact_context 工具的 TUI renderResult（折叠态一行摘要 / 展开态 content 纯文本）。
+ */
+
+import type { Theme } from "@earendil-works/pi-coding-agent";
+import { Text } from "@earendil-works/pi-tui";
+import { firstContentText } from "@zhushanwen/extension-protocol";
+
+/** content[0] 的 text 纯文本（content 缺失或非 text 类型时兜底空串）。 */
+function contentText(result: unknown): string {
+	const content = fieldValue(result, "content");
+	if (!Array.isArray(content)) return "";
+	return firstContentText({ content });
+}
+
+/** unknown 值按字段名安全取值（index-signature 收窄，替代全可选结构断言）。 */
+function fieldValue(v: unknown, key: string): unknown {
+	if (typeof v !== "object" || v === null) return undefined;
+	return (v as Record<string, unknown>)[key];
+}
+
+/** 从 details 安全读取 summary（容错非对象形态与旧版本 entry）。 */
+function readDetailsSummary(result: unknown): string | undefined {
+	const summary = fieldValue(fieldValue(result, "details"), "summary");
+	return typeof summary === "string" && summary !== "" ? summary : undefined;
+}
+
+/** 折叠态回退：content 文本首行（保持一行摘要形态）。 */
+function firstLine(text: string): string {
+	const idx = text.indexOf("\n");
+	return idx === -1 ? text : text.slice(0, idx);
+}
+
+/** 折叠态（expanded=false）= details.summary 一行（dim）；展开态 = content 纯文本。 */
+export function renderCompactContextResult(
+	result: unknown,
+	options: { expanded: boolean },
+	theme: Theme,
+): Text {
+	if (options.expanded) return new Text(contentText(result), 0, 0);
+	const summary = readDetailsSummary(result) ?? firstLine(contentText(result));
+	return new Text(theme.fg("dim", summary), 0, 0);
+}

package/src/tool.ts

@@ -14,12 +14,13 @@
  * 「manual + 非 runtime 发起 + compaction_start 前 turn 被掐断」后经投递内核追加续跑投递。
  */
 
-import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
+import type { ExtensionAPI, ExtensionContext, Theme } from "@earendil-works/pi-coding-agent";
 import { guardStaleCtx, isRecord, toErrorMessage } from "@zhushanwen/pi-ext-guards";
 import { Type } from "typebox";
 
 import { debugLog } from "./compact-handler.js";
 import { sendSmartContextNotice } from "./notices.js";
+import { renderCompactContextResult } from "./render-result.js";
 import { buildDegradationHintLine } from "./reminder.js";
 import {
 	DEGRADATION_HINT_MIN_COMPACTIONS,
@@ -63,6 +64,16 @@ export interface CompactContextDetails {
 	/** 压缩已触发（结果经注入消息送达，不在工具结果里）。 */
 	launched: boolean;
 	fellBack: boolean;
+	/** 收起态一行摘要（buildCompactContextSummary 产出；GUI 收起态与 TUI 折叠态共用）。 */
+	summary?: string;
+}
+
+/**
+ * 收起态一行摘要。R2 降级态下工具返回时压缩尚未完成（结果异步注入），无保留比例
+ * 数据可取——恒为 "Context compacted"；完成后的比例数据走注入消息，不进本摘要。
+ */
+export function buildCompactContextSummary(): string {
+	return "Context compacted";
 }
 
 /** 工具描述（§3.5：三条件自查引导；条件即纪律，无第三层复述——D1 单源化核对通过）。 */
@@ -236,8 +247,13 @@ export function registerCompactContextTool(
 					fellBack: false,
 					compactionCount: compactionCount + 1,
 					launched: true,
+					summary: buildCompactContextSummary(),
 				} satisfies CompactContextDetails,
 			};
 		},
+
+		renderResult(result: unknown, options: { expanded: boolean }, theme: Theme) {
+			return renderCompactContextResult(result, options, theme);
+		},
 	});
 }