@posthog/agent 2.3.11 → 2.3.15

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@posthog/agent",
-  "version": "2.3.11",
+  "version": "2.3.15",
   "repository": "https://github.com/PostHog/code",
   "description": "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   "exports": {

package/src/adapters/acp-connection.ts

@@ -1,17 +1,17 @@
 import { AgentSideConnection, ndJsonStream } from "@agentclientprotocol/sdk";
-import { POSTHOG_NOTIFICATIONS } from "../acp-extensions.js";
-import type { SessionLogWriter } from "../session-log-writer.js";
-import type { ProcessSpawnedCallback } from "../types.js";
-import { Logger } from "../utils/logger.js";
+import { POSTHOG_NOTIFICATIONS } from "../acp-extensions";
+import type { SessionLogWriter } from "../session-log-writer";
+import type { ProcessSpawnedCallback } from "../types";
+import { Logger } from "../utils/logger";
 import {
   createBidirectionalStreams,
   createTappedWritableStream,
   nodeReadableToWebReadable,
   nodeWritableToWebWritable,
   type StreamPair,
-} from "../utils/streams.js";
-import { ClaudeAcpAgent } from "./claude/claude-agent.js";
-import { type CodexProcessOptions, spawnCodexProcess } from "./codex/spawn.js";
+} from "../utils/streams";
+import { ClaudeAcpAgent } from "./claude/claude-agent";
+import { type CodexProcessOptions, spawnCodexProcess } from "./codex/spawn";
 
 type AgentAdapter = "claude" | "codex";
 

package/src/adapters/base-acp-agent.ts

@@ -21,9 +21,9 @@ import {
   fetchGatewayModels,
   formatGatewayModelName,
   isAnthropicModel,
-} from "../gateway-models.js";
-import { Logger } from "../utils/logger.js";
-import type { SettingsManager } from "./claude/session/settings.js";
+} from "../gateway-models";
+import { Logger } from "../utils/logger";
+import type { SettingsManager } from "./claude/session/settings";
 
 export interface BaseSession {
   notificationHistory: SessionNotification[];

package/src/adapters/claude/claude-agent.ts

@@ -44,39 +44,40 @@ import {
 } from "@anthropic-ai/claude-agent-sdk";
 import { v7 as uuidv7 } from "uuid";
 import packageJson from "../../../package.json" with { type: "json" };
-import { unreachable, withTimeout } from "../../utils/common.js";
-import { Logger } from "../../utils/logger.js";
-import { Pushable } from "../../utils/streams.js";
-import { BaseAcpAgent } from "../base-acp-agent.js";
-import { promptToClaude } from "./conversion/acp-to-sdk.js";
+import { unreachable, withTimeout } from "../../utils/common";
+import { Logger } from "../../utils/logger";
+import { Pushable } from "../../utils/streams";
+import { BaseAcpAgent } from "../base-acp-agent";
+import { promptToClaude } from "./conversion/acp-to-sdk";
 import {
   handleResultMessage,
   handleStreamEvent,
   handleSystemMessage,
   handleUserAssistantMessage,
-} from "./conversion/sdk-to-acp.js";
-import { fetchMcpToolMetadata } from "./mcp/tool-metadata.js";
-import { canUseTool } from "./permissions/permission-handlers.js";
-import { getAvailableSlashCommands } from "./session/commands.js";
-import { parseMcpServers } from "./session/mcp-config.js";
-import { DEFAULT_MODEL, toSdkModelId } from "./session/models.js";
+} from "./conversion/sdk-to-acp";
+import { fetchMcpToolMetadata } from "./mcp/tool-metadata";
+import { canUseTool } from "./permissions/permission-handlers";
+import { getAvailableSlashCommands } from "./session/commands";
+import { parseMcpServers } from "./session/mcp-config";
+import { DEFAULT_MODEL, toSdkModelId } from "./session/models";
 import {
   buildSessionOptions,
   buildSystemPrompt,
   type ProcessSpawnedInfo,
-} from "./session/options.js";
-import { SettingsManager } from "./session/settings.js";
+} from "./session/options";
+import { SettingsManager } from "./session/settings";
 import {
   CODE_EXECUTION_MODES,
   type CodeExecutionMode,
   getAvailableModes,
-} from "./tools.js";
+} from "./tools";
 import type {
   BackgroundTerminal,
+  EffortLevel,
   NewSessionMeta,
   Session,
   ToolUseCache,
-} from "./types.js";
+} from "./types";
 
 const SESSION_VALIDATION_TIMEOUT_MS = 10_000;
 const MAX_TITLE_LENGTH = 256;
