@posthog/agent 1.28.0 → 1.30.0

package/dist/index.d.ts

@@ -37,6 +37,8 @@ declare const POSTHOG_NOTIFICATIONS: {
     readonly CONSOLE: "_posthog/console";
     /** SDK session ID notification (for resumption) */
     readonly SDK_SESSION: "_posthog/sdk_session";
+    /** Sandbox execution output (stdout/stderr from Modal or Docker) */
+    readonly SANDBOX_OUTPUT: "_posthog/sandbox_output";
 };
 type PostHogNotificationType = (typeof POSTHOG_NOTIFICATIONS)[keyof typeof POSTHOG_NOTIFICATIONS];
 interface ArtifactNotificationPayload {
@@ -92,7 +94,16 @@ interface SdkSessionPayload {
     sessionId: string;
     sdkSessionId: string;
 }
-type PostHogNotificationPayload = ArtifactNotificationPayload | PhaseNotificationPayload | BranchCreatedPayload | PrCreatedPayload | RunStartedPayload | TaskCompletePayload | ErrorNotificationPayload | ConsoleNotificationPayload | SdkSessionPayload;
+/**
+ * Sandbox execution output
+ */
+interface SandboxOutputPayload {
+    sessionId: string;
+    stdout: string;
+    stderr: string;
+    exitCode: number;
+}
+type PostHogNotificationPayload = ArtifactNotificationPayload | PhaseNotificationPayload | BranchCreatedPayload | PrCreatedPayload | RunStartedPayload | TaskCompletePayload | ErrorNotificationPayload | ConsoleNotificationPayload | SdkSessionPayload | SandboxOutputPayload;
 
 /**
  * Stored custom notification following ACP extensibility model.
@@ -217,7 +228,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 +238,7 @@ interface AgentConfig {
 }
 interface PostHogAPIConfig {
     apiUrl: string;
-    apiKey: string;
+    getApiKey: () => string;
     projectId: number;
 }
 type ResourceType = "error" | "experiment" | "insight" | "feature_flag" | "generic";
@@ -454,7 +465,7 @@ 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;
@@ -482,6 +493,7 @@ 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;
 }
@@ -1234,6 +1246,8 @@ declare class ToolRegistry {
     getByCategory(category: string): Tool[];
 }
 
+declare function getLlmGatewayUrl(posthogHost: string): string;
+
 interface WorktreeConfig {
     mainRepoPath: string;
     worktreeBasePath?: string;
@@ -1255,7 +1269,9 @@ declare class WorktreeManager {
     ensureArrayDirIgnored(): Promise<void>;
     private generateUniqueWorktreeName;
     private getDefaultBranch;
-    createWorktree(): Promise<WorktreeInfo>;
+    createWorktree(options?: {
+        baseBranch?: string;
+    }): Promise<WorktreeInfo>;
     deleteWorktree(worktreePath: string): Promise<void>;
     getWorktreeInfo(worktreePath: string): Promise<WorktreeInfo | null>;
     listWorktrees(): Promise<WorktreeInfo[]>;
@@ -1271,4 +1287,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 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, getLlmGatewayUrl, parseAgentEvent, parseAgentEvents };

package/dist/index.js

@@ -65,7 +65,9 @@ var POSTHOG_NOTIFICATIONS = {
   /** Console/log output */
   CONSOLE: "_posthog/console",
   /** SDK session ID notification (for resumption) */
-  SDK_SESSION: "_posthog/sdk_session"
+  SDK_SESSION: "_posthog/sdk_session",
+  /** Sandbox execution output (stdout/stderr from Modal or Docker) */
+  SANDBOX_OUTPUT: "_posthog/sandbox_output"
 };
 
 // src/adapters/claude/claude.ts
