compressed-context 0.1.0 → 0.2.0

package/src/compression/select.ts

@@ -1,123 +1,123 @@
-import { stringify, toolInput, toolResult, toolResultText } from "../shared/parts.ts"
-
-/** 媒体 source 的最小结构；bytes/base64 可落盘，url/ref 只保留引用。 */
-export interface MediaSourceLike {
-  readonly type: string
-  readonly data?: string | Uint8Array | undefined
-  readonly mediaType?: string | undefined
-  readonly url?: string | undefined
-  readonly provider?: string | undefined
-  readonly id?: string | undefined
-  readonly expiresAt?: number | undefined
-}
-
-/** 单条内容 part 的最小结构，避免绑定 @opencode/ai 的具体类。 */
-export interface ContentPartLike {
-  readonly type: string
-  readonly text?: string | null | undefined
-  readonly name?: string | undefined
-  readonly input?: unknown
-  readonly result?: unknown
-  readonly state?:
-    | {
-        readonly input?: unknown
-        readonly content?: unknown
-        readonly output?: unknown
-      }
-    | undefined
-  readonly filename?: string | undefined
-  readonly media?:
-    | {
-        readonly source?: MediaSourceLike | undefined
-        readonly mediaType?: string | undefined
-        readonly info?:
-          | {
-              readonly width?: number | undefined
-              readonly height?: number | undefined
-              readonly durationSeconds?: number | undefined
-            }
-          | undefined
-        readonly providerMetadata?: unknown
-      }
-    | undefined
-}
-
-/** 单条消息的最小结构。 */
-export interface MessageLike {
-  readonly id?: string | undefined
-  readonly role: string
-  readonly content: readonly ContentPartLike[]
-  /** v2 用户消息把正文放在 text，而不是 content 数组。 */
-  readonly text?: string | null | undefined
-}
-
-export const CHECKPOINT_PREFIX = "<conversation-checkpoint>"
-
-/** 输入：消息；输出：纯文本（只拼 text part）。 */
-export function messageText(message: MessageLike): string {
-  const direct = typeof message.text === "string" ? message.text : ""
-  const content: readonly ContentPartLike[] = Array.isArray(message.content)
-    ? message.content
-    : []
-  const parts = content
-    .filter((part) => part.type === "text")
-    .map((part) => part.text ?? "")
-    .join("\n")
-  return [direct, parts].filter((text) => text.length > 0).join("\n")
-}
-
-/** 输入：消息；输出：是否是 checkpoint 消息。 */
-export function isCheckpoint(message: MessageLike): boolean {
-  return messageText(message).trimStart().startsWith(CHECKPOINT_PREFIX)
-}
-
-/**
- * 输入：当前请求的消息列表。
- * 输出：可压缩的消息（最后一条 user 之前，排除 checkpoint）。
- * 预期：没有 user 消息时返回空数组。
- */
-export function eligibleMessages(messages: readonly MessageLike[]): readonly MessageLike[] {
-  const lastUser = messages.findLastIndex((message) => message.role === "user")
-  if (lastUser <= 0) return []
-  return messages.slice(0, lastUser).filter((message) => !isCheckpoint(message))
-}
-
-/**
- * 输入：compaction 钩子的 older 消息。
- * 输出：可压缩消息（排除 checkpoint）。
- * 预期：older 里没有“当前 user”可排除，不能复用 eligibleMessages。
- */
-export function olderMessages(messages: readonly MessageLike[]): readonly MessageLike[] {
-  return messages.filter((message) => !isCheckpoint(message))
-}
-
-/** 输入：消息列表；输出：首尾带 id 的消息 id；没有可用 id 时返回 undefined。 */
-export function messageRange(
-  messages: readonly MessageLike[],
-): { readonly fromId: string; readonly toId: string } | undefined {
-  const first = messages.find((message) => message.id !== undefined)
-  const last = [...messages].reverse().find((message) => message.id !== undefined)
-  if (first?.id === undefined || last?.id === undefined) return undefined
-  return { fromId: first.id, toId: last.id }
-}
-
-/** 输入：消息列表；输出：所有 text part 的总字符数。 */
-export function textChars(messages: readonly MessageLike[]): number {
-  let total = 0
-  for (const message of messages) {
-    if (typeof message.text === "string") total += message.text.length
-    const content: readonly ContentPartLike[] = Array.isArray(message.content)
-      ? message.content
-      : []
-    for (const part of content) {
-      if (part.type === "text") {
-        total += (part.text ?? "").length
-        continue
-      }
-      if (part.type === "tool" || part.type === "tool-call" || part.type === "tool-result") {
-        total += `${stringify(toolInput(part))}${toolResultText(toolResult(part))}`.length
-      }
-    }
-  }
-  return total
-}
+import { stringify, toolInput, toolResult, toolResultText } from "../shared/parts.ts"
+
+/** 媒体 source 的最小结构；bytes/base64 可落盘，url/ref 只保留引用。 */
+export interface MediaSourceLike {
+  readonly type: string
+  readonly data?: string | Uint8Array | undefined
+  readonly mediaType?: string | undefined
+  readonly url?: string | undefined
+  readonly provider?: string | undefined
+  readonly id?: string | undefined
+  readonly expiresAt?: number | undefined
+}
+
+/** 单条内容 part 的最小结构，避免绑定 @opencode/ai 的具体类。 */
+export interface ContentPartLike {
+  readonly type: string
+  readonly text?: string | null | undefined
+  readonly name?: string | undefined
+  readonly input?: unknown
+  readonly result?: unknown
+  readonly state?:
+    | {
+        readonly input?: unknown
+        readonly content?: unknown
+        readonly output?: unknown
+      }
+    | undefined
+  readonly filename?: string | undefined
+  readonly media?:
+    | {
+        readonly source?: MediaSourceLike | undefined
+        readonly mediaType?: string | undefined
+        readonly info?:
+          | {
+              readonly width?: number | undefined
+              readonly height?: number | undefined
+              readonly durationSeconds?: number | undefined
+            }
+          | undefined
+        readonly providerMetadata?: unknown
+      }
+    | undefined
+}
+
+/** 单条消息的最小结构。 */
+export interface MessageLike {
+  readonly id?: string | undefined
+  readonly role: string
+  readonly content: readonly ContentPartLike[]
+  /** v2 用户消息把正文放在 text，而不是 content 数组。 */
+  readonly text?: string | null | undefined
+}
+
+export const CHECKPOINT_PREFIX = "<conversation-checkpoint>"
+
+/** 输入：消息；输出：纯文本（只拼 text part）。 */
+export function messageText(message: MessageLike): string {
+  const direct = typeof message.text === "string" ? message.text : ""
+  const content: readonly ContentPartLike[] = Array.isArray(message.content)
+    ? message.content
+    : []
+  const parts = content
+    .filter((part) => part.type === "text")
+    .map((part) => part.text ?? "")
+    .join("\n")
+  return [direct, parts].filter((text) => text.length > 0).join("\n")
+}
+
+/** 输入：消息；输出：是否是 checkpoint 消息。 */
+export function isCheckpoint(message: MessageLike): boolean {
+  return messageText(message).trimStart().startsWith(CHECKPOINT_PREFIX)
+}
+
+/**
+ * 输入：当前请求的消息列表。
+ * 输出：可压缩的消息（最后一条 user 之前，排除 checkpoint）。
+ * 预期：没有 user 消息时返回空数组。
+ */
+export function eligibleMessages(messages: readonly MessageLike[]): readonly MessageLike[] {
+  const lastUser = messages.findLastIndex((message) => message.role === "user")
+  if (lastUser <= 0) return []
+  return messages.slice(0, lastUser).filter((message) => !isCheckpoint(message))
+}
+
+/**
+ * 输入：compaction 钩子的 older 消息。
+ * 输出：可压缩消息（排除 checkpoint）。
+ * 预期：older 里没有“当前 user”可排除，不能复用 eligibleMessages。
+ */
+export function olderMessages(messages: readonly MessageLike[]): readonly MessageLike[] {
+  return messages.filter((message) => !isCheckpoint(message))
+}
+
+/** 输入：消息列表；输出：首尾带 id 的消息 id；没有可用 id 时返回 undefined。 */
+export function messageRange(
+  messages: readonly MessageLike[],
+): { readonly fromId: string; readonly toId: string } | undefined {
+  const first = messages.find((message) => message.id !== undefined)
+  const last = [...messages].reverse().find((message) => message.id !== undefined)
+  if (first?.id === undefined || last?.id === undefined) return undefined
+  return { fromId: first.id, toId: last.id }
+}
+
+/** 输入：消息列表；输出：所有 text part 的总字符数。 */
+export function textChars(messages: readonly MessageLike[]): number {
+  let total = 0
+  for (const message of messages) {
+    if (typeof message.text === "string") total += message.text.length
+    const content: readonly ContentPartLike[] = Array.isArray(message.content)
+      ? message.content
+      : []
+    for (const part of content) {
+      if (part.type === "text") {
+        total += (part.text ?? "").length
+        continue
+      }
+      if (part.type === "tool" || part.type === "tool-call" || part.type === "tool-result") {
+        total += `${stringify(toolInput(part))}${toolResultText(toolResult(part))}`.length
+      }
+    }
+  }
+  return total
+}