@@ -558,6 +559,10 @@ export class ClaudeAcpAgent extends BaseAcpAgent {
       const sdkModelId = toSdkModelId(params.value);
       await this.session.query.setModel(sdkModelId);
       this.session.modelId = params.value;
+    } else if (params.configId === "effort") {
+      const newEffort = params.value as EffortLevel;
+      this.session.effort = newEffort;
+      this.session.queryOptions.effort = newEffort;
     }
 
     this.session.configOptions = this.session.configOptions.map((o) =>
@@ -623,6 +628,7 @@ export class ClaudeAcpAgent extends BaseAcpAgent {
 
     const meta = params._meta as NewSessionMeta | undefined;
     const taskId = meta?.persistence?.taskId;
+    const effort = meta?.claudeCode?.options?.effort as EffortLevel | undefined;
 
     // We want to create a new session id unless it is resume,
     // but not resume + forkSession.
@@ -673,6 +679,7 @@ export class ClaudeAcpAgent extends BaseAcpAgent {
       onModeChange: this.createOnModeChange(),
       onProcessSpawned: this.options?.onProcessSpawned,
       onProcessExited: this.options?.onProcessExited,
+      effort,
     });
 
     // Use the same abort controller that buildSessionOptions gave to the query
@@ -682,6 +689,7 @@ export class ClaudeAcpAgent extends BaseAcpAgent {
 
     const session: Session = {
       query: q,
+      queryOptions: options,
       input,
       cancelled: false,
       settingsManager,
@@ -693,6 +701,7 @@ export class ClaudeAcpAgent extends BaseAcpAgent {
         cachedReadTokens: 0,
         cachedWriteTokens: 0,
       },
+      effort,
       configOptions: [],
       promptRunning: false,
       pendingMessages: new Map(),
@@ -790,7 +799,11 @@ export class ClaudeAcpAgent extends BaseAcpAgent {
       ),
     };
 
-    const configOptions = this.buildConfigOptions(permissionMode, modelOptions);
+    const configOptions = this.buildConfigOptions(
+      permissionMode,
+      modelOptions,
+      effort ?? "high",
+    );
     session.configOptions = configOptions;
 
     if (!creationOpts.skipBackgroundFetches) {
@@ -844,6 +857,7 @@ export class ClaudeAcpAgent extends BaseAcpAgent {
       currentModelId: string;
       options: SessionConfigSelectOption[];
     },
+    currentEffort: EffortLevel = "high",
   ): SessionConfigOption[] {
     const modeOptions = getAvailableModes().map((mode) => ({
       value: mode.id,
@@ -871,6 +885,20 @@ export class ClaudeAcpAgent extends BaseAcpAgent {
         category: "model" as SessionConfigOptionCategory,
         description: "Choose which model Claude should use",
       },
+      {
+        id: "effort",
+        name: "Effort",
+        type: "select",
+        currentValue: currentEffort,
+        options: [
+          { value: "low", name: "Low" },
+          { value: "medium", name: "Medium" },
+          { value: "high", name: "High" },
+          { value: "max", name: "Max" },
+        ],
+        category: "thought_level" as SessionConfigOptionCategory,
+        description: "Controls how much effort Claude puts into its response",
+      },
     ];
   }
 

package/src/adapters/claude/conversion/sdk-to-acp.ts

@@ -17,18 +17,18 @@ import type {
   BetaContentBlock,
   BetaRawContentBlockDelta,
 } from "@anthropic-ai/sdk/resources/beta.mjs";
-import { image, text } from "../../../utils/acp-content.js";
-import { unreachable } from "../../../utils/common.js";
-import type { Logger } from "../../../utils/logger.js";
-import { registerHookCallback } from "../hooks.js";
-import type { Session, ToolUpdateMeta, ToolUseCache } from "../types.js";
+import { image, text } from "../../../utils/acp-content";
+import { unreachable } from "../../../utils/common";
+import type { Logger } from "../../../utils/logger";
+import { registerHookCallback } from "../hooks";
+import type { Session, ToolUpdateMeta, ToolUseCache } from "../types";
 import {
   type ClaudePlanEntry,
   planEntries,
   toolInfoFromToolUse,
   toolUpdateFromEditToolResponse,
   toolUpdateFromToolResult,
-} from "./tool-use-to-acp.js";
+} from "./tool-use-to-acp";
 
 type AnthropicContentChunk =
   | ContentBlockParam

package/src/adapters/claude/conversion/tool-use-to-acp.ts

@@ -29,8 +29,8 @@ function stripSystemReminders(value: string): string {
   return value.replace(SYSTEM_REMINDER_REGEX, "");
 }
 
-import { resourceLink, text, toolContent } from "../../../utils/acp-content.js";
-import { getMcpToolMetadata } from "../mcp/tool-metadata.js";
+import { resourceLink, text, toolContent } from "../../../utils/acp-content";
+import { getMcpToolMetadata } from "../mcp/tool-metadata";
 
 type ToolInfo = Pick<ToolCall, "title" | "kind" | "content" | "locations">;
 

package/src/adapters/claude/hooks.ts

@@ -1,7 +1,7 @@
 import type { HookCallback, HookInput } from "@anthropic-ai/claude-agent-sdk";
-import type { Logger } from "../../utils/logger.js";
-import type { SettingsManager } from "./session/settings.js";
-import type { CodeExecutionMode } from "./tools.js";
+import type { Logger } from "../../utils/logger";
+import type { SettingsManager } from "./session/settings";
+import type { CodeExecutionMode } from "./tools";
 
 const toolUseCallbacks: {
   [toolUseId: string]: {

package/src/adapters/claude/mcp/tool-metadata.ts

@@ -1,5 +1,5 @@
 import type { McpServerStatus, Query } from "@anthropic-ai/claude-agent-sdk";
-import { Logger } from "../../../utils/logger.js";
+import { Logger } from "../../../utils/logger";
 
 export interface McpToolMetadata {
   readOnly: boolean;

package/src/adapters/claude/permissions/permission-handlers.ts

@@ -3,27 +3,27 @@ import type {
   RequestPermissionResponse,
 } from "@agentclientprotocol/sdk";
 import type { PermissionUpdate } from "@anthropic-ai/claude-agent-sdk";
-import { text } from "../../../utils/acp-content.js";
-import type { Logger } from "../../../utils/logger.js";
-import { toolInfoFromToolUse } from "../conversion/tool-use-to-acp.js";
+import { text } from "../../../utils/acp-content";
+import type { Logger } from "../../../utils/logger";
+import { toolInfoFromToolUse } from "../conversion/tool-use-to-acp";
 import {
   getClaudePlansDir,
   getLatestAssistantText,
   isClaudePlanFilePath,
   isPlanReady,
-} from "../plan/utils.js";
+} from "../plan/utils";
 import {
   type AskUserQuestionInput,
   normalizeAskUserQuestionInput,
   OPTION_PREFIX,
   type QuestionItem,
-} from "../questions/utils.js";
-import { isToolAllowedForMode, WRITE_TOOLS } from "../tools.js";
-import type { Session } from "../types.js";
+} from "../questions/utils";
+import { isToolAllowedForMode, WRITE_TOOLS } from "../tools";
+import type { Session } from "../types";
 import {
   buildExitPlanModePermissionOptions,
   buildPermissionOptions,
-} from "./permission-options.js";
+} from "./permission-options";
 
 export type ToolPermissionResult =
   | {

package/src/adapters/claude/permissions/permission-options.ts

@@ -1,5 +1,5 @@
 import type { PermissionUpdate } from "@anthropic-ai/claude-agent-sdk";
-import { BASH_TOOLS, READ_TOOLS, SEARCH_TOOLS, WRITE_TOOLS } from "../tools.js";
+import { BASH_TOOLS, READ_TOOLS, SEARCH_TOOLS, WRITE_TOOLS } from "../tools";
 
 export interface PermissionOption {
   kind: "allow_once" | "allow_always" | "reject_once" | "reject_always";

package/src/adapters/claude/questions/utils.ts

@@ -1,6 +1,6 @@
 import type { ToolCallContent, ToolKind } from "@agentclientprotocol/sdk";
 import { z } from "zod";
-import type { PermissionOption } from "../permissions/permission-options.js";
+import type { PermissionOption } from "../permissions/permission-options";
 
 export const OPTION_PREFIX = "option_";
 

package/src/adapters/claude/session/jsonl-hydration.test.ts

@@ -1,10 +1,10 @@
 import { describe, expect, it } from "vitest";
-import type { StoredEntry } from "../../../types.js";
+import type { StoredEntry } from "../../../types";
 import {
   conversationTurnsToJsonlEntries,
   getSessionJsonlPath,
   rebuildConversation,
-} from "./jsonl-hydration.js";
+} from "./jsonl-hydration";
 
 function entry(
   sessionUpdate: string,

package/src/adapters/claude/session/jsonl-hydration.ts

@@ -3,8 +3,8 @@ import * as fs from "node:fs/promises";
 import * as os from "node:os";
 import * as path from "node:path";
 import type { ContentBlock } from "@agentclientprotocol/sdk";
-import type { PostHogAPIClient } from "../../../posthog-api.js";
-import type { StoredEntry } from "../../../types.js";
+import type { PostHogAPIClient } from "../../../posthog-api";
+import type { StoredEntry } from "../../../types";
 
 interface ConversationTurn {
   role: "user" | "assistant";

package/src/adapters/claude/session/options.ts

@@ -9,16 +9,17 @@ import type {
   SpawnedProcess,
   SpawnOptions,
 } from "@anthropic-ai/claude-agent-sdk";
-import { IS_ROOT } from "../../../utils/common.js";
-import type { Logger } from "../../../utils/logger.js";
+import { IS_ROOT } from "../../../utils/common";
+import type { Logger } from "../../../utils/logger";
 import {
   createPostToolUseHook,
   createPreToolUseHook,
   type OnModeChange,
-} from "../hooks.js";
-import type { CodeExecutionMode } from "../tools.js";
-import { DEFAULT_MODEL } from "./models.js";
-import type { SettingsManager } from "./settings.js";
+} from "../hooks";
+import type { CodeExecutionMode } from "../tools";
+import type { EffortLevel } from "../types";
+import { DEFAULT_MODEL } from "./models";
+import type { SettingsManager } from "./settings";
 
 export interface ProcessSpawnedInfo {
   pid: number;
@@ -43,6 +44,7 @@ export interface BuildOptionsParams {
   onModeChange?: OnModeChange;
   onProcessSpawned?: (info: ProcessSpawnedInfo) => void;
   onProcessExited?: (pid: number) => void;
+  effort?: EffortLevel;
 }
 
 const BRANCH_NAMING_INSTRUCTIONS = `
@@ -295,6 +297,10 @@ export function buildSessionOptions(params: BuildOptionsParams): Options {
     options.additionalDirectories = params.additionalDirectories;
   }
 
+  if (params.effort) {
+    options.effort = params.effort;
+  }
+
   clearStatsigCache();
   return options;
 }

package/src/adapters/claude/tools.ts

@@ -3,10 +3,10 @@ export {
   type CodeExecutionMode,
   getAvailableModes,
   type ModeInfo,
-} from "../../execution-mode.js";
+} from "../../execution-mode";
 
-import type { CodeExecutionMode } from "../../execution-mode.js";
-import { isMcpToolReadOnly } from "./mcp/tool-metadata.js";
+import type { CodeExecutionMode } from "../../execution-mode";
+import { isMcpToolReadOnly } from "./mcp/tool-metadata";
 
 export const READ_TOOLS: Set<string> = new Set(["Read", "NotebookRead"]);
 

package/src/adapters/claude/types.ts

@@ -8,10 +8,12 @@ import type {
   Query,
   SDKUserMessage,
 } from "@anthropic-ai/claude-agent-sdk";
-import type { Pushable } from "../../utils/streams.js";
-import type { BaseSession } from "../base-acp-agent.js";
-import type { SettingsManager } from "./session/settings.js";
-import type { CodeExecutionMode } from "./tools.js";
+import type { Pushable } from "../../utils/streams";
+import type { BaseSession } from "../base-acp-agent";
+import type { SettingsManager } from "./session/settings";
+import type { CodeExecutionMode } from "./tools";
+
+export type EffortLevel = "low" | "medium" | "high" | "max";
 
 export type AccumulatedUsage = {
   inputTokens: number;
@@ -38,6 +40,8 @@ export type PendingMessage = {
 
 export type Session = BaseSession & {
   query: Query;
+  /** The Options object passed to query() — mutating it affects subsequent prompts */
+  queryOptions: Options;
   input: Pushable<SDKUserMessage>;
   settingsManager: SettingsManager;
   permissionMode: CodeExecutionMode;
@@ -46,6 +50,7 @@ export type Session = BaseSession & {
   taskRunId?: string;
   lastPlanFilePath?: string;
   lastPlanContent?: string;
+  effort?: EffortLevel;
   configOptions: SessionConfigOption[];
   accumulatedUsage: AccumulatedUsage;
   promptRunning: boolean;

package/src/adapters/codex/spawn.ts

@@ -2,8 +2,8 @@ import { type ChildProcess, spawn } from "node:child_process";
 import { existsSync } from "node:fs";
 import { delimiter, dirname } from "node:path";
 import type { Readable, Writable } from "node:stream";
-import type { ProcessSpawnedCallback } from "../../types.js";
-import { Logger } from "../../utils/logger.js";
+import type { ProcessSpawnedCallback } from "../../types";
+import { Logger } from "../../utils/logger";
 
 export interface CodexProcessOptions {
   cwd?: string;

package/src/agent.ts

@@ -1,16 +1,16 @@
 import {
   createAcpConnection,
   type InProcessAcpConnection,
-} from "./adapters/acp-connection.js";
+} from "./adapters/acp-connection";
 import {
   BLOCKED_MODELS,
   DEFAULT_GATEWAY_MODEL,
   fetchArrayModels,
-} from "./gateway-models.js";
-import { PostHogAPIClient, type TaskRunUpdate } from "./posthog-api.js";
-import { SessionLogWriter } from "./session-log-writer.js";
-import type { AgentConfig, TaskExecutionOptions } from "./types.js";
-import { Logger } from "./utils/logger.js";
+} from "./gateway-models";
+import { PostHogAPIClient, type TaskRunUpdate } from "./posthog-api";
+import { SessionLogWriter } from "./session-log-writer";
+import type { AgentConfig, TaskExecutionOptions } from "./types";
+import { Logger } from "./utils/logger";
 
 export class Agent {
   private posthogAPI?: PostHogAPIClient;

package/src/execution-mode.ts

@@ -1,4 +1,4 @@
-import { IS_ROOT } from "./utils/common.js";
+import { IS_ROOT } from "./utils/common";
 
 export interface ModeInfo {
   id: CodeExecutionMode;

package/src/index.ts

@@ -1 +1 @@
-export { isMcpToolReadOnly } from "./adapters/claude/mcp/tool-metadata.js";
+export { isMcpToolReadOnly } from "./adapters/claude/mcp/tool-metadata";

package/src/otel-log-writer.test.ts

@@ -1,6 +1,6 @@
 import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
-import { OtelLogWriter } from "./otel-log-writer.js";
-import type { StoredNotification } from "./types.js";
+import { OtelLogWriter } from "./otel-log-writer";
+import type { StoredNotification } from "./types";
 
 // Mock the OTEL exporter
 const mockExport = vi.fn((_logs, callback) => {

package/src/otel-log-writer.ts

@@ -6,8 +6,8 @@ import {
   LoggerProvider,
 } from "@opentelemetry/sdk-logs";
 import { ATTR_SERVICE_NAME } from "@opentelemetry/semantic-conventions";
-import type { StoredNotification } from "./types.js";
-import type { Logger } from "./utils/logger.js";
+import type { StoredNotification } from "./types";
+import type { Logger } from "./utils/logger";
 
 export interface OtelLogConfig {
   /** PostHog ingest host, e.g., "https://us.i.posthog.com" */

package/src/posthog-api.ts

@@ -6,8 +6,8 @@ import type {
   Task,
   TaskRun,
   TaskRunArtifact,
-} from "./types.js";
-import { getLlmGatewayUrl } from "./utils/gateway.js";
+} from "./types";
+import { getLlmGatewayUrl } from "./utils/gateway";
 
 export { getLlmGatewayUrl };
 

package/src/resume.ts

@@ -16,10 +16,10 @@
  */
 
 import type { ContentBlock } from "@agentclientprotocol/sdk";
-import type { PostHogAPIClient } from "./posthog-api.js";
-import { ResumeSaga } from "./sagas/resume-saga.js";
-import type { DeviceInfo, TreeSnapshotEvent } from "./types.js";
-import { Logger } from "./utils/logger.js";
+import type { PostHogAPIClient } from "./posthog-api";
+import { ResumeSaga } from "./sagas/resume-saga";
+import type { DeviceInfo, TreeSnapshotEvent } from "./types";
+import { Logger } from "./utils/logger";
 
 export interface ResumeState {
   conversation: ConversationTurn[];

package/src/sagas/apply-snapshot-saga.test.ts

@@ -1,7 +1,7 @@
 import { join } from "node:path";
 import type { SagaLogger } from "@posthog/shared";
 import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
-import { ApplySnapshotSaga } from "./apply-snapshot-saga.js";
+import { ApplySnapshotSaga } from "./apply-snapshot-saga";
 import {
   createArchiveBuffer,
   createMockApiClient,
@@ -9,7 +9,7 @@ import {
   createSnapshot,
   createTestRepo,
   type TestRepo,
-} from "./test-fixtures.js";
+} from "./test-fixtures";
 
 describe("ApplySnapshotSaga", () => {
   let repo: TestRepo;

package/src/sagas/apply-snapshot-saga.ts

@@ -2,8 +2,8 @@ import { mkdir, rm, writeFile } from "node:fs/promises";
 import { join } from "node:path";
 import { ApplyTreeSaga as GitApplyTreeSaga } from "@posthog/git/sagas/tree";
 import { Saga } from "@posthog/shared";
-import type { PostHogAPIClient } from "../posthog-api.js";
-import type { TreeSnapshot } from "../types.js";
+import type { PostHogAPIClient } from "../posthog-api";
+import type { TreeSnapshot } from "../types";
 
 export interface ApplySnapshotInput {
   snapshot: TreeSnapshot;

package/src/sagas/capture-tree-saga.test.ts

@@ -1,15 +1,15 @@
 import { join } from "node:path";
 import type { SagaLogger } from "@posthog/shared";
 import { afterEach, beforeEach, describe, expect, it, type vi } from "vitest";
-import { isCommitOnRemote, validateForCloudHandoff } from "../tree-tracker.js";
-import { CaptureTreeSaga } from "./capture-tree-saga.js";
+import { isCommitOnRemote, validateForCloudHandoff } from "../tree-tracker";
+import { CaptureTreeSaga } from "./capture-tree-saga";
 import {
   createMockApiClient,
   createMockLogger,
   createSnapshot,
   createTestRepo,
   type TestRepo,
-} from "./test-fixtures.js";
+} from "./test-fixtures";
 
 describe("CaptureTreeSaga", () => {
   let repo: TestRepo;

package/src/sagas/capture-tree-saga.ts

@@ -3,8 +3,8 @@ import { readFile, rm } from "node:fs/promises";
 import { join } from "node:path";
 import { CaptureTreeSaga as GitCaptureTreeSaga } from "@posthog/git/sagas/tree";
 import { Saga } from "@posthog/shared";
-import type { PostHogAPIClient } from "../posthog-api.js";
-import type { TreeSnapshot } from "../types.js";
+import type { PostHogAPIClient } from "../posthog-api";
+import type { TreeSnapshot } from "../types";
 
 export interface CaptureTreeInput {
   repositoryPath: string;

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

@@ -1,8 +1,8 @@
 import type { SagaLogger } from "@posthog/shared";
 import { afterEach, beforeEach, describe, expect, it, type vi } from "vitest";
-import { POSTHOG_NOTIFICATIONS } from "../acp-extensions.js";
-import type { PostHogAPIClient } from "../posthog-api.js";
-import { ResumeSaga } from "./resume-saga.js";
+import { POSTHOG_NOTIFICATIONS } from "../acp-extensions";
+import type { PostHogAPIClient } from "../posthog-api";
+import { ResumeSaga } from "./resume-saga";
 import {
   createAgentChunk,
   createAgentMessage,
@@ -17,7 +17,7 @@ import {
   createTreeSnapshotNotification,
   createUserMessage,
   type TestRepo,
-} from "./test-fixtures.js";
+} from "./test-fixtures";
 
 describe("ResumeSaga", () => {
   let repo: TestRepo;

package/src/sagas/resume-saga.ts

@@ -1,14 +1,14 @@
 import type { ContentBlock } from "@agentclientprotocol/sdk";
 import { Saga } from "@posthog/shared";
-import { POSTHOG_NOTIFICATIONS } from "../acp-extensions.js";
-import type { PostHogAPIClient } from "../posthog-api.js";
-import { TreeTracker } from "../tree-tracker.js";
+import { POSTHOG_NOTIFICATIONS } from "../acp-extensions";
+import type { PostHogAPIClient } from "../posthog-api";
+import { TreeTracker } from "../tree-tracker";
 import type {
   DeviceInfo,
   StoredNotification,
   TreeSnapshotEvent,
-} from "../types.js";
-import { Logger } from "../utils/logger.js";
+} from "../types";
+import { Logger } from "../utils/logger";
 
 export interface ConversationTurn {
   role: "user" | "assistant";

package/src/sagas/test-fixtures.ts

@@ -7,9 +7,9 @@ import { promisify } from "node:util";
 import type { SagaLogger } from "@posthog/shared";
 import * as tar from "tar";
 import { vi } from "vitest";
-import { POSTHOG_NOTIFICATIONS } from "../acp-extensions.js";
-import type { PostHogAPIClient } from "../posthog-api.js";
-import type { StoredNotification, TaskRun, TreeSnapshot } from "../types.js";
+import { POSTHOG_NOTIFICATIONS } from "../acp-extensions";
+import type { PostHogAPIClient } from "../posthog-api";
+import type { StoredNotification, TaskRun, TreeSnapshot } from "../types";
 
 const execFileAsync = promisify(execFile);