opencode-lcm 0.13.2 → 0.13.5

package/src/store.ts

@@ -72,6 +72,7 @@ import type {
   ScopeName,
   SearchResult,
   StoreStats,
+  SummaryStrategyName,
 } from './types.js';
 import {
   asRecord,
@@ -101,6 +102,7 @@ type SummaryNodeData = {
   endIndex: number;
   messageIDs: string[];
   summaryText: string;
+  strategy: SummaryStrategyName;
   createdAt: number;
 };
 
@@ -379,6 +381,7 @@ function filterValidConversationMessages(
   messages: ConversationMessage[],
   context?: MessageValidationContext,
 ): ConversationMessage[] {
+  if (context?.operation === 'transformMessages') return messages;
   const valid = messages.filter((message) => Boolean(getValidMessageInfo(message?.info)));
   const dropped = messages.length - valid.length;
   if (dropped > 0 && context) {
@@ -416,14 +419,21 @@ function getDeferredPartUpdateKey(event: Event): string | undefined {
   return `${event.properties.part.sessionID}:${event.properties.part.messageID}:${event.properties.part.id}`;
 }
 
-function compareMessages(a: ConversationMessage, b: ConversationMessage): number {
-  const aInfo = getValidMessageInfo(a.info);
-  const bInfo = getValidMessageInfo(b.info);
-  if (!aInfo && !bInfo) return 0;
-  if (!aInfo) return 1;
-  if (!bInfo) return -1;
+function messageCreatedAt(message: ConversationMessage | undefined): number {
+  const created = message?.info?.time?.created;
+  return typeof created === 'number' && Number.isFinite(created) ? created : 0;
+}
+
+function messageParts(message: ConversationMessage | undefined): Part[] {
+  return Array.isArray(message?.parts) ? message.parts : [];
+}
 
-  return aInfo.time.created - bInfo.time.created || aInfo.id.localeCompare(bInfo.id);
+function signatureString(value: unknown, fallback = ''): string {
+  return typeof value === 'string' ? value : fallback;
+}
+
+function compareMessages(a: ConversationMessage, b: ConversationMessage): number {
+  return messageCreatedAt(a) - messageCreatedAt(b);
 }
 
 function emptySession(sessionID: string): NormalizedSession {
@@ -501,19 +511,21 @@ function isSyntheticLcmTextPart(part: Part, markers?: string[]): boolean {
 function guessMessageText(message: ConversationMessage, ignoreToolPrefixes: string[]): string {
   const segments: string[] = [];
 
-  for (const part of message.parts) {
+  for (const part of messageParts(message)) {
     switch (part.type) {
       case 'text': {
         if (isSyntheticLcmTextPart(part, ['archive-summary', 'retrieved-context', 'archived-part']))
           break;
-        if (part.text.startsWith('[Archived by opencode-lcm:')) break;
-        const sanitized = sanitizeAutomaticRetrievalSourceText(part.text);
+        const text = typeof part.text === 'string' ? part.text : '';
+        if (text.startsWith('[Archived by opencode-lcm:')) break;
+        const sanitized = sanitizeAutomaticRetrievalSourceText(text);
         if (sanitized) segments.push(sanitized);
         break;
       }
       case 'reasoning': {
-        if (part.text.startsWith('[Archived by opencode-lcm:')) break;
-        const sanitized = sanitizeAutomaticRetrievalSourceText(part.text);
+        const text = typeof part.text === 'string' ? part.text : '';
+        if (text.startsWith('[Archived by opencode-lcm:')) break;
+        const sanitized = sanitizeAutomaticRetrievalSourceText(text);
         if (sanitized) segments.push(sanitized);
         break;
       }
@@ -525,12 +537,13 @@ function guessMessageText(message: ConversationMessage, ignoreToolPrefixes: stri
         break;
       }
       case 'tool': {
-        if (ignoreToolPrefixes.some((prefix) => part.tool.startsWith(prefix))) break;
+        const toolName = typeof part.tool === 'string' ? part.tool : '';
+        if (ignoreToolPrefixes.some((prefix) => toolName.startsWith(prefix))) break;
         const state = part.state;
-        if (state.status === 'completed') segments.push(`${part.tool}: ${state.output}`);
-        if (state.status === 'error') segments.push(`${part.tool}: ${state.error}`);
+        if (state.status === 'completed') segments.push(`${toolName}: ${state.output}`);
+        if (state.status === 'error') segments.push(`${toolName}: ${state.error}`);
         if (state.status === 'pending' || state.status === 'running') {
-          segments.push(`${part.tool}: ${JSON.stringify(state.input)}`);
+          segments.push(`${toolName}: ${JSON.stringify(state.input)}`);
         }
         if (state.status === 'completed' && state.attachments && state.attachments.length > 0) {
           const attachmentNames = state.attachments
@@ -538,7 +551,7 @@ function guessMessageText(message: ConversationMessage, ignoreToolPrefixes: stri
             .filter(Boolean)
             .slice(0, 4);
           if (attachmentNames.length > 0)
-            segments.push(`${part.tool} attachments: ${attachmentNames.join(', ')}`);
+            segments.push(`${toolName} attachments: ${attachmentNames.join(', ')}`);
         }
         break;
       }
@@ -613,6 +626,9 @@ type CaptureHydrationOptions = {
   isBunRuntime?: boolean;
   platform?: string | undefined;
 };
+type ReadMessageOptions = {
+  hydrateArtifacts?: boolean;
+};
 
 function normalizeSqliteRuntimeOverride(value: string | undefined): SqliteRuntime | 'auto' {
   const normalized = value?.trim().toLowerCase();
@@ -652,6 +668,17 @@ export function resolveCaptureHydrationMode(
   return isBunRuntime && platform === 'win32' ? 'full' : 'targeted';
 }
 
+function shouldUseLightweightPartCapture(
+  event: CapturedEvent,
+  options?: CaptureHydrationOptions,
+): boolean {
+  const isBunRuntime =
+    options?.isBunRuntime ?? (typeof globalThis === 'object' && 'Bun' in globalThis);
+  const platform = options?.platform ?? process.platform;
+  if (!isBunRuntime || platform !== 'win32') return false;
+  return event.type === 'message.part.updated' || event.type === 'message.part.removed';
+}
+
 function isSqliteRuntimeImportError(runtime: SqliteRuntime, error: unknown): boolean {
   const code =
     typeof error === 'object' && error && 'code' in error && typeof error.code === 'string'
@@ -1051,6 +1078,7 @@ export class SqliteLcmStore {
         end_index INTEGER NOT NULL,
         message_ids_json TEXT NOT NULL,
         summary_text TEXT NOT NULL,
+        strategy TEXT NOT NULL DEFAULT 'deterministic-v1',
         created_at INTEGER NOT NULL,
         FOREIGN KEY (session_id) REFERENCES sessions(session_id) ON DELETE CASCADE
       );
@@ -1107,6 +1135,7 @@ export class SqliteLcmStore {
 
       this.ensureSessionColumnsSync();
       this.ensureSummaryStateColumnsSync();
+      this.ensureSummaryNodeColumnsSync();
       this.ensureArtifactColumnsSync();
       logStartupPhase('open-db:create-indexes');
       db.exec('CREATE INDEX IF NOT EXISTS idx_artifacts_content_hash ON artifacts(content_hash)');
@@ -1281,8 +1310,9 @@ export class SqliteLcmStore {
 
       if (!normalized.sessionID || !shouldPersistSession) return;
 
-      const session =
-        resolveCaptureHydrationMode() === 'targeted'
+      const session = shouldUseLightweightPartCapture(normalized)
+        ? this.readSessionForCaptureSync(normalized, { hydrateArtifacts: false })
+        : resolveCaptureHydrationMode() === 'targeted'
           ? this.readSessionForCaptureSync(normalized)
           : this.readSessionSync(normalized.sessionID);
       const previousParentSessionID = session.parentSessionID;
@@ -1695,7 +1725,7 @@ export class SqliteLcmStore {
       return issues.length > 0 ? { sessionID: session.sessionID, issues } : undefined;
     }
 
-    const latestMessageCreated = archived.at(-1)?.info.time.created ?? 0;
+    const latestMessageCreated = messageCreatedAt(archived.at(-1));
     const archivedSignature = this.buildArchivedSignature(archived);
     const rootIDs = state ? parseJson<string[]>(state.root_node_ids_json) : [];
     const roots = rootIDs
@@ -1848,7 +1878,7 @@ export class SqliteLcmStore {
           return this.isMessageArchivedSync(
             session.sessionID,
             existing.info.id,
-            existing.info.time.created,
+            messageCreatedAt(existing),
           );
         }
 
@@ -1873,7 +1903,7 @@ export class SqliteLcmStore {
         return this.isMessageArchivedSync(
           session.sessionID,
           message.info.id,
-          message.info.time.created,
+          messageCreatedAt(message),
         );
       }
       case 'message.part.removed': {
@@ -1884,7 +1914,7 @@ export class SqliteLcmStore {
         return this.isMessageArchivedSync(
           session.sessionID,
           message.info.id,
-          message.info.time.created,
+          messageCreatedAt(message),
         );
       }
       default:
@@ -3573,6 +3603,16 @@ export class SqliteLcmStore {
   private summarizeMessages(
     messages: ConversationMessage[],
     limit = SUMMARY_NODE_CHAR_LIMIT,
+  ): string {
+    const strategy = this.options.summaryV2?.strategy ?? 'deterministic-v1';
+    return strategy === 'deterministic-v2'
+      ? this.summarizeMessagesDeterministicV2(messages, limit)
+      : this.summarizeMessagesDeterministicV1(messages, limit);
+  }
+
+  private summarizeMessagesDeterministicV1(
+    messages: ConversationMessage[],
+    limit = SUMMARY_NODE_CHAR_LIMIT,
   ): string {
     const goals = messages
       .filter((message) => message.info.role === 'user')
@@ -3602,6 +3642,71 @@ export class SqliteLcmStore {
     return truncate(segments.join(' || '), limit);
   }
 
+  private summarizeMessagesDeterministicV2(
+    messages: ConversationMessage[],
+    limit = SUMMARY_NODE_CHAR_LIMIT,
+  ): string {
+    const perMsgBudget = this.options.summaryV2?.perMessageBudget ?? 110;
+    const ignoreToolPrefixes = this.options.interop.ignoreToolPrefixes;
+    const userTexts = messages
+      .filter((message) => message.info.role === 'user')
+      .map((message) => guessMessageText(message, ignoreToolPrefixes))
+      .filter(Boolean);
+    const assistantTexts = messages
+      .filter((message) => message.info.role === 'assistant')
+      .map((message) => guessMessageText(message, ignoreToolPrefixes))
+      .filter(Boolean);
+    const allFiles = [...new Set(messages.flatMap(listFiles))];
+    const allTools = [...new Set(this.listTools(messages))];
+    const hasErrors = messages.some((message) =>
+      message.parts.some(
+        (part) =>
+          (part.type === 'tool' && 'state' in part && part.state?.status === 'error') ||
+          (part.type === 'text' &&
+            /\b(?:error|exception|fail(?:ed|ure)?)\b/i.test(part.text ?? '')),
+      ),
+    );
+
+    const segments: string[] = [];
+    if (userTexts.length > 0) {
+      const first = truncate(userTexts[0], perMsgBudget);
+      if (userTexts.length > 1) {
+        const last = truncate(userTexts[userTexts.length - 1], perMsgBudget);
+        segments.push(`Goals: ${first} → ${last}`);
+      } else {
+        segments.push(`Goals: ${first}`);
+      }
+    }
+
+    if (assistantTexts.length > 0) {
+      const recent = assistantTexts
+        .slice(-2)
+        .map((text) => truncate(text, perMsgBudget))
+        .join(' | ');
+      segments.push(`Work: ${recent}`);
+    }
+
+    if (allFiles.length > 0) {
+      const shown = allFiles.slice(0, 6).join(', ');
+      segments.push(
+        allFiles.length > 6 ? `Files[${allFiles.length}]: ${shown}` : `Files: ${shown}`,
+      );
+    }
+
+    if (allTools.length > 0) {
+      const shown = allTools.slice(0, 6).join(', ');
+      segments.push(
+        allTools.length > 6 ? `Tools[${allTools.length}]: ${shown}` : `Tools: ${shown}`,
+      );
+    }
+
+    if (hasErrors) segments.push('⚠err');
+    segments.push(`${messages.length}msg(u:${userTexts.length}/a:${assistantTexts.length})`);
+
+    if (segments.length === 0) return truncate(`Archived ${messages.length} messages`, limit);
+    return truncate(segments.join(' || '), limit);
+  }
+
   private listTools(messages: ConversationMessage[]): string[] {
     const tools: string[] = [];
     for (const message of messages) {
@@ -3617,13 +3722,13 @@ export class SqliteLcmStore {
   private buildArchivedSignature(messages: ConversationMessage[]): string {
     const hash = createHash('sha256');
     for (const message of messages) {
-      hash.update(message.info.id);
-      hash.update(message.info.role);
-      hash.update(String(message.info.time.created));
-      hash.update(guessMessageText(message, this.options.interop.ignoreToolPrefixes));
+      hash.update(signatureString(message.info?.id, 'unknown-message'));
+      hash.update(signatureString(message.info?.role, 'unknown-role'));
+      hash.update(String(messageCreatedAt(message)));
+      hash.update(String(guessMessageText(message, this.options.interop.ignoreToolPrefixes) ?? ''));
       hash.update(JSON.stringify(listFiles(message)));
       hash.update(JSON.stringify(this.listTools([message])));
-      hash.update(String(message.parts.length));
+      hash.update(String(messageParts(message).length));
     }
     return hash.digest('hex');
   }
@@ -3650,7 +3755,7 @@ export class SqliteLcmStore {
       return [];
     }
 
-    const latestMessageCreated = archivedMessages.at(-1)?.info.time.created ?? 0;
+    const latestMessageCreated = messageCreatedAt(archivedMessages.at(-1));
     const archivedSignature = this.buildArchivedSignature(archivedMessages);
     const state = safeQueryOne<SummaryStateRow>(
       this.getDb().prepare('SELECT * FROM summary_state WHERE session_id = ?'),
@@ -3714,6 +3819,7 @@ export class SqliteLcmStore {
         archivedMessages.slice(node.startIndex, node.endIndex + 1),
       );
       if (node.summaryText !== expectedSummaryText) return false;
+      if (node.strategy !== (this.options.summaryV2?.strategy ?? 'deterministic-v1')) return false;
 
       const children = this.readSummaryChildrenSync(node.nodeID);
       if (node.nodeKind === 'leaf') {
@@ -3753,6 +3859,7 @@ export class SqliteLcmStore {
     archivedSignature: string,
   ): SummaryNodeData[] {
     const now = Date.now();
+    const summaryStrategy = this.options.summaryV2?.strategy ?? 'deterministic-v1';
     let level = 0;
     const nodes: SummaryNodeData[] = [];
     const edges: Array<{
@@ -3779,6 +3886,7 @@ export class SqliteLcmStore {
       endIndex: input.endIndex,
       messageIDs: input.messageIDs,
       summaryText: input.summaryText,
+      strategy: summaryStrategy,
       createdAt: now,
     });
 
@@ -3842,8 +3950,8 @@ export class SqliteLcmStore {
 
       const insertNode = db.prepare(
         `INSERT INTO summary_nodes
-         (node_id, session_id, level, node_kind, start_index, end_index, message_ids_json, summary_text, created_at)
-         VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)`,
+         (node_id, session_id, level, node_kind, start_index, end_index, message_ids_json, summary_text, strategy, created_at)
+         VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
       );
       const insertEdge = db.prepare(
         `INSERT INTO summary_edges (session_id, parent_id, child_id, child_position)
@@ -3863,6 +3971,7 @@ export class SqliteLcmStore {
           node.endIndex,
           JSON.stringify(node.messageIDs),
           node.summaryText,
+          node.strategy,
           node.createdAt,
         );
         insertSummaryFts.run(
@@ -3890,7 +3999,7 @@ export class SqliteLcmStore {
       ).run(
         sessionID,
         archivedMessages.length,
-        archivedMessages.at(-1)?.info.time.created ?? 0,
+        messageCreatedAt(archivedMessages.at(-1)),
         archivedSignature,
         JSON.stringify(roots.map((node) => node.nodeID)),
         now,
@@ -3917,6 +4026,7 @@ export class SqliteLcmStore {
       endIndex: row.end_index,
       messageIDs: parseJson<string[]>(row.message_ids_json),
       summaryText: row.summary_text,
+      strategy: row.strategy,
       createdAt: row.created_at,
     };
   }
@@ -4355,6 +4465,17 @@ export class SqliteLcmStore {
     db.exec("ALTER TABLE summary_state ADD COLUMN archived_signature TEXT NOT NULL DEFAULT ''");
   }
 
+  private ensureSummaryNodeColumnsSync(): void {
+    const db = this.getDb();
+    const columns = db.prepare('PRAGMA table_info(summary_nodes)').all() as Array<{ name: string }>;
+    const names = new Set(columns.map((column) => column.name));
+    if (names.has('strategy')) return;
+
+    db.exec(
+      "ALTER TABLE summary_nodes ADD COLUMN strategy TEXT NOT NULL DEFAULT 'deterministic-v1'",
+    );
+  }
+
   private ensureArtifactColumnsSync(): void {
     const db = this.getDb();
     const columns = db.prepare('PRAGMA table_info(artifacts)').all() as Array<{ name: string }>;
@@ -4872,7 +4993,10 @@ export class SqliteLcmStore {
     return parentSessionID;
   }
 
-  private readSessionForCaptureSync(event: CapturedEvent): NormalizedSession {
+  private readSessionForCaptureSync(
+    event: CapturedEvent,
+    options?: ReadMessageOptions,
+  ): NormalizedSession {
     const sessionID = event.sessionID;
     if (!sessionID) return emptySession('');
 
@@ -4881,17 +5005,17 @@ export class SqliteLcmStore {
 
     switch (payload.type) {
       case 'message.updated': {
-        const message = this.readMessageSync(sessionID, payload.properties.info.id);
+        const message = this.readMessageSync(sessionID, payload.properties.info.id, options);
         if (message) session.messages = [message];
         return session;
       }
       case 'message.part.updated': {
-        const message = this.readMessageSync(sessionID, payload.properties.part.messageID);
+        const message = this.readMessageSync(sessionID, payload.properties.part.messageID, options);
         if (message) session.messages = [message];
         return session;
       }
       case 'message.part.removed': {
-        const message = this.readMessageSync(sessionID, payload.properties.messageID);
+        const message = this.readMessageSync(sessionID, payload.properties.messageID, options);
         if (message) session.messages = [message];
         return session;
       }
@@ -4900,7 +5024,11 @@ export class SqliteLcmStore {
     }
   }
 
-  private readMessageSync(sessionID: string, messageID: string): ConversationMessage | undefined {
+  private readMessageSync(
+    sessionID: string,
+    messageID: string,
+    options?: ReadMessageOptions,
+  ): ConversationMessage | undefined {
     const db = this.getDb();
     const row = safeQueryOne<MessageRow>(
       db.prepare('SELECT * FROM messages WHERE session_id = ? AND message_id = ?'),
@@ -4909,11 +5037,14 @@ export class SqliteLcmStore {
     );
     if (!row) return undefined;
 
+    const hydrateArtifacts = options?.hydrateArtifacts ?? true;
     const artifactsByPart = new Map<string, ArtifactData[]>();
-    for (const artifact of this.readArtifactsForMessageSync(messageID)) {
-      const list = artifactsByPart.get(artifact.partID) ?? [];
-      list.push(artifact);
-      artifactsByPart.set(artifact.partID, list);
+    if (hydrateArtifacts) {
+      for (const artifact of this.readArtifactsForMessageSync(messageID)) {
+        const list = artifactsByPart.get(artifact.partID) ?? [];
+        list.push(artifact);
+        artifactsByPart.set(artifact.partID, list);
+      }
     }
 
     const parts = db
@@ -4939,7 +5070,7 @@ export class SqliteLcmStore {
       info,
       parts: parts.map((partRow) => {
         const part = parseJson<Part>(partRow.part_json);
-        hydratePartFromArtifacts(part, artifactsByPart.get(part.id) ?? []);
+        if (hydrateArtifacts) hydratePartFromArtifacts(part, artifactsByPart.get(part.id) ?? []);
         return part;
       }),
     };
@@ -5002,15 +5133,19 @@ export class SqliteLcmStore {
         const message = session.messages.find(
           (entry) => entry.info.id === payload.properties.part.messageID,
         );
+        const preservedArtifacts = message
+          ? this.readArtifactsForMessageSync(message.info.id).filter(
+              (artifact) => artifact.partID !== payload.properties.part.id,
+            )
+          : [];
         const externalized = message ? await this.externalizeMessage(message) : undefined;
         withTransaction(this.getDb(), 'capture', () => {
           this.upsertSessionRowSync(session);
           if (externalized) {
-            this.replaceStoredMessageSync(
-              session.sessionID,
-              externalized.storedMessage,
-              externalized.artifacts,
-            );
+            this.replaceStoredMessageSync(session.sessionID, externalized.storedMessage, [
+              ...preservedArtifacts,
+              ...externalized.artifacts,
+            ]);
           }
         });
         return;
@@ -5019,15 +5154,19 @@ export class SqliteLcmStore {
         const message = session.messages.find(
           (entry) => entry.info.id === payload.properties.messageID,
         );
+        const preservedArtifacts = message
+          ? this.readArtifactsForMessageSync(message.info.id).filter(
+              (artifact) => artifact.partID !== payload.properties.partID,
+            )
+          : [];
         const externalized = message ? await this.externalizeMessage(message) : undefined;
         withTransaction(this.getDb(), 'capture', () => {
           this.upsertSessionRowSync(session);
           if (externalized) {
-            this.replaceStoredMessageSync(
-              session.sessionID,
-              externalized.storedMessage,
-              externalized.artifacts,
-            );
+            this.replaceStoredMessageSync(session.sessionID, externalized.storedMessage, [
+              ...preservedArtifacts,
+              ...externalized.artifacts,
+            ]);
           }
         });
         return;

package/src/types.ts

@@ -55,6 +55,13 @@ export type AutomaticRetrievalOptions = {
   stop: AutomaticRetrievalStopOptions;
 };
 
+export type SummaryStrategyName = 'deterministic-v1' | 'deterministic-v2';
+
+export type SummaryV2Options = {
+  strategy: SummaryStrategyName;
+  perMessageBudget: number;
+};
+
 export type OpencodeLcmOptions = {
   interop: InteropOptions;
   scopeDefaults: ScopeDefaults;
@@ -74,6 +81,7 @@ export type OpencodeLcmOptions = {
   artifactViewChars: number;
   binaryPreviewProviders: string[];
   previewBytePeek: number;
+  summaryV2: SummaryV2Options;
 };
 
 export type CapturedEvent = {