mirror of
https://github.com/openclaw/openclaw.git
synced 2026-02-19 18:39:20 -05:00
fix(agent): search all agent stores when resolving --session-id (#13579)
* fix(agent): search all agent stores when resolving --session-id When `--session-id` was provided without `--to` or `--agent`, the reverse lookup only searched the default agent's session store. Sessions created under a specific agent (e.g. `--agent mybot`) live in that agent's store file, so the lookup silently failed and the session was not reused. Now `resolveSessionKeyForRequest` iterates all configured agent stores when the primary store doesn't contain the requested sessionId. Fixes #12881 * fix: search other agent stores when --to key does not match --session-id When --to derives a session key whose stored sessionId doesn't match the requested --session-id, the cross-store search now also runs. This handles the case where a user provides both --to and --session-id targeting a session in a different agent's store.
This commit is contained in:
227
src/commands/agent/session.test.ts
Normal file
227
src/commands/agent/session.test.ts
Normal file
@@ -0,0 +1,227 @@
|
||||
import { beforeEach, describe, expect, it, vi } from "vitest";
|
||||
import type { OpenClawConfig } from "../../config/config.js";
|
||||
|
||||
const mocks = vi.hoisted(() => ({
|
||||
loadSessionStore: vi.fn(),
|
||||
resolveStorePath: vi.fn(),
|
||||
listAgentIds: vi.fn(),
|
||||
}));
|
||||
|
||||
vi.mock("../../config/sessions.js", async () => {
|
||||
const actual = await vi.importActual<typeof import("../../config/sessions.js")>(
|
||||
"../../config/sessions.js",
|
||||
);
|
||||
return {
|
||||
...actual,
|
||||
loadSessionStore: mocks.loadSessionStore,
|
||||
resolveStorePath: mocks.resolveStorePath,
|
||||
};
|
||||
});
|
||||
|
||||
vi.mock("../../agents/agent-scope.js", () => ({
|
||||
listAgentIds: mocks.listAgentIds,
|
||||
}));
|
||||
|
||||
describe("resolveSessionKeyForRequest", () => {
|
||||
beforeEach(() => {
|
||||
vi.clearAllMocks();
|
||||
mocks.listAgentIds.mockReturnValue(["main"]);
|
||||
});
|
||||
|
||||
async function importFresh() {
|
||||
return await import("./session.js");
|
||||
}
|
||||
|
||||
const baseCfg: OpenClawConfig = {};
|
||||
|
||||
it("returns sessionKey when --to resolves a session key via context", async () => {
|
||||
const { resolveSessionKeyForRequest } = await importFresh();
|
||||
|
||||
mocks.resolveStorePath.mockReturnValue("/tmp/main-store.json");
|
||||
mocks.loadSessionStore.mockReturnValue({
|
||||
"agent:main:main": { sessionId: "sess-1", updatedAt: 0 },
|
||||
});
|
||||
|
||||
const result = resolveSessionKeyForRequest({
|
||||
cfg: baseCfg,
|
||||
to: "+15551234567",
|
||||
});
|
||||
expect(result.sessionKey).toBe("agent:main:main");
|
||||
});
|
||||
|
||||
it("finds session by sessionId via reverse lookup in primary store", async () => {
|
||||
const { resolveSessionKeyForRequest } = await importFresh();
|
||||
|
||||
mocks.resolveStorePath.mockReturnValue("/tmp/main-store.json");
|
||||
mocks.loadSessionStore.mockReturnValue({
|
||||
"agent:main:main": { sessionId: "target-session-id", updatedAt: 0 },
|
||||
});
|
||||
|
||||
const result = resolveSessionKeyForRequest({
|
||||
cfg: baseCfg,
|
||||
sessionId: "target-session-id",
|
||||
});
|
||||
expect(result.sessionKey).toBe("agent:main:main");
|
||||
});
|
||||
|
||||
it("finds session by sessionId in non-primary agent store", async () => {
|
||||
const { resolveSessionKeyForRequest } = await importFresh();
|
||||
|
||||
mocks.listAgentIds.mockReturnValue(["main", "mybot"]);
|
||||
mocks.resolveStorePath.mockImplementation(
|
||||
(_store: string | undefined, opts?: { agentId?: string }) => {
|
||||
if (opts?.agentId === "mybot") {
|
||||
return "/tmp/mybot-store.json";
|
||||
}
|
||||
return "/tmp/main-store.json";
|
||||
},
|
||||
);
|
||||
mocks.loadSessionStore.mockImplementation((storePath: string) => {
|
||||
if (storePath === "/tmp/mybot-store.json") {
|
||||
return {
|
||||
"agent:mybot:main": { sessionId: "target-session-id", updatedAt: 0 },
|
||||
};
|
||||
}
|
||||
return {};
|
||||
});
|
||||
|
||||
const result = resolveSessionKeyForRequest({
|
||||
cfg: baseCfg,
|
||||
sessionId: "target-session-id",
|
||||
});
|
||||
expect(result.sessionKey).toBe("agent:mybot:main");
|
||||
expect(result.storePath).toBe("/tmp/mybot-store.json");
|
||||
});
|
||||
|
||||
it("returns correct sessionStore when session found in non-primary agent store", async () => {
|
||||
const { resolveSessionKeyForRequest } = await importFresh();
|
||||
|
||||
const mybotStore = {
|
||||
"agent:mybot:main": { sessionId: "target-session-id", updatedAt: 0 },
|
||||
};
|
||||
mocks.listAgentIds.mockReturnValue(["main", "mybot"]);
|
||||
mocks.resolveStorePath.mockImplementation(
|
||||
(_store: string | undefined, opts?: { agentId?: string }) => {
|
||||
if (opts?.agentId === "mybot") {
|
||||
return "/tmp/mybot-store.json";
|
||||
}
|
||||
return "/tmp/main-store.json";
|
||||
},
|
||||
);
|
||||
mocks.loadSessionStore.mockImplementation((storePath: string) => {
|
||||
if (storePath === "/tmp/mybot-store.json") {
|
||||
return { ...mybotStore };
|
||||
}
|
||||
return {};
|
||||
});
|
||||
|
||||
const result = resolveSessionKeyForRequest({
|
||||
cfg: baseCfg,
|
||||
sessionId: "target-session-id",
|
||||
});
|
||||
expect(result.sessionStore["agent:mybot:main"]?.sessionId).toBe("target-session-id");
|
||||
});
|
||||
|
||||
it("returns undefined sessionKey when sessionId not found in any store", async () => {
|
||||
const { resolveSessionKeyForRequest } = await importFresh();
|
||||
|
||||
mocks.listAgentIds.mockReturnValue(["main", "mybot"]);
|
||||
mocks.resolveStorePath.mockImplementation(
|
||||
(_store: string | undefined, opts?: { agentId?: string }) => {
|
||||
if (opts?.agentId === "mybot") {
|
||||
return "/tmp/mybot-store.json";
|
||||
}
|
||||
return "/tmp/main-store.json";
|
||||
},
|
||||
);
|
||||
mocks.loadSessionStore.mockReturnValue({});
|
||||
|
||||
const result = resolveSessionKeyForRequest({
|
||||
cfg: baseCfg,
|
||||
sessionId: "nonexistent-id",
|
||||
});
|
||||
expect(result.sessionKey).toBeUndefined();
|
||||
});
|
||||
|
||||
it("does not search other stores when explicitSessionKey is set", async () => {
|
||||
const { resolveSessionKeyForRequest } = await importFresh();
|
||||
|
||||
mocks.listAgentIds.mockReturnValue(["main", "mybot"]);
|
||||
mocks.resolveStorePath.mockReturnValue("/tmp/main-store.json");
|
||||
mocks.loadSessionStore.mockReturnValue({
|
||||
"agent:main:main": { sessionId: "other-id", updatedAt: 0 },
|
||||
});
|
||||
|
||||
const result = resolveSessionKeyForRequest({
|
||||
cfg: baseCfg,
|
||||
sessionKey: "agent:main:main",
|
||||
sessionId: "target-session-id",
|
||||
});
|
||||
// explicitSessionKey is set, so sessionKey comes from it, not from sessionId lookup
|
||||
expect(result.sessionKey).toBe("agent:main:main");
|
||||
});
|
||||
|
||||
it("searches other stores when --to derives a key that does not match --session-id", async () => {
|
||||
const { resolveSessionKeyForRequest } = await importFresh();
|
||||
|
||||
mocks.listAgentIds.mockReturnValue(["main", "mybot"]);
|
||||
mocks.resolveStorePath.mockImplementation(
|
||||
(_store: string | undefined, opts?: { agentId?: string }) => {
|
||||
if (opts?.agentId === "mybot") {
|
||||
return "/tmp/mybot-store.json";
|
||||
}
|
||||
return "/tmp/main-store.json";
|
||||
},
|
||||
);
|
||||
mocks.loadSessionStore.mockImplementation((storePath: string) => {
|
||||
if (storePath === "/tmp/main-store.json") {
|
||||
return {
|
||||
"agent:main:main": { sessionId: "other-session-id", updatedAt: 0 },
|
||||
};
|
||||
}
|
||||
if (storePath === "/tmp/mybot-store.json") {
|
||||
return {
|
||||
"agent:mybot:main": { sessionId: "target-session-id", updatedAt: 0 },
|
||||
};
|
||||
}
|
||||
return {};
|
||||
});
|
||||
|
||||
const result = resolveSessionKeyForRequest({
|
||||
cfg: baseCfg,
|
||||
to: "+15551234567",
|
||||
sessionId: "target-session-id",
|
||||
});
|
||||
// --to derives agent:main:main, but its sessionId doesn't match target-session-id,
|
||||
// so the cross-store search finds it in the mybot store
|
||||
expect(result.sessionKey).toBe("agent:mybot:main");
|
||||
expect(result.storePath).toBe("/tmp/mybot-store.json");
|
||||
});
|
||||
|
||||
it("skips already-searched primary store when iterating agents", async () => {
|
||||
const { resolveSessionKeyForRequest } = await importFresh();
|
||||
|
||||
mocks.listAgentIds.mockReturnValue(["main", "mybot"]);
|
||||
mocks.resolveStorePath.mockImplementation(
|
||||
(_store: string | undefined, opts?: { agentId?: string }) => {
|
||||
if (opts?.agentId === "mybot") {
|
||||
return "/tmp/mybot-store.json";
|
||||
}
|
||||
return "/tmp/main-store.json";
|
||||
},
|
||||
);
|
||||
mocks.loadSessionStore.mockReturnValue({});
|
||||
|
||||
resolveSessionKeyForRequest({
|
||||
cfg: baseCfg,
|
||||
sessionId: "nonexistent-id",
|
||||
});
|
||||
|
||||
// loadSessionStore should be called twice: once for main, once for mybot
|
||||
// (not twice for main)
|
||||
const storePaths = mocks.loadSessionStore.mock.calls.map((call: [string]) => call[0]);
|
||||
expect(storePaths).toHaveLength(2);
|
||||
expect(storePaths).toContain("/tmp/main-store.json");
|
||||
expect(storePaths).toContain("/tmp/mybot-store.json");
|
||||
});
|
||||
});
|
||||
@@ -1,6 +1,7 @@
|
||||
import crypto from "node:crypto";
|
||||
import type { MsgContext } from "../../auto-reply/templating.js";
|
||||
import type { OpenClawConfig } from "../../config/config.js";
|
||||
import { listAgentIds } from "../../agents/agent-scope.js";
|
||||
import {
|
||||
normalizeThinkLevel,
|
||||
normalizeVerboseLevel,
|
||||
@@ -78,6 +79,31 @@ export function resolveSessionKeyForRequest(opts: {
|
||||
}
|
||||
}
|
||||
|
||||
// When sessionId was provided but not found in the primary store, search all agent stores.
|
||||
// Sessions created under a specific agent live in that agent's store file; the primary
|
||||
// store (derived from the default agent) won't contain them.
|
||||
// Also covers the case where --to derived a sessionKey that doesn't match the requested sessionId.
|
||||
if (
|
||||
opts.sessionId &&
|
||||
!explicitSessionKey &&
|
||||
(!sessionKey || sessionStore[sessionKey]?.sessionId !== opts.sessionId)
|
||||
) {
|
||||
const allAgentIds = listAgentIds(opts.cfg);
|
||||
for (const agentId of allAgentIds) {
|
||||
if (agentId === storeAgentId) {
|
||||
continue;
|
||||
}
|
||||
const altStorePath = resolveStorePath(sessionCfg?.store, { agentId });
|
||||
const altStore = loadSessionStore(altStorePath);
|
||||
const foundKey = Object.keys(altStore).find(
|
||||
(key) => altStore[key]?.sessionId === opts.sessionId,
|
||||
);
|
||||
if (foundKey) {
|
||||
return { sessionKey: foundKey, sessionStore: altStore, storePath: altStorePath };
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return { sessionKey, sessionStore, storePath };
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user