@selesai/code 0.3.10 → 0.5.0

package/dist/core/agent-session-auto-handoff.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=agent-session-auto-handoff.test.d.ts.map

package/dist/core/agent-session-auto-handoff.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-session-auto-handoff.test.d.ts","sourceRoot":"","sources":["../../src/core/agent-session-auto-handoff.test.ts"],"names":[],"mappings":"","sourcesContent":["import { describe, expect, it, vi } from \"vitest\";\nimport { AgentSession } from \"./agent-session.ts\";\nimport type { ExtensionRunner } from \"./extensions/runner.ts\";\nimport type { ResolvedCommand } from \"./extensions/types.ts\";\nimport { SettingsManager } from \"./settings-manager.ts\";\n\nfunction createMockSession({\n\tenabled,\n\tthreshold = 128_000,\n\ttokens,\n\tmode = \"tui\",\n\tcommand,\n}: {\n\tenabled: boolean;\n\tthreshold?: number;\n\ttokens: number | null;\n\tmode?: \"tui\" | \"print\" | \"rpc\";\n\tcommand?: ResolvedCommand | undefined;\n}) {\n\tconst settings = SettingsManager.inMemory({\n\t\tautoHandoff: { enabled, thresholdTokens: threshold },\n\t});\n\n\tconst getCommand = vi.fn(() => command);\n\tconst handler = command?.handler ?? vi.fn();\n\tconst createCommandContext = vi.fn(() => ({ mock: \"ctx\" }) as unknown as never);\n\tconst emitError = vi.fn();\n\n\tconst extensionRunner = {\n\t\tgetCommand,\n\t\tcreateCommandContext,\n\t\temitError,\n\t\temit: vi.fn(),\n\t} as unknown as ExtensionRunner;\n\n\tconst resourceLoader = {\n\t\tgetExtensions: () => ({ extensions: [], errors: [], runtime: { flagValues: new Map(), pendingProviderRegistrations: [], assertActive: () => {} } }),\n\t\tgetSkills: () => ({ skills: [], diagnostics: [] }),\n\t\tgetPrompts: () => ({ prompts: [], diagnostics: [] }),\n\t\tgetThemes: () => ({ themes: [], diagnostics: [] }),\n\t\tgetAgents: () => ({ agents: [], diagnostics: [] }),\n\t\tgetAgentsFiles: () => ({ agentsFiles: [] }),\n\t\tgetSystemPrompt: () => undefined,\n\t\tgetAppendSystemPrompt: () => [],\n\t\tgetExtensionDiagnostics: () => [],\n\t\textendResources: () => {},\n\t\treload: vi.fn(),\n\t};\n\n\tconst modelRegistry = {\n\t\tregisterProvider: vi.fn(),\n\t\tunregisterProvider: vi.fn(),\n\t\tgetProvider: vi.fn(),\n\t\tgetProviders: vi.fn(() => []),\n\t\tgetModels: vi.fn(() => []),\n\t};\n\n\tconst agent = {\n\t\tstate: { messages: [], tools: [] },\n\t\tsubscribe: vi.fn(() => () => {}),\n\t};\n\n\tconst session = new AgentSession({\n\t\tagent: agent as unknown as never,\n\t\tsessionManager: { getBranch: () => [] } as unknown as never,\n\t\tsettingsManager: settings,\n\t\tcwd: \"/tmp\",\n\t\tresourceLoader: resourceLoader as unknown as never,\n\t\tmodelRegistry: modelRegistry as unknown as never,\n\t});\n\n\t(session as any)._extensionRunner = extensionRunner;\n\t(session as any)._extensionMode = mode;\n\t(session as any).getContextUsage = () =>\n\t\ttokens === null ? null : { tokens, contextWindow: 200_000, percent: tokens / 200_000 };\n\n\treturn { session, getCommand, createCommandContext, handler, emitError, settings };\n}\n\ndescribe(\"AgentSession auto handoff\", () => {\n\tit(\"invokes handoff-new once when enabled and threshold reached in tui mode\", async () => {\n\t\tconst command = { handler: vi.fn() } as unknown as ResolvedCommand;\n\t\tconst { session, getCommand, createCommandContext, emitError } = createMockSession({\n\t\t\tenabled: true,\n\t\t\ttokens: 128_000,\n\t\t\tcommand,\n\t\t});\n\n\t\tawait (session as any)._checkAutoHandoff();\n\t\tawait (session as any)._checkAutoHandoff();\n\n\t\texpect(getCommand).toHaveBeenCalledWith(\"handoff-new\");\n\t\texpect(createCommandContext).toHaveBeenCalled();\n\t\texpect(command.handler).toHaveBeenCalledTimes(1);\n\t\texpect(command.handler).toHaveBeenCalledWith(\"\", expect.anything());\n\t\texpect(emitError).not.toHaveBeenCalled();\n\t});\n\n\tit(\"dispatches auto handoff via _emitAgentSettled in tui mode\", async () => {\n\t\tconst command = { handler: vi.fn() } as unknown as ResolvedCommand;\n\t\tconst { session } = createMockSession({ enabled: true, tokens: 200_000, command });\n\t\tconst events: string[] = [];\n\t\tsession.subscribe((event) => {\n\t\t\tif (event.type === \"agent_settled\") events.push(event.type);\n\t\t});\n\n\t\tawait (session as any)._emitAgentSettled();\n\n\t\texpect(events).toEqual([\"agent_settled\"]);\n\t\texpect(command.handler).toHaveBeenCalledTimes(1);\n\t});\n\n\tit(\"_emitAgentSettled does not auto-handoff in non-tui mode\", async () => {\n\t\tconst command = { handler: vi.fn() } as unknown as ResolvedCommand;\n\t\tconst { session } = createMockSession({\n\t\t\tenabled: true,\n\t\t\ttokens: 200_000,\n\t\t\tmode: \"print\",\n\t\t\tcommand,\n\t\t});\n\t\tconst events: string[] = [];\n\t\tsession.subscribe((event) => {\n\t\t\tif (event.type === \"agent_settled\") events.push(event.type);\n\t\t});\n\n\t\tawait (session as any)._emitAgentSettled();\n\n\t\texpect(events).toEqual([\"agent_settled\"]);\n\t\texpect(command.handler).not.toHaveBeenCalled();\n\t});\n\n\tit(\"does nothing when disabled\", async () => {\n\t\tconst command = { handler: vi.fn() } as unknown as ResolvedCommand;\n\t\tconst { session, getCommand } = createMockSession({ enabled: false, tokens: 200_000, command });\n\n\t\tawait (session as any)._checkAutoHandoff();\n\n\t\texpect(getCommand).not.toHaveBeenCalled();\n\t});\n\n\tit(\"does nothing when tokens are below threshold\", async () => {\n\t\tconst command = { handler: vi.fn() } as unknown as ResolvedCommand;\n\t\tconst { session, getCommand } = createMockSession({ enabled: true, tokens: 127_999, command });\n\n\t\tawait (session as any)._checkAutoHandoff();\n\n\t\texpect(getCommand).not.toHaveBeenCalled();\n\t});\n\n\tit(\"does nothing when tokens are null\", async () => {\n\t\tconst command = { handler: vi.fn() } as unknown as ResolvedCommand;\n\t\tconst { session, getCommand } = createMockSession({ enabled: true, tokens: null, command });\n\n\t\tawait (session as any)._checkAutoHandoff();\n\n\t\texpect(getCommand).not.toHaveBeenCalled();\n\t});\n\n\tit(\"does nothing in non-tui mode\", async () => {\n\t\tconst command = { handler: vi.fn() } as unknown as ResolvedCommand;\n\t\tconst { session, getCommand } = createMockSession({\n\t\t\tenabled: true,\n\t\t\ttokens: 200_000,\n\t\t\tmode: \"rpc\",\n\t\t\tcommand,\n\t\t});\n\n\t\tawait (session as any)._checkAutoHandoff();\n\n\t\texpect(getCommand).not.toHaveBeenCalled();\n\t});\n\n\tit(\"does nothing when handoff-new command is not registered\", async () => {\n\t\tconst { session, getCommand } = createMockSession({ enabled: true, tokens: 200_000, command: undefined });\n\n\t\tawait (session as any)._checkAutoHandoff();\n\n\t\texpect(getCommand).toHaveBeenCalledWith(\"handoff-new\");\n\t});\n\n\tit(\"emits error when handoff-new throws but does not rethrow\", async () => {\n\t\tconst command = {\n\t\t\thandler: vi.fn(() => Promise.reject(new Error(\"boom\"))),\n\t\t} as unknown as ResolvedCommand;\n\t\tconst { session, emitError } = createMockSession({ enabled: true, tokens: 200_000, command });\n\n\t\tawait expect((session as any)._checkAutoHandoff()).resolves.toBeUndefined();\n\n\t\texpect(emitError).toHaveBeenCalledWith({\n\t\t\textensionPath: \"command:handoff-new\",\n\t\t\tevent: \"auto-handoff\",\n\t\t\terror: \"boom\",\n\t\t});\n\t});\n\n\tit(\"resets triggered flag when tokens drop below threshold\", async () => {\n\t\tconst command = { handler: vi.fn() } as unknown as ResolvedCommand;\n\t\tconst { session, getCommand } = createMockSession({ enabled: true, tokens: 200_000, command });\n\n\t\tawait (session as any)._checkAutoHandoff();\n\t\texpect(command.handler).toHaveBeenCalledTimes(1);\n\n\t\t// Simulate tokens dropping below threshold (e.g. compaction/new session)\n\t\t(session as any).getContextUsage = () => ({ tokens: 1000, contextWindow: 200_000, percent: 0.5 });\n\t\tawait (session as any)._checkAutoHandoff();\n\t\texpect(command.handler).toHaveBeenCalledTimes(1); // still once\n\n\t\t// Tokens rise above threshold again\n\t\t(session as any).getContextUsage = () => ({ tokens: 200_000, contextWindow: 200_000, percent: 100 });\n\t\tawait (session as any)._checkAutoHandoff();\n\t\texpect(command.handler).toHaveBeenCalledTimes(2);\n\t});\n});\n"]}

package/dist/core/agent-session-auto-handoff.test.js

@@ -0,0 +1,161 @@
+import { describe, expect, it, vi } from "vitest";
+import { AgentSession } from "./agent-session.js";
+import { SettingsManager } from "./settings-manager.js";
+function createMockSession({ enabled, threshold = 128_000, tokens, mode = "tui", command, }) {
+    const settings = SettingsManager.inMemory({
+        autoHandoff: { enabled, thresholdTokens: threshold },
+    });
+    const getCommand = vi.fn(() => command);
+    const handler = command?.handler ?? vi.fn();
+    const createCommandContext = vi.fn(() => ({ mock: "ctx" }));
+    const emitError = vi.fn();
+    const extensionRunner = {
+        getCommand,
+        createCommandContext,
+        emitError,
+        emit: vi.fn(),
+    };
+    const resourceLoader = {
+        getExtensions: () => ({ extensions: [], errors: [], runtime: { flagValues: new Map(), pendingProviderRegistrations: [], assertActive: () => { } } }),
+        getSkills: () => ({ skills: [], diagnostics: [] }),
+        getPrompts: () => ({ prompts: [], diagnostics: [] }),
+        getThemes: () => ({ themes: [], diagnostics: [] }),
+        getAgents: () => ({ agents: [], diagnostics: [] }),
+        getAgentsFiles: () => ({ agentsFiles: [] }),
+        getSystemPrompt: () => undefined,
+        getAppendSystemPrompt: () => [],
+        getExtensionDiagnostics: () => [],
+        extendResources: () => { },
+        reload: vi.fn(),
+    };
+    const modelRegistry = {
+        registerProvider: vi.fn(),
+        unregisterProvider: vi.fn(),
+        getProvider: vi.fn(),
+        getProviders: vi.fn(() => []),
+        getModels: vi.fn(() => []),
+    };
+    const agent = {
+        state: { messages: [], tools: [] },
+        subscribe: vi.fn(() => () => { }),
+    };
+    const session = new AgentSession({
+        agent: agent,
+        sessionManager: { getBranch: () => [] },
+        settingsManager: settings,
+        cwd: "/tmp",
+        resourceLoader: resourceLoader,
+        modelRegistry: modelRegistry,
+    });
+    session._extensionRunner = extensionRunner;
+    session._extensionMode = mode;
+    session.getContextUsage = () => tokens === null ? null : { tokens, contextWindow: 200_000, percent: tokens / 200_000 };
+    return { session, getCommand, createCommandContext, handler, emitError, settings };
+}
+describe("AgentSession auto handoff", () => {
+    it("invokes handoff-new once when enabled and threshold reached in tui mode", async () => {
+        const command = { handler: vi.fn() };
+        const { session, getCommand, createCommandContext, emitError } = createMockSession({
+            enabled: true,
+            tokens: 128_000,
+            command,
+        });
+        await session._checkAutoHandoff();
+        await session._checkAutoHandoff();
+        expect(getCommand).toHaveBeenCalledWith("handoff-new");
+        expect(createCommandContext).toHaveBeenCalled();
+        expect(command.handler).toHaveBeenCalledTimes(1);
+        expect(command.handler).toHaveBeenCalledWith("", expect.anything());
+        expect(emitError).not.toHaveBeenCalled();
+    });
+    it("dispatches auto handoff via _emitAgentSettled in tui mode", async () => {
+        const command = { handler: vi.fn() };
+        const { session } = createMockSession({ enabled: true, tokens: 200_000, command });
+        const events = [];
+        session.subscribe((event) => {
+            if (event.type === "agent_settled")
+                events.push(event.type);
+        });
+        await session._emitAgentSettled();
+        expect(events).toEqual(["agent_settled"]);
+        expect(command.handler).toHaveBeenCalledTimes(1);
+    });
+    it("_emitAgentSettled does not auto-handoff in non-tui mode", async () => {
+        const command = { handler: vi.fn() };
+        const { session } = createMockSession({
+            enabled: true,
+            tokens: 200_000,
+            mode: "print",
+            command,
+        });
+        const events = [];
+        session.subscribe((event) => {
+            if (event.type === "agent_settled")
+                events.push(event.type);
+        });
+        await session._emitAgentSettled();
+        expect(events).toEqual(["agent_settled"]);
+        expect(command.handler).not.toHaveBeenCalled();
+    });
+    it("does nothing when disabled", async () => {
+        const command = { handler: vi.fn() };
+        const { session, getCommand } = createMockSession({ enabled: false, tokens: 200_000, command });
+        await session._checkAutoHandoff();
+        expect(getCommand).not.toHaveBeenCalled();
+    });
+    it("does nothing when tokens are below threshold", async () => {
+        const command = { handler: vi.fn() };
+        const { session, getCommand } = createMockSession({ enabled: true, tokens: 127_999, command });
+        await session._checkAutoHandoff();
+        expect(getCommand).not.toHaveBeenCalled();
+    });
+    it("does nothing when tokens are null", async () => {
+        const command = { handler: vi.fn() };
+        const { session, getCommand } = createMockSession({ enabled: true, tokens: null, command });
+        await session._checkAutoHandoff();
+        expect(getCommand).not.toHaveBeenCalled();
+    });
+    it("does nothing in non-tui mode", async () => {
+        const command = { handler: vi.fn() };
+        const { session, getCommand } = createMockSession({
+            enabled: true,
+            tokens: 200_000,
+            mode: "rpc",
+            command,
+        });
+        await session._checkAutoHandoff();
+        expect(getCommand).not.toHaveBeenCalled();
+    });
+    it("does nothing when handoff-new command is not registered", async () => {
+        const { session, getCommand } = createMockSession({ enabled: true, tokens: 200_000, command: undefined });
+        await session._checkAutoHandoff();
+        expect(getCommand).toHaveBeenCalledWith("handoff-new");
+    });
+    it("emits error when handoff-new throws but does not rethrow", async () => {
+        const command = {
+            handler: vi.fn(() => Promise.reject(new Error("boom"))),
+        };
+        const { session, emitError } = createMockSession({ enabled: true, tokens: 200_000, command });
+        await expect(session._checkAutoHandoff()).resolves.toBeUndefined();
+        expect(emitError).toHaveBeenCalledWith({
+            extensionPath: "command:handoff-new",
+            event: "auto-handoff",
+            error: "boom",
+        });
+    });
+    it("resets triggered flag when tokens drop below threshold", async () => {
+        const command = { handler: vi.fn() };
+        const { session, getCommand } = createMockSession({ enabled: true, tokens: 200_000, command });
+        await session._checkAutoHandoff();
+        expect(command.handler).toHaveBeenCalledTimes(1);
+        // Simulate tokens dropping below threshold (e.g. compaction/new session)
+        session.getContextUsage = () => ({ tokens: 1000, contextWindow: 200_000, percent: 0.5 });
+        await session._checkAutoHandoff();
+        expect(command.handler).toHaveBeenCalledTimes(1); // still once
+        // Tokens rise above threshold again
+        session.getContextUsage = () => ({ tokens: 200_000, contextWindow: 200_000, percent: 100 });
+        await session._checkAutoHandoff();
+        expect(command.handler).toHaveBeenCalledTimes(2);
+    });
+});
+//# sourceMappingURL=agent-session-auto-handoff.test.js.map

package/dist/core/agent-session-auto-handoff.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-session-auto-handoff.test.js","sourceRoot":"","sources":["../../src/core/agent-session-auto-handoff.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAGlD,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAExD,SAAS,iBAAiB,CAAC,EAC1B,OAAO,EACP,SAAS,GAAG,OAAO,EACnB,MAAM,EACN,IAAI,GAAG,KAAK,EACZ,OAAO,GAOP,EAAE;IACF,MAAM,QAAQ,GAAG,eAAe,CAAC,QAAQ,CAAC;QACzC,WAAW,EAAE,EAAE,OAAO,EAAE,eAAe,EAAE,SAAS,EAAE;KACpD,CAAC,CAAC;IAEH,MAAM,UAAU,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,CAAC;IACxC,MAAM,OAAO,GAAG,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;IAC5C,MAAM,oBAAoB,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAqB,CAAC,CAAC;IAChF,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAE1B,MAAM,eAAe,GAAG;QACvB,UAAU;QACV,oBAAoB;QACpB,SAAS;QACT,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;KACiB,CAAC;IAEhC,MAAM,cAAc,GAAG;QACtB,aAAa,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,UAAU,EAAE,IAAI,GAAG,EAAE,EAAE,4BAA4B,EAAE,EAAE,EAAE,YAAY,EAAE,GAAG,EAAE,CAAC,EAAC,CAAC,EAAE,EAAE,CAAC;QACnJ,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC;QAClD,UAAU,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC;QACpD,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC;QAClD,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC;QAClD,cAAc,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC;QAC3C,eAAe,EAAE,GAAG,EAAE,CAAC,SAAS;QAChC,qBAAqB,EAAE,GAAG,EAAE,CAAC,EAAE;QAC/B,uBAAuB,EAAE,GAAG,EAAE,CAAC,EAAE;QACjC,eAAe,EAAE,GAAG,EAAE,CAAC,EAAC,CAAC;QACzB,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;KACf,CAAC;IAEF,MAAM,aAAa,GAAG;QACrB,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE;QACzB,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE;QAC3B,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;QACpB,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;QAC7B,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;KAC1B,CAAC;IAEF,MAAM,KAAK,GAAG;QACb,KAAK,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE;QAClC,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,GAAG,EAAE,CAAC,EAAC,CAAC,CAAC;KAChC,CAAC;IAEF,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC;QAChC,KAAK,EAAE,KAAyB;QAChC,cAAc,EAAE,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC,EAAE,EAAsB;QAC3D,eAAe,EAAE,QAAQ;QACzB,GAAG,EAAE,MAAM;QACX,cAAc,EAAE,cAAkC;QAClD,aAAa,EAAE,aAAiC;KAChD,CAAC,CAAC;IAEF,OAAe,CAAC,gBAAgB,GAAG,eAAe,CAAC;IACnD,OAAe,CAAC,cAAc,GAAG,IAAI,CAAC;IACtC,OAAe,CAAC,eAAe,GAAG,GAAG,EAAE,CACvC,MAAM,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,aAAa,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,EAAE,CAAC;IAExF,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,oBAAoB,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC;AAAA,CACnF;AAED,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE,CAAC;IAC3C,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE,CAAC;QACzF,MAAM,OAAO,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,EAAgC,CAAC;QACnE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,oBAAoB,EAAE,SAAS,EAAE,GAAG,iBAAiB,CAAC;YAClF,OAAO,EAAE,IAAI;YACb,MAAM,EAAE,OAAO;YACf,OAAO;SACP,CAAC,CAAC;QAEH,MAAO,OAAe,CAAC,iBAAiB,EAAE,CAAC;QAC3C,MAAO,OAAe,CAAC,iBAAiB,EAAE,CAAC;QAE3C,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,aAAa,CAAC,CAAC;QACvD,MAAM,CAAC,oBAAoB,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAChD,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACjD,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,EAAE,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;QACpE,MAAM,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAAA,CACzC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE,CAAC;QAC3E,MAAM,OAAO,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,EAAgC,CAAC;QACnE,MAAM,EAAE,OAAO,EAAE,GAAG,iBAAiB,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QACnF,MAAM,MAAM,GAAa,EAAE,CAAC;QAC5B,OAAO,CAAC,SAAS,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC;YAC5B,IAAI,KAAK,CAAC,IAAI,KAAK,eAAe;gBAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAAA,CAC5D,CAAC,CAAC;QAEH,MAAO,OAAe,CAAC,iBAAiB,EAAE,CAAC;QAE3C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC;QAC1C,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAAA,CACjD,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE,CAAC;QACzE,MAAM,OAAO,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,EAAgC,CAAC;QACnE,MAAM,EAAE,OAAO,EAAE,GAAG,iBAAiB,CAAC;YACrC,OAAO,EAAE,IAAI;YACb,MAAM,EAAE,OAAO;YACf,IAAI,EAAE,OAAO;YACb,OAAO;SACP,CAAC,CAAC;QACH,MAAM,MAAM,GAAa,EAAE,CAAC;QAC5B,OAAO,CAAC,SAAS,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC;YAC5B,IAAI,KAAK,CAAC,IAAI,KAAK,eAAe;gBAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAAA,CAC5D,CAAC,CAAC;QAEH,MAAO,OAAe,CAAC,iBAAiB,EAAE,CAAC;QAE3C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC;QAC1C,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAAA,CAC/C,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE,CAAC;QAC5C,MAAM,OAAO,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,EAAgC,CAAC;QACnE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,iBAAiB,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QAEhG,MAAO,OAAe,CAAC,iBAAiB,EAAE,CAAC;QAE3C,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAAA,CAC1C,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE,CAAC;QAC9D,MAAM,OAAO,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,EAAgC,CAAC;QACnE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,iBAAiB,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QAE/F,MAAO,OAAe,CAAC,iBAAiB,EAAE,CAAC;QAE3C,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAAA,CAC1C,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE,CAAC;QACnD,MAAM,OAAO,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,EAAgC,CAAC;QACnE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,iBAAiB,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;QAE5F,MAAO,OAAe,CAAC,iBAAiB,EAAE,CAAC;QAE3C,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAAA,CAC1C,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE,CAAC;QAC9C,MAAM,OAAO,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,EAAgC,CAAC;QACnE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,iBAAiB,CAAC;YACjD,OAAO,EAAE,IAAI;YACb,MAAM,EAAE,OAAO;YACf,IAAI,EAAE,KAAK;YACX,OAAO;SACP,CAAC,CAAC;QAEH,MAAO,OAAe,CAAC,iBAAiB,EAAE,CAAC;QAE3C,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAAA,CAC1C,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE,CAAC;QACzE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,iBAAiB,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;QAE1G,MAAO,OAAe,CAAC,iBAAiB,EAAE,CAAC;QAE3C,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,aAAa,CAAC,CAAC;IAAA,CACvD,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE,CAAC;QAC1E,MAAM,OAAO,GAAG;YACf,OAAO,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;SACzB,CAAC;QAChC,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,GAAG,iBAAiB,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QAE9F,MAAM,MAAM,CAAE,OAAe,CAAC,iBAAiB,EAAE,CAAC,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;QAE5E,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC;YACtC,aAAa,EAAE,qBAAqB;YACpC,KAAK,EAAE,cAAc;YACrB,KAAK,EAAE,MAAM;SACb,CAAC,CAAC;IAAA,CACH,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE,CAAC;QACxE,MAAM,OAAO,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,EAAgC,CAAC;QACnE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,iBAAiB,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QAE/F,MAAO,OAAe,CAAC,iBAAiB,EAAE,CAAC;QAC3C,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAEjD,yEAAyE;QACxE,OAAe,CAAC,eAAe,GAAG,GAAG,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;QAClG,MAAO,OAAe,CAAC,iBAAiB,EAAE,CAAC;QAC3C,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa;QAE/D,oCAAoC;QACnC,OAAe,CAAC,eAAe,GAAG,GAAG,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,aAAa,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;QACrG,MAAO,OAAe,CAAC,iBAAiB,EAAE,CAAC;QAC3C,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAAA,CACjD,CAAC,CAAC;AAAA,CACH,CAAC,CAAC","sourcesContent":["import { describe, expect, it, vi } from \"vitest\";\nimport { AgentSession } from \"./agent-session.ts\";\nimport type { ExtensionRunner } from \"./extensions/runner.ts\";\nimport type { ResolvedCommand } from \"./extensions/types.ts\";\nimport { SettingsManager } from \"./settings-manager.ts\";\n\nfunction createMockSession({\n\tenabled,\n\tthreshold = 128_000,\n\ttokens,\n\tmode = \"tui\",\n\tcommand,\n}: {\n\tenabled: boolean;\n\tthreshold?: number;\n\ttokens: number | null;\n\tmode?: \"tui\" | \"print\" | \"rpc\";\n\tcommand?: ResolvedCommand | undefined;\n}) {\n\tconst settings = SettingsManager.inMemory({\n\t\tautoHandoff: { enabled, thresholdTokens: threshold },\n\t});\n\n\tconst getCommand = vi.fn(() => command);\n\tconst handler = command?.handler ?? vi.fn();\n\tconst createCommandContext = vi.fn(() => ({ mock: \"ctx\" }) as unknown as never);\n\tconst emitError = vi.fn();\n\n\tconst extensionRunner = {\n\t\tgetCommand,\n\t\tcreateCommandContext,\n\t\temitError,\n\t\temit: vi.fn(),\n\t} as unknown as ExtensionRunner;\n\n\tconst resourceLoader = {\n\t\tgetExtensions: () => ({ extensions: [], errors: [], runtime: { flagValues: new Map(), pendingProviderRegistrations: [], assertActive: () => {} } }),\n\t\tgetSkills: () => ({ skills: [], diagnostics: [] }),\n\t\tgetPrompts: () => ({ prompts: [], diagnostics: [] }),\n\t\tgetThemes: () => ({ themes: [], diagnostics: [] }),\n\t\tgetAgents: () => ({ agents: [], diagnostics: [] }),\n\t\tgetAgentsFiles: () => ({ agentsFiles: [] }),\n\t\tgetSystemPrompt: () => undefined,\n\t\tgetAppendSystemPrompt: () => [],\n\t\tgetExtensionDiagnostics: () => [],\n\t\textendResources: () => {},\n\t\treload: vi.fn(),\n\t};\n\n\tconst modelRegistry = {\n\t\tregisterProvider: vi.fn(),\n\t\tunregisterProvider: vi.fn(),\n\t\tgetProvider: vi.fn(),\n\t\tgetProviders: vi.fn(() => []),\n\t\tgetModels: vi.fn(() => []),\n\t};\n\n\tconst agent = {\n\t\tstate: { messages: [], tools: [] },\n\t\tsubscribe: vi.fn(() => () => {}),\n\t};\n\n\tconst session = new AgentSession({\n\t\tagent: agent as unknown as never,\n\t\tsessionManager: { getBranch: () => [] } as unknown as never,\n\t\tsettingsManager: settings,\n\t\tcwd: \"/tmp\",\n\t\tresourceLoader: resourceLoader as unknown as never,\n\t\tmodelRegistry: modelRegistry as unknown as never,\n\t});\n\n\t(session as any)._extensionRunner = extensionRunner;\n\t(session as any)._extensionMode = mode;\n\t(session as any).getContextUsage = () =>\n\t\ttokens === null ? null : { tokens, contextWindow: 200_000, percent: tokens / 200_000 };\n\n\treturn { session, getCommand, createCommandContext, handler, emitError, settings };\n}\n\ndescribe(\"AgentSession auto handoff\", () => {\n\tit(\"invokes handoff-new once when enabled and threshold reached in tui mode\", async () => {\n\t\tconst command = { handler: vi.fn() } as unknown as ResolvedCommand;\n\t\tconst { session, getCommand, createCommandContext, emitError } = createMockSession({\n\t\t\tenabled: true,\n\t\t\ttokens: 128_000,\n\t\t\tcommand,\n\t\t});\n\n\t\tawait (session as any)._checkAutoHandoff();\n\t\tawait (session as any)._checkAutoHandoff();\n\n\t\texpect(getCommand).toHaveBeenCalledWith(\"handoff-new\");\n\t\texpect(createCommandContext).toHaveBeenCalled();\n\t\texpect(command.handler).toHaveBeenCalledTimes(1);\n\t\texpect(command.handler).toHaveBeenCalledWith(\"\", expect.anything());\n\t\texpect(emitError).not.toHaveBeenCalled();\n\t});\n\n\tit(\"dispatches auto handoff via _emitAgentSettled in tui mode\", async () => {\n\t\tconst command = { handler: vi.fn() } as unknown as ResolvedCommand;\n\t\tconst { session } = createMockSession({ enabled: true, tokens: 200_000, command });\n\t\tconst events: string[] = [];\n\t\tsession.subscribe((event) => {\n\t\t\tif (event.type === \"agent_settled\") events.push(event.type);\n\t\t});\n\n\t\tawait (session as any)._emitAgentSettled();\n\n\t\texpect(events).toEqual([\"agent_settled\"]);\n\t\texpect(command.handler).toHaveBeenCalledTimes(1);\n\t});\n\n\tit(\"_emitAgentSettled does not auto-handoff in non-tui mode\", async () => {\n\t\tconst command = { handler: vi.fn() } as unknown as ResolvedCommand;\n\t\tconst { session } = createMockSession({\n\t\t\tenabled: true,\n\t\t\ttokens: 200_000,\n\t\t\tmode: \"print\",\n\t\t\tcommand,\n\t\t});\n\t\tconst events: string[] = [];\n\t\tsession.subscribe((event) => {\n\t\t\tif (event.type === \"agent_settled\") events.push(event.type);\n\t\t});\n\n\t\tawait (session as any)._emitAgentSettled();\n\n\t\texpect(events).toEqual([\"agent_settled\"]);\n\t\texpect(command.handler).not.toHaveBeenCalled();\n\t});\n\n\tit(\"does nothing when disabled\", async () => {\n\t\tconst command = { handler: vi.fn() } as unknown as ResolvedCommand;\n\t\tconst { session, getCommand } = createMockSession({ enabled: false, tokens: 200_000, command });\n\n\t\tawait (session as any)._checkAutoHandoff();\n\n\t\texpect(getCommand).not.toHaveBeenCalled();\n\t});\n\n\tit(\"does nothing when tokens are below threshold\", async () => {\n\t\tconst command = { handler: vi.fn() } as unknown as ResolvedCommand;\n\t\tconst { session, getCommand } = createMockSession({ enabled: true, tokens: 127_999, command });\n\n\t\tawait (session as any)._checkAutoHandoff();\n\n\t\texpect(getCommand).not.toHaveBeenCalled();\n\t});\n\n\tit(\"does nothing when tokens are null\", async () => {\n\t\tconst command = { handler: vi.fn() } as unknown as ResolvedCommand;\n\t\tconst { session, getCommand } = createMockSession({ enabled: true, tokens: null, command });\n\n\t\tawait (session as any)._checkAutoHandoff();\n\n\t\texpect(getCommand).not.toHaveBeenCalled();\n\t});\n\n\tit(\"does nothing in non-tui mode\", async () => {\n\t\tconst command = { handler: vi.fn() } as unknown as ResolvedCommand;\n\t\tconst { session, getCommand } = createMockSession({\n\t\t\tenabled: true,\n\t\t\ttokens: 200_000,\n\t\t\tmode: \"rpc\",\n\t\t\tcommand,\n\t\t});\n\n\t\tawait (session as any)._checkAutoHandoff();\n\n\t\texpect(getCommand).not.toHaveBeenCalled();\n\t});\n\n\tit(\"does nothing when handoff-new command is not registered\", async () => {\n\t\tconst { session, getCommand } = createMockSession({ enabled: true, tokens: 200_000, command: undefined });\n\n\t\tawait (session as any)._checkAutoHandoff();\n\n\t\texpect(getCommand).toHaveBeenCalledWith(\"handoff-new\");\n\t});\n\n\tit(\"emits error when handoff-new throws but does not rethrow\", async () => {\n\t\tconst command = {\n\t\t\thandler: vi.fn(() => Promise.reject(new Error(\"boom\"))),\n\t\t} as unknown as ResolvedCommand;\n\t\tconst { session, emitError } = createMockSession({ enabled: true, tokens: 200_000, command });\n\n\t\tawait expect((session as any)._checkAutoHandoff()).resolves.toBeUndefined();\n\n\t\texpect(emitError).toHaveBeenCalledWith({\n\t\t\textensionPath: \"command:handoff-new\",\n\t\t\tevent: \"auto-handoff\",\n\t\t\terror: \"boom\",\n\t\t});\n\t});\n\n\tit(\"resets triggered flag when tokens drop below threshold\", async () => {\n\t\tconst command = { handler: vi.fn() } as unknown as ResolvedCommand;\n\t\tconst { session, getCommand } = createMockSession({ enabled: true, tokens: 200_000, command });\n\n\t\tawait (session as any)._checkAutoHandoff();\n\t\texpect(command.handler).toHaveBeenCalledTimes(1);\n\n\t\t// Simulate tokens dropping below threshold (e.g. compaction/new session)\n\t\t(session as any).getContextUsage = () => ({ tokens: 1000, contextWindow: 200_000, percent: 0.5 });\n\t\tawait (session as any)._checkAutoHandoff();\n\t\texpect(command.handler).toHaveBeenCalledTimes(1); // still once\n\n\t\t// Tokens rise above threshold again\n\t\t(session as any).getContextUsage = () => ({ tokens: 200_000, contextWindow: 200_000, percent: 100 });\n\t\tawait (session as any)._checkAutoHandoff();\n\t\texpect(command.handler).toHaveBeenCalledTimes(2);\n\t});\n});\n"]}

package/dist/core/agent-session.d.ts

@@ -237,6 +237,8 @@ export declare class AgentSession {
     private _getIdleWaitPromise;
     private _resolveIdleWaitIfIdle;
     private _emitAgentSettled;
+    private _autoHandoffTriggered;
+    private _checkAutoHandoff;
     private _lastAssistantMessage;
     /** Internal handler for agent events - shared by subscribe and reconnect */
     private _handleAgentEvent;
@@ -490,6 +492,8 @@ export declare class AgentSession {
     setAutoCompactionEnabled(enabled: boolean): void;
     /** Whether auto-compaction is enabled */
     get autoCompactionEnabled(): boolean;
+    get autoHandoffEnabled(): boolean;
+    get autoHandoffThresholdTokens(): number;
     bindExtensions(bindings: ExtensionBindings): Promise<void>;
     private extendResourcesFromExtensions;
     private buildExtensionResourcePaths;