@posthog/agent 1.29.0 → 2.0.0

package/dist/index.d.ts

@@ -17,16 +17,8 @@ import { z } from 'zod';
  * Used with AgentSideConnection.extNotification() or Client.extNotification()
  */
 declare const POSTHOG_NOTIFICATIONS: {
-    /** Artifact produced during task execution (research, plan, etc.) */
-    readonly ARTIFACT: "_posthog/artifact";
-    /** Phase has started (research, plan, build, etc.) */
-    readonly PHASE_START: "_posthog/phase_start";
-    /** Phase has completed */
-    readonly PHASE_COMPLETE: "_posthog/phase_complete";
     /** Git branch was created */
     readonly BRANCH_CREATED: "_posthog/branch_created";
-    /** Pull request was created */
-    readonly PR_CREATED: "_posthog/pr_created";
     /** Task run has started */
     readonly RUN_STARTED: "_posthog/run_started";
     /** Task has completed */
@@ -39,27 +31,9 @@ declare const POSTHOG_NOTIFICATIONS: {
     readonly SDK_SESSION: "_posthog/sdk_session";
 };
 type PostHogNotificationType = (typeof POSTHOG_NOTIFICATIONS)[keyof typeof POSTHOG_NOTIFICATIONS];
-interface ArtifactNotificationPayload {
-    sessionId: string;
-    kind: "research_evaluation" | "research_questions" | "plan" | "pr_body" | string;
-    content: unknown;
-}
-interface PhaseNotificationPayload {
-    sessionId: string;
-    phase: "research" | "plan" | "build" | "finalize" | string;
-    [key: string]: unknown;
-}
 interface BranchCreatedPayload {
-    sessionId: string;
     branch: string;
 }
-/**
- * Payload for PR created notification
- */
-interface PrCreatedPayload {
-    sessionId: string;
-    prUrl: string;
-}
 interface RunStartedPayload {
     sessionId: string;
     runId: string;
@@ -92,7 +66,7 @@ interface SdkSessionPayload {
     sessionId: string;
     sdkSessionId: string;
 }
-type PostHogNotificationPayload = ArtifactNotificationPayload | PhaseNotificationPayload | BranchCreatedPayload | PrCreatedPayload | RunStartedPayload | TaskCompletePayload | ErrorNotificationPayload | ConsoleNotificationPayload | SdkSessionPayload;
+type PostHogNotificationPayload = BranchCreatedPayload | RunStartedPayload | TaskCompletePayload | ErrorNotificationPayload | ConsoleNotificationPayload | SdkSessionPayload;
 
 /**
  * Stored custom notification following ACP extensibility model.
@@ -190,6 +164,9 @@ interface TaskExecutionOptions {
     queryOverrides?: Record<string, unknown>;
     canUseTool?: CanUseTool;
     skipGitBranch?: boolean;
+    framework?: "claude";
+    task?: Task;
+    isReconnect?: boolean;
 }
 interface ExecutionResult {
     results: any[];
@@ -217,7 +194,7 @@ type OnLogCallback = (level: LogLevel$1, scope: string, message: string, data?:
 interface AgentConfig {
     workingDirectory?: string;
     posthogApiUrl?: string;
-    posthogApiKey?: string;
+    getPosthogApiKey?: () => string;
     posthogProjectId?: number;
     posthogMcpUrl?: string;
     mcpServers?: Record<string, McpServerConfig>;
@@ -227,7 +204,7 @@ interface AgentConfig {
 }
 interface PostHogAPIConfig {
     apiUrl: string;
-    apiKey: string;
+    getApiKey: () => string;
     projectId: number;
 }
 type ResourceType = "error" | "experiment" | "insight" | "feature_flag" | "generic";
@@ -245,25 +222,6 @@ interface UrlMention {
     id?: string;
     label?: string;
 }
-interface ResearchQuestion {
-    id: string;
-    question: string;
-    options: string[];
-}
-interface ResearchAnswer {
-    questionId: string;
-    selectedOption: string;
-    customInput?: string;
-}
-interface ResearchEvaluation {
-    actionabilityScore: number;
-    context: string;
-    keyFiles: string[];
-    blockers?: string[];
-    questions?: ResearchQuestion[];
-    answered?: boolean;
-    answers?: ResearchAnswer[];
-}
 interface WorktreeInfo {
     worktreePath: string;
     worktreeName: string;
@@ -416,13 +374,14 @@ type StreamPair = {
 };
 
 /**
- * The claude adapter has been based on the original claude-code-acp adapter,
- * and could use some cleanup.
+ * Shared ACP connection factory.
  *
- * https://github.com/zed-industries/claude-code-acp
+ * Creates ACP connections for the Claude Code agent.
  */
 
+type AgentFramework = "claude";
 type AcpConnectionConfig = {
+    framework?: AgentFramework;
     sessionStore?: SessionStore;
     sessionId?: string;
     taskId?: string;
@@ -431,6 +390,12 @@ type InProcessAcpConnection = {
     agentConnection: AgentSideConnection;
     clientStreams: StreamPair;
 };
+/**
+ * Creates an ACP connection with the specified agent framework.
+ *
+ * @param config - Configuration including framework selection
+ * @returns Connection with agent and client streams
+ */
 declare function createAcpConnection(config?: AcpConnectionConfig): InProcessAcpConnection;
 
 declare class Agent {
@@ -439,10 +404,8 @@ declare class Agent {
     private posthogAPI?;
     private fileManager;
     private gitManager;
-    private templateManager;
     private logger;
     private acpConnection?;
-    private promptBuilder;
     private mcpServers?;
     private canUseTool?;
     private currentRunId?;
@@ -454,11 +417,14 @@ declare class Agent {
      */
     setDebug(enabled: boolean): void;
     /**
-     * Configure LLM gateway environment variables for Claude Code CLI
+     * Configure LLM gateway environment variables for Claude Code CLI.
      */
     private _configureLlmGateway;
-    private getOrCreateConnection;
-    runTask(taskId: string, taskRunId: string, options?: TaskExecutionOptions): Promise<void>;
+    /**
+     * @deprecated Use runTaskV2() for local execution or runTaskCloud() for cloud execution.
+     * This method used the old workflow system which has been removed.
+     */
+    runTask(_taskId: string, _taskRunId: string, _options?: TaskExecutionOptions): Promise<void>;
     /**
      * Creates an in-process ACP connection for client communication.
      * Sets up git branch for the task, configures LLM gateway.
@@ -482,8 +448,8 @@ declare class Agent {
     getTaskExecutionStatus(taskId: string): string | null;
     private prepareTaskBranch;
     private ensureOpenAIGatewayEnv;
+    private ensureGeminiGatewayEnv;
     runTaskCloud(taskId: string, taskRunId: string, options?: TaskExecutionOptions): Promise<void>;
-    private ensurePullRequest;
 }
 
 declare const TokenEventSchema: z.ZodObject<{
@@ -699,13 +665,13 @@ declare const AgentEventSchema: z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
 }, "strip", z.ZodTypeAny, {
     type: "message_start";
     ts: number;
-    model?: string | undefined;
     messageId?: string | undefined;
+    model?: string | undefined;
 }, {
     type: "message_start";
     ts: number;
-    model?: string | undefined;
     messageId?: string | undefined;
+    model?: string | undefined;
 }>, z.ZodObject<{
     ts: z.ZodNumber;
 } & {
@@ -723,18 +689,18 @@ declare const AgentEventSchema: z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
     type: "message_delta";
     ts: number;
     stopReason?: string | undefined;
+    stopSequence?: string | undefined;
     usage?: {
         outputTokens: number;
     } | undefined;
-    stopSequence?: string | undefined;
 }, {
     type: "message_delta";
     ts: number;
     stopReason?: string | undefined;
+    stopSequence?: string | undefined;
     usage?: {
         outputTokens: number;
     } | undefined;
-    stopSequence?: string | undefined;
 }>, z.ZodObject<{
     ts: z.ZodNumber;
 } & {
@@ -819,33 +785,33 @@ declare const AgentEventSchema: z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
 }, "strip", z.ZodTypeAny, {
     type: "init";
     tools: string[];
-    cwd: string;
+    ts: number;
     model: string;
     permissionMode: string;
+    cwd: string;
     apiKeySource: string;
-    ts: number;
+    agents?: string[] | undefined;
+    slashCommands?: string[] | undefined;
+    outputStyle?: string | undefined;
     mcpServers?: {
         status: string;
         name: string;
     }[] | undefined;
-    agents?: string[] | undefined;
-    slashCommands?: string[] | undefined;
-    outputStyle?: string | undefined;
 }, {
     type: "init";
     tools: string[];
-    cwd: string;
+    ts: number;
     model: string;
     permissionMode: string;
+    cwd: string;
     apiKeySource: string;
-    ts: number;
+    agents?: string[] | undefined;
+    slashCommands?: string[] | undefined;
+    outputStyle?: string | undefined;
     mcpServers?: {
         status: string;
         name: string;
     }[] | undefined;
-    agents?: string[] | undefined;
-    slashCommands?: string[] | undefined;
-    outputStyle?: string | undefined;
 }>, z.ZodObject<{
     ts: z.ZodNumber;
 } & {
@@ -919,18 +885,22 @@ declare const AgentEventSchema: z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
         tool_input: z.ZodRecord<z.ZodString, z.ZodUnknown>;
     }, "strip", z.ZodTypeAny, {
         tool_use_id: string;
-        tool_name: string;
         tool_input: Record<string, unknown>;
+        tool_name: string;
     }, {
         tool_use_id: string;
-        tool_name: string;
         tool_input: Record<string, unknown>;
+        tool_name: string;
     }>, "many">>;
 }, "strip", z.ZodTypeAny, {
     type: "done";
     ts: number;
     result?: string | undefined;
     usage?: unknown;
+    durationMs?: number | undefined;
+    durationApiMs?: number | undefined;
+    numTurns?: number | undefined;
+    totalCostUsd?: number | undefined;
     modelUsage?: Record<string, {
         outputTokens: number;
         inputTokens: number;
@@ -940,20 +910,20 @@ declare const AgentEventSchema: z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
         costUSD: number;
         contextWindow: number;
     }> | undefined;
-    durationMs?: number | undefined;
-    durationApiMs?: number | undefined;
-    numTurns?: number | undefined;
-    totalCostUsd?: number | undefined;
     permissionDenials?: {
         tool_use_id: string;
-        tool_name: string;
         tool_input: Record<string, unknown>;
+        tool_name: string;
     }[] | undefined;
 }, {
     type: "done";
     ts: number;
     result?: string | undefined;
     usage?: unknown;
+    durationMs?: number | undefined;
+    durationApiMs?: number | undefined;
+    numTurns?: number | undefined;
+    totalCostUsd?: number | undefined;
     modelUsage?: Record<string, {
         outputTokens: number;
         inputTokens: number;
@@ -963,14 +933,10 @@ declare const AgentEventSchema: z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
         costUSD: number;
         contextWindow: number;
     }> | undefined;
-    durationMs?: number | undefined;
-    durationApiMs?: number | undefined;
-    numTurns?: number | undefined;
-    totalCostUsd?: number | undefined;
     permissionDenials?: {
         tool_use_id: string;
-        tool_name: string;
         tool_input: Record<string, unknown>;
+        tool_name: string;
     }[] | undefined;
 }>, z.ZodObject<{
     ts: z.ZodNumber;
@@ -1093,8 +1059,6 @@ declare class PostHogFileManager {
     readContext(taskId: string): Promise<string | null>;
     writeRequirements(taskId: string, requirements: string): Promise<void>;
     readRequirements(taskId: string): Promise<string | null>;
-    writeResearch(taskId: string, data: ResearchEvaluation): Promise<void>;
-    readResearch(taskId: string): Promise<ResearchEvaluation | null>;
     writeTodos(taskId: string, data: unknown): Promise<void>;
     readTodos(taskId: string): Promise<unknown | null>;
     getTaskFiles(taskId: string): Promise<SupportingFile[]>;
@@ -1200,6 +1164,10 @@ interface ExitPlanModeTool extends Tool {
     name: "ExitPlanMode";
     category: "assistant";
 }
+interface AskUserQuestionTool extends Tool {
+    name: "AskUserQuestion";
+    category: "assistant";
+}
 interface SlashCommandTool extends Tool {
     name: "SlashCommand";
     category: "assistant";
@@ -1208,7 +1176,7 @@ interface SlashCommandTool extends Tool {
  * Union type of all known tool types.
  * Useful for discriminated unions and type narrowing.
  */
-type KnownTool = ReadTool | WriteTool | EditTool | GlobTool | NotebookEditTool | BashTool | BashOutputTool | KillShellTool | WebFetchTool | WebSearchTool | GrepTool | TaskTool | TodoWriteTool | ExitPlanModeTool | SlashCommandTool;
+type KnownTool = ReadTool | WriteTool | EditTool | GlobTool | NotebookEditTool | BashTool | BashOutputTool | KillShellTool | WebFetchTool | WebSearchTool | GrepTool | TaskTool | TodoWriteTool | ExitPlanModeTool | AskUserQuestionTool | SlashCommandTool;
 
 /**
  * Tool registry for looking up tool definitions by name.
@@ -1234,6 +1202,8 @@ declare class ToolRegistry {
     getByCategory(category: string): Tool[];
 }
 
+declare function getLlmGatewayUrl(posthogHost: string): string;
+
 interface WorktreeConfig {
     mainRepoPath: string;
     worktreeBasePath?: string;
@@ -1273,4 +1243,4 @@ declare class WorktreeManager {
     }>;
 }
 
-export { type AcpConnectionConfig, Agent, type AgentConfig, type AgentEvent, type ArtifactNotificationPayload, type BashOutputTool, type BashTool, type BranchCreatedPayload, type ConsoleEvent, type ConsoleNotificationPayload, type EditTool, type ErrorEvent, type ErrorNotificationPayload, type ExecutionResult, type ExitPlanModeTool, type GlobTool, type GrepTool, type InProcessAcpConnection, type KillShellTool, type KnownTool, LogLevel, type LogLevel$1 as LogLevelType, Logger, type LoggerConfig, type McpServerConfig, type NotebookEditTool, type OnLogCallback, POSTHOG_NOTIFICATIONS, PermissionMode, type PhaseNotificationPayload, type PostHogNotificationPayload, type PostHogNotificationType, type PrCreatedPayload, type ReadTool, type ResearchEvaluation, type RunStartedPayload, type SdkSessionPayload, type SessionPersistenceConfig, SessionStore, type SlashCommandTool, type StatusEvent, type StoredEntry, type StoredNotification, type SupportingFile, type Task, type TaskCompletePayload, type TaskRun, type TaskTool, type TodoItem, type TodoList, TodoManager, type TodoWriteTool, type TokenEvent, type Tool, type ToolCategory, ToolRegistry, type WebFetchTool, type WebSearchTool, type WorktreeConfig, type WorktreeInfo, WorktreeManager, type WriteTool, createAcpConnection, parseAgentEvent, parseAgentEvents };
+export { type AcpConnectionConfig, Agent, type AgentConfig, type AgentEvent, type AgentFramework, type AskUserQuestionTool, type BashOutputTool, type BashTool, type BranchCreatedPayload, type ConsoleEvent, type ConsoleNotificationPayload, type EditTool, type ErrorEvent, type ErrorNotificationPayload, type ExecutionResult, type ExitPlanModeTool, type GlobTool, type GrepTool, type InProcessAcpConnection, type KillShellTool, type KnownTool, LogLevel, type LogLevel$1 as LogLevelType, Logger, type LoggerConfig, type McpServerConfig, type NotebookEditTool, type OnLogCallback, POSTHOG_NOTIFICATIONS, PermissionMode, type PostHogNotificationPayload, type PostHogNotificationType, type ReadTool, type RunStartedPayload, type SdkSessionPayload, type SessionPersistenceConfig, SessionStore, type SlashCommandTool, type StatusEvent, type StoredEntry, type StoredNotification, type SupportingFile, type Task, type TaskCompletePayload, type TaskRun, type TaskTool, type TodoItem, type TodoList, TodoManager, type TodoWriteTool, type TokenEvent, type Tool, type ToolCategory, ToolRegistry, type WebFetchTool, type WebSearchTool, type WorktreeConfig, type WorktreeInfo, WorktreeManager, type WriteTool, createAcpConnection, getLlmGatewayUrl, parseAgentEvent, parseAgentEvents };