metaphorics
18b2bef00e
Adds a fastModeScope setting (both|openai|claude, default both). setFastMode(true) derives the service tier from it (both->priority, openai->openai-only, claude->claude-only) instead of hardcoding unscoped priority; the off path and the already-on no-op are unchanged. /fast status now reports the active scope. Default both preserves existing behavior.
106 lines
3.1 KiB
TypeScript
106 lines
3.1 KiB
TypeScript
import { afterEach, beforeEach, describe, expect, it } from "bun:test";
|
|
import * as path from "node:path";
|
|
import { Agent } from "@oh-my-pi/pi-agent-core";
|
|
import { getBundledModel } from "@oh-my-pi/pi-catalog/models";
|
|
import { ModelRegistry } from "@oh-my-pi/pi-coding-agent/config/model-registry";
|
|
import { Settings } from "@oh-my-pi/pi-coding-agent/config/settings";
|
|
import { AgentSession } from "@oh-my-pi/pi-coding-agent/session/agent-session";
|
|
import { AuthStorage } from "@oh-my-pi/pi-coding-agent/session/auth-storage";
|
|
import { SessionManager } from "@oh-my-pi/pi-coding-agent/session/session-manager";
|
|
import { TempDir } from "@oh-my-pi/pi-utils";
|
|
|
|
type FastModeScope = "both" | "openai" | "claude";
|
|
|
|
describe("fast mode scope", () => {
|
|
let tempDir: TempDir;
|
|
let authStorage: AuthStorage;
|
|
let session: AgentSession;
|
|
let modelRegistry: ModelRegistry;
|
|
|
|
beforeEach(() => {
|
|
tempDir = TempDir.createSync("@pi-fast-mode-scope-");
|
|
});
|
|
|
|
afterEach(async () => {
|
|
if (session) {
|
|
await session.dispose();
|
|
}
|
|
authStorage?.close();
|
|
tempDir.removeSync();
|
|
});
|
|
|
|
async function createSession(fastModeScope?: FastModeScope): Promise<AgentSession> {
|
|
const model = getBundledModel("anthropic", "claude-sonnet-4-5");
|
|
if (!model) {
|
|
throw new Error("Expected bundled test model to exist");
|
|
}
|
|
|
|
const settings = fastModeScope === undefined ? Settings.isolated() : Settings.isolated({ fastModeScope });
|
|
const agent = new Agent({
|
|
initialState: {
|
|
model,
|
|
systemPrompt: ["Test"],
|
|
tools: [],
|
|
messages: [],
|
|
},
|
|
});
|
|
|
|
authStorage = await AuthStorage.create(path.join(tempDir.path(), "testauth.db"));
|
|
authStorage.setRuntimeApiKey(model.provider, "anthropic-token");
|
|
modelRegistry = new ModelRegistry(authStorage, path.join(tempDir.path(), "models.yml"));
|
|
|
|
session = new AgentSession({
|
|
agent,
|
|
sessionManager: SessionManager.inMemory(),
|
|
settings,
|
|
modelRegistry,
|
|
});
|
|
session.subscribe(() => {});
|
|
return session;
|
|
}
|
|
|
|
it("scopes enabled fast mode to OpenAI when configured", async () => {
|
|
const session = await createSession("openai");
|
|
|
|
session.setFastMode(true);
|
|
|
|
expect(session.serviceTier).toBe("openai-only");
|
|
});
|
|
|
|
it("scopes enabled fast mode to Claude when configured", async () => {
|
|
const session = await createSession("claude");
|
|
|
|
session.setFastMode(true);
|
|
|
|
expect(session.serviceTier).toBe("claude-only");
|
|
});
|
|
|
|
it("defaults enabled fast mode to priority for both providers", async () => {
|
|
const session = await createSession();
|
|
|
|
session.setFastMode(true);
|
|
|
|
expect(session.serviceTier).toBe("priority");
|
|
});
|
|
|
|
it("clears the service tier when disabled", async () => {
|
|
const session = await createSession("openai");
|
|
session.setFastMode(true);
|
|
|
|
session.setFastMode(false);
|
|
|
|
expect(session.serviceTier).toBeUndefined();
|
|
});
|
|
|
|
it("does not broaden an already enabled scoped tier", async () => {
|
|
const session = await createSession("claude");
|
|
session.setFastMode(true);
|
|
expect(session.serviceTier).toBe("claude-only");
|
|
session.settings.set("fastModeScope", "both");
|
|
|
|
session.setFastMode(true);
|
|
|
|
expect(session.serviceTier).toBe("claude-only");
|
|
});
|
|
});
|