fix(sessions): preserve allow-any subagent model overrides (#21088) (thanks @Slats24)

This commit is contained in:
Peter Steinberger
2026-02-24 04:15:18 +00:00
parent 87dd896963
commit cd3927ad67
2 changed files with 43 additions and 0 deletions

View File

@@ -367,6 +367,48 @@ describe("agentCommand", () => {
});
});
it("keeps stored session model override when models allowlist is empty", async () => {
await withTempHome(async (home) => {
const store = path.join(home, "sessions.json");
writeSessionStoreSeed(store, {
"agent:main:subagent:allow-any": {
sessionId: "session-allow-any",
updatedAt: Date.now(),
providerOverride: "openai",
modelOverride: "gpt-custom-foo",
},
});
mockConfig(home, store, {
model: { primary: "anthropic/claude-opus-4-5" },
models: {},
});
vi.mocked(loadModelCatalog).mockResolvedValueOnce([
{ id: "claude-opus-4-5", name: "Opus", provider: "anthropic" },
]);
await agentCommand(
{
message: "hi",
sessionKey: "agent:main:subagent:allow-any",
},
runtime,
);
const callArgs = vi.mocked(runEmbeddedPiAgent).mock.calls.at(-1)?.[0];
expect(callArgs?.provider).toBe("openai");
expect(callArgs?.model).toBe("gpt-custom-foo");
const saved = JSON.parse(fs.readFileSync(store, "utf-8")) as Record<
string,
{ providerOverride?: string; modelOverride?: string }
>;
expect(saved["agent:main:subagent:allow-any"]?.providerOverride).toBe("openai");
expect(saved["agent:main:subagent:allow-any"]?.modelOverride).toBe("gpt-custom-foo");
});
});
it("keeps explicit sessionKey even when sessionId exists elsewhere", async () => {
await withTempHome(async (home) => {
const store = path.join(home, "sessions.json");