package/src/compression/session-message.ts

@@ -1,112 +1,112 @@
-import type { ContentPartLike, MediaSourceLike, MessageLike } from "./select.ts"
-
-/** 输入：任意值；输出：对象记录或 undefined。 */
-function asRecord(value: unknown): Record<string, unknown> | undefined {
-  return typeof value === "object" && value !== null ? (value as Record<string, unknown>) : undefined
-}
-
-/** 输入：user.files 项；输出：media part；形状不合法时 undefined。 */
-function userFileToPart(value: unknown): ContentPartLike | undefined {
-  const file = asRecord(value)
-  if (file === undefined || typeof file["mime"] !== "string") return undefined
-  const source = asRecord(file["source"])
-  const data = typeof file["data"] === "string" ? file["data"] : undefined
-  const uri = typeof source?.["uri"] === "string" ? source["uri"] : undefined
-  const sourceKind = typeof source?.["type"] === "string" ? source["type"] : "inline"
-  const media: MediaSourceLike =
-    sourceKind === "uri" && uri !== undefined
-      ? { type: "url", url: uri, mediaType: file["mime"] }
-      : { type: "base64", ...(data === undefined ? {} : { data }), mediaType: file["mime"] }
-  return {
-    type: "media",
-    ...(typeof file["name"] === "string" ? { filename: file["name"] } : {}),
-    media: { source: media },
-  }
-}
-
-/** 输入：assistant.content 项；输出：可读 parts；reasoning 跳过。 */
-function assistantPartToParts(value: unknown): readonly ContentPartLike[] {
-  const part = asRecord(value)
-  if (part === undefined) return []
-  if (part["type"] === "text" && typeof part["text"] === "string") {
-    return [{ type: "text", text: part["text"] }]
-  }
-  if (part["type"] === "tool" && typeof part["name"] === "string") {
-    const state = asRecord(part["state"])
-    const parts: ContentPartLike[] = [
-      {
-        type: "tool-call",
-        name: part["name"],
-        input: state?.["input"],
-      },
-    ]
-    if (state?.["status"] === "completed") {
-      parts.push({
-        type: "tool-result",
-        name: part["name"],
-        result: state["content"] ?? state["input"],
-      })
-    }
-    return parts
-  }
-  if (part["type"] === "media") {
-    const media = asRecord(part["media"])
-    const source = asRecord(media?.["source"])
-    if (source === undefined || typeof source["type"] !== "string") return []
-    return [
-      {
-        type: "media",
-        ...(typeof part["filename"] === "string" ? { filename: part["filename"] } : {}),
-        media: {
-          source: {
-            type: source["type"],
-            ...(typeof source["data"] === "string" ? { data: source["data"] } : {}),
-            ...(source["data"] instanceof Uint8Array ? { data: source["data"] } : {}),
-            ...(typeof source["mediaType"] === "string" ? { mediaType: source["mediaType"] } : {}),
-            ...(typeof source["url"] === "string" ? { url: source["url"] } : {}),
-            ...(typeof source["provider"] === "string" ? { provider: source["provider"] } : {}),
-            ...(typeof source["id"] === "string" ? { id: source["id"] } : {}),
-            ...(typeof source["expiresAt"] === "number" ? { expiresAt: source["expiresAt"] } : {}),
-          },
-        },
-      },
-    ]
-  }
-  return []
-}
-
-/**
- * 输入：ctx.session.context 返回的持久化 session 消息。
- * 输出：MessageLike 列表；user.text 转 text，user.files 转 media，assistant.content 转 parts。
- * 预期：agent/model/compaction 等非对话项跳过；reasoning 不进入压缩 transcript。
- */
-export function sessionMessagesToMessageLikes(messages: readonly unknown[]): readonly MessageLike[] {
-  const out: MessageLike[] = []
-  for (const item of messages) {
-    const record = asRecord(item)
-    if (record === undefined) continue
-    const id = typeof record["id"] === "string" ? record["id"] : undefined
-    if (record["type"] === "user") {
-      const content: ContentPartLike[] = []
-      if (typeof record["text"] === "string" && record["text"].length > 0) {
-        content.push({ type: "text", text: record["text"] })
-      }
-      if (Array.isArray(record["files"])) {
-        for (const file of record["files"]) {
-          const part = userFileToPart(file)
-          if (part !== undefined) content.push(part)
-        }
-      }
-      out.push({ ...(id === undefined ? {} : { id }), role: "user", content })
-      continue
-    }
-    if (record["type"] === "assistant" && Array.isArray(record["content"])) {
-      const content: ContentPartLike[] = []
-      for (const part of record["content"]) {
-        content.push(...assistantPartToParts(part))
-      }
-      out.push({ ...(id === undefined ? {} : { id }), role: "assistant", content })
-    }
-  }
-  return out
-}
+import type { ContentPartLike, MediaSourceLike, MessageLike } from "./select.ts"
+
+/** 输入：任意值；输出：对象记录或 undefined。 */
+function asRecord(value: unknown): Record<string, unknown> | undefined {
+  return typeof value === "object" && value !== null ? (value as Record<string, unknown>) : undefined
+}
+
+/** 输入：user.files 项；输出：media part；形状不合法时 undefined。 */
+function userFileToPart(value: unknown): ContentPartLike | undefined {
+  const file = asRecord(value)
+  if (file === undefined || typeof file["mime"] !== "string") return undefined
+  const source = asRecord(file["source"])
+  const data = typeof file["data"] === "string" ? file["data"] : undefined
+  const uri = typeof source?.["uri"] === "string" ? source["uri"] : undefined
+  const sourceKind = typeof source?.["type"] === "string" ? source["type"] : "inline"
+  const media: MediaSourceLike =
+    sourceKind === "uri" && uri !== undefined
+      ? { type: "url", url: uri, mediaType: file["mime"] }
+      : { type: "base64", ...(data === undefined ? {} : { data }), mediaType: file["mime"] }
+  return {
+    type: "media",
+    ...(typeof file["name"] === "string" ? { filename: file["name"] } : {}),
+    media: { source: media },
+  }
+}
+
+/** 输入：assistant.content 项；输出：可读 parts；reasoning 跳过。 */
+function assistantPartToParts(value: unknown): readonly ContentPartLike[] {
+  const part = asRecord(value)
+  if (part === undefined) return []
+  if (part["type"] === "text" && typeof part["text"] === "string") {
+    return [{ type: "text", text: part["text"] }]
+  }
+  if (part["type"] === "tool" && typeof part["name"] === "string") {
+    const state = asRecord(part["state"])
+    const parts: ContentPartLike[] = [
+      {
+        type: "tool-call",
+        name: part["name"],
+        input: state?.["input"],
+      },
+    ]
+    if (state?.["status"] === "completed") {
+      parts.push({
+        type: "tool-result",
+        name: part["name"],
+        result: state["content"] ?? state["input"],
+      })
+    }
+    return parts
+  }
+  if (part["type"] === "media") {
+    const media = asRecord(part["media"])
+    const source = asRecord(media?.["source"])
+    if (source === undefined || typeof source["type"] !== "string") return []
+    return [
+      {
+        type: "media",
+        ...(typeof part["filename"] === "string" ? { filename: part["filename"] } : {}),
+        media: {
+          source: {
+            type: source["type"],
+            ...(typeof source["data"] === "string" ? { data: source["data"] } : {}),
+            ...(source["data"] instanceof Uint8Array ? { data: source["data"] } : {}),
+            ...(typeof source["mediaType"] === "string" ? { mediaType: source["mediaType"] } : {}),
+            ...(typeof source["url"] === "string" ? { url: source["url"] } : {}),
+            ...(typeof source["provider"] === "string" ? { provider: source["provider"] } : {}),
+            ...(typeof source["id"] === "string" ? { id: source["id"] } : {}),
+            ...(typeof source["expiresAt"] === "number" ? { expiresAt: source["expiresAt"] } : {}),
+          },
+        },
+      },
+    ]
+  }
+  return []
+}
+
+/**
+ * 输入：ctx.session.context 返回的持久化 session 消息。
+ * 输出：MessageLike 列表；user.text 转 text，user.files 转 media，assistant.content 转 parts。
+ * 预期：agent/model/compaction 等非对话项跳过；reasoning 不进入压缩 transcript。
+ */
+export function sessionMessagesToMessageLikes(messages: readonly unknown[]): readonly MessageLike[] {
+  const out: MessageLike[] = []
+  for (const item of messages) {
+    const record = asRecord(item)
+    if (record === undefined) continue
+    const id = typeof record["id"] === "string" ? record["id"] : undefined
+    if (record["type"] === "user") {
+      const content: ContentPartLike[] = []
+      if (typeof record["text"] === "string" && record["text"].length > 0) {
+        content.push({ type: "text", text: record["text"] })
+      }
+      if (Array.isArray(record["files"])) {
+        for (const file of record["files"]) {
+          const part = userFileToPart(file)
+          if (part !== undefined) content.push(part)
+        }
+      }
+      out.push({ ...(id === undefined ? {} : { id }), role: "user", content })
+      continue
+    }
+    if (record["type"] === "assistant" && Array.isArray(record["content"])) {
+      const content: ContentPartLike[] = []
+      for (const part of record["content"]) {
+        content.push(...assistantPartToParts(part))
+      }
+      out.push({ ...(id === undefined ? {} : { id }), role: "assistant", content })
+    }
+  }
+  return out
+}