smartcompact 0.1.4 → 0.1.5

package/adapters/pi/index.ts

@@ -12,6 +12,7 @@ import type { Lines } from '../../plugin/src/core/lines.ts'
 import { resolveParams } from '../../plugin/src/core/params.ts'
 import { countDecisions, renderReport } from '../../plugin/src/core/report.ts'
 import {
+  cycleReads,
   type DecisionLogEntry,
   expectedAfterTokens,
   growthSinceMilestone,
@@ -19,6 +20,7 @@ import {
   parseState,
   type SessionState,
   withCompaction,
+  withRead,
   withSample,
   withSignals,
 } from '../../plugin/src/core/state.ts'
@@ -163,8 +165,9 @@ export default function smartCompact(pi: ExtensionAPI) {
     if (event.toolName === CHECKPOINT_TOOL) return
     sample(ctx, lines(ctx))
     const text = event.content.flatMap(block => (block.type === 'text' ? [block.text] : [])).join('\n')
-    const { signals, ledger } = recognizePi({ toolName: event.toolName, input: event.input, isError: event.isError, text }, state.ledger)
+    const { signals, ledger, read } = recognizePi({ toolName: event.toolName, input: event.input, isError: event.isError, text }, state.ledger)
     state = withSignals({ ...state, ledger }, signals)
+    if (read) state = withRead(state, read)
   })
 
   pi.on('before_agent_start', async (_event, ctx) => {
@@ -216,10 +219,11 @@ export default function smartCompact(pi: ExtensionAPI) {
     const fromPlugin = event.reason === 'manual' && ours
     ours = false
     const summaryTokens = Math.round(event.compactionEntry.summary.length / 4)
+    const reads = cycleReads(state)
     // 原生按阈值的压缩、溢出恢复和用户的 /compact，没有插件时也会发生；只有插件发起的压缩计入节省。
     state = withCompaction(state, event.compactionEntry.tokensBefore, summaryTokens, fromPlugin)
     persist()
-    log({ event: 'compacted', trigger: fromPlugin ? 'plugin' : event.reason === 'threshold' ? 'auto' : event.reason, tokensBefore: event.compactionEntry.tokensBefore })
+    log({ event: 'compacted', trigger: fromPlugin ? 'plugin' : event.reason === 'threshold' ? 'auto' : event.reason, tokensBefore: event.compactionEntry.tokensBefore, ...reads })
     pi.sendMessage(
       { customType: LEDGER_MESSAGE, content: renderLedger(state.ledger, params(ctx).ledgerMaxChars), display: false },
       { deliverAs: ctx.isIdle() ? 'nextTurn' : 'steer' },

package/adapters/pi/recognize.ts

@@ -14,7 +14,7 @@ export type PiToolOutcome = {
 
 const EDIT_TOOLS = new Set(['edit', 'write'])
 
-export function recognizePi(outcome: PiToolOutcome, ledger: Ledger): { signals: Signal[]; ledger: Ledger } {
+export function recognizePi(outcome: PiToolOutcome, ledger: Ledger): { signals: Signal[]; ledger: Ledger; read?: string } {
   const signals: Signal[] = [{ kind: 'tool-result', isError: outcome.isError }]
   let next = ledger
   const { path, command } = outcome.input
@@ -26,5 +26,6 @@ export function recognizePi(outcome: PiToolOutcome, ledger: Ledger): { signals:
     signals.push(...classified.signals)
     next = classified.ledger
   }
-  return { signals, ledger: next }
+  const read = outcome.toolName === 'read' && !outcome.isError && typeof path === 'string' && path ? path : undefined
+  return { signals, ledger: next, read }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "smartcompact",
-  "version": "0.1.4",
+  "version": "0.1.5",
   "type": "module",
   "description": "Compact long agent sessions earlier, at a finished piece of work, to cut cost without losing results (Claude Code, pi, Codex)",
   "license": "MIT",

package/plugin/src/core/ledger.ts

@@ -128,9 +128,20 @@ export function applyCheckpoint(ledger: Ledger, input: CheckpointInput): Ledger
   }
 }
 
-/** 排版成补回上下文的文本，超长时截断。 */
+/** 排版成补回上下文的文本，超长时先丢最旧的决定、再丢最早改过的文件，仍超长才截断。 */
 export function renderLedger(ledger: Ledger, maxChars: number): string {
-  const { facts, notes } = ledger
+  let decisions = ledger.notes.decisions
+  let files = ledger.facts.filesChanged
+  let rendered = layoutLedger(ledger, decisions, files)
+  while (rendered.length > maxChars && (decisions.length || files.length)) {
+    if (decisions.length) decisions = decisions.slice(1)
+    else files = files.slice(1)
+    rendered = layoutLedger(ledger, decisions, files)
+  }
+  return rendered.length <= maxChars ? rendered : `${rendered.slice(0, maxChars - 20)}\n…(truncated)`
+}
+
+function layoutLedger({ facts, notes }: Ledger, decisions: readonly string[], files: readonly string[]): string {
   const lines = [
     LEDGER_MARKER,
     'Generated by the SmartCompact plugin to keep continuity across compaction. It is not a message from the user.',
@@ -139,19 +150,19 @@ export function renderLedger(ledger: Ledger, maxChars: number): string {
     const rows = body.filter((row): row is string => Boolean(row))
     if (rows.length) lines.push('', `## ${title}`, ...rows)
   }
+  const omitted = (dropped: number, what: string) => (dropped ? `- …(${dropped} earlier ${what} omitted)` : undefined)
   section('Goal', [notes.goal])
   section('Current state', [notes.current])
   section('Next', [notes.next])
-  section('Decisions', notes.decisions.map(decision => `- ${decision}`))
-  section('Tasks in progress', facts.tasksInProgress.map(task => `- ${task}`))
-  section('Files changed', facts.filesChanged.map(file => `- ${file}`))
   section('Latest results', [
     facts.lastTest && `- tests ${facts.lastTest.passed ? 'passed' : 'FAILED'}: ${facts.lastTest.command}`,
     facts.lastBuild && `- build ${facts.lastBuild.passed ? 'passed' : 'FAILED'}: ${facts.lastBuild.command}`,
     facts.lastCommit && `- last commit: ${facts.lastCommit.summary}`,
   ])
-  const rendered = lines.join('\n')
-  return rendered.length <= maxChars ? rendered : `${rendered.slice(0, maxChars - 20)}\n…(truncated)`
+  section('Tasks in progress', facts.tasksInProgress.map(task => `- ${task}`))
+  section('Decisions', [omitted(notes.decisions.length - decisions.length, 'decisions'), ...decisions.map(decision => `- ${decision}`)])
+  section('Files changed', [omitted(facts.filesChanged.length - files.length, 'files'), ...files.map(file => `- ${file}`)])
+  return lines.join('\n')
 }
 
 export function isLedgerText(value: string): boolean {

package/plugin/src/core/state.ts

@@ -20,8 +20,24 @@ export type SessionState = {
   /** 各种信号出现过的次数，用于确认识别是否生效 */
   seen: SignalsSeen
   usage: UsageTally
+  /** 读文件的统计；平台认不出读文件时没有这一项 */
+  reads?: ReadStats
 }
 
+/** 用来看压缩后要花多少次读取重新熟悉代码。 */
+export type ReadStats = {
+  /** 上次压缩之前读过或改过的文件 */
+  known: string[]
+  /** 上次压缩以来读过的文件 */
+  current: string[]
+  /** 上次压缩以来读文件的次数 */
+  reads: number
+  /** 其中读的是压缩前已经看过的文件 */
+  rereads: number
+}
+
+const MAX_KNOWN_FILES = 300
+
 export const INITIAL_SESSION_STATE: SessionState = Object.freeze({
   work: INITIAL_WORK_STATE,
   ledger: EMPTY_LEDGER,
@@ -73,8 +89,15 @@ export function withSample(state: SessionState, tokens: number, ceiling: number,
  * @param attributed 是否算作插件带来的压缩；用户手动 /compact 没有插件时也会发生，不计入节省
  */
 export function withCompaction(state: SessionState, before: number, summary: number, attributed = true): SessionState {
+  const reads = state.reads && {
+    known: [...new Set([...state.reads.known, ...state.reads.current, ...state.ledger.facts.filesChanged])].slice(-MAX_KNOWN_FILES),
+    current: [],
+    reads: 0,
+    rereads: 0,
+  }
   return {
     ...state,
+    ...(reads && { reads }),
     work: afterCompaction(state.work),
     compactions: state.compactions + 1,
     notesUpdated: false,
@@ -83,6 +106,25 @@ export function withCompaction(state: SessionState, before: number, summary: num
   }
 }
 
+/** 记一次读文件：读的是上次压缩前看过的文件就算重读。 */
+export function withRead(state: SessionState, path: string): SessionState {
+  const stats = state.reads ?? { known: [], current: [], reads: 0, rereads: 0 }
+  return {
+    ...state,
+    reads: {
+      known: stats.known,
+      current: [...stats.current.filter(file => file !== path), path].slice(-MAX_KNOWN_FILES),
+      reads: stats.reads + 1,
+      rereads: stats.rereads + (stats.known.includes(path) ? 1 : 0),
+    },
+  }
+}
+
+/** 压缩时记进日志的本周期读文件次数；平台认不出读文件时为空。 */
+export function cycleReads(state: SessionState): Pick<DecisionLogEntry, 'reads' | 'rereads'> {
+  return state.reads ? { reads: state.reads.reads, rereads: state.reads.rereads } : {}
+}
+
 export type DecisionLogEntry = {
   at: string
   /** 会话 ID，报告据此只统计当前会话 */
@@ -95,6 +137,9 @@ export type DecisionLogEntry = {
   tokensAfter?: number
   /** 决定压缩时，距离最近的阶段节点上下文涨了多少 */
   sinceMilestoneTokens?: number
+  /** 压缩时：上次压缩以来读文件的次数，以及其中重读压缩前看过的文件的次数 */
+  reads?: number
+  rereads?: number
   trigger?: string
   detail?: string
 }