fix: persist and restore registered context engines in plugin cache

张小波 · 张小波 · commit 7d74d56bf570 · 2026-05-04T11:26:13.000+08:00
Fixes issue #77063: lossless-claw selected and enabled but not registered as context engine after plugin cache restore. Root cause: When a cached plugin registry is reused, the cached state restoration restores agent harnesses, commands, compaction providers, and memory embedding providers, but NOT registered context engines. This causes context engines registered by plugins (like lossless-claw) to be lost when the plugin loader uses cached state. Changes: - src/context-engine/registry.ts: Add RegisteredContextEngineEntry type, listRegisteredContextEngines(), and restoreRegisteredContextEngines() functions for cache snapshot/restore support - src/plugins/loader.ts: Add contextEngines field to CachedPluginState, call restoreRegisteredContextEngines() when using cached state, and save contextEngines to cache when storing registry
diff --git a/src/context-engine/registry.ts b/src/context-engine/registry.ts
@@ -424,6 +424,49 @@ export function clearContextEnginesForOwner(owner: string): void {
   }
 }
 
+/**
+ * A serializable snapshot of a registered context engine for cache restore.
+ */
+export type RegisteredContextEngineEntry = {
+  id: string;
+  factory: ContextEngineFactory;
+  owner: string;
+};
+
+/**
+ * List all registered context engines with their factories and owners.
+ * Used for caching and restoring plugin state.
+ */
+export function listRegisteredContextEngines(): RegisteredContextEngineEntry[] {
+  const entries: RegisteredContextEngineEntry[] = [];
+  for (const [id, entry] of getContextEngineRegistryState().engines.entries()) {
+    entries.push({ id, factory: entry.factory, owner: entry.owner });
+  }
+  return entries;
+}
+
+/**
+ * Restore previously cached context engines. Clears existing plugin-owned engines
+ * first, then re-registers all entries from the cache.
+ */
+export function restoreRegisteredContextEngines(entries: RegisteredContextEngineEntry[]): void {
+  // Clear all plugin-owned context engines first
+  const registry = getContextEngineRegistryState().engines;
+  for (const [id, entry] of registry.entries()) {
+    if (entry.owner.startsWith("plugin:")) {
+      registry.delete(id);
+    }
+  }
+  // Restore all cached entries
+  for (const { id, factory, owner } of entries) {
+    if (id === defaultSlotIdForKey("contextEngine") && owner !== CORE_CONTEXT_ENGINE_OWNER) {
+      // Skip restoring the default engine slot with a non-core owner
+      continue;
+    }
+    registry.set(id, { factory, owner });
+  }
+}
+
 function describeResolvedContextEngineContractError(
   engineId: string,
   engine: unknown,
diff --git a/src/plugins/loader.ts b/src/plugins/loader.ts
@@ -8,6 +8,11 @@ import {
 } from "../agents/harness/registry.js";
 import type { OpenClawConfig } from "../config/types.openclaw.js";
 import type { PluginInstallRecord } from "../config/types.plugins.js";
+import {
+  listRegisteredContextEngines,
+  restoreRegisteredContextEngines,
+  type RegisteredContextEngineEntry,
+} from "../context-engine/registry.js";
 import type { GatewayRequestHandler } from "../gateway/server-methods/types.js";
 import { openBoundaryFileSync } from "../infra/boundary-file-read.js";
 import { createSubsystemLogger } from "../logging/subsystem.js";
@@ -244,6 +249,7 @@ type CachedPluginState = {
   compactionProviders: ReturnType<typeof listRegisteredCompactionProviders>;
   memoryEmbeddingProviders: ReturnType<typeof listRegisteredMemoryEmbeddingProviders>;
   memoryPromptSupplements: ReturnType<typeof listMemoryPromptSupplements>;
+  contextEngines: RegisteredContextEngineEntry[];
 };
 
 const MAX_PLUGIN_REGISTRY_CACHE_ENTRIES = 128;
@@ -1482,6 +1488,7 @@ export function loadOpenClawPlugins(options: PluginLoadOptions = {}): PluginRegi
         );
         restorePluginInteractiveHandlers(cached.state.interactiveHandlers ?? []);
         restoreRegisteredMemoryEmbeddingProviders(cached.state.memoryEmbeddingProviders);
+        restoreRegisteredContextEngines(cached.state.contextEngines);
         restoreMemoryPluginState({
           capability: cached.state.memoryCapability,
           corpusSupplements: cached.state.memoryCorpusSupplements,
@@ -2400,6 +2407,7 @@ export function loadOpenClawPlugins(options: PluginLoadOptions = {}): PluginRegi
           compactionProviders: listRegisteredCompactionProviders(),
           memoryEmbeddingProviders: listRegisteredMemoryEmbeddingProviders(),
           memoryPromptSupplements: listMemoryPromptSupplements(),
+          contextEngines: listRegisteredContextEngines(),
         },
         onlyPluginIds,
       );