@@ -196,7 +198,7 @@ function createTappedWritableStream(underlying, options) {
 // package.json
 var package_default = {
   name: "@posthog/agent",
-  version: "1.28.0",
+  version: "1.30.0",
   repository: "https://github.com/PostHog/array",
   description: "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   main: "./dist/index.js",
@@ -1729,6 +1731,18 @@ var ClaudeAcpAgent = class {
     }
     const availableCommands = await getAvailableSlashCommands(q);
     const models = await getAvailableModels(q);
+    const requestedModel = params._meta?.model;
+    if (requestedModel) {
+      try {
+        await q.setModel(requestedModel);
+        this.logger.info("Set initial model", { model: requestedModel });
+      } catch (err) {
+        this.logger.warn("Failed to set initial model, using default", {
+          requestedModel,
+          error: err
+        });
+      }
+    }
     setTimeout(() => {
       this.client.sessionUpdate({
         sessionId,
@@ -1901,8 +1915,11 @@ var ClaudeAcpAgent = class {
             throw RequestError.authRequired();
           }
           const content = message.message.content;
+          const contentToProcess = Array.isArray(content) ? content.filter(
+            (block) => block.type !== "text" && block.type !== "thinking"
+          ) : content;
           for (const notification of toAcpNotifications(
-            content,
+            contentToProcess,
             message.message.role,
             params.sessionId,
             this.toolUseCache,
@@ -2121,6 +2138,11 @@ var ClaudeAcpAgent = class {
       await this.resumeSession(params);
       return {};
     }
+    if (method === "session/setModel") {
+      const { sessionId, modelId } = params;
+      await this.setSessionModel({ sessionId, modelId });
+      return {};
+    }
     throw RequestError.methodNotFound(method);
   }
   /**
@@ -2860,13 +2882,9 @@ import { promisify } from "util";
 var execAsync = promisify(exec);
 var GitManager = class {
   repositoryPath;
-  authorName;
-  authorEmail;
   logger;
   constructor(config) {
     this.repositoryPath = config.repositoryPath;
-    this.authorName = config.authorName;
-    this.authorEmail = config.authorEmail;
     this.logger = config.logger || new Logger({ debug: false, prefix: "[GitManager]" });
   }
   escapeShellArg(str) {
@@ -3040,11 +3058,6 @@ ${error}`);
     if (options?.allowEmpty) {
       command += " --allow-empty";
     }
-    const authorName = options?.authorName || this.authorName;
-    const authorEmail = options?.authorEmail || this.authorEmail;
-    if (authorName && authorEmail) {
-      command += ` --author="${authorName} <${authorEmail}>"`;
-    }
     return command;
   }
   async getRemoteUrl() {
@@ -3164,7 +3177,6 @@ ${error}`);
     const message = `\u{1F4CB} Add plan for task: ${taskTitle}
 
 Task ID: ${taskId}
-Generated by PostHog Agent
 
 This commit contains the implementation plan and supporting documentation
 for the task. Review the plan before proceeding with implementation.`;
@@ -3181,8 +3193,7 @@ for the task. Review the plan before proceeding with implementation.`;
     }
     let message = `\u2728 Implement task: ${taskTitle}
 
-Task ID: ${taskId}
-Generated by PostHog Agent`;
+Task ID: ${taskId}`;
     if (planSummary) {
       message += `
 
@@ -3286,6 +3297,18 @@ This commit implements the changes described in the task plan.`;
   }
 };
 
+// src/utils/gateway.ts
+function getLlmGatewayUrl(posthogHost) {
+  const url = new URL(posthogHost);
+  const hostname = url.hostname;
+  if (hostname === "localhost" || hostname === "127.0.0.1") {
+    return `${url.protocol}//localhost:3308`;
+  }
+  const regionMatch = hostname.match(/^(us|eu)\.posthog\.com$/);
+  const region = regionMatch ? regionMatch[1] : "us";
+  return `https://gateway.${region}.posthog.com`;
+}
+
 // src/posthog-api.ts
 var PostHogAPIClient = class {
   config;
@@ -3298,7 +3321,7 @@ var PostHogAPIClient = class {
   }
   get headers() {
     return {
-      Authorization: `Bearer ${this.config.apiKey}`,
+      Authorization: `Bearer ${this.config.getApiKey()}`,
       "Content-Type": "application/json"
     };
   }
@@ -3330,11 +3353,10 @@ var PostHogAPIClient = class {
     return this.baseUrl;
   }
   getApiKey() {
-    return this.config.apiKey;
+    return this.config.getApiKey();
   }
   getLlmGatewayUrl() {
-    const teamId = this.getTeamId();
-    return `${this.baseUrl}/api/projects/${teamId}/llm_gateway`;
+    return getLlmGatewayUrl(this.baseUrl);
   }
   async fetchTask(taskId) {
     const teamId = this.getTeamId();
@@ -3725,14 +3747,16 @@ var PromptBuilder = class {
     if (filePaths.length === 0 || !repositoryPath) {
       return { description, referencedFiles };
     }
+    const successfulPaths = /* @__PURE__ */ new Set();
     for (const filePath of filePaths) {
       const content = await this.readFileContent(repositoryPath, filePath);
       if (content !== null) {
         referencedFiles.push({ path: filePath, content });
+        successfulPaths.add(filePath);
       }
     }
     let processedDescription = description;
-    for (const filePath of filePaths) {
+    for (const filePath of successfulPaths) {
       const fileName = filePath.split("/").pop() || filePath;
       processedDescription = processedDescription.replace(
         new RegExp(
@@ -4447,7 +4471,7 @@ Placeholder content for ${file.name}`;
   generatePostHogReadme() {
     return `# PostHog Task Files
 
-This directory contains task-related files generated by the PostHog Agent.
+This directory contains task-related files.
 
 ## Structure
 
@@ -4463,7 +4487,7 @@ Each task has its own subdirectory: \`.posthog/{task-id}/\`
 
 These files are:
 - Version controlled alongside your code
-- Used by the PostHog Agent for context
+- Used for task context and planning
 - Available for review in pull requests
 - Organized by task ID for easy reference
 
@@ -4473,10 +4497,6 @@ Customize \`.posthog/.gitignore\` to control which files are committed:
 - Include plans and documentation by default
 - Exclude temporary files and sensitive data
 - Customize based on your team's needs
-
----
-
-*Generated by PostHog Agent*
 `;
   }
 };
@@ -5633,8 +5653,8 @@ var Agent = class {
     this.debug = config.debug || false;
     const posthogMcpUrl = config.posthogMcpUrl || process.env.POSTHOG_MCP_URL || "https://mcp.posthog.com/mcp";
     const headers = {};
-    if (config.posthogApiKey) {
-      headers.Authorization = `Bearer ${config.posthogApiKey}`;
+    if (config.getPosthogApiKey) {
+      headers.Authorization = `Bearer ${config.getPosthogApiKey()}`;
     }
     const defaultMcpServers = {
       posthog: {
@@ -5662,10 +5682,10 @@ var Agent = class {
       logger: this.logger.child("GitManager")
     });
     this.templateManager = new TemplateManager();
-    if (config.posthogApiUrl && config.posthogApiKey && config.posthogProjectId) {
+    if (config.posthogApiUrl && config.getPosthogApiKey && config.posthogProjectId) {
       this.posthogAPI = new PostHogAPIClient({
         apiUrl: config.posthogApiUrl,
-        apiKey: config.posthogApiKey,
+        getApiKey: config.getPosthogApiKey,
         projectId: config.posthogProjectId
       });
       this.sessionStore = new SessionStore(
@@ -5688,7 +5708,7 @@ var Agent = class {
     this.logger.setDebug(enabled);
   }
   /**
-   * Configure LLM gateway environment variables for Claude Code CLI
+   * Configure LLM gateway environment variables for Claude Code CLI.
    */
   async _configureLlmGateway() {
     if (!this.posthogAPI) {
@@ -5700,6 +5720,7 @@ var Agent = class {
       process.env.ANTHROPIC_BASE_URL = gatewayUrl;
       process.env.ANTHROPIC_AUTH_TOKEN = apiKey;
       this.ensureOpenAIGatewayEnv(gatewayUrl, apiKey);
+      this.ensureGeminiGatewayEnv(gatewayUrl, apiKey);
     } catch (error) {
       this.logger.error("Failed to configure LLM gateway", error);
       throw error;
@@ -5883,9 +5904,7 @@ var Agent = class {
 **Description**: ${taskDescription}
 
 ## Changes
-This PR implements the changes described in the task.
-
-Generated by PostHog Agent`;
+This PR implements the changes described in the task.`;
     const prBody = customBody || defaultBody;
     const prUrl = await this.gitManager.createPullRequest(
       branchName,
@@ -6004,6 +6023,16 @@ Generated by PostHog Agent`;
       process.env.OPENAI_API_KEY = resolvedToken;
     }
   }
+  ensureGeminiGatewayEnv(gatewayUrl, token) {
+    const resolvedGatewayUrl = gatewayUrl || process.env.ANTHROPIC_BASE_URL;
+    const resolvedToken = token || process.env.ANTHROPIC_AUTH_TOKEN;
+    if (resolvedGatewayUrl) {
+      process.env.GEMINI_BASE_URL = resolvedGatewayUrl;
+    }
+    if (resolvedToken) {
+      process.env.GEMINI_API_KEY = resolvedToken;
+    }
+  }
   async runTaskCloud(taskId, taskRunId, options = {}) {
     await this._configureLlmGateway();
     const task = await this.fetchTask(taskId);
@@ -7083,7 +7112,7 @@ ${ignorePattern}
       }
     }
   }
-  async createWorktree() {
+  async createWorktree(options) {
     if (!this.usesExternalPath()) {
       await this.ensureArrayDirIgnored();
     }
@@ -7094,7 +7123,7 @@ ${ignorePattern}
     const worktreeName = await this.generateUniqueWorktreeName();
     const worktreePath = this.getWorktreePath(worktreeName);
     const branchName = `array/${worktreeName}`;
-    const baseBranch = await this.getDefaultBranch();
+    const baseBranch = options?.baseBranch ?? await this.getDefaultBranch();
     this.logger.info("Creating worktree", {
       worktreeName,
       worktreePath,
@@ -7321,6 +7350,7 @@ export {
   ToolRegistry,
   WorktreeManager,
   createAcpConnection,
+  getLlmGatewayUrl,
   parseAgentEvent,
   parseAgentEvents
 };