@posthog/agent 1.24.2 → 1.25.0

package/dist/src/agent.js

@@ -1,82 +1,85 @@
-import { query } from '@anthropic-ai/claude-agent-sdk';
-import { TaskManager } from './task-manager.js';
-import { PostHogAPIClient } from './posthog-api.js';
+import { POSTHOG_NOTIFICATIONS } from './acp-extensions.js';
+import { createAcpConnection } from './adapters/claude/claude.js';
 import { PostHogFileManager } from './file-manager.js';
 import { GitManager } from './git-manager.js';
+import { PostHogAPIClient } from './posthog-api.js';
+import { PromptBuilder } from './prompt-builder.js';
+import { SessionStore } from './session-store.js';
+import { TaskManager } from './task-manager.js';
 import { TemplateManager } from './template-manager.js';
-import { ClaudeAdapter } from './adapters/claude/claude-adapter.js';
 import { Logger } from './utils/logger.js';
-import { PromptBuilder } from './prompt-builder.js';
-import { TaskProgressReporter } from './task-progress-reporter.js';
 import { TASK_WORKFLOW } from './workflow/config.js';
 export { PermissionMode } from './types.js';
 
 class Agent {
     workingDirectory;
-    onEvent;
     taskManager;
     posthogAPI;
     fileManager;
     gitManager;
     templateManager;
-    adapter;
     logger;
-    progressReporter;
+    acpConnection;
     promptBuilder;
     mcpServers;
     canUseTool;
+    currentRunId;
+    sessionStore;
     debug;
     constructor(config) {
         this.workingDirectory = config.workingDirectory || process.cwd();
-        this.onEvent = config.onEvent;
         this.canUseTool = config.canUseTool;
         this.debug = config.debug || false;
         // Build default PostHog MCP server configuration
-        const posthogMcpUrl = config.posthogMcpUrl
-            || process.env.POSTHOG_MCP_URL
-            || 'https://mcp.posthog.com/mcp';
+        const posthogMcpUrl = config.posthogMcpUrl ||
+            process.env.POSTHOG_MCP_URL ||
+            "https://mcp.posthog.com/mcp";
         // Add auth if API key provided
         const headers = {};
         if (config.posthogApiKey) {
-            headers['Authorization'] = `Bearer ${config.posthogApiKey}`;
+            headers.Authorization = `Bearer ${config.posthogApiKey}`;
         }
         const defaultMcpServers = {
             posthog: {
-                type: 'http',
+                type: "http",
                 url: posthogMcpUrl,
                 ...(Object.keys(headers).length > 0 ? { headers } : {}),
-            }
+            },
         };
         // Merge default PostHog MCP with user-provided servers (user config takes precedence)
         this.mcpServers = {
             ...defaultMcpServers,
-            ...config.mcpServers
+            ...config.mcpServers,
         };
-        this.logger = new Logger({ debug: this.debug, prefix: '[PostHog Agent]' });
+        this.logger = new Logger({
+            debug: this.debug,
+            prefix: "[PostHog Agent]",
+            onLog: config.onLog,
+        });
         this.taskManager = new TaskManager();
-        // Hardcode Claude adapter for now - extensible for other providers later
-        this.adapter = new ClaudeAdapter();
-        this.fileManager = new PostHogFileManager(this.workingDirectory, this.logger.child('FileManager'));
+        this.fileManager = new PostHogFileManager(this.workingDirectory, this.logger.child("FileManager"));
         this.gitManager = new GitManager({
             repositoryPath: this.workingDirectory,
-            logger: this.logger.child('GitManager')
-            // TODO: Add author config from environment or config
+            logger: this.logger.child("GitManager"),
         });
         this.templateManager = new TemplateManager();
-        if (config.posthogApiUrl && config.posthogApiKey) {
+        if (config.posthogApiUrl &&
+            config.posthogApiKey &&
+            config.posthogProjectId) {
             this.posthogAPI = new PostHogAPIClient({
                 apiUrl: config.posthogApiUrl,
                 apiKey: config.posthogApiKey,
                 projectId: config.posthogProjectId,
             });
+            // Create SessionStore from the API client for ACP connection
+            this.sessionStore = new SessionStore(this.posthogAPI, this.logger.child("SessionStore"));
         }
         this.promptBuilder = new PromptBuilder({
             getTaskFiles: (taskId) => this.getTaskFiles(taskId),
             generatePlanTemplate: (vars) => this.templateManager.generatePlan(vars),
             posthogClient: this.posthogAPI,
-            logger: this.logger.child('PromptBuilder')
+            logger: this.logger.child("PromptBuilder"),
         });
-        this.progressReporter = new TaskProgressReporter(this.posthogAPI, this.logger);
     }
     /**
      * Enable or disable debug logging
@@ -98,30 +101,52 @@ class Agent {
             process.env.ANTHROPIC_BASE_URL = gatewayUrl;
             process.env.ANTHROPIC_AUTH_TOKEN = apiKey;
             this.ensureOpenAIGatewayEnv(gatewayUrl, apiKey);
-            this.logger.debug('Configured LLM gateway', { gatewayUrl });
         }
         catch (error) {
-            this.logger.error('Failed to configure LLM gateway', error);
+            this.logger.error("Failed to configure LLM gateway", error);
             throw error;
         }
     }
+    getOrCreateConnection() {
+        if (!this.acpConnection) {
+            this.acpConnection = createAcpConnection({
+                sessionStore: this.sessionStore,
+            });
+        }
+        return this.acpConnection;
+    }
     // Adaptive task execution orchestrated via workflow steps
-    async runTask(taskOrId, options = {}) {
-        await this._configureLlmGateway();
-        const task = typeof taskOrId === 'string' ? await this.fetchTask(taskOrId) : taskOrId;
+    async runTask(taskId, taskRunId, options = {}) {
+        // await this._configureLlmGateway();
+        const task = await this.fetchTask(taskId);
         const cwd = options.repositoryPath || this.workingDirectory;
         const isCloudMode = options.isCloudMode ?? false;
         const taskSlug = task.slug || task.id;
-        this.logger.info('Starting adaptive task execution', { taskId: task.id, taskSlug, isCloudMode });
-        // Initialize progress reporter for task run tracking (needed for PR attachment)
-        await this.progressReporter.start(task.id, { totalSteps: TASK_WORKFLOW.length });
-        this.emitEvent(this.adapter.createStatusEvent('run_started', { runId: this.progressReporter.runId }));
-        await this.prepareTaskBranch(taskSlug, isCloudMode);
+        // Use taskRunId as sessionId - they are the same identifier
+        this.currentRunId = taskRunId;
+        this.logger.info("Starting adaptive task execution", {
+            taskId: task.id,
+            taskSlug,
+            taskRunId,
+            isCloudMode,
+        });
+        const connection = this.getOrCreateConnection();
+        // Create sendNotification using ACP connection's extNotification
+        const sendNotification = async (method, params) => {
+            this.logger.debug(`Notification: ${method}`, params);
+            await connection.agentConnection.extNotification?.(method, params);
+        };
+        await sendNotification(POSTHOG_NOTIFICATIONS.RUN_STARTED, {
+            sessionId: taskRunId,
+            runId: taskRunId,
+        });
+        await this.prepareTaskBranch(taskSlug, isCloudMode, sendNotification);
         let taskError;
         try {
             const workflowContext = {
                 task,
                 taskSlug,
+                runId: taskRunId,
                 cwd,
                 isCloudMode,
                 options,
@@ -129,11 +154,11 @@ class Agent {
                 fileManager: this.fileManager,
                 gitManager: this.gitManager,
                 promptBuilder: this.promptBuilder,
-                progressReporter: this.progressReporter,
-                adapter: this.adapter,
+                connection: connection.agentConnection,
+                sessionId: taskRunId,
+                sendNotification,
                 mcpServers: this.mcpServers,
                 posthogAPI: this.posthogAPI,
-                emitEvent: (event) => this.emitEvent(event),
                 stepResults: {},
             };
             for (const step of TASK_WORKFLOW) {
@@ -144,63 +169,78 @@ class Agent {
             }
             const shouldCreatePR = options.createPR ?? isCloudMode;
             if (shouldCreatePR) {
-                await this.ensurePullRequest(task, workflowContext.stepResults);
+                await this.ensurePullRequest(task, workflowContext.stepResults, sendNotification);
             }
-            this.logger.info('Task execution complete', { taskId: task.id });
-            this.emitEvent(this.adapter.createStatusEvent('task_complete', { taskId: task.id }));
+            this.logger.info("Task execution complete", { taskId: task.id });
+            await sendNotification(POSTHOG_NOTIFICATIONS.TASK_COMPLETE, {
+                sessionId: taskRunId,
+                taskId: task.id,
+            });
         }
         catch (error) {
             taskError = error instanceof Error ? error : new Error(String(error));
-            this.logger.error('Task execution failed', { taskId: task.id, error: taskError.message });
-        }
-        finally {
-            if (taskError) {
-                await this.progressReporter.fail(taskError);
-                throw taskError;
-            }
-            else {
-                await this.progressReporter.complete();
-            }
+            this.logger.error("Task execution failed", {
+                taskId: task.id,
+                error: taskError.message,
+            });
+            await sendNotification(POSTHOG_NOTIFICATIONS.ERROR, {
+                sessionId: taskRunId,
+                message: taskError.message,
+            });
+            throw taskError;
         }
     }
-    // Direct prompt execution - still supported for low-level usage
-    async run(prompt, options = {}) {
+    /**
+     * Creates an in-process ACP connection for client communication.
+     * Sets up git branch for the task, configures LLM gateway.
+     * The client handles all prompting/querying via the returned streams.
+     *
+     * @returns InProcessAcpConnection with clientStreams for the client to use
+     */
+    async runTaskV2(taskId, taskRunId, options = {}) {
         await this._configureLlmGateway();
-        const baseOptions = {
-            model: "claude-sonnet-4-5-20250929",
-            cwd: options.repositoryPath || this.workingDirectory,
-            permissionMode: options.permissionMode || "default",
-            settingSources: ["local"],
-            mcpServers: this.mcpServers,
+        const task = await this.fetchTask(taskId);
+        const taskSlug = task.slug || task.id;
+        const isCloudMode = options.isCloudMode ?? false;
+        options.repositoryPath || this.workingDirectory;
+        // Use taskRunId as sessionId - they are the same identifier
+        this.currentRunId = taskRunId;
+        this.acpConnection = createAcpConnection({
+            sessionStore: this.sessionStore,
+            sessionId: taskRunId,
+            taskId: task.id,
+        });
+        const sendNotification = async (method, params) => {
+            this.logger.debug(`Notification: ${method}`, params);
+            await this.acpConnection?.agentConnection.extNotification?.(method, params);
         };
-        // Add canUseTool hook if provided (options take precedence over instance config)
-        const canUseTool = options.canUseTool || this.canUseTool;
-        if (canUseTool) {
-            baseOptions.canUseTool = canUseTool;
-        }
-        const response = query({
-            prompt,
-            options: { ...baseOptions, ...(options.queryOverrides || {}) },
+        await sendNotification(POSTHOG_NOTIFICATIONS.RUN_STARTED, {
+            sessionId: taskRunId,
+            runId: taskRunId,
         });
-        const results = [];
-        for await (const message of response) {
-            this.logger.debug('Received message in direct run', message);
-            // Emit raw SDK event
-            this.emitEvent(this.adapter.createRawSDKEvent(message));
-            const transformedEvents = this.adapter.transform(message);
-            for (const event of transformedEvents) {
-                this.emitEvent(event);
+        if (!options.skipGitBranch) {
+            try {
+                await this.prepareTaskBranch(taskSlug, isCloudMode, sendNotification);
+            }
+            catch (error) {
+                const errorMessage = error instanceof Error ? error.message : String(error);
+                this.logger.error("Failed to prepare task branch", {
+                    error: errorMessage,
+                });
+                await sendNotification(POSTHOG_NOTIFICATIONS.ERROR, {
+                    sessionId: taskRunId,
+                    message: errorMessage,
+                });
+                throw error;
             }
-            results.push(message);
         }
-        return { results };
+        return this.acpConnection;
     }
     // PostHog task operations
     async fetchTask(taskId) {
-        this.logger.debug('Fetching task from PostHog', { taskId });
         if (!this.posthogAPI) {
-            const error = new Error('PostHog API not configured. Provide posthogApiUrl and posthogApiKey in constructor.');
-            this.logger.error('PostHog API not configured', error);
+            const error = new Error("PostHog API not configured. Provide posthogApiUrl and posthogApiKey in constructor.");
+            this.logger.error("PostHog API not configured", error);
             throw error;
         }
         return this.posthogAPI.fetchTask(taskId);
@@ -208,62 +248,18 @@ class Agent {
     getPostHogClient() {
         return this.posthogAPI;
     }
-    async listTasks(filters) {
-        if (!this.posthogAPI) {
-            throw new Error('PostHog API not configured. Provide posthogApiUrl and posthogApiKey in constructor.');
-        }
-        return this.posthogAPI.listTasks(filters);
-    }
-    // File system operations for task artifacts
-    async writeTaskFile(taskId, fileName, content, type = 'reference') {
-        this.logger.debug('Writing task file', { taskId, fileName, type, contentLength: content.length });
-        await this.fileManager.writeTaskFile(taskId, { name: fileName, content, type });
-    }
-    async readTaskFile(taskId, fileName) {
-        this.logger.debug('Reading task file', { taskId, fileName });
-        return await this.fileManager.readTaskFile(taskId, fileName);
-    }
     async getTaskFiles(taskId) {
-        this.logger.debug('Getting task files', { taskId });
+        this.logger.debug("Getting task files", { taskId });
         const files = await this.fileManager.getTaskFiles(taskId);
-        this.logger.debug('Found task files', { taskId, fileCount: files.length });
+        this.logger.debug("Found task files", { taskId, fileCount: files.length });
         return files;
     }
-    async writePlan(taskId, plan) {
-        this.logger.info('Writing plan', { taskId, planLength: plan.length });
-        await this.fileManager.writePlan(taskId, plan);
-    }
-    async readPlan(taskId) {
-        this.logger.debug('Reading plan', { taskId });
-        return await this.fileManager.readPlan(taskId);
-    }
-    // Git operations for task execution
-    async createPlanningBranch(taskId) {
-        this.logger.info('Creating planning branch', { taskId });
-        const branchName = await this.gitManager.createTaskPlanningBranch(taskId);
-        this.logger.debug('Planning branch created', { taskId, branchName });
-        return branchName;
-    }
-    async commitPlan(taskId, taskTitle) {
-        this.logger.info('Committing plan', { taskId, taskTitle });
-        const commitHash = await this.gitManager.commitPlan(taskId, taskTitle);
-        this.logger.debug('Plan committed', { taskId, commitHash });
-        return commitHash;
-    }
-    async createImplementationBranch(taskId, planningBranchName) {
-        this.logger.info('Creating implementation branch', { taskId, fromBranch: planningBranchName });
-        const branchName = await this.gitManager.createTaskImplementationBranch(taskId, planningBranchName);
-        this.logger.debug('Implementation branch created', { taskId, branchName });
-        return branchName;
-    }
-    async commitImplementation(taskId, taskTitle, planSummary) {
-        this.logger.info('Committing implementation', { taskId, taskTitle });
-        const commitHash = await this.gitManager.commitImplementation(taskId, taskTitle, planSummary);
-        this.logger.debug('Implementation committed', { taskId, commitHash });
-        return commitHash;
-    }
     async createPullRequest(taskId, branchName, taskTitle, taskDescription, customBody) {
-        this.logger.info('Creating pull request', { taskId, branchName, taskTitle });
+        this.logger.info("Creating pull request", {
+            taskId,
+            branchName,
+            taskTitle,
+        });
         const defaultBody = `## Task Details
 **Task ID**: ${taskId}
 **Description**: ${taskDescription}
@@ -274,40 +270,50 @@ This PR implements the changes described in the task.
 Generated by PostHog Agent`;
         const prBody = customBody || defaultBody;
         const prUrl = await this.gitManager.createPullRequest(branchName, taskTitle, prBody);
-        this.logger.info('Pull request created', { taskId, prUrl });
+        this.logger.info("Pull request created", { taskId, prUrl });
         return prUrl;
     }
     async attachPullRequestToTask(taskId, prUrl, branchName) {
-        this.logger.info('Attaching PR to task run', { taskId, prUrl, branchName });
-        if (!this.posthogAPI || !this.progressReporter.runId) {
-            const error = new Error('PostHog API not configured or no active run. Cannot attach PR to task.');
-            this.logger.error('PostHog API not configured', error);
+        this.logger.info("Attaching PR to task run", { taskId, prUrl, branchName });
+        if (!this.posthogAPI || !this.currentRunId) {
+            const error = new Error("PostHog API not configured or no active run. Cannot attach PR to task.");
+            this.logger.error("PostHog API not configured", error);
             throw error;
         }
         const updates = {
-            output: { pr_url: prUrl }
+            output: { pr_url: prUrl },
         };
         if (branchName) {
             updates.branch = branchName;
         }
-        await this.posthogAPI.updateTaskRun(taskId, this.progressReporter.runId, updates);
-        this.logger.debug('PR attached to task run', { taskId, runId: this.progressReporter.runId, prUrl });
+        await this.posthogAPI.updateTaskRun(taskId, this.currentRunId, updates);
+        this.logger.debug("PR attached to task run", {
+            taskId,
+            runId: this.currentRunId,
+            prUrl,
+        });
     }
     async updateTaskBranch(taskId, branchName) {
-        this.logger.info('Updating task run branch', { taskId, branchName });
-        if (!this.posthogAPI || !this.progressReporter.runId) {
-            const error = new Error('PostHog API not configured or no active run. Cannot update branch.');
-            this.logger.error('PostHog API not configured', error);
+        this.logger.info("Updating task run branch", { taskId, branchName });
+        if (!this.posthogAPI || !this.currentRunId) {
+            const error = new Error("PostHog API not configured or no active run. Cannot update branch.");
+            this.logger.error("PostHog API not configured", error);
             throw error;
         }
-        await this.posthogAPI.updateTaskRun(taskId, this.progressReporter.runId, { branch: branchName });
-        this.logger.debug('Task run branch updated', { taskId, runId: this.progressReporter.runId, branchName });
+        await this.posthogAPI.updateTaskRun(taskId, this.currentRunId, {
+            branch: branchName,
+        });
+        this.logger.debug("Task run branch updated", {
+            taskId,
+            runId: this.currentRunId,
+            branchName,
+        });
     }
     // Execution management
     cancelTask(taskId) {
         // Find the execution for this task and cancel it
-        for (const [executionId, execution] of this.taskManager['executionStates']) {
-            if (execution.taskId === taskId && execution.status === 'running') {
+        for (const [executionId, execution] of this.taskManager.executionStates) {
+            if (execution.taskId === taskId && execution.status === "running") {
                 this.taskManager.cancelExecution(executionId);
                 break;
             }
@@ -315,26 +321,43 @@ Generated by PostHog Agent`;
     }
     getTaskExecutionStatus(taskId) {
         // Find the execution for this task
-        for (const execution of this.taskManager['executionStates'].values()) {
+        for (const execution of this.taskManager.executionStates.values()) {
             if (execution.taskId === taskId) {
                 return execution.status;
             }
         }
         return null;
     }
-    async prepareTaskBranch(taskSlug, isCloudMode) {
+    async prepareTaskBranch(taskSlug, isCloudMode, sendNotification) {
         if (await this.gitManager.hasChanges()) {
-            throw new Error('Cannot start task with uncommitted changes. Please commit or stash your changes first.');
+            throw new Error("Cannot start task with uncommitted changes. Please commit or stash your changes first.");
+        }
+        // If we're running in a worktree, we're already on the correct branch
+        // (the worktree was created with its own branch). Skip branch creation.
+        const isWorktree = await this.gitManager.isWorktree();
+        if (isWorktree) {
+            const currentBranch = await this.gitManager.getCurrentBranch();
+            this.logger.info("Running in worktree, using existing branch", {
+                branch: currentBranch,
+            });
+            await sendNotification(POSTHOG_NOTIFICATIONS.BRANCH_CREATED, {
+                branch: currentBranch,
+            });
+            return;
         }
         await this.gitManager.resetToDefaultBranchIfNeeded();
         const existingBranch = await this.gitManager.getTaskBranch(taskSlug);
         if (!existingBranch) {
             const branchName = await this.gitManager.createTaskBranch(taskSlug);
-            this.emitEvent(this.adapter.createStatusEvent('branch_created', { branch: branchName }));
+            await sendNotification(POSTHOG_NOTIFICATIONS.BRANCH_CREATED, {
+                branch: branchName,
+            });
             await this.gitManager.addAllPostHogFiles();
             // Only commit if there are changes or we're in cloud mode
             if (isCloudMode) {
-                await this.gitManager.commitAndPush(`Initialize task ${taskSlug}`, { allowEmpty: true });
+                await this.gitManager.commitAndPush(`Initialize task ${taskSlug}`, {
+                    allowEmpty: true,
+                });
             }
             else {
                 // Check if there are any changes before committing
@@ -345,7 +368,9 @@ Generated by PostHog Agent`;
             }
         }
         else {
-            this.logger.info('Switching to existing task branch', { branch: existingBranch });
+            this.logger.info("Switching to existing task branch", {
+                branch: existingBranch,
+            });
             await this.gitManager.switchToBranch(existingBranch);
         }
     }
@@ -359,46 +384,41 @@ Generated by PostHog Agent`;
             process.env.OPENAI_API_KEY = resolvedToken;
         }
     }
-    async ensurePullRequest(task, stepResults) {
+    async ensurePullRequest(task, stepResults, sendNotification) {
         const latestRun = task.latest_run;
-        const existingPr = latestRun?.output && typeof latestRun.output === 'object'
+        const existingPr = latestRun?.output && typeof latestRun.output === "object"
             ? latestRun.output.pr_url
             : null;
         if (existingPr) {
-            this.logger.info('PR already exists, skipping creation', { taskId: task.id, prUrl: existingPr });
+            this.logger.info("PR already exists, skipping creation", {
+                taskId: task.id,
+                prUrl: existingPr,
+            });
             return;
         }
-        const buildResult = stepResults['build'];
+        const buildResult = stepResults.build;
         if (!buildResult?.commitCreated) {
-            this.logger.warn('Build step did not produce a commit; skipping PR creation', { taskId: task.id });
+            this.logger.warn("Build step did not produce a commit; skipping PR creation", { taskId: task.id });
             return;
         }
         const branchName = await this.gitManager.getCurrentBranch();
-        const finalizeResult = stepResults['finalize'];
+        const finalizeResult = stepResults.finalize;
         const prBody = finalizeResult?.prBody;
-        const prUrl = await this.createPullRequest(task.id, branchName, task.title, task.description ?? '', prBody);
-        this.emitEvent(this.adapter.createStatusEvent('pr_created', { prUrl }));
+        const prUrl = await this.createPullRequest(task.id, branchName, task.title, task.description ?? "", prBody);
+        await sendNotification(POSTHOG_NOTIFICATIONS.PR_CREATED, { prUrl });
         try {
             await this.attachPullRequestToTask(task.id, prUrl, branchName);
-            this.logger.info('PR attached to task successfully', { taskId: task.id, prUrl });
+            this.logger.info("PR attached to task successfully", {
+                taskId: task.id,
+                prUrl,
+            });
         }
         catch (error) {
-            this.logger.warn('Could not attach PR to task', {
+            this.logger.warn("Could not attach PR to task", {
                 error: error instanceof Error ? error.message : String(error),
             });
         }
     }
-    emitEvent(event) {
-        if (this.debug && event.type !== 'token') {
-            // Log all events except tokens (too verbose)
-            this.logger.debug('Emitting event', { type: event.type, ts: event.ts });
-        }
-        const persistPromise = this.progressReporter.recordEvent(event);
-        if (persistPromise && typeof persistPromise.then === 'function') {
-            persistPromise.catch((error) => this.logger.debug('Failed to persist agent event', { message: error.message }));
-        }
-        this.onEvent?.(event);
-    }
 }
 
 export { Agent };