fix(memory): initialize candidates and enforce lifecycle rollout

2026-07-21 22:55:19 +08:00
parent 0dd9331e5b
commit bfb1f6fea9
8 changed files with 207 additions and 14 deletions
@@ -33,6 +33,15 @@ export function resolveMemoryV2LifecyclePolicy(env = process.env) {
  };
 }

+export function resolveMemoryV2LifecycleWorkerScopes(policy = {}) {
+  if (!policy.enabled) return [];
+  if (policy.rolloutMode === 'active') return [null];
+  if (policy.rolloutMode === 'canary') {
+    return [...new Set((policy.rolloutUserIds ?? []).map(String).filter(Boolean))];
+  }
+  return [];
+}
+
 function normalizeItem(row) {
  return {
    id: String(row.id),
@@ -71,7 +80,7 @@ export function createMemoryV2LifecycleService({ pool = null, env = process.env,
  }

  async function forgetMemory({ userId, memoryId } = {}) {
-    if (!canRun(userId) && !policy.forgettingEnabled) return { ok: true, updated: false, skipped: true, reason: 'disabled' };
+    if (!policy.forgettingEnabled || !canRun(userId)) return { ok: true, updated: false, skipped: true, reason: 'disabled' };
    if (!pool?.query || !userId || !memoryId) return { ok: false, updated: false, reason: 'invalid_input' };
    const [result] = await pool.query(
      `UPDATE ${MEMORY_TABLE} SET status = 'deleted', updated_at = ? WHERE id = ? AND user_id = ? AND status <> 'deleted'`,
@@ -82,7 +91,9 @@ export function createMemoryV2LifecycleService({ pool = null, env = process.env,
  }

  async function expire({ userId = null } = {}) {
-    if (!pool?.query || !policy.forgettingEnabled) return { ok: true, skipped: true, reason: 'disabled', expired: 0 };
+    if (!pool?.query || !policy.forgettingEnabled || !canRun(userId)) {
+      return { ok: true, skipped: true, reason: 'disabled', expired: 0 };
+    }
    const cutoff = now() - policy.retentionDays * 86400000;
    const params = [cutoff];
    let scope = '';