getcompanion-ai/co-mono
1
0
Fork 0
mirror of https://github.com/getcompanion-ai/co-mono.git synced 2026-04-15 21:03:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(coding-agent): add switchSession to Extension API (#1191)

Browse source 
Allows extension commands to programmatically switch to a different session file via ctx.switchSession(sessionPath).

fixes #1187
This commit is contained in:
Juan Ibiapina 2026-02-02 18:03:26 +01:00 • committed by GitHub
parent 0d934091f4
commit d0228412d6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 24 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
packages/coding-agent/src/core/extensions/index.ts
View file

@ -14,6 +14,7 @@ export type {
NavigateTreeHandler,
NewSessionHandler,
ShutdownHandler,
SwitchSessionHandler,
} from "./runner.js";
export { ExtensionRunner } from "./runner.js";
export type {

6
packages/coding-agent/src/core/extensions/runner.ts
View file

@ -106,6 +106,8 @@ export type NavigateTreeHandler = (
options?: { summarize?: boolean; customInstructions?: string; replaceInstructions?: boolean; label?: string },
) => Promise<{ cancelled: boolean }>;
export type SwitchSessionHandler = (sessionPath: string) => Promise<{ cancelled: boolean }>;
export type ShutdownHandler = () => void;
/**
@ -165,6 +167,7 @@ export class ExtensionRunner {
private newSessionHandler: NewSessionHandler = async () => ({ cancelled: false });
private forkHandler: ForkHandler = async () => ({ cancelled: false });
private navigateTreeHandler: NavigateTreeHandler = async () => ({ cancelled: false });
private switchSessionHandler: SwitchSessionHandler = async () => ({ cancelled: false });
private shutdownHandler: ShutdownHandler = () => {};
private shortcutDiagnostics: ResourceDiagnostic[] = [];
@ -221,6 +224,7 @@ export class ExtensionRunner {
this.newSessionHandler = actions.newSession;
this.forkHandler = actions.fork;
this.navigateTreeHandler = actions.navigateTree;
this.switchSessionHandler = actions.switchSession;
return;
}
@ -228,6 +232,7 @@ export class ExtensionRunner {
this.newSessionHandler = async () => ({ cancelled: false });
this.forkHandler = async () => ({ cancelled: false });
this.navigateTreeHandler = async () => ({ cancelled: false });
this.switchSessionHandler = async () => ({ cancelled: false });
}
setUIContext(uiContext?: ExtensionUIContext): void {
@ -436,6 +441,7 @@ export class ExtensionRunner {
newSession: (options) => this.newSessionHandler(options),
fork: (entryId) => this.forkHandler(entryId),
navigateTree: (targetId, options) => this.navigateTreeHandler(targetId, options),
switchSession: (sessionPath) => this.switchSessionHandler(sessionPath),
};
}

4
packages/coding-agent/src/core/extensions/types.ts
View file

@ -293,6 +293,9 @@ export interface ExtensionCommandContext extends ExtensionContext {
targetId: string,
options?: { summarize?: boolean; customInstructions?: string; replaceInstructions?: boolean; label?: string },
): Promise<{ cancelled: boolean }>;
/** Switch to a different session file. */
switchSession(sessionPath: string): Promise<{ cancelled: boolean }>;
}
// ============================================================================
@ -1214,6 +1217,7 @@ export interface ExtensionCommandContextActions {
targetId: string,
options?: { summarize?: boolean; customInstructions?: string; replaceInstructions?: boolean; label?: string },
) => Promise<{ cancelled: boolean }>;
switchSession: (sessionPath: string) => Promise<{ cancelled: boolean }>;
}
/**

4
packages/coding-agent/src/modes/interactive/interactive-mode.ts
View file

@ -1059,6 +1059,10 @@ export class InteractiveMode {
return { cancelled: false };
},
switchSession: async (sessionPath) => {
await this.handleResumeSession(sessionPath);
return { cancelled: false };
},
},
shutdownHandler: () => {
this.shutdownRequested = true;

4
packages/coding-agent/src/modes/print-mode.ts
View file

@ -59,6 +59,10 @@ export async function runPrintMode(session: AgentSession, options: PrintModeOpti
});
return { cancelled: result.cancelled };
},
switchSession: async (sessionPath) => {
const success = await session.switchSession(sessionPath);
return { cancelled: !success };
},
},
onError: (err) => {
console.error(`Extension error (${err.extensionPath}): ${err.error}`);

4
packages/coding-agent/src/modes/rpc/rpc-mode.ts
View file

@ -277,6 +277,10 @@ export async function runRpcMode(session: AgentSession): Promise<never> {
});
return { cancelled: result.cancelled };
},
switchSession: async (sessionPath) => {
const success = await session.switchSession(sessionPath);
return { cancelled: !success };
},
},
shutdownHandler: () => {
shutdownRequested = true;