Fix type issues

Author: ameliakuang

src/browser.js

@@ -138,6 +138,7 @@ export function createMemoryBank(config = {}) {
             }, doc, options);
         },
         compact: () => compactor.compactAll(),
+        pruneExpired: () => compactor.pruneExpired(),
         storage: {
             read: (path) => backend.read(path),
             resolvePath: (path) => backend.resolvePath ? backend.resolvePath(path) : Promise.resolve(null),

src/internal/llm-client/openai.js

@@ -275,29 +275,39 @@ function getRetryDelay(attempt, retryAfterMs = null) {
     return Math.min(exponential + jitter, MAX_DELAY_MS);
 }
 
+/**
+ * @param {number} ms
+ * @param {AbortSignal | null} [signal]
+ * @returns {Promise<void>}
+ */
 function sleep(ms, signal = null) {
-    return new Promise((resolve, reject) => {
+    const abortSignal = signal;
+    return /** @type {Promise<void>} */ (new Promise((resolve, reject) => {
         const timeoutId = setTimeout(() => {
             cleanup();
             resolve();
         }, ms);
 
         const cleanup = () => {
             clearTimeout(timeoutId);
-            signal?.removeEventListener?.('abort', onAbort);
+            if (abortSignal) {
+                abortSignal.removeEventListener('abort', onAbort);
+            }
         };
         const onAbort = () => {
             cleanup();
             reject(createAbortError('OpenAI API request aborted.'));
         };
 
-        if (signal?.aborted) {
+        if (abortSignal?.aborted) {
             onAbort();
             return;
         }
 
-        signal?.addEventListener?.('abort', onAbort, { once: true });
-    });
+        if (abortSignal) {
+            abortSignal.addEventListener('abort', onAbort, { once: true });
+        }
+    }));
 }
 
 async function createHttpError(response, attempt = 1) {

src/internal/llm-client/tinfoil.js

@@ -30,7 +30,7 @@ let tinfoilModulePromise = null;
  * Load TinfoilAI class. If a pre-loaded module is provided (e.g. a browser
  * bundle), use it directly. Otherwise fall back to `import('tinfoil')`.
  *
- * @param {object} [providedModule] — pre-loaded tinfoil module (must export TinfoilAI)
+ * @param {object} [providedModule] pre-loaded tinfoil module (must export TinfoilAI)
  */
 async function loadTinfoilAI(providedModule) {
     if (providedModule) {

src/internal/toolLoop.js

@@ -8,6 +8,9 @@
  * uses non-streaming requests for reliable tool call parsing.
  */
 /** @import { ToolLoopOptions, ToolLoopResult, ChatCompletionResponse, ToolCall, LLMMessage } from '../types.js' */
+/**
+ * @typedef {Error & { isUserAbort?: boolean }} AbortableError
+ */
 
 const DEFAULT_MAX_ITERATIONS = 10;
 const DEFAULT_MAX_OUTPUT_TOKENS = 500;
@@ -233,7 +236,7 @@ export async function runAgenticToolLoop(options) {
 }
 
 function createAbortError() {
-    const error = new Error('Tool loop aborted.');
+    const error = /** @type {AbortableError} */ (new Error('Tool loop aborted.'));
     error.name = 'AbortError';
     error.isUserAbort = true;
     return error;

src/tools/compaction.js

@@ -103,8 +103,9 @@ class MemoryCompactor {
             if (toExpire.length === 0) continue;
 
             const expireLineIndexes = new Set(toExpire.map(b => b.lineIndex));
+            /** @type {import('../types.js').Bullet[]} */
             const marked = bullets.map(b =>
-                expireLineIndexes.has(b.lineIndex) ? { ...b, status: 'expired' } : b
+                expireLineIndexes.has(b.lineIndex) ? { ...b, status: /** @type {'expired'} */ ('expired') } : b
             );
 
             const defaultTopic = inferTopicFromPath(file.path);

src/tools/ingestion.js

@@ -5,6 +5,9 @@
  * agentic loop to decide whether to create/append/update memory files.
  */
 /** @import { IngestOptions, IngestResult, LLMClient, Message, StorageBackend, ToolDefinition } from '../types.js' */
+/**
+ * @typedef {Error & { isUserAbort?: boolean }} AbortableError
+ */
 import { runAgenticToolLoop } from '../internal/toolLoop.js';
 import { createExtractionExecutors } from './executors.js';
 import { resolvePromptSet } from '../prompts/index.js';
@@ -274,7 +277,7 @@ class MemoryIngester {
 }
 
 function createAbortError() {
-    const error = new Error('Memory ingestion aborted.');
+    const error = /** @type {AbortableError} */ (new Error('Memory ingestion aborted.'));
     error.name = 'AbortError';
     error.isUserAbort = true;
     return error;

src/types.js

@@ -65,6 +65,7 @@
 /**
  * @typedef {object} CompactBulletsOptions
  * @property {string} [today]
+ * @property {string} [now]
  * @property {number} [maxActivePerTopic]
  * @property {string} [defaultTopic]
  */
@@ -220,6 +221,7 @@
  * @property {{ path: string; content: string }[]} files
  * @property {string[]} paths
  * @property {string | null} assembledContext
+ * @property {string} [displayText]
  * @property {boolean} skipped - true when existing context already covered the query
  * @property {string} [skipReason] - explanation when skipped=true
  */
@@ -240,6 +242,7 @@
  * @property {string | null} reviewPrompt
  * @property {string | null} apiPrompt
  * @property {string | null} assembledContext
+ * @property {string} [displayText]
  * @property {boolean} skipped - true when existing context already covered the query
  * @property {string} [skipReason] - explanation when skipped=true
  */
@@ -360,6 +363,7 @@
  * @property {string} [configRepo]
  * @property {string} [attestationBundleURL]
  * @property {'ehbp' | 'tls'} [transport]
+ * @property {Record<string, any>} [tinfoilModule]
  */
 
 /**
@@ -528,6 +532,7 @@
  * @property {(doc: OmfDocument, options?: OmfImportOptions) => Promise<OmfImportPreview>} previewOmfImport
  * @property {(doc: OmfDocument, options?: OmfImportOptions) => Promise<OmfImportResult>} importOmf
  * @property {() => Promise<{filesChanged: number, filesTotal: number} | undefined>} compact
+ * @property {() => Promise<{archived: number, filesChanged: number}>} pruneExpired
  * @property {(query: string, options?: {deep?: boolean, mode?: string}) => Promise<{status: string, deleteCalls: number, writes: Array<any>}>} [deleteContent]
  * @property {StorageFacade} storage
  * @property {() => Promise<string>} serialize