feat(memory): enforce daemon schema compatibility

Author: nguyenthdat

Cargo.lock

@@ -23,7 +23,7 @@ fs2 = "0.4"
 hex = "0.4"
 hf-hub = { version = "=1.0.0", features = ["blocking"] }
 ignore = "0.4.31"
-llama-cpp-2 = { version = "=0.1.152", default-features = false }
+llama-cpp-2 = { version = "=0.1.153", default-features = false }
 prost = "=0.14.4"
 regex = "1.13.1"
 self_cell = "=1.3.0"
@@ -33,7 +33,7 @@ sha2 = "0.11"
 unicode-normalization = "0.1"
 rustix = { version = "1.1", features = ["fs", "process"] }
 tokio = { version = "1", features = ["io-util", "net", "rt-multi-thread", "sync", "time"] }
-xberg = { version = "=1.0.0", default-features = false, features = ["tokio-runtime", "simd-utf8", "pdf"] }
+xberg = { version = "=1.0.3", default-features = false, features = ["tokio-runtime", "simd-utf8", "pdf"] }
 zvec-rust = "=0.6.0"
 
 [build-dependencies]

Cargo.toml

@@ -1,6 +1,7 @@
 import { describe, expect, test } from "bun:test";
 import {
   assertDaemonVersionCompatible,
+  assertDaemonSchemaCompatible,
   isRetrySafeMemoryMethod,
   NativeMemoryClient,
   NativeMemoryClientPool,
@@ -46,6 +47,25 @@ describe("shared daemon client", () => {
     expect(() => assertDaemonVersionCompatible("development", "0.6.0-beta.0")).not.toThrow();
   });
 
+  test("reports both domain schema generations for a stale daemon", () => {
+    expect(() =>
+      assertDaemonSchemaCompatible(
+        "0.6.0",
+        { daemonVersion: "0.6.0", domainSchemaGeneration: 2, pid: 64346 },
+        "/tmp/opencode-memory/daemon.sock",
+      ),
+    ).toThrow(
+      "Native memory daemon domain schema mismatch at /tmp/opencode-memory/daemon.sock: client 4, daemon 2 (plugin 0.6.0, daemon 0.6.0, pid 64346)",
+    );
+    expect(() =>
+      assertDaemonSchemaCompatible(
+        "0.6.0",
+        { daemonVersion: "0.6.0", domainSchemaGeneration: 4, pid: 64346 },
+        "/tmp/opencode-memory/daemon.sock",
+      ),
+    ).not.toThrow();
+  });
+
   test("preserves model retry classifications", () => {
     expect(isRetrySafeMemoryMethod("model_profiles")).toBe(true);
     expect(isRetrySafeMemoryMethod("model_switch_status")).toBe(true);

opencode-memory/src/daemon-client.test.ts

@@ -372,6 +372,7 @@ class DaemonProjectClient implements NativeMemoryRequester {
       if (response.body.case !== "getDaemonInfo") {
         throw new Error("Native memory daemon omitted GetDaemonInfo");
       }
+      assertDaemonSchemaCompatible(this.pluginVersion, response.body.value, this.endpoint);
       const hello = create(OpenSessionRequestSchema, {
         clientInstanceId: randomUUID(),
         minimumProtocolGeneration: DAEMON_PROTOCOL_GENERATION,
@@ -473,10 +474,11 @@ class DaemonProjectClient implements NativeMemoryRequester {
       throw new Error(
         `Native memory daemon protocol mismatch at ${this.endpoint}: ` +
           `client supports ${DAEMON_PROTOCOL_GENERATION}, daemon ${daemon.daemonVersion} supports ` +
-          `${daemon.minimumProtocolGeneration}-${daemon.maximumProtocolGeneration}. ` +
-          "Close active OpenCode processes and restart the native memory daemon.",
+          `${daemon.minimumProtocolGeneration}-${daemon.maximumProtocolGeneration} (pid ${daemon.pid}). ` +
+          "Close all OpenCode processes using memory and restart the native memory daemon.",
       );
     }
+    assertDaemonSchemaCompatible(this.pluginVersion, daemon, this.endpoint);
     assertDaemonVersionCompatible(this.pluginVersion, daemon.daemonVersion, daemon.pid);
     const hello = create(OpenSessionRequestSchema, {
       clientInstanceId: randomUUID(),
@@ -909,6 +911,20 @@ export function assertDaemonVersionCompatible(
   );
 }
 
+export function assertDaemonSchemaCompatible(
+  pluginVersion: string,
+  daemon: Pick<GetDaemonInfoResponse, "daemonVersion" | "domainSchemaGeneration" | "pid">,
+  endpoint: string,
+): void {
+  if (daemon.domainSchemaGeneration === DOMAIN_SCHEMA_GENERATION) return;
+  throw new Error(
+    `Native memory daemon domain schema mismatch at ${endpoint}: ` +
+      `client ${DOMAIN_SCHEMA_GENERATION}, daemon ${daemon.domainSchemaGeneration} ` +
+      `(plugin ${pluginVersion}, daemon ${daemon.daemonVersion}, pid ${daemon.pid}). ` +
+      "Close all OpenCode processes using memory and restart the native memory daemon.",
+  );
+}
+
 function resolvePackagedBinary(packageName: string, binaryName: string): string | undefined {
   try {
     return require.resolve(`${packageName}/bin/${binaryName}`);