getcompanion-ai/co-mono
1
0
Fork 0
mirror of https://github.com/getcompanion-ai/co-mono.git synced 2026-04-16 06:02:42 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Refactor SessionEventBase to pass sessionManager and modelRegistry

Browse source 
Breaking changes to hook types:
- SessionEventBase now passes sessionManager and modelRegistry directly
- before_compact: passes preparation, previousCompactions (newest first)
- before_switch: has targetSessionFile; switch: has previousSessionFile
- Removed resolveApiKey (use modelRegistry.getApiKey())
- getSessionFile() returns string | undefined for in-memory sessions

Updated:
- All session event emissions in agent-session.ts
- Hook examples (custom-compaction.ts, auto-commit-on-exit.ts, confirm-destructive.ts)
- Tests (compaction-hooks.test.ts, compaction-hooks-example.test.ts)
- export-html.ts guards for in-memory sessions
This commit is contained in:
Mario Zechner 2025-12-26 22:22:43 +01:00
parent d96375b5e5
commit 9bba388ec5
14 changed files with 145 additions and 177 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -421,7 +421,7 @@ export function findMostRecentSession(sessionDir: string): string | null {
*/
export class SessionManager {
private sessionId: string = "";
private sessionFile: string = "";
private sessionFile: string | undefined;
private sessionDir: string;
private cwd: string;
private persist: boolean;
@ -434,10 +434,10 @@ export class SessionManager {
private constructor(cwd: string, sessionDir: string, sessionFile: string | null, persist: boolean) {
this.cwd = cwd;
this.sessionDir = sessionDir;
this.persist = persist;
if (persist && sessionDir && !existsSync(sessionDir)) {
mkdirSync(sessionDir, { recursive: true });
}
this.persist = persist;
if (sessionFile) {
this.setSessionFile(sessionFile);
@ -479,8 +479,8 @@ export class SessionManager {
this.byId.clear();
this.leafId = "";
this.flushed = false;
// Only generate filename if not already set (e.g., via --session flag)
if (!this.sessionFile) {
// Only generate filename if persisting and not already set (e.g., via --session flag)
if (this.persist && !this.sessionFile) {
const fileTimestamp = timestamp.replace(/[:.]/g, "-");
this.sessionFile = join(this.getSessionDir(), `${fileTimestamp}_${this.sessionId}.jsonl`);
}
@ -505,7 +505,7 @@ export class SessionManager {
}
private _rewriteFile(): void {
if (!this.persist) return;
if (!this.persist || !this.sessionFile) return;
const content = `${this.fileEntries.map((e) => JSON.stringify(e)).join("\n")}\n`;
writeFileSync(this.sessionFile, content);
}
@ -526,12 +526,12 @@ export class SessionManager {
return this.sessionId;
}
getSessionFile(): string {
getSessionFile(): string | undefined {
return this.sessionFile;
}
_persist(entry: SessionEntry): void {
if (!this.persist) return;
if (!this.persist || !this.sessionFile) return;
const hasAssistant = this.fileEntries.some((e) => e.type === "message" && e.message.role === "assistant");
if (!hasAssistant) return;