npm - @foolsecret/pi-prompt - Versions diffs - 0.4.0 → 0.4.9 - Mend

@foolsecret/pi-prompt 0.4.0 → 0.4.9

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (15) hide show

package/CHANGELOG.md +203 -0
package/README.md +43 -5
package/package.json +55 -55
package/src/auto.ts +109 -13
package/src/command.ts +10 -2
package/src/compare.ts +228 -0
package/src/config.ts +178 -1
package/src/context.ts +136 -0
package/src/draft.ts +189 -0
package/src/info-page.ts +115 -0
package/src/prompt-extension.ts +350 -70
package/src/prompts.ts +48 -6
package/src/stats.ts +50 -5
package/src/tool-output.ts +128 -0
package/src/ui.ts +598 -96

package/src/prompts.ts CHANGED Viewed

@@ -12,6 +12,7 @@
  */
 import type { DoMode, RuntimeShowMode, WriteMode } from "./modes.ts";
+import { createHash } from "node:crypto";
 /** 单个注入块策略接口（每轴每档 = 一个策略类） */
 interface BlobStrategy {
@@ -181,17 +182,58 @@ export class PromptRegistry {
 	}
 	/**
-	 * 组合注入文案：固定顺序拼接（show→write→do），头部一行档位标识。
-	 * 组合确定性 → 每个 (show,write,do) 三元组都是恒定字符串（缓存友好）。
+	 * 组合注入文案：三块独立拼接，每块带自己的固定边界标记。
+	 *
+	 * 缓存友好设计（v0.4.1）：三轴各自成块、块内字节只随自己的档位变 ——
+	 * 调 write/do **不再改变 show 块的字节**，把前缀失效面收缩到最小。
+	 * 每块格式恒定：`[PI-PROMPT:<AXIS> <mode>]\n<body>`；不注入的轴整块省略。
+	 *
+	 * 注意：三块仍是同一个字符串，拼在系统提示的同一处 —— 中间块变动仍会
+	 * 连带失效其后的内容（含 pi 追加的 project_context/skills/cwd）。真正的
+	 * 隔离要靠"注入移到 system prompt 末尾"（见 prompt-extension）。
+	 *
 	 * 三块全空（show normal + write normal + do normal）返回 undefined（不注入）。
 	 */
 	compose(show: RuntimeShowMode, write: WriteMode, doMode: DoMode): string | undefined {
 		const blocks = [
-			this.textForShow(show),
-			this.textForWrite(write),
-			this.textForDo(doMode),
+			blockFor("SHOW", show, this.textForShow(show)),
+			blockFor("WRITE", write, this.textForWrite(write)),
+			blockFor("DO", doMode, this.textForDo(doMode)),
 		].filter((block): block is string => block !== undefined);
 		if (blocks.length === 0) return undefined;
-		return `[PI-PROMPT ${show}|${write}|${doMode}]\n${blocks.join("\n")}`;
+		// 直接拼接（非 join）：每块自带尾部分隔符，块字节与邻块无关 ——
+		// 保证"调 write/do 不改 show 块字节"（否则 join 的 \n 会让末尾块少一个分隔符）。
+		return blocks.join("");
 	}
+	/**
+	 * 单轴独立块（供希望按轴分别拼接的调用方使用）：返回该轴的恒定块或 undefined。
+	 * 块字节只依赖该轴档位 —— 这是"调 write 不动 show 字节"的保证点。
+	 */
+	composeBlock(axis: "SHOW" | "WRITE" | "DO", mode: RuntimeShowMode | WriteMode | DoMode): string | undefined {
+		if (axis === "SHOW") return blockFor("SHOW", mode, this.textForShow(mode as RuntimeShowMode));
+		if (axis === "WRITE") return blockFor("WRITE", mode, this.textForWrite(mode as WriteMode));
+		return blockFor("DO", mode, this.textForDo(mode as DoMode));
+	}
+}
+/**
+ * 构造单轴注入块：`[PI-PROMPT:<AXIS> <mode>]\n<body>\n`。
+ * 尾部自带换行 → 块字节与邻块彻底解耦（拼接顺序无关），
+ * 这是"调 write 不动 show 块字节"的保证点。
+ * body 为空（不注入的档）时整块返回 undefined，调用方据此省略该轴。
+ */
+function blockFor(axis: "SHOW" | "WRITE" | "DO", mode: string, body: string | undefined): string | undefined {
+	if (body === undefined) return undefined;
+	return `[PI-PROMPT:${axis} ${mode}]\n${body}\n`;
+}
+/**
+ * 注入串的短哈希（前 8 位）—— 学 Reasonix 的 CaptureShape：
+ * 让用户一眼看出"前缀有没有变"（换档/改档都会变），是排查缓存 miss 的第一现场。
+ * 纯函数，可直接单测。空串（不注入）返回 "(none)"。
+ */
+export function prefixHash(text: string | undefined): string {
+	if (text === undefined || text === "") return "(none)";
+	return createHash("sha256").update(text, "utf8").digest("hex").slice(0, 8);
 }

package/src/stats.ts CHANGED Viewed

@@ -26,11 +26,33 @@ export interface TurnUsage {
 	cacheWrite: number;
 }
+/**
+ * 台账 schema 版本（写入记录时的格式版本）。
+ * v0=历史（无此字段）· v1 起记录 v/pv。字段只增不改，读取时按版本兼容。
+ */
+export const USAGE_SCHEMA_VERSION: number = 1;
+/** 当前插件版本（写入台账的 pv 字段；从 package.json 读，失败回退 "unknown"） */
+export function currentPluginVersion(): string {
+	try {
+		// 相对本模块向上找 package.json（src/ 与发布后的 dist/ 均可）
+		const url = new URL("../package.json", import.meta.url);
+		const pkg = JSON.parse(readFileSync(url, "utf8")) as { version?: string };
+		return pkg.version ?? "unknown";
+	} catch {
+		return "unknown";
+	}
+}
 /** 任务粗略分桶（按用户请求 token 估算，µ 估计与校准分组用） */
 export type TaskBin = "tiny" | "short" | "mid" | "long" | "xlong";
 /** 一条台账记录（v0.2 字段；history 兼容 records 的 mode 字段） */
 export interface UsageRecord extends TurnUsage {
+	/** schema 版本（历史记录无此字段 → 读取时补 0） */
+	v?: number;
+	/** 插件版本（历史记录无此字段 → 读取时补 "legacy"） */
+	pv?: string;
 	ts: number;
 	provider: string;
 	model: string;
@@ -275,11 +297,23 @@ export interface UsageRow {
 	savedCNY: number;
 }
+/**
+ * 缓存命中率（纯函数，百分数 0~100）：cacheRead / (input + cacheRead)。
+ * 分母 = 全部输入 token（未命中 input + 命中 cacheRead）。
+ * 口径说明：pi 的 usage 里 input 只计未命中部分（GLM 同理，见认知修正 #8），
+ * 故 input + cacheRead = prompt 总量；无输入时返回 0（不出现 NaN）。
+ */
+export function cacheHitRate(input: number, cacheRead: number): number {
+	const denom = input + cacheRead;
+	if (denom <= 0) return 0;
+	return (100 * cacheRead) / denom;
+}
 /** /prompt usage 结果：聚合行 + 总览 + 按会话明细 */
 export interface UsageSummary {
 	rows: UsageRow[];
-	total: { turns: number; input: number; cacheRead: number; output: number; costCNY: number };
-	sessions: Array<{ session: string; turns: number; input: number; output: number; costCNY: number; savedCNY: number }>;
+	total: { turns: number; input: number; cacheRead: number; output: number; costCNY: number; hitRate: number };
+	sessions: Array<{ session: string; turns: number; input: number; cacheRead: number; output: number; costCNY: number; savedCNY: number; hitRate: number }>;
 }
 /**
@@ -298,7 +332,7 @@ export function recordCost(record: UsageRecord, price: PricingResolver = activeR
  */
 export function summarizeUsage(records: readonly UsageRecord[], price: PricingResolver = activeResolver): UsageSummary {
 	const byDayModel = new Map<string, UsageRow>();
-	const bySession = new Map<string, { session: string; turns: number; input: number; output: number; costCNY: number; savedCNY: number }>();
+	const bySession = new Map<string, { session: string; turns: number; input: number; cacheRead: number; output: number; costCNY: number; savedCNY: number; hitRate: number }>();
 	let totalInput = 0;
 	let totalCache = 0;
 	let totalOutput = 0;
@@ -326,12 +360,14 @@ export function summarizeUsage(records: readonly UsageRecord[], price: PricingRe
 		byDayModel.set(key, row);
 		if (record.session !== undefined) {
-			const sess = bySession.get(record.session) ?? { session: record.session, turns: 0, input: 0, output: 0, costCNY: 0, savedCNY: 0 };
+			const sess = bySession.get(record.session) ?? { session: record.session, turns: 0, input: 0, cacheRead: 0, output: 0, costCNY: 0, savedCNY: 0, hitRate: 0 };
 			sess.turns += 1;
 			sess.input += record.input;
+			sess.cacheRead += record.cacheRead;
 			sess.output += record.output;
 			sess.costCNY += cost;
 			sess.savedCNY += ((record.output * reduction) / 1_000_000) * price(record.model, record.provider, record.ts).output;
+			sess.hitRate = cacheHitRate(sess.input, sess.cacheRead);
 			bySession.set(record.session, sess);
 		}
 	}
@@ -340,7 +376,14 @@ export function summarizeUsage(records: readonly UsageRecord[], price: PricingRe
 	const sessions = [...bySession.values()].sort((a, b) => a.session.localeCompare(b.session));
 	return {
 		rows,
-		total: { turns: totalTurns, input: totalInput, cacheRead: totalCache, output: totalOutput, costCNY: totalCost },
+		total: {
+			turns: totalTurns,
+			input: totalInput,
+			cacheRead: totalCache,
+			output: totalOutput,
+			costCNY: totalCost,
+			hitRate: cacheHitRate(totalInput, totalCache),
+		},
 		sessions,
 	};
 }
@@ -376,6 +419,8 @@ export class UsageLedger {
 				// v0.1 记录只有 mode：迁移成 show 轴（normalizeRuntimeShow 顺带把
 				// v0.2 历史 "off" 归一为 normal）；无法识别的旧档（review 等）回退 normal
 				records.push({
+					v: typeof parsed.v === "number" ? parsed.v : 0,
+					pv: typeof parsed.pv === "string" ? parsed.pv : "legacy",
 					ts: typeof parsed.ts === "number" ? parsed.ts : 0,
 					provider: typeof parsed.provider === "string" ? parsed.provider : "unknown",
 					model: typeof parsed.model === "string" ? parsed.model : "unknown",

package/src/tool-output.ts ADDED Viewed

@@ -0,0 +1,128 @@
+/**
+ * 工具输出截断（纯函数，确定性；对齐 Reasonix 的 tool-aware 方案）。
+ *
+ * 背景：
+ *   - 实测工具输出占上下文 **27~65%**（大会话），是输入成本的隐性大项。
+ *   - pi 自己已把工具输出截到 **50KB**（read/grep/bash/find 全覆盖），
+ *     但 50KB 之上仍有大量冗余：实测 grep 均 13.6KB、read p95 9.5KB。
+ *   - 本模块提供**更激进**的可选上限（默认关），按工具分层。
+ *
+ * 缓存友好性（硬要求）：
+ *   - 在 `tool_result` hook **创建时**截断一次，此后字节不变（不改历史）。
+ *   - **确定性**：同输入必得同输出（否则破坏前缀缓存）。
+ *   - 未超限时**原样返回**（零影响）。
+ *
+ * 为何分层（不照抄 Reasonix 的 32KB）：
+ *   - Reasonix 的 32KB 是**触发阈值**，实际净保留仅 11~16KB（head/tail 剪）。
+ *   - read 是模型主动要看的内容，砍狠会引发重读 → 反而增加（最贵的）输出成本。
+ *     实测 read p50=1.3KB、p90=5.4KB → 8K 只影响 6% 的 read 调用，风险可控。
+ */
+/** 截断结果 */
+export interface ToolTruncateResult {
+	/** 截断后的文本（未超限时原样） */
+	text: string;
+	/** 是否发生截断 */
+	truncated: boolean;
+	/** 省下的字节数（未截断为 0） */
+	savedBytes: number;
+}
+/** 各工具的默认上限（由调用方从 config 传入；这里只作兵底） */
+export interface ToolCaps {
+	grep?: number;
+	read?: number;
+	bash?: number;
+	default?: number;
+}
+/** 错误关键字（命中则多留尾部——错误信息多在输出末尾） */
+const ERROR_MARKERS = ["error:", "panic:", "fatal:", "traceback", "exception"];
+/**
+ * 把字节上限换算成安全的字符串切点：UTF-8 字符不会被截半。
+ * 从 from 向 to 方向找最近的合法边界。
+ */
+function snapToBoundary(text: string, index: number, dir: "start" | "end"): number {
+	if (index <= 0) return 0;
+	if (index >= text.length) return text.length;
+	let i = index;
+	if (dir === "start") {
+		// 向后退到非续字节（0b10xxxxxx）之后
+		while (i > 0 && (text.charCodeAt(i) & 0xc0) === 0x80) i -= 1;
+		return i;
+	}
+	while (i < text.length && (text.charCodeAt(i) & 0xc0) === 0x80) i += 1;
+	return i;
+}
+/**
+ * 用 `Buffer.byteLength` 计字节（字符串 .length 是 UTF-16 码元，不等于字节）。
+ */
+function bytes(text: string): number {
+	return Buffer.byteLength(text, "utf8");
+}
+/**
+ * 工具感知截断（纯函数、确定性）。
+ * 策略：保留 head + 中段标记 + tail（借用 Reasonix 的 head/tail 思路）。
+ *   - bash：head/tail 各半（错误多在末尾）
+ *   - read/grep/其他：head 为主、tail 少量（代码/匹配通常看开头）
+ * @param text 原始工具输出
+ * @param tool 工具名（"read"/"grep"/"bash"/其他）
+ * @param cap 字节上限（<=0 = 不截）
+ */
+export function truncateToolOutput(text: string, tool: string, cap: number): ToolTruncateResult {
+	if (!cap || cap <= 0) return { text, truncated: false, savedBytes: 0 };
+	const total = bytes(text);
+	if (total <= cap) return { text, truncated: false, savedBytes: 0 };
+	// 留出标记文本的空间，避免截断后反而超限（小 cap 时按比例留，不硬扣 128）
+	const MARKER_RESERVE = Math.min(128, Math.floor(cap / 4));
+	const budget = Math.max(8, cap - MARKER_RESERVE);
+	// 分层比例：bash 偏尾（错误在末尾），其余偏头
+	let headChars: number;
+	let tailChars: number;
+	if (tool === "bash") {
+		headChars = Math.floor(budget / 2);
+		tailChars = budget - headChars;
+	} else if (tool === "read" || tool === "grep") {
+		headChars = Math.floor(budget * 0.85);
+		tailChars = budget - headChars;
+	} else {
+		headChars = Math.floor(budget * 0.75);
+		tailChars = budget - headChars;
+	}
+	// 错误关键字 → 多留尾部（翻倍，但不超过预算）
+	const lower = text.slice(-4096).toLowerCase();
+	if (ERROR_MARKERS.some((m) => lower.includes(m))) {
+		const extra = Math.min(tailChars, Math.floor(budget / 3));
+		if (headChars + tailChars + extra <= budget + extra) {
+			tailChars += extra;
+			headChars = Math.max(0, budget - tailChars);
+		}
+	}
+	// 按字节切（先粗切字符数，再向边界吸附）
+	const headRaw = text.slice(0, headChars);
+	const headEnd = headRaw.length >= text.length ? text.length : snapToBoundary(text, headRaw.length, "end");
+	const tailStartCandidate = text.length - tailChars;
+	const tailStart = tailStartCandidate <= headEnd ? headEnd : snapToBoundary(text, tailStartCandidate, "start");
+	const head = text.slice(0, headEnd);
+	const tail = text.slice(tailStart);
+	// 标记要极短（ASCII）且能进 MARKER_RESERVE
+	const marker = `\n[truncated ${total}>${cap}B]\n`;
+	const result = head + marker + tail;
+	// 双保险：若仍超限（多字节字符/标记开销），硬切到 cap（吸附到字符边界）
+	if (bytes(result) > cap) {
+		let cut = Math.min(cap, result.length);
+		while (cut > 0 && bytes(result.slice(0, cut)) > cap) cut -= 1;
+		const sliced = result.slice(0, cut);
+		return { text: sliced, truncated: true, savedBytes: total - bytes(sliced) };
+	}
+	return { text: result, truncated: true, savedBytes: total - bytes(result) };
+}