@posthog/agent 2.3.13 → 2.3.18

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@posthog/agent",
-  "version": "2.3.13",
+  "version": "2.3.18",
   "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,40 +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;

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/instructions.ts

@@ -0,0 +1,19 @@
+const BRANCH_NAMING = `
+# Branch Naming
+
+When working in a detached HEAD state, create a descriptive branch name based on the work being done before committing. Do this automatically without asking the user.
+`;
+
+const PLAN_MODE = `
+# Plan Mode
+
+Only enter plan mode (EnterPlanMode) when the user is requesting a significant change in approach or direction mid-task. Do NOT enter plan mode for:
+- Confirmations or approvals ("yes", "looks good", "continue", "go ahead")
+- Minor clarifications or small adjustments
+- Answers to questions you asked (unless you are still in the initial planning phase and have not yet started executing)
+- Feedback that does not require replanning
+
+When in doubt, continue executing and incorporate the feedback inline.
+`;
+
+export const APPENDED_INSTRUCTIONS = BRANCH_NAMING + PLAN_MODE;

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,17 +9,18 @@ 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 type { EffortLevel } from "../types.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 { APPENDED_INSTRUCTIONS } from "./instructions";
+import { DEFAULT_MODEL } from "./models";
+import type { SettingsManager } from "./settings";
 
 export interface ProcessSpawnedInfo {
   pid: number;
@@ -47,19 +48,13 @@ export interface BuildOptionsParams {
   effort?: EffortLevel;
 }
 
-const BRANCH_NAMING_INSTRUCTIONS = `
-# Branch Naming
-
-When working in a detached HEAD state, create a descriptive branch name based on the work being done before committing. Do this automatically without asking the user.
-`;
-
 export function buildSystemPrompt(
   customPrompt?: unknown,
 ): Options["systemPrompt"] {
   const defaultPrompt: Options["systemPrompt"] = {
     type: "preset",
     preset: "claude_code",
-    append: BRANCH_NAMING_INSTRUCTIONS,
+    append: APPENDED_INSTRUCTIONS,
   };
 
   if (!customPrompt) {
@@ -67,7 +62,7 @@ export function buildSystemPrompt(
   }
 
   if (typeof customPrompt === "string") {
-    return customPrompt + BRANCH_NAMING_INSTRUCTIONS;
+    return customPrompt + APPENDED_INSTRUCTIONS;
   }
 
   if (
@@ -78,7 +73,7 @@ export function buildSystemPrompt(
   ) {
     return {
       ...defaultPrompt,
-      append: customPrompt.append + BRANCH_NAMING_INSTRUCTIONS,
+      append: customPrompt.append + APPENDED_INSTRUCTIONS,
     };
   }
 

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,10 @@ 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";
 

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);
 

package/src/server/agent-server.test.ts

@@ -9,11 +9,11 @@ import {
   expect,
   it,
 } from "vitest";
-import { createTestRepo, type TestRepo } from "../test/fixtures/api.js";
-import { createPostHogHandlers } from "../test/mocks/msw-handlers.js";
-import type { TaskRun } from "../types.js";
-import { AgentServer } from "./agent-server.js";
-import { type JwtPayload, SANDBOX_CONNECTION_AUDIENCE } from "./jwt.js";
+import { createTestRepo, type TestRepo } from "../test/fixtures/api";
+import { createPostHogHandlers } from "../test/mocks/msw-handlers";
+import type { TaskRun } from "../types";
+import { AgentServer } from "./agent-server";
+import { type JwtPayload, SANDBOX_CONNECTION_AUDIENCE } from "./jwt";
 
 interface TestableServer {
   getInitialPromptOverride(run: TaskRun): string | null;