diff --git a/packages/server/src/server/persistence-hooks.test.ts b/packages/server/src/server/persistence-hooks.test.ts index a0bbb49035..144c76a231 100644 --- a/packages/server/src/server/persistence-hooks.test.ts +++ b/packages/server/src/server/persistence-hooks.test.ts @@ -214,3 +214,14 @@ describe("persistence hooks", () => { expect(handle).toBeNull(); }); }); + +test("restores optional MCP requirements for continuation", () => { + const record = createRecord({ + config: { + mcpServerRequirements: { hub: "optional" }, + mcpServers: { hub: { type: "http", url: "https://hub.test/mcp" } }, + }, + }); + expect(buildConfigOverrides(record).mcpServerRequirements).toEqual({ hub: "optional" }); + expect(buildSessionConfig(record)?.mcpServerRequirements).toEqual({ hub: "optional" }); +}); diff --git a/packages/server/src/server/persistence-hooks.ts b/packages/server/src/server/persistence-hooks.ts index e95743d6be..cdb8485c84 100644 --- a/packages/server/src/server/persistence-hooks.ts +++ b/packages/server/src/server/persistence-hooks.ts @@ -78,6 +78,7 @@ export function buildConfigOverrides(record: StoredAgentRecord): Partial