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

Replace custom tool dispose() with shutdown session event

Browse source 
Breaking change: CustomAgentTool.dispose() removed. Use onSession with
reason 'shutdown' instead for cleanup.

- Add 'shutdown' to SessionEvent.reason for custom tools
- Remove dispose() method from CustomAgentTool interface
- Make emitToolSessionEvent() public on AgentSession
- Emit shutdown event to tools in InteractiveMode.shutdown()
- Update custom-tools.md with new API and examples
This commit is contained in:
Mario Zechner 2025-12-31 02:55:45 +01:00
parent 450d77fb79
commit ff78ac2f84
4 changed files with 46 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

14
packages/coding-agent/src/core/agent-session.ts
View file

@ -698,7 +698,7 @@ export class AgentSession {
}
// Emit session event to custom tools
await this._emitToolSessionEvent("new", previousSessionFile);
await this.emitToolSessionEvent("new", previousSessionFile);
return true;
}
@ -1473,7 +1473,7 @@ export class AgentSession {
}
// Emit session event to custom tools
await this._emitToolSessionEvent("switch", previousSessionFile);
await this.emitToolSessionEvent("switch", previousSessionFile);
this.agent.replaceMessages(sessionContext.messages);
@ -1550,7 +1550,7 @@ export class AgentSession {
}
// Emit session event to custom tools (with reason "branch")
await this._emitToolSessionEvent("branch", previousSessionFile);
await this.emitToolSessionEvent("branch", previousSessionFile);
if (!skipConversationRestore) {
this.agent.replaceMessages(sessionContext.messages);
@ -1720,7 +1720,7 @@ export class AgentSession {
}
// Emit to custom tools
await this._emitToolSessionEvent("tree", this.sessionFile);
await this.emitToolSessionEvent("tree", this.sessionFile);
this._branchSummaryAbortController = undefined;
return { editorText, cancelled: false, summaryEntry };
@ -1875,11 +1875,11 @@ export class AgentSession {
/**
* Emit session event to all custom tools.
* Called on session switch, branch, and clear.
* Called on session switch, branch, tree navigation, and shutdown.
*/
private async _emitToolSessionEvent(
async emitToolSessionEvent(
reason: ToolSessionEvent["reason"],
previousSessionFile: string | undefined,
previousSessionFile?: string | undefined,
): Promise<void> {
const event: ToolSessionEvent = {
entries: this.sessionManager.getEntries(),