can1357
7f1d071843
Rewrote SessionManager to use streaming NDJSON writes with proper backpressure handling instead of synchronous file appends. Key changes: - SessionManager.open() and continueRecent() are now async for proper init - Session writes use atomic rename pattern with fsync for crash safety - Added flush() method for explicit control over pending write completion - Large content blocks (>500KB) are truncated before writing to prevent OOM - Images are replaced with placeholder thumbnails during persistence
49 lines
1.7 KiB
TypeScript
49 lines
1.7 KiB
TypeScript
/**
|
|
* Session Management
|
|
*
|
|
* Control session persistence: in-memory, new file, continue, or open specific.
|
|
*/
|
|
|
|
import { createAgentSession, SessionManager } from "@oh-my-pi/pi-coding-agent";
|
|
|
|
// In-memory (no persistence)
|
|
const { session: inMemory } = await createAgentSession({
|
|
sessionManager: SessionManager.inMemory(),
|
|
});
|
|
console.log("In-memory session:", inMemory.sessionFile ?? "(none)");
|
|
|
|
// New persistent session
|
|
const { session: newSession } = await createAgentSession({
|
|
sessionManager: SessionManager.create(process.cwd()),
|
|
});
|
|
console.log("New session file:", newSession.sessionFile);
|
|
|
|
// Continue most recent session (or create new if none)
|
|
const { session: continued, modelFallbackMessage } = await createAgentSession({
|
|
sessionManager: await SessionManager.continueRecent(process.cwd()),
|
|
});
|
|
if (modelFallbackMessage) console.log("Note:", modelFallbackMessage);
|
|
console.log("Continued session:", continued.sessionFile);
|
|
|
|
// List and open specific session
|
|
const sessions = SessionManager.list(process.cwd());
|
|
console.log(`\nFound ${sessions.length} sessions:`);
|
|
for (const info of sessions.slice(0, 3)) {
|
|
console.log(` ${info.id.slice(0, 8)}... - "${info.firstMessage.slice(0, 30)}..."`);
|
|
}
|
|
|
|
if (sessions.length > 0) {
|
|
const { session: opened } = await createAgentSession({
|
|
sessionManager: await SessionManager.open(sessions[0].path),
|
|
});
|
|
console.log(`\nOpened: ${opened.sessionId}`);
|
|
}
|
|
|
|
// Custom session directory (no cwd encoding)
|
|
// const customDir = "/path/to/my-sessions";
|
|
// const { session } = await createAgentSession({
|
|
// sessionManager: SessionManager.create(process.cwd(), customDir),
|
|
// });
|
|
// SessionManager.list(process.cwd(), customDir);
|
|
// SessionManager.continueRecent(process.cwd(), customDir);
|