@fleetagent/pi-coding-agent 0.0.1 → 0.0.2

package/docs/extensions.md

@@ -306,13 +306,13 @@ user sends another prompt ◄─────────────────
 /new (new session) or /resume (switch session)
   ├─► session_before_switch (can cancel)
   ├─► session_shutdown
-  ├─► session_start { reason: "new" | "resume", previousSessionFile? }
+  ├─► session_start { reason: "new" | "resume", previousSessionReference? }
   └─► resources_discover { reason: "startup" }
 
 /fork or /clone
   ├─► session_before_fork (can cancel)
   ├─► session_shutdown
-  ├─► session_start { reason: "fork", previousSessionFile }
+  ├─► session_start { reason: "fork", previousSessionReference }
   └─► resources_discover { reason: "startup" }
 
 /compact or auto-compaction
@@ -355,7 +355,7 @@ pi.on("resources_discover", async (event, _ctx) => {
 
 ### Session Events
 
-See [Session Format](session-format.md) for session storage internals and the SessionManager API.
+See [Session Format](session-format.md) for session storage internals, active `Session` APIs, and `SessionManager` lifecycle APIs.
 
 #### session_start
 
@@ -364,8 +364,9 @@ Fired when a session is started, loaded, or reloaded.
 ```typescript
 pi.on("session_start", async (event, ctx) => {
   // event.reason - "startup" | "reload" | "new" | "resume" | "fork"
-  // event.previousSessionFile - present for "new", "resume", and "fork"
-  ctx.ui.notify(`Session: ${ctx.sessionManager.getSessionFile() ?? "ephemeral"}`, "info");
+  // event.previousSessionReference - present for "new", "resume", and "fork"
+  // event.previousSessionFile - deprecated alias for local-file-oriented compatibility
+  ctx.ui.notify(`Session: ${ctx.session.getSessionReference() ?? "ephemeral"}`, "info");
 });
 ```
 
@@ -376,7 +377,8 @@ Fired before starting a new session (`/new`) or switching sessions (`/resume`).
 ```typescript
 pi.on("session_before_switch", async (event, ctx) => {
   // event.reason - "new" or "resume"
-  // event.targetSessionFile - session we're switching to (only for "resume")
+  // event.targetSessionReference - session we're switching to (only for "resume")
+  // event.targetSessionFile - deprecated alias for local-file-oriented compatibility
 
   if (event.reason === "new") {
     const ok = await ctx.ui.confirm("Clear?", "Delete all messages?");
@@ -385,7 +387,7 @@ pi.on("session_before_switch", async (event, ctx) => {
 });
 ```
 
-After a successful switch or new-session action, pi emits `session_shutdown` for the old extension instance, reloads and rebinds extensions for the new session, then emits `session_start` with `reason: "new" | "resume"` and `previousSessionFile`.
+After a successful switch or new-session action, pi emits `session_shutdown` for the old extension instance, reloads and rebinds extensions for the new session, then emits `session_start` with `reason: "new" | "resume"` and `previousSessionReference`.
 Do cleanup work in `session_shutdown`, then reestablish any in-memory state in `session_start`.
 
 #### session_before_fork
@@ -402,7 +404,7 @@ pi.on("session_before_fork", async (event, ctx) => {
 });
 ```
 
-After a successful fork or clone, pi emits `session_shutdown` for the old extension instance, reloads and rebinds extensions for the new session, then emits `session_start` with `reason: "fork"` and `previousSessionFile`.
+After a successful fork or clone, pi emits `session_shutdown` for the old extension instance, reloads and rebinds extensions for the new session, then emits `session_start` with `reason: "fork"` and `previousSessionReference`.
 Do cleanup work in `session_shutdown`, then reestablish any in-memory state in `session_start`.
 
 #### session_before_compact / session_compact
@@ -456,7 +458,8 @@ Fired before an extension runtime is torn down.
 ```typescript
 pi.on("session_shutdown", async (event, ctx) => {
   // event.reason - "quit" | "reload" | "new" | "resume" | "fork"
-  // event.targetSessionFile - destination session for session replacement flows
+  // event.targetSessionReference - destination session for session replacement flows
+  // event.targetSessionFile - deprecated alias for local-file-oriented compatibility
   // Cleanup, save state, etc.
 });
 ```
@@ -675,9 +678,9 @@ Use this to update extension UI when `pi.setThinkingLevel()`, model changes, or
 
 Fired after `tool_execution_start`, before the tool executes. **Can block.** Use `isToolCallEventType` to narrow and get typed inputs.
 
-Before `tool_call` runs, pi waits for previously emitted Agent events to finish draining through `AgentSession`. This means `ctx.sessionManager` is up to date through the current assistant tool-calling message.
+Before `tool_call` runs, pi waits for previously emitted Agent events to finish draining through `AgentSession`. This means `ctx.session` is up to date through the current assistant tool-calling message.
 
-In the default parallel tool execution mode, sibling tool calls from the same assistant message are preflighted sequentially, then executed concurrently. `tool_call` is not guaranteed to see sibling tool results from that same assistant message in `ctx.sessionManager`.
+In the default parallel tool execution mode, sibling tool calls from the same assistant message are preflighted sequentially, then executed concurrently. `tool_call` is not guaranteed to see sibling tool results from that same assistant message in `ctx.session`.
 
 `event.input` is mutable. Mutate it in place to patch tool arguments before execution.
 
@@ -865,16 +868,16 @@ UI methods for user interaction. See [Custom UI](#custom-ui) for full details.
 
 Current working directory.
 
-### ctx.sessionManager
+### ctx.session
 
-Read-only access to session state. See [Session Format](session-format.md) for the full SessionManager API and entry types.
+Read-only access to active session state. See [Session Format](session-format.md) for the full `Session` API and entry types.
 
 For `tool_call`, this state is synchronized through the current assistant message before handlers run. In parallel tool execution mode it is still not guaranteed to include sibling tool results from the same assistant message.
 
 ```typescript
-ctx.sessionManager.getEntries()       // All entries
-ctx.sessionManager.getBranch()        // Current branch
-ctx.sessionManager.getLeafId()        // Current leaf entry ID
+ctx.session.getEntries()       // All entries
+ctx.session.getBranch()        // Current branch
+ctx.session.getLeafId()        // Current leaf entry ID
 ```
 
 ### ctx.modelRegistry / ctx.model
@@ -993,13 +996,13 @@ pi.registerCommand("my-cmd", {
 Create a new session:
 
 ```typescript
-const parentSession = ctx.sessionManager.getSessionFile();
+const parentSession = ctx.session.getSessionReference();
 const kickoff = "Continue in the replacement session";
 
 const result = await ctx.newSession({
   parentSession,
-  setup: async (sm) => {
-    sm.appendMessage({
+  setup: async (session) => {
+    session.appendMessage({
       role: "user",
       content: [{ type: "text", text: "Context from previous session..." }],
       timestamp: Date.now(),
@@ -1018,7 +1021,7 @@ if (result.cancelled) {
 
 Options:
 - `parentSession`: parent session file to record in the new session header
-- `setup`: mutate the new session's `SessionManager` before `withSession` runs
+- `setup`: mutate the new active `Session` before `withSession` runs
 - `withSession`: run post-switch work against a fresh replacement-session context. Do not use captured old `pi` / command `ctx`; see [Session replacement lifecycle and footguns](#session-replacement-lifecycle-and-footguns).
 
 ### ctx.fork(entryId, options?)
@@ -1084,19 +1087,19 @@ if (result.cancelled) {
 Options:
 - `withSession`: run post-switch work against a fresh replacement-session context. Do not use captured old `pi` / command `ctx`; see [Session replacement lifecycle and footguns](#session-replacement-lifecycle-and-footguns).
 
-To discover available sessions, use the static `LocalSessionManager.list()` or `LocalSessionManager.listAll()` methods:
+To discover available sessions, create a `LocalSessionManager` and call `list()` or `listAll()`:
 
 ```typescript
-import { SessionManager } from "@fleetagent/pi-coding-agent";
+import { LocalSessionManager } from "@fleetagent/pi-coding-agent";
 
 pi.registerCommand("switch", {
   description: "Switch to another session",
   handler: async (args, ctx) => {
-    const sessions = await LocalSessionManager.list(ctx.cwd);
+    const sessions = await new LocalSessionManager({ cwd: ctx.cwd }).list();
     if (sessions.length === 0) return;
     const choice = await ctx.ui.select(
       "Pick session:",
-      sessions.map(s => s.file),
+      sessions.map(s => s.reference),
     );
     if (choice) {
       await ctx.switchSession(choice, {
@@ -1117,7 +1120,7 @@ Lifecycle and footguns:
 - `withSession` runs only after the old session has emitted `session_shutdown`, the old runtime has been torn down, the replacement session has been rebound, and the new extension instance has already received `session_start`.
 - The callback still executes in the original closure, not inside the new extension instance. That means your old extension instance may already have run its shutdown cleanup before `withSession` starts.
 - Captured old `pi` / old command `ctx` session-bound objects are stale after replacement and will throw if used. Use only the `ctx` passed to `withSession` for session-bound work.
-- Previously extracted raw objects are still your responsibility. For example, if you capture `const sm = ctx.sessionManager` before replacement, `sm` is still the old `SessionManager` object. Do not reuse it after replacement.
+- Previously extracted raw objects are still your responsibility. For example, if you capture `const oldSession = ctx.session` before replacement, `oldSession` is still the old `Session` object. Do not reuse it after replacement.
 - Code in `withSession` should assume any state invalidated by your `session_shutdown` handler is already gone. Only capture plain data that survives shutdown cleanly, such as strings, ids, and serialized config.
 
 Safe pattern:
@@ -1140,11 +1143,11 @@ Unsafe pattern:
 ```typescript
 pi.registerCommand("handoff", {
   handler: async (_args, ctx) => {
-    const oldSessionManager = ctx.sessionManager;
+    const oldSession = ctx.session;
     await ctx.newSession({
       withSession: async (_ctx) => {
         // stale old objects: do not do this
-        oldSessionManager.getSessionFile();
+        oldSession.getSessionReference();
         pi.sendUserMessage("wrong");
       },
     });
@@ -1325,7 +1328,7 @@ pi.appendEntry("my-state", { count: 42 });
 
 // Restore on reload
 pi.on("session_start", async (_event, ctx) => {
-  for (const entry of ctx.sessionManager.getEntries()) {
+  for (const entry of ctx.session.getEntries()) {
     if (entry.type === "custom" && entry.customType === "my-state") {
       // Reconstruct from entry.data
     }
@@ -1363,8 +1366,8 @@ pi.setLabel(entryId, "checkpoint-before-refactor");
 // Clear a label
 pi.setLabel(entryId, undefined);
 
-// Read labels via sessionManager
-const label = ctx.sessionManager.getLabel(entryId);
+// Read labels via the active session
+const label = ctx.session.getLabel(entryId);
 ```
 
 Labels persist in the session and survive restarts. Use them to mark important points (turns, checkpoints) in the conversation tree.
@@ -1379,7 +1382,7 @@ If multiple extensions register the same command name, pi keeps them all and ass
 pi.registerCommand("stats", {
   description: "Show session statistics",
   handler: async (args, ctx) => {
-    const count = ctx.sessionManager.getEntries().length;
+    const count = ctx.session.getEntries().length;
     ctx.ui.notify(`${count} entries`, "info");
   }
 });
@@ -1634,7 +1637,7 @@ export default function (pi: ExtensionAPI) {
   // Reconstruct state from session
   pi.on("session_start", async (_event, ctx) => {
     items = [];
-    for (const entry of ctx.sessionManager.getBranch()) {
+    for (const entry of ctx.session.getBranch()) {
       if (entry.type === "message" && entry.message.role === "toolResult") {
         if (entry.message.toolName === "my_tool") {
           items = entry.message.details?.items ?? [];

package/docs/index.md

@@ -65,7 +65,7 @@ For the full first-run flow, see [Quickstart](quickstart.md).
 
 ## Reference
 
-- [Session format](session-format.md) - JSONL session file format, entry types, and SessionManager API.
+- [Session format](session-format.md) - JSONL session file format, entry types, and session APIs.
 
 ## Platform setup
 

package/docs/sdk.md

@@ -69,6 +69,8 @@ Use `pi.createAgentSession({ session })` when you already have an opened or crea
 
 `AgentSession` owns active conversation behavior: prompts, message history, model state, compaction, event streaming, and in-place tree navigation.
 
+Use `session.sessionReference` when you need the backend-neutral active session reference. For local JSONL sessions this is the session file path; for in-memory sessions it is `undefined`. `session.sessionFile` remains available as a local-file-oriented alias.
+
 Session replacement APIs such as new-session, resume, fork, clone, and import live on `PiAgent`, not on `AgentSession`.
 
 ### SessionManager and Session

package/docs/session-format.md

@@ -29,7 +29,7 @@ Existing sessions are automatically migrated to the current version (v3) when lo
 ## Source Files
 
 Source on GitHub ([pi-mono](https://github.com/earendil-works/pi-mono)):
-- [`packages/coding-agent/src/core/session-manager.ts`](https://github.com/earendil-works/pi-mono/blob/main/packages/coding-agent/src/core/session-manager.ts) - Session entry types and SessionManager
+- [`packages/coding-agent/src/core/session-manager.ts`](https://github.com/earendil-works/pi-mono/blob/main/packages/coding-agent/src/core/session-manager.ts) - Session entry types, `Session`, and `SessionManager`
 - [`packages/coding-agent/src/core/messages.ts`](https://github.com/earendil-works/pi-mono/blob/main/packages/coding-agent/src/core/messages.ts) - Extended message types (BashExecutionMessage, CustomMessage, etc.)
 - [`packages/ai/src/types.ts`](https://github.com/earendil-works/pi-mono/blob/main/packages/ai/src/types.ts) - Base message types (UserMessage, AssistantMessage, ToolResultMessage)
 - [`packages/agent/src/types.ts`](https://github.com/earendil-works/pi-mono/blob/main/packages/agent/src/types.ts) - AgentMessage union type
@@ -358,27 +358,27 @@ for (const line of lines) {
 }
 ```
 
-## SessionManager API
+## Programmatic APIs
 
-Key methods for working with sessions programmatically.
+`SessionManager` handles lifecycle and discovery. It creates, opens, forks, imports, and lists `Session` objects. `Session` represents one active conversation tree and owns entry append/read/navigation operations.
 
-### Static Creation Methods
-- `LocalSessionManager.create(cwd, sessionDir?)` - New session
-- `LocalSessionManager.openReference(path, sessionDir?)` - Open existing session file
-- `LocalSessionManager.continueRecent(cwd, sessionDir?)` - Continue most recent or create new
-- `InMemorySessionManager.create(cwd?)` - No file persistence
-- `LocalSessionManager.forkFrom(sourcePath, targetCwd, sessionDir?)` - Fork session from another project
+### SessionManager Methods
+- `create(options?)` - Create a new session
+- `openReference(reference, options?)` - Open an existing session reference
+- `continueRecent()` - Continue the most recent session or create a new one
+- `forkFrom(reference)` - Fork from an existing session reference
+- `forkSession(source, targetLeafId)` - Fork from an active `Session`
+- `importJsonl(inputPath, options?)` - Import a JSONL session file
+- `list(onProgress?)` - List sessions for the manager's scope
+- `listAll(onProgress?)` - List all sessions across scopes supported by the backend
 
-### Static Listing Methods
-- `LocalSessionManager.list(cwd, sessionDir?, onProgress?)` - List sessions for a directory
-- `LocalSessionManager.listAll(onProgress?)` - List all sessions across all projects
+### Session Methods - Lifecycle Helpers
+- `createSubSession(options?)` - Create a child session via the session's private manager
+- `forkSubSession(targetLeafId)` - Fork the active session via the session's private manager
+- `copyBranchFrom(source, leafId, parentSession?)` - Copy one branch into this session
+- `createBranchedSession(leafId)` - Extract branch to a new session reference; this mutates the current `Session` to point at the new reference
 
-### Instance Methods - Session Management
-- `newSession(options?)` - Start a new session (options: `{ parentSession?: string }`)
-- `setSessionFile(path)` - Switch to a different session file
-- `createBranchedSession(leafId)` - Extract branch to new session file
-
-### Instance Methods - Appending (all return entry ID)
+### Session Methods - Appending (all return entry ID)
 - `appendMessage(message)` - Add message
 - `appendThinkingLevelChange(level)` - Record thinking change
 - `appendModelChange(provider, modelId)` - Record model change
@@ -388,7 +388,7 @@ Key methods for working with sessions programmatically.
 - `appendCustomMessageEntry(customType, content, display, details?)` - Extension message (in context)
 - `appendLabelChange(targetId, label)` - Set/clear label
 
-### Instance Methods - Tree Navigation
+### Session Methods - Tree Navigation
 - `getLeafId()` - Current position
 - `getLeafEntry()` - Get current leaf entry
 - `getEntry(id)` - Get entry by ID
@@ -400,7 +400,7 @@ Key methods for working with sessions programmatically.
 - `resetLeaf()` - Reset leaf to null (before any entries)
 - `branchWithSummary(entryId, summary, details?, fromHook?)` - Branch with context summary
 
-### Instance Methods - Context & Info
+### Session Methods - Context & Info
 - `buildSessionContext()` - Get messages, thinkingLevel, and model for LLM
 - `getEntries()` - All entries (excluding header)
 - `getHeader()` - Session header metadata
@@ -408,5 +408,5 @@ Key methods for working with sessions programmatically.
 - `getCwd()` - Working directory
 - `getSessionDir()` - Session storage directory
 - `getSessionId()` - Session UUID
-- `getSessionFile()` - Session file path (undefined for in-memory)
+- `getSessionReference()` - Backend session reference (local file path for JSONL, remote reference for remote sessions, undefined for in-memory)
 - `isPersisted()` - Whether session is saved to disk

package/docs/sessions.md

@@ -16,7 +16,7 @@ pi --fork <path|id>    # Fork a session file or partial session ID into a new se
 
 Use `/session` in interactive mode to see the current session file, session ID, message count, tokens, and cost.
 
-For the JSONL file format and SessionManager API, see [Session Format](session-format.md).
+For the JSONL file format, active `Session` API, and `SessionManager` lifecycle API, see [Session Format](session-format.md).
 
 ## Session Commands
 
@@ -134,4 +134,4 @@ See [Compaction](compaction.md) for branch summarization internals and extension
 
 Session files are JSONL and contain message entries, model changes, thinking-level changes, labels, compactions, branch summaries, and extension entries.
 
-For parsers, extensions, SDK usage, and the full SessionManager API, see [Session Format](session-format.md).
+For parsers, extensions, SDK usage, and the full session APIs, see [Session Format](session-format.md).

package/docs/tui.md

@@ -813,7 +813,7 @@ ctx.ui.setFooter((tui, theme, footerData) => ({
 ctx.ui.setFooter(undefined); // restore default
 ```
 
-Token stats available via `ctx.sessionManager.getBranch()` and `ctx.model`.
+Token stats available via `ctx.session.getBranch()` and `ctx.model`.
 
 **Examples:** [custom-footer.ts](../examples/extensions/custom-footer.ts)
 

package/examples/README.md

@@ -18,6 +18,9 @@ Example extensions demonstrating:
 - External integrations (SSH, file watchers, system theme sync)
 - Custom providers (Anthropic with custom streaming, GitLab Duo)
 
+### [remote-session-server/](remote-session-server/)
+Hono-based remote session storage server. It uses static bearer auth and stores sessions as JSONL files for debugging the remote session protocol end-to-end.
+
 ## Documentation
 
 - [SDK Reference](sdk/README.md)

package/examples/extensions/README.md

@@ -198,7 +198,7 @@ return {
 
 // Reconstruct on session events
 pi.on("session_start", async (_event, ctx) => {
-  for (const entry of ctx.sessionManager.getBranch()) {
+  for (const entry of ctx.session.getBranch()) {
     if (entry.type === "message" && entry.message.toolName === "my_tool") {
       const details = entry.message.details;
       // Reconstruct state from details

package/examples/extensions/auto-commit-on-exit.ts

@@ -18,7 +18,7 @@ export default function (pi: ExtensionAPI) {
 		}
 
 		// Find the last assistant message for commit context
-		const entries = ctx.sessionManager.getEntries();
+		const entries = ctx.session.getEntries();
 		let lastAssistantText = "";
 		for (let i = entries.length - 1; i >= 0; i--) {
 			const entry = entries[i];

package/examples/extensions/bookmark.ts

@@ -16,7 +16,7 @@ export default function (pi: ExtensionAPI) {
 			const label = args.trim() || `bookmark-${Date.now()}`;
 
 			// Find the last assistant message entry
-			const entries = ctx.sessionManager.getEntries();
+			const entries = ctx.session.getEntries();
 			for (let i = entries.length - 1; i >= 0; i--) {
 				const entry = entries[i];
 				if (entry.type === "message" && entry.message.role === "assistant") {
@@ -34,10 +34,10 @@ export default function (pi: ExtensionAPI) {
 	pi.registerCommand("unbookmark", {
 		description: "Remove bookmark from last labeled entry",
 		handler: async (_args, ctx) => {
-			const entries = ctx.sessionManager.getEntries();
+			const entries = ctx.session.getEntries();
 			for (let i = entries.length - 1; i >= 0; i--) {
 				const entry = entries[i];
-				const label = ctx.sessionManager.getLabel(entry.id);
+				const label = ctx.session.getLabel(entry.id);
 				if (label) {
 					pi.setLabel(entry.id, undefined);
 					ctx.ui.notify(`Removed bookmark: ${label}`, "info");

package/examples/extensions/confirm-destructive.ts

@@ -25,7 +25,7 @@ export default function (pi: ExtensionAPI) {
 		}
 
 		// reason === "resume" - check if there are unsaved changes (messages since last assistant response)
-		const entries = ctx.sessionManager.getEntries();
+		const entries = ctx.session.getEntries();
 		const hasUnsavedWork = entries.some(
 			(e): e is SessionMessageEntry => e.type === "message" && e.message.role === "user",
 		);

package/examples/extensions/custom-compaction.ts

@@ -108,7 +108,7 @@ ${conversationText}
 				return;
 			}
 
-			// Return compaction content - SessionManager adds id/parentId
+			// Return compaction content - Session adds id/parentId
 			// Use firstKeptEntryId from preparation to keep recent messages
 			return {
 				compaction: {

package/examples/extensions/custom-footer.ts

@@ -5,7 +5,7 @@
  * - getGitBranch(): current git branch
  * - getExtensionStatuses(): texts from ctx.ui.setStatus()
  *
- * Token stats come from ctx.sessionManager/ctx.model (already accessible).
+ * Token stats come from ctx.session/ctx.model (already accessible).
  */
 
 import type { AssistantMessage } from "@fleetagent/pi-ai";
@@ -32,7 +32,7 @@ export default function (pi: ExtensionAPI) {
 							let input = 0,
 								output = 0,
 								cost = 0;
-							for (const e of ctx.sessionManager.getBranch()) {
+							for (const e of ctx.session.getBranch()) {
 								if (e.type === "message" && e.message.role === "assistant") {
 									const m = e.message as AssistantMessage;
 									input += m.usage.input;

package/examples/extensions/custom-provider-anthropic/package.json

@@ -1,7 +1,7 @@
 {
 	"name": "pi-extension-custom-provider-anthropic",
 	"private": true,
-	"version": "0.0.1",
+	"version": "0.0.2",
 	"type": "module",
 	"scripts": {
 		"clean": "echo 'nothing to clean'",

package/examples/extensions/custom-provider-gitlab-duo/package.json

@@ -1,7 +1,7 @@
 {
 	"name": "pi-extension-custom-provider-gitlab-duo",
 	"private": true,
-	"version": "0.0.1",
+	"version": "0.0.2",
 	"type": "module",
 	"scripts": {
 		"clean": "echo 'nothing to clean'",

package/examples/extensions/git-checkpoint.ts

@@ -13,7 +13,7 @@ export default function (pi: ExtensionAPI) {
 
 	// Track the current entry ID when user messages are saved
 	pi.on("tool_result", async (_event, ctx) => {
-		const leaf = ctx.sessionManager.getLeafEntry();
+		const leaf = ctx.session.getLeafEntry();
 		if (leaf) currentEntryId = leaf.id;
 	});
 

package/examples/extensions/handoff.ts

@@ -99,7 +99,7 @@ export default function (pi: ExtensionAPI) {
 
 			// Gather conversation context from current branch. If the branch was compacted,
 			// include the compaction summary plus entries from firstKeptEntryId onward.
-			const messages = getHandoffMessages(ctx.sessionManager.getBranch());
+			const messages = getHandoffMessages(ctx.session.getBranch());
 
 			if (messages.length === 0) {
 				ctx.ui.notify("No conversation to hand off", "error");
@@ -109,7 +109,7 @@ export default function (pi: ExtensionAPI) {
 			// Convert to LLM format and serialize
 			const llmMessages = convertToLlm(messages);
 			const conversationText = serializeConversation(llmMessages);
-			const currentSessionFile = ctx.sessionManager.getSessionReference();
+			const currentSessionFile = ctx.session.getSessionReference();
 
 			// Generate the handoff prompt with loader UI
 			const result = await ctx.ui.custom<string | null>((tui, theme, _kb, done) => {

package/examples/extensions/plan-mode/index.ts

@@ -293,7 +293,7 @@ After completing a step, include a [DONE:n] tag in your response.`,
 			planModeEnabled = true;
 		}
 
-		const entries = ctx.sessionManager.getEntries();
+		const entries = ctx.session.getEntries();
 
 		// Restore persisted state
 		const planModeEntry = entries

package/examples/extensions/preset.ts

@@ -404,7 +404,7 @@ export default function presetExtension(pi: ExtensionAPI) {
 		}
 
 		// Restore preset from session state
-		const entries = ctx.sessionManager.getEntries();
+		const entries = ctx.session.getEntries();
 		const presetEntry = entries
 			.filter((e: { type: string; customType?: string }) => e.type === "custom" && e.customType === "preset-state")
 			.pop() as { data?: { name: string } } | undefined;

package/examples/extensions/qna.ts

@@ -42,7 +42,7 @@ export default function (pi: ExtensionAPI) {
 			}
 
 			// Find the last assistant message on the current branch
-			const branch = ctx.sessionManager.getBranch();
+			const branch = ctx.session.getBranch();
 			let lastAssistantText: string | undefined;
 
 			for (let i = branch.length - 1; i >= 0; i--) {

package/examples/extensions/sandbox/package.json

@@ -1,7 +1,7 @@
 {
 	"name": "pi-extension-sandbox",
 	"private": true,
-	"version": "0.0.1",
+	"version": "0.0.2",
 	"type": "module",
 	"scripts": {
 		"clean": "echo 'nothing to clean'",

package/examples/extensions/snake.ts

@@ -317,7 +317,7 @@ export default function (pi: ExtensionAPI) {
 			}
 
 			// Load saved state from session
-			const entries = ctx.sessionManager.getEntries();
+			const entries = ctx.session.getEntries();
 			let savedState: GameState | undefined;
 			for (let i = entries.length - 1; i >= 0; i--) {
 				const entry = entries[i];

package/examples/extensions/space-invaders.ts

@@ -535,7 +535,7 @@ export default function (pi: ExtensionAPI) {
 			}
 
 			// Load saved state from session
-			const entries = ctx.sessionManager.getEntries();
+			const entries = ctx.session.getEntries();
 			let savedState: GameState | undefined;
 			for (let i = entries.length - 1; i >= 0; i--) {
 				const entry = entries[i];

package/examples/extensions/summarize.ts

@@ -146,7 +146,7 @@ export default function (pi: ExtensionAPI) {
 	pi.registerCommand("summarize", {
 		description: "Summarize the current conversation in a custom UI",
 		handler: async (_args, ctx) => {
-			const branch = ctx.sessionManager.getBranch();
+			const branch = ctx.session.getBranch();
 			const conversationText = buildConversationText(branch);
 
 			if (!conversationText.trim()) {

package/examples/extensions/tic-tac-toe.ts

@@ -628,7 +628,7 @@ function reconstructState(ctx: ExtensionContext): void {
 	gameState = createInitialState();
 	gameActive = false;
 
-	for (const entry of ctx.sessionManager.getBranch()) {
+	for (const entry of ctx.session.getBranch()) {
 		if (entry.type !== "message") continue;
 		const msg = entry.message;
 		if (msg.role !== "toolResult") continue;

package/examples/extensions/todo.ts

@@ -115,7 +115,7 @@ export default function (pi: ExtensionAPI) {
 		todos = [];
 		nextId = 1;
 
-		for (const entry of ctx.sessionManager.getBranch()) {
+		for (const entry of ctx.session.getBranch()) {
 			if (entry.type !== "message") continue;
 			const msg = entry.message;
 			if (msg.role !== "toolResult" || msg.toolName !== "todo") continue;

package/examples/extensions/tools.ts

@@ -40,7 +40,7 @@ export default function toolsExtension(pi: ExtensionAPI) {
 		allTools = pi.getAllTools();
 
 		// Get entries in current branch only
-		const branchEntries = ctx.sessionManager.getBranch();
+		const branchEntries = ctx.session.getBranch();
 		let savedTools: string[] | undefined;
 
 		for (const entry of branchEntries) {

package/examples/extensions/with-deps/package.json

@@ -1,7 +1,7 @@
 {
 	"name": "pi-extension-with-deps",
 	"private": true,
-	"version": "0.0.1",
+	"version": "0.0.2",
 	"type": "module",
 	"scripts": {
 		"clean": "echo 'nothing to clean'",