@posthog/agent 2.3.398 → 2.3.403

package/dist/types.d.ts

@@ -41,7 +41,7 @@ interface Task {
     };
     latest_run?: TaskRun;
 }
-type ArtifactType = "plan" | "context" | "reference" | "output" | "artifact" | "tree_snapshot" | "user_attachment";
+type ArtifactType = "plan" | "context" | "reference" | "output" | "artifact" | "user_attachment";
 interface TaskRunArtifact {
     id?: string;
     name: string;
@@ -137,17 +137,6 @@ interface FileChange {
     path: string;
     status: FileStatus;
 }
-interface TreeSnapshot {
-    treeHash: string;
-    baseCommit: string | null;
-    archiveUrl?: string;
-    changes: FileChange[];
-    timestamp: string;
-    interrupted?: boolean;
-}
-interface TreeSnapshotEvent extends TreeSnapshot {
-    device?: DeviceInfo;
-}
 type HandoffLocalGitState = HandoffLocalGitState$1;
 interface GitCheckpoint extends GitHandoffCheckpoint {
     artifactPath?: string;
@@ -164,4 +153,4 @@ interface GitCheckpointEvent extends GitCheckpoint {
  */
 declare const AGENT_TYPES_MODULE = true;
 
-export { AGENT_TYPES_MODULE, type AgentConfig, type AgentMode, type ArtifactType, type DeviceInfo, type FileChange, type FileStatus, type GitCheckpoint, type GitCheckpointEvent, type HandoffLocalGitState, type LogLevel, type OnLogCallback, type OtelTransportConfig, type PostHogAPIConfig, type ProcessSpawnedCallback, type StoredEntry, type StoredNotification, type Task, type TaskExecutionOptions, type TaskRun, type TaskRunArtifact, type TaskRunEnvironment, type TaskRunStatus, type TreeSnapshot, type TreeSnapshotEvent };
+export { AGENT_TYPES_MODULE, type AgentConfig, type AgentMode, type ArtifactType, type DeviceInfo, type FileChange, type FileStatus, type GitCheckpoint, type GitCheckpointEvent, type HandoffLocalGitState, type LogLevel, type OnLogCallback, type OtelTransportConfig, type PostHogAPIConfig, type ProcessSpawnedCallback, type StoredEntry, type StoredNotification, type Task, type TaskExecutionOptions, type TaskRun, type TaskRunArtifact, type TaskRunEnvironment, type TaskRunStatus };

package/dist/types.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/types.ts"],"sourcesContent":["import type {\n  GitHandoffCheckpoint,\n  HandoffLocalGitState as GitHandoffLocalGitState,\n} from \"@posthog/git/handoff\";\n\n/**\n * Stored custom notification following ACP extensibility model.\n * Custom notifications use underscore-prefixed methods (e.g., `_posthog/phase_start`).\n * See: https://agentclientprotocol.com/docs/extensibility\n */\nexport interface StoredNotification {\n  type: \"notification\";\n  /** When this notification was stored */\n  timestamp: string;\n  /** JSON-RPC 2.0 notification (no id field = notification, not request) */\n  notification: {\n    jsonrpc: \"2.0\";\n    method: string;\n    params?: Record<string, unknown>;\n  };\n}\n\n/**\n * Type alias for stored log entries.\n */\nexport type StoredEntry = StoredNotification;\n\n// PostHog Task model (matches PostHog Code's OpenAPI schema)\nexport interface Task {\n  id: string;\n  task_number?: number;\n  slug?: string;\n  title: string;\n  description: string;\n  origin_product:\n    | \"error_tracking\"\n    | \"eval_clusters\"\n    | \"user_created\"\n    | \"support_queue\"\n    | \"session_summaries\";\n  github_integration?: number | null;\n  repository: string; // Format: \"organization/repository\" (e.g., \"posthog/posthog-js\")\n  json_schema?: Record<string, unknown> | null; // JSON schema for task output validation\n  created_at: string;\n  updated_at: string;\n  created_by?: {\n    id: number;\n    uuid: string;\n    distinct_id: string;\n    first_name: string;\n    email: string;\n  };\n  latest_run?: TaskRun;\n}\n\n// Log entry structure for TaskRun.log\n\nexport type ArtifactType =\n  | \"plan\"\n  | \"context\"\n  | \"reference\"\n  | \"output\"\n  | \"artifact\"\n  | \"tree_snapshot\"\n  | \"user_attachment\";\n\nexport interface TaskRunArtifact {\n  id?: string;\n  name: string;\n  type: ArtifactType;\n  source?: string;\n  size?: number;\n  content_type?: string;\n  storage_path?: string;\n  uploaded_at?: string;\n}\n\nexport type TaskRunStatus =\n  | \"not_started\"\n  | \"queued\"\n  | \"in_progress\"\n  | \"completed\"\n  | \"failed\"\n  | \"cancelled\";\n\nexport type TaskRunEnvironment = \"local\" | \"cloud\";\n\n// TaskRun model - represents individual execution runs of tasks\nexport interface TaskRun {\n  id: string;\n  task: string; // Task ID\n  team: number;\n  branch: string | null;\n  stage: string | null; // Current stage (e.g., 'research', 'plan', 'build')\n  environment: TaskRunEnvironment;\n  status: TaskRunStatus;\n  log_url: string;\n  error_message: string | null;\n  output: Record<string, unknown> | null; // Structured output (PR URL, commit SHA, etc.)\n  state: Record<string, unknown>; // Intermediate run state (defaults to {}, never null)\n  artifacts?: TaskRunArtifact[];\n  created_at: string;\n  updated_at: string;\n  completed_at: string | null;\n}\n\nexport interface ProcessSpawnedCallback {\n  onProcessSpawned?: (info: {\n    pid: number;\n    command: string;\n    sessionId?: string;\n  }) => void;\n  onProcessExited?: (pid: number) => void;\n  onMcpServersReady?: (serverNames: string[]) => void;\n}\n\nexport interface TaskExecutionOptions {\n  repositoryPath?: string;\n  adapter?: \"claude\" | \"codex\";\n  model?: string;\n  gatewayUrl?: string;\n  codexBinaryPath?: string;\n  instructions?: string;\n  processCallbacks?: ProcessSpawnedCallback;\n  /** Callback invoked when the agent calls the create_output tool for structured output */\n  onStructuredOutput?: (output: Record<string, unknown>) => Promise<void>;\n}\n\nexport type LogLevel = \"debug\" | \"info\" | \"warn\" | \"error\";\n\nexport type OnLogCallback = (\n  level: LogLevel,\n  scope: string,\n  message: string,\n  data?: unknown,\n) => void;\n\nexport interface PostHogAPIConfig {\n  apiUrl: string;\n  getApiKey: () => string | Promise<string>;\n  refreshApiKey?: () => string | Promise<string>;\n  projectId: number;\n  userAgent?: string;\n}\n\nexport interface OtelTransportConfig {\n  /** PostHog ingest host, e.g., \"https://us.i.posthog.com\" */\n  host: string;\n  /** Project API key */\n  apiKey: string;\n  /** Override the logs endpoint path (default: /i/v1/logs) */\n  logsPath?: string;\n}\n\nexport interface AgentConfig {\n  posthog?: PostHogAPIConfig;\n  /** OTEL transport config for shipping logs to PostHog Logs */\n  otelTransport?: OtelTransportConfig;\n  /** Skip session log persistence (e.g. for preview sessions with no real task) */\n  skipLogPersistence?: boolean;\n  /** Local cache path for instant log loading (e.g., ~/.posthog-code) */\n  localCachePath?: string;\n  /**\n   * Annotate files the agent reads with PostHog enrichment (event volume,\n   * flag rollout/staleness, experiment links). Defaults to enabled when\n   * `posthog` config is present; set `{ enabled: false }` to opt out.\n   */\n  enricher?: { enabled?: boolean };\n  debug?: boolean;\n  onLog?: OnLogCallback;\n}\n\n// Device info for tracking where work happens\nexport interface DeviceInfo {\n  type: \"local\" | \"cloud\";\n  name?: string;\n}\n\n// Agent execution mode - for tracking interactive vs background runs, when backgrounded an agent will continue working without asking questions\nexport type AgentMode = \"interactive\" | \"background\";\n\n// Git file status codes\nexport type FileStatus = \"A\" | \"M\" | \"D\";\n\nexport interface FileChange {\n  path: string;\n  status: FileStatus;\n}\n\n// Tree snapshot - what TreeTracker captures\nexport interface TreeSnapshot {\n  treeHash: string;\n  baseCommit: string | null;\n  archiveUrl?: string;\n  changes: FileChange[];\n  timestamp: string;\n  interrupted?: boolean;\n}\n\n// Tree snapshot event - includes device info when sent as notification\nexport interface TreeSnapshotEvent extends TreeSnapshot {\n  device?: DeviceInfo;\n}\n\nexport type HandoffLocalGitState = GitHandoffLocalGitState;\n\nexport interface GitCheckpoint extends GitHandoffCheckpoint {\n  artifactPath?: string;\n  indexArtifactPath?: string;\n}\n\nexport interface GitCheckpointEvent extends GitCheckpoint {\n  device?: DeviceInfo;\n}\n\n/**\n * Keeps the emitted `@posthog/agent/types` entrypoint as a runtime ESM module.\n *\n * `export {}` is stripped by tsup in this package, which leaves `dist/types.js`\n * empty and breaks downstream type resolution for the exported subpath.\n */\nexport const AGENT_TYPES_MODULE = true;\n"],"mappings":";AA6NO,IAAM,qBAAqB;","names":[]}
+{"version":3,"sources":["../src/types.ts"],"sourcesContent":["import type {\n  GitHandoffCheckpoint,\n  HandoffLocalGitState as GitHandoffLocalGitState,\n} from \"@posthog/git/handoff\";\n\n/**\n * Stored custom notification following ACP extensibility model.\n * Custom notifications use underscore-prefixed methods (e.g., `_posthog/phase_start`).\n * See: https://agentclientprotocol.com/docs/extensibility\n */\nexport interface StoredNotification {\n  type: \"notification\";\n  /** When this notification was stored */\n  timestamp: string;\n  /** JSON-RPC 2.0 notification (no id field = notification, not request) */\n  notification: {\n    jsonrpc: \"2.0\";\n    method: string;\n    params?: Record<string, unknown>;\n  };\n}\n\n/**\n * Type alias for stored log entries.\n */\nexport type StoredEntry = StoredNotification;\n\n// PostHog Task model (matches PostHog Code's OpenAPI schema)\nexport interface Task {\n  id: string;\n  task_number?: number;\n  slug?: string;\n  title: string;\n  description: string;\n  origin_product:\n    | \"error_tracking\"\n    | \"eval_clusters\"\n    | \"user_created\"\n    | \"support_queue\"\n    | \"session_summaries\";\n  github_integration?: number | null;\n  repository: string; // Format: \"organization/repository\" (e.g., \"posthog/posthog-js\")\n  json_schema?: Record<string, unknown> | null; // JSON schema for task output validation\n  created_at: string;\n  updated_at: string;\n  created_by?: {\n    id: number;\n    uuid: string;\n    distinct_id: string;\n    first_name: string;\n    email: string;\n  };\n  latest_run?: TaskRun;\n}\n\n// Log entry structure for TaskRun.log\n\nexport type ArtifactType =\n  | \"plan\"\n  | \"context\"\n  | \"reference\"\n  | \"output\"\n  | \"artifact\"\n  | \"user_attachment\";\n\nexport interface TaskRunArtifact {\n  id?: string;\n  name: string;\n  type: ArtifactType;\n  source?: string;\n  size?: number;\n  content_type?: string;\n  storage_path?: string;\n  uploaded_at?: string;\n}\n\nexport type TaskRunStatus =\n  | \"not_started\"\n  | \"queued\"\n  | \"in_progress\"\n  | \"completed\"\n  | \"failed\"\n  | \"cancelled\";\n\nexport type TaskRunEnvironment = \"local\" | \"cloud\";\n\n// TaskRun model - represents individual execution runs of tasks\nexport interface TaskRun {\n  id: string;\n  task: string; // Task ID\n  team: number;\n  branch: string | null;\n  stage: string | null; // Current stage (e.g., 'research', 'plan', 'build')\n  environment: TaskRunEnvironment;\n  status: TaskRunStatus;\n  log_url: string;\n  error_message: string | null;\n  output: Record<string, unknown> | null; // Structured output (PR URL, commit SHA, etc.)\n  state: Record<string, unknown>; // Intermediate run state (defaults to {}, never null)\n  artifacts?: TaskRunArtifact[];\n  created_at: string;\n  updated_at: string;\n  completed_at: string | null;\n}\n\nexport interface ProcessSpawnedCallback {\n  onProcessSpawned?: (info: {\n    pid: number;\n    command: string;\n    sessionId?: string;\n  }) => void;\n  onProcessExited?: (pid: number) => void;\n  onMcpServersReady?: (serverNames: string[]) => void;\n}\n\nexport interface TaskExecutionOptions {\n  repositoryPath?: string;\n  adapter?: \"claude\" | \"codex\";\n  model?: string;\n  gatewayUrl?: string;\n  codexBinaryPath?: string;\n  instructions?: string;\n  processCallbacks?: ProcessSpawnedCallback;\n  /** Callback invoked when the agent calls the create_output tool for structured output */\n  onStructuredOutput?: (output: Record<string, unknown>) => Promise<void>;\n}\n\nexport type LogLevel = \"debug\" | \"info\" | \"warn\" | \"error\";\n\nexport type OnLogCallback = (\n  level: LogLevel,\n  scope: string,\n  message: string,\n  data?: unknown,\n) => void;\n\nexport interface PostHogAPIConfig {\n  apiUrl: string;\n  getApiKey: () => string | Promise<string>;\n  refreshApiKey?: () => string | Promise<string>;\n  projectId: number;\n  userAgent?: string;\n}\n\nexport interface OtelTransportConfig {\n  /** PostHog ingest host, e.g., \"https://us.i.posthog.com\" */\n  host: string;\n  /** Project API key */\n  apiKey: string;\n  /** Override the logs endpoint path (default: /i/v1/logs) */\n  logsPath?: string;\n}\n\nexport interface AgentConfig {\n  posthog?: PostHogAPIConfig;\n  /** OTEL transport config for shipping logs to PostHog Logs */\n  otelTransport?: OtelTransportConfig;\n  /** Skip session log persistence (e.g. for preview sessions with no real task) */\n  skipLogPersistence?: boolean;\n  /** Local cache path for instant log loading (e.g., ~/.posthog-code) */\n  localCachePath?: string;\n  /**\n   * Annotate files the agent reads with PostHog enrichment (event volume,\n   * flag rollout/staleness, experiment links). Defaults to enabled when\n   * `posthog` config is present; set `{ enabled: false }` to opt out.\n   */\n  enricher?: { enabled?: boolean };\n  debug?: boolean;\n  onLog?: OnLogCallback;\n}\n\n// Device info for tracking where work happens\nexport interface DeviceInfo {\n  type: \"local\" | \"cloud\";\n  name?: string;\n}\n\n// Agent execution mode - for tracking interactive vs background runs, when backgrounded an agent will continue working without asking questions\nexport type AgentMode = \"interactive\" | \"background\";\n\n// Git file status codes\nexport type FileStatus = \"A\" | \"M\" | \"D\";\n\nexport interface FileChange {\n  path: string;\n  status: FileStatus;\n}\n\nexport type HandoffLocalGitState = GitHandoffLocalGitState;\n\nexport interface GitCheckpoint extends GitHandoffCheckpoint {\n  artifactPath?: string;\n  indexArtifactPath?: string;\n}\n\nexport interface GitCheckpointEvent extends GitCheckpoint {\n  device?: DeviceInfo;\n}\n\n/**\n * Keeps the emitted `@posthog/agent/types` entrypoint as a runtime ESM module.\n *\n * `export {}` is stripped by tsup in this package, which leaves `dist/types.js`\n * empty and breaks downstream type resolution for the exported subpath.\n */\nexport const AGENT_TYPES_MODULE = true;\n"],"mappings":";AA6MO,IAAM,qBAAqB;","names":[]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@posthog/agent",
-  "version": "2.3.398",
+  "version": "2.3.403",
   "repository": "https://github.com/PostHog/code",
   "description": "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   "exports": {
@@ -68,10 +68,6 @@
       "types": "./dist/handoff-checkpoint.d.ts",
       "import": "./dist/handoff-checkpoint.js"
     },
-    "./tree-tracker": {
-      "types": "./dist/tree-tracker.d.ts",
-      "import": "./dist/tree-tracker.js"
-    },
     "./server": {
       "types": "./dist/server/agent-server.d.ts",
       "import": "./dist/server/agent-server.js"
@@ -106,9 +102,9 @@
     "tsx": "^4.20.6",
     "typescript": "^5.5.0",
     "vitest": "^2.1.8",
+    "@posthog/enricher": "1.0.0",
     "@posthog/shared": "1.0.0",
-    "@posthog/git": "1.0.0",
-    "@posthog/enricher": "1.0.0"
+    "@posthog/git": "1.0.0"
   },
   "dependencies": {
     "@agentclientprotocol/sdk": "0.19.0",

package/src/acp-extensions.ts

@@ -34,9 +34,6 @@ export const POSTHOG_NOTIFICATIONS = {
   /** Maps taskRunId to agent's sessionId and adapter type (for resumption) */
   SDK_SESSION: "_posthog/sdk_session",
 
-  /** Tree state snapshot captured (git tree hash + file archive) */
-  TREE_SNAPSHOT: "_posthog/tree_snapshot",
-
   /** Git checkpoint captured for handoff */
   GIT_CHECKPOINT: "_posthog/git_checkpoint",
 

package/src/handoff-checkpoint.test.ts

@@ -20,8 +20,6 @@ interface HandoffRepos {
   localGitState: HandoffLocalGitState;
 }
 
-const WORKTREE_FILES = ["tracked.txt", "unstaged.txt", "untracked.txt"];
-
 function createMockApi(store: BundleStore) {
   return {
     uploadTaskArtifacts: async (
@@ -64,7 +62,7 @@ function createBundleStore(): BundleStore {
     artifacts: {},
     manifest: [
       {
-        storage_path: "gs://bucket/handoff-0-existing-tree_snapshot.tar.gz",
+        storage_path: "gs://bucket/handoff-0-existing-checkpoint.pack",
       },
     ],
   };
@@ -128,15 +126,6 @@ async function makeCloudChanges(repo: TestRepo): Promise<void> {
   await repo.writeFile("untracked.txt", "untracked\n");
 }
 
-async function mirrorRestoredWorktree(
-  cloudRepo: TestRepo,
-  localRepo: TestRepo,
-): Promise<void> {
-  for (const file of WORKTREE_FILES) {
-    await localRepo.writeFile(file, await cloudRepo.readFile(file));
-  }
-}
-
 describe("HandoffCheckpointTracker", () => {
   const cleanups: Array<() => Promise<void>> = [];
 
@@ -144,7 +133,7 @@ describe("HandoffCheckpointTracker", () => {
     await Promise.all(cleanups.splice(0).map((cleanup) => cleanup()));
   });
 
-  it("restores head commit and index state for handoff replay", async () => {
+  it("restores head, worktree, and index state for handoff replay", async () => {
     const { cloudRepo, localRepo, branch, localGitState } =
       await prepareHandoffRepos(cleanups);
     await makeCloudChanges(cloudRepo);
@@ -162,10 +151,6 @@ describe("HandoffCheckpointTracker", () => {
     const applyTracker = createTracker(localRepo.path, apiClient);
     await applyTracker.applyFromHandoff(checkpoint);
 
-    // The handoff service restores files separately via tree_snapshot.
-    // Mirror that here so the restored git index can be validated.
-    await mirrorRestoredWorktree(cloudRepo, localRepo);
-
     expect(await localRepo.git(["rev-parse", "HEAD"])).toBe(checkpoint.head);
     expect(await localRepo.git(["rev-parse", "--abbrev-ref", "HEAD"])).toBe(
       branch,
@@ -179,5 +164,6 @@ describe("HandoffCheckpointTracker", () => {
     expect(status).toContain("M  tracked.txt");
     expect(status).toContain(" M unstaged.txt");
     expect(status).toContain("?? untracked.txt");
+    expect(localRepo.exists(".posthog/tmp")).toBe(false);
   });
 });

package/src/handoff-checkpoint.ts

@@ -1,5 +1,6 @@
-import { mkdir, readFile, rm, writeFile } from "node:fs/promises";
-import { join } from "node:path";
+import { mkdtemp, readFile, rm, writeFile } from "node:fs/promises";
+import { tmpdir } from "node:os";
+import { dirname, join } from "node:path";
 import {
   type GitHandoffBranchDivergence,
   type GitHandoffCheckpoint,
@@ -100,8 +101,12 @@ export class HandoffCheckpointTracker {
         indexArtifactPath: uploads.index?.storagePath,
       };
     } finally {
+      const tempDir = capture.headPack?.path
+        ? dirname(capture.headPack.path)
+        : dirname(capture.indexFile.path);
       await this.removeIfPresent(capture.headPack?.path);
       await this.removeIfPresent(capture.indexFile.path);
+      await rm(tempDir, { recursive: true, force: true }).catch(() => {});
     }
   }
 
@@ -121,8 +126,9 @@ export class HandoffCheckpointTracker {
     }
 
     const gitTracker = this.createGitTracker();
-    const tmpDir = join(this.repositoryPath, ".posthog", "tmp");
-    await mkdir(tmpDir, { recursive: true });
+    const tmpDir = await mkdtemp(
+      join(tmpdir(), `posthog-code-handoff-${checkpoint.checkpointId}-`),
+    );
 
     const packPath = join(tmpDir, `${checkpoint.checkpointId}.pack`);
     const indexPath = join(tmpDir, `${checkpoint.checkpointId}.index`);
@@ -161,6 +167,7 @@ export class HandoffCheckpointTracker {
     } finally {
       await this.removeIfPresent(packPath);
       await this.removeIfPresent(indexPath);
+      await rm(tmpDir, { recursive: true, force: true }).catch(() => {});
     }
   }
 
@@ -294,52 +301,24 @@ export class HandoffCheckpointTracker {
     uploads: Uploads,
   ): void {
     this.logger.info("Captured handoff checkpoint", {
-      checkpointId: checkpoint.checkpointId,
       branch: checkpoint.branch,
-      head: checkpoint.head,
-      artifactPath: uploads.pack?.storagePath,
-      indexArtifactPath: uploads.index?.storagePath,
-      ...this.buildMetricPayload(uploads),
+      head: checkpoint.head?.slice(0, 7),
+      totalBytes: this.sumRawBytes(uploads.pack, uploads.index),
     });
   }
 
   private logApplyMetrics(
     checkpoint: GitCheckpoint,
-    downloads: Downloads,
+    _downloads: Downloads,
     totalBytes: number,
   ): void {
     this.logger.info("Applied handoff checkpoint", {
-      checkpointId: checkpoint.checkpointId,
-      commit: checkpoint.commit,
       branch: checkpoint.branch,
-      head: checkpoint.head,
-      packBytes: downloads.pack?.rawBytes ?? 0,
-      packWireBytes: downloads.pack?.wireBytes ?? 0,
-      indexBytes: downloads.index?.rawBytes ?? 0,
-      indexWireBytes: downloads.index?.wireBytes ?? 0,
+      head: checkpoint.head?.slice(0, 7),
       totalBytes,
-      totalWireBytes: this.sumWireBytes(downloads.pack, downloads.index),
     });
   }
 
-  private buildMetricPayload(metrics: ArtifactSlotMap<object>): {
-    packBytes: number;
-    packWireBytes: number;
-    indexBytes: number;
-    indexWireBytes: number;
-    totalBytes: number;
-    totalWireBytes: number;
-  } {
-    return {
-      packBytes: metrics.pack?.rawBytes ?? 0,
-      packWireBytes: metrics.pack?.wireBytes ?? 0,
-      indexBytes: metrics.index?.rawBytes ?? 0,
-      indexWireBytes: metrics.index?.wireBytes ?? 0,
-      totalBytes: this.sumRawBytes(metrics.pack, metrics.index),
-      totalWireBytes: this.sumWireBytes(metrics.pack, metrics.index),
-    };
-  }
-
   private sumRawBytes(
     ...artifacts: Array<{ rawBytes: number } | undefined>
   ): number {
@@ -349,15 +328,6 @@ export class HandoffCheckpointTracker {
     );
   }
 
-  private sumWireBytes(
-    ...artifacts: Array<{ wireBytes: number } | undefined>
-  ): number {
-    return artifacts.reduce(
-      (total, artifact) => total + (artifact?.wireBytes ?? 0),
-      0,
-    );
-  }
-
   private async removeIfPresent(filePath: string | undefined): Promise<void> {
     if (!filePath) {
       return;

package/src/resume.ts

@@ -3,15 +3,15 @@
  *
  * Handles resuming a task from any point:
  * - Fetches log via the PostHog API
- * - Finds latest tree_snapshot event
+ * - Finds latest git_checkpoint event
  * - Rebuilds conversation from log events
- * - Restores working tree from snapshot
+ * - Restores working tree from checkpoint
  *
  * Uses Saga pattern for atomic operations with clear success/failure tracking.
  *
  * The log is the single source of truth for:
  * - Conversation history (user_message, agent_message_chunk, tool_call, tool_result)
- * - Working tree state (tree_snapshot events)
+ * - Working tree state (git_checkpoint events)
  * - Session metadata (device info, mode changes)
  */
 
@@ -19,16 +19,11 @@ import type { ContentBlock } from "@agentclientprotocol/sdk";
 import { selectRecentTurns } from "./adapters/claude/session/jsonl-hydration";
 import type { PostHogAPIClient } from "./posthog-api";
 import { ResumeSaga } from "./sagas/resume-saga";
-import type {
-  DeviceInfo,
-  GitCheckpointEvent,
-  TreeSnapshotEvent,
-} from "./types";
+import type { DeviceInfo, GitCheckpointEvent } from "./types";
 import { Logger } from "./utils/logger";
 
 export interface ResumeState {
   conversation: ConversationTurn[];
-  latestSnapshot: TreeSnapshotEvent | null;
   latestGitCheckpoint: GitCheckpointEvent | null;
   interrupted: boolean;
   lastDevice?: DeviceInfo;
@@ -59,7 +54,7 @@ export interface ResumeConfig {
 /**
  * Resume a task from its persisted log.
  * Returns the rebuilt state for the agent to continue from.
- * Snapshot and checkpoint application happens in the agent server after SSE connects.
+ * Checkpoint application happens in the agent server after SSE connects.
  */
 export async function resumeFromLog(
   config: ResumeConfig,
@@ -94,7 +89,6 @@ export async function resumeFromLog(
 
   return {
     conversation: result.data.conversation as ConversationTurn[],
-    latestSnapshot: result.data.latestSnapshot,
     latestGitCheckpoint: result.data.latestGitCheckpoint,
     interrupted: result.data.interrupted,
     lastDevice: result.data.lastDevice,

package/src/sagas/resume-saga.test.ts

@@ -1,19 +1,17 @@
 import type { SagaLogger } from "@posthog/shared";
 import { afterEach, beforeEach, describe, expect, it, type vi } from "vitest";
-import { POSTHOG_NOTIFICATIONS } from "../acp-extensions";
 import type { PostHogAPIClient } from "../posthog-api";
 import { ResumeSaga } from "./resume-saga";
 import {
   createAgentChunk,
   createAgentMessage,
+  createGitCheckpointNotification,
   createMockApiClient,
   createMockLogger,
-  createNotification,
   createTaskRun,
   createTestRepo,
   createToolCall,
   createToolResult,
-  createTreeSnapshotNotification,
   createUserMessage,
   type TestRepo,
 } from "./test-fixtures";
@@ -50,7 +48,7 @@ describe("ResumeSaga", () => {
       expect(result.success).toBe(true);
       if (result.success) {
         expect(result.data.conversation).toHaveLength(0);
-        expect(result.data.latestSnapshot).toBeNull();
+        expect(result.data.latestGitCheckpoint).toBeNull();
         expect(result.data.logEntryCount).toBe(0);
       }
     });
@@ -412,76 +410,24 @@ describe("ResumeSaga", () => {
     });
   });
 
-  describe("snapshot finding", () => {
-    it("finds latest tree snapshot", async () => {
+  describe("checkpoint finding", () => {
+    it("finds latest git checkpoint", async () => {
       (mockApiClient.getTaskRun as ReturnType<typeof vi.fn>).mockResolvedValue(
         createTaskRun(),
       );
       (
         mockApiClient.fetchTaskRunLogs as ReturnType<typeof vi.fn>
       ).mockResolvedValue([
-        createTreeSnapshotNotification("hash-1"),
-        createUserMessage("continue"),
-        createTreeSnapshotNotification("hash-2", "gs://bucket/hash-2.tar.gz"),
-      ]);
-
-      const saga = new ResumeSaga(mockLogger);
-      const result = await saga.run({
-        taskId: "task-1",
-        runId: "run-1",
-        repositoryPath: repo.path,
-        apiClient: mockApiClient,
-      });
-
-      expect(result.success).toBe(true);
-      if (!result.success) return;
-
-      expect(result.data.latestSnapshot?.treeHash).toBe("hash-2");
-    });
-
-    it("returns interrupted flag from snapshot", async () => {
-      (mockApiClient.getTaskRun as ReturnType<typeof vi.fn>).mockResolvedValue(
-        createTaskRun(),
-      );
-      (
-        mockApiClient.fetchTaskRunLogs as ReturnType<typeof vi.fn>
-      ).mockResolvedValue([
-        createTreeSnapshotNotification("hash-1", "gs://bucket/file.tar.gz", {
-          interrupted: true,
+        createGitCheckpointNotification({
+          checkpointId: "checkpoint-1",
+          checkpointRef: "refs/posthog-code-checkpoint/checkpoint-1",
+          head: "head-1",
         }),
-      ]);
-
-      const saga = new ResumeSaga(mockLogger);
-      const result = await saga.run({
-        taskId: "task-1",
-        runId: "run-1",
-        repositoryPath: repo.path,
-        apiClient: mockApiClient,
-      });
-
-      expect(result.success).toBe(true);
-      if (!result.success) return;
-
-      expect(result.data.interrupted).toBe(true);
-    });
-  });
-
-  describe("snapshot metadata", () => {
-    it("returns latest snapshot metadata when archive URL present", async () => {
-      const baseCommit = await repo.git(["rev-parse", "HEAD"]);
-
-      (mockApiClient.getTaskRun as ReturnType<typeof vi.fn>).mockResolvedValue(
-        createTaskRun(),
-      );
-      (
-        mockApiClient.fetchTaskRunLogs as ReturnType<typeof vi.fn>
-      ).mockResolvedValue([
-        createNotification(POSTHOG_NOTIFICATIONS.TREE_SNAPSHOT, {
-          treeHash: "hash-1",
-          baseCommit,
-          archiveUrl: "gs://bucket/hash-1.tar.gz",
-          changes: [{ path: "restored.ts", status: "A" }],
-          timestamp: new Date().toISOString(),
+        createUserMessage("continue"),
+        createGitCheckpointNotification({
+          checkpointId: "checkpoint-2",
+          checkpointRef: "refs/posthog-code-checkpoint/checkpoint-2",
+          head: "head-2",
         }),
       ]);
 
@@ -496,21 +442,22 @@ describe("ResumeSaga", () => {
       expect(result.success).toBe(true);
       if (!result.success) return;
 
-      expect(result.data.latestSnapshot?.treeHash).toBe("hash-1");
-      expect(result.data.latestSnapshot?.archiveUrl).toBe(
-        "gs://bucket/hash-1.tar.gz",
+      expect(result.data.latestGitCheckpoint?.checkpointId).toBe(
+        "checkpoint-2",
       );
     });
 
-    it("returns snapshot metadata even when no archive URL", async () => {
+    it("does not mark resume as interrupted from checkpoint state", async () => {
       (mockApiClient.getTaskRun as ReturnType<typeof vi.fn>).mockResolvedValue(
         createTaskRun(),
       );
       (
         mockApiClient.fetchTaskRunLogs as ReturnType<typeof vi.fn>
       ).mockResolvedValue([
-        createTreeSnapshotNotification("hash-1"),
-        createUserMessage("hello"),
+        createGitCheckpointNotification({
+          checkpointId: "checkpoint-1",
+          checkpointRef: "refs/posthog-code-checkpoint/checkpoint-1",
+        }),
       ]);
 
       const saga = new ResumeSaga(mockLogger);
@@ -524,8 +471,7 @@ describe("ResumeSaga", () => {
       expect(result.success).toBe(true);
       if (!result.success) return;
 
-      expect(result.data.latestSnapshot).not.toBeNull();
-      expect(result.data.conversation).toHaveLength(1);
+      expect(result.data.interrupted).toBe(false);
     });
   });
 
@@ -537,10 +483,14 @@ describe("ResumeSaga", () => {
       (
         mockApiClient.fetchTaskRunLogs as ReturnType<typeof vi.fn>
       ).mockResolvedValue([
-        createTreeSnapshotNotification("hash-1", undefined, {
+        createGitCheckpointNotification({
+          checkpointId: "checkpoint-1",
+          checkpointRef: "refs/posthog-code-checkpoint/checkpoint-1",
           device: { type: "local" },
         }),
-        createTreeSnapshotNotification("hash-2", undefined, {
+        createGitCheckpointNotification({
+          checkpointId: "checkpoint-2",
+          checkpointRef: "refs/posthog-code-checkpoint/checkpoint-2",
           device: { type: "cloud" },
         }),
       ]);

package/src/sagas/resume-saga.ts

@@ -1,12 +1,11 @@
 import type { ContentBlock } from "@agentclientprotocol/sdk";
 import { Saga } from "@posthog/shared";
-import { isNotification, POSTHOG_NOTIFICATIONS } from "../acp-extensions";
+import { POSTHOG_NOTIFICATIONS } from "../acp-extensions";
 import type { PostHogAPIClient } from "../posthog-api";
 import type {
   DeviceInfo,
   GitCheckpointEvent,
   StoredNotification,
-  TreeSnapshotEvent,
 } from "../types";
 import type { Logger } from "../utils/logger";
 
@@ -33,7 +32,6 @@ export interface ResumeInput {
 
 export interface ResumeOutput {
   conversation: ConversationTurn[];
-  latestSnapshot: TreeSnapshotEvent | null;
   latestGitCheckpoint: GitCheckpointEvent | null;
   interrupted: boolean;
   lastDevice?: DeviceInfo;
@@ -68,25 +66,11 @@ export class ResumeSaga extends Saga<ResumeInput, ResumeOutput> {
 
     this.log.info("Fetched log entries", { count: entries.length });
 
-    // Step 3: Find latest snapshot (read-only, pure computation)
-    const latestSnapshot = await this.readOnlyStep("find_snapshot", () =>
-      Promise.resolve(this.findLatestTreeSnapshot(entries)),
-    );
-
     const latestGitCheckpoint = await this.readOnlyStep(
       "find_git_checkpoint",
       () => Promise.resolve(this.findLatestGitCheckpoint(entries)),
     );
 
-    // Step 4: Apply snapshot if present (wrapped in step for consistent logging)
-    if (latestSnapshot) {
-      this.log.info("Found tree snapshot", {
-        treeHash: latestSnapshot.treeHash,
-        hasArchiveUrl: !!latestSnapshot.archiveUrl,
-        changes: latestSnapshot.changes?.length ?? 0,
-      });
-    }
-
     if (latestGitCheckpoint) {
       this.log.info("Found git checkpoint", {
         checkpointId: latestGitCheckpoint.checkpointId,
@@ -105,16 +89,14 @@ export class ResumeSaga extends Saga<ResumeInput, ResumeOutput> {
 
     this.log.info("Resume state rebuilt", {
       turns: conversation.length,
-      hasSnapshot: !!latestSnapshot,
       hasGitCheckpoint: !!latestGitCheckpoint,
-      interrupted: latestSnapshot?.interrupted ?? false,
+      interrupted: false,
     });
 
     return {
       conversation,
-      latestSnapshot,
       latestGitCheckpoint,
-      interrupted: latestSnapshot?.interrupted ?? false,
+      interrupted: false,
       lastDevice,
       logEntryCount: entries.length,
     };
@@ -123,35 +105,12 @@ export class ResumeSaga extends Saga<ResumeInput, ResumeOutput> {
   private emptyResult(): ResumeOutput {
     return {
       conversation: [],
-      latestSnapshot: null,
       latestGitCheckpoint: null,
       interrupted: false,
       logEntryCount: 0,
     };
   }
 
-  private findLatestTreeSnapshot(
-    entries: StoredNotification[],
-  ): TreeSnapshotEvent | null {
-    for (let i = entries.length - 1; i >= 0; i--) {
-      const entry = entries[i];
-      if (
-        isNotification(
-          entry.notification?.method,
-          POSTHOG_NOTIFICATIONS.TREE_SNAPSHOT,
-        )
-      ) {
-        const params = entry.notification.params as
-          | TreeSnapshotEvent
-          | undefined;
-        if (params?.treeHash) {
-          return params;
-        }
-      }
-    }
-    return null;
-  }
-
   private findLatestGitCheckpoint(
     entries: StoredNotification[],
   ): GitCheckpointEvent | null {