@posthog/agent 1.28.0 → 1.30.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "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",

package/src/acp-extensions.ts

@@ -33,6 +33,8 @@ export const POSTHOG_NOTIFICATIONS = {
   CONSOLE: "_posthog/console",
   /** SDK session ID notification (for resumption) */
   SDK_SESSION: "_posthog/sdk_session",
+  /** Sandbox execution output (stdout/stderr from Modal or Docker) */
+  SANDBOX_OUTPUT: "_posthog/sandbox_output",
 } as const;
 
 export type PostHogNotificationType =
@@ -105,6 +107,16 @@ export interface SdkSessionPayload {
   sdkSessionId: string;
 }
 
+/**
+ * Sandbox execution output
+ */
+export interface SandboxOutputPayload {
+  sessionId: string;
+  stdout: string;
+  stderr: string;
+  exitCode: number;
+}
+
 export type PostHogNotificationPayload =
   | ArtifactNotificationPayload
   | PhaseNotificationPayload
@@ -114,4 +126,5 @@ export type PostHogNotificationPayload =
   | TaskCompletePayload
   | ErrorNotificationPayload
   | ConsoleNotificationPayload
-  | SdkSessionPayload;
+  | SdkSessionPayload
+  | SandboxOutputPayload;

package/src/adapters/claude/claude.ts

@@ -135,6 +135,8 @@ export type NewSessionMeta = {
      */
     options?: Options;
   };
+  /** Initial model to use for the session (e.g., 'claude-opus-4-5', 'gpt-5.1') */
+  model?: string;
 };
 
 /**
@@ -443,6 +445,20 @@ export class ClaudeAcpAgent implements Agent {
     const availableCommands = await getAvailableSlashCommands(q);
     const models = await getAvailableModels(q);
 
+    // Set initial model if provided via _meta (must be after getAvailableModels which resets to default)
+    const requestedModel = (params._meta as NewSessionMeta | undefined)?.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,
+        });
+      }
+    }
+
     // Needs to happen after we return the session
     setTimeout(() => {
       this.client.sessionUpdate({
@@ -657,12 +673,16 @@ export class ClaudeAcpAgent implements Agent {
             throw RequestError.authRequired();
           }
 
-          // For assistant messages, text/thinking are normally streamed via stream_event.
-          // But some gateways (like LiteLLM) don't stream, so we process all content.
+          // Text/thinking is streamed via stream_event, so skip them here to avoid duplication.
           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 as typeof content,
             message.message.role,
             params.sessionId,
             this.toolUseCache,
@@ -922,6 +942,15 @@ export class ClaudeAcpAgent implements Agent {
       return {};
     }
 
+    if (method === "session/setModel") {
+      const { sessionId, modelId } = params as {
+        sessionId: string;
+        modelId: string;
+      };
+      await this.setSessionModel({ sessionId, modelId });
+      return {};
+    }
+
     throw RequestError.methodNotFound(method);
   }
 

package/src/agent.ts

@@ -57,8 +57,8 @@ export class Agent {
 
     // Add auth if API key provided
     const headers: Record<string, string> = {};
-    if (config.posthogApiKey) {
-      headers.Authorization = `Bearer ${config.posthogApiKey}`;
+    if (config.getPosthogApiKey) {
+      headers.Authorization = `Bearer ${config.getPosthogApiKey()}`;
     }
 
     const defaultMcpServers = {
@@ -93,12 +93,12 @@ export class Agent {
 
     if (
       config.posthogApiUrl &&
-      config.posthogApiKey &&
+      config.getPosthogApiKey &&
       config.posthogProjectId
     ) {
       this.posthogAPI = new PostHogAPIClient({
         apiUrl: config.posthogApiUrl,
-        apiKey: config.posthogApiKey,
+        getApiKey: config.getPosthogApiKey,
         projectId: config.posthogProjectId,
       });
 
@@ -126,7 +126,7 @@ export class Agent {
   }
 
   /**
-   * Configure LLM gateway environment variables for Claude Code CLI
+   * Configure LLM gateway environment variables for Claude Code CLI.
    */
   private async _configureLlmGateway(): Promise<void> {
     if (!this.posthogAPI) {
@@ -139,6 +139,7 @@ export class Agent {
       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;
@@ -373,9 +374,7 @@ export class Agent {
 **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(
@@ -529,6 +528,19 @@ Generated by PostHog Agent`;
     }
   }
 
+  private ensureGeminiGatewayEnv(gatewayUrl?: string, token?: string): void {
+    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: string,
     taskRunId: string,

package/src/git-manager.ts

@@ -6,8 +6,6 @@ const execAsync = promisify(exec);
 
 export interface GitConfig {
   repositoryPath: string;
-  authorName?: string;
-  authorEmail?: string;
   logger?: Logger;
 }
 
@@ -19,14 +17,10 @@ export interface BranchInfo {
 
 export class GitManager {
   private repositoryPath: string;
-  private authorName?: string;
-  private authorEmail?: string;
   private logger: Logger;
 
   constructor(config: GitConfig) {
     this.repositoryPath = config.repositoryPath;
-    this.authorName = config.authorName;
-    this.authorEmail = config.authorEmail;
     this.logger =
       config.logger || new Logger({ debug: false, prefix: "[GitManager]" });
   }
@@ -170,8 +164,7 @@ export class GitManager {
   async commitChanges(
     message: string,
     options?: {
-      authorName?: string;
-      authorEmail?: string;
+      allowEmpty?: boolean;
     },
   ): Promise<string> {
     const command = this.buildCommitCommand(message, options);
@@ -244,8 +237,6 @@ export class GitManager {
     message: string,
     options?: {
       allowEmpty?: boolean;
-      authorName?: string;
-      authorEmail?: string;
     },
   ): string {
     let command = `commit -m "${this.escapeShellArg(message)}"`;
@@ -254,13 +245,6 @@ export class GitManager {
       command += " --allow-empty";
     }
 
-    const authorName = options?.authorName || this.authorName;
-    const authorEmail = options?.authorEmail || this.authorEmail;
-
-    if (authorName && authorEmail) {
-      command += ` --author="${authorName} <${authorEmail}>"`;
-    }
-
     return command;
   }
 
@@ -427,7 +411,6 @@ export class GitManager {
     const message = `📋 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.`;
@@ -452,8 +435,7 @@ for the task. Review the plan before proceeding with implementation.`;
 
     let message = `✨ Implement task: ${taskTitle}
 
-Task ID: ${taskId}
-Generated by PostHog Agent`;
+Task ID: ${taskId}`;
 
     if (planSummary) {
       message += `\n\nPlan Summary:\n${planSummary}`;

package/src/posthog-api.ts

@@ -8,6 +8,7 @@ import type {
   TaskRunArtifact,
   UrlMention,
 } from "./types.js";
+import { getLlmGatewayUrl } from "./utils/gateway.js";
 
 interface PostHogApiResponse<T> {
   results?: T[];
@@ -42,7 +43,7 @@ export class PostHogAPIClient {
 
   private get headers(): Record<string, string> {
     return {
-      Authorization: `Bearer ${this.config.apiKey}`,
+      Authorization: `Bearer ${this.config.getApiKey()}`,
       "Content-Type": "application/json",
     };
   }
@@ -84,12 +85,11 @@ export class PostHogAPIClient {
   }
 
   getApiKey(): string {
-    return this.config.apiKey;
+    return this.config.getApiKey();
   }
 
   getLlmGatewayUrl(): string {
-    const teamId = this.getTeamId();
-    return `${this.baseUrl}/api/projects/${teamId}/llm_gateway`;
+    return getLlmGatewayUrl(this.baseUrl);
   }
 
   async fetchTask(taskId: string): Promise<Task> {

package/src/prompt-builder.ts

@@ -215,17 +215,19 @@ export class PromptBuilder {
       return { description, referencedFiles };
     }
 
-    // Read all referenced files
+    // Read all referenced files, tracking which ones succeed
+    const successfulPaths = new Set<string>();
     for (const filePath of filePaths) {
       const content = await this.readFileContent(repositoryPath, filePath);
       if (content !== null) {
         referencedFiles.push({ path: filePath, content });
+        successfulPaths.add(filePath);
       }
     }
 
-    // Replace file tags with just the filename for readability
+    // Only replace tags for files that were successfully read
     let processedDescription = description;
-    for (const filePath of filePaths) {
+    for (const filePath of successfulPaths) {
       const fileName = filePath.split("/").pop() || filePath;
       processedDescription = processedDescription.replace(
         new RegExp(

package/src/template-manager.ts

@@ -205,7 +205,7 @@ export class TemplateManager {
   generatePostHogReadme(): string {
     return `# PostHog Task Files
 
-This directory contains task-related files generated by the PostHog Agent.
+This directory contains task-related files.
 
 ## Structure
 
@@ -221,7 +221,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
 
@@ -231,10 +231,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*
 `;
   }
 }

package/src/types.ts

@@ -197,7 +197,7 @@ export interface AgentConfig {
 
   // PostHog API configuration (optional - enables PostHog integration when provided)
   posthogApiUrl?: string;
-  posthogApiKey?: string;
+  getPosthogApiKey?: () => string;
   posthogProjectId?: number;
 
   // PostHog MCP configuration
@@ -219,7 +219,7 @@ export interface AgentConfig {
 
 export interface PostHogAPIConfig {
   apiUrl: string;
-  apiKey: string;
+  getApiKey: () => string;
   projectId: number;
 }
 

package/src/utils/gateway.ts

@@ -0,0 +1,15 @@
+export function getLlmGatewayUrl(posthogHost: string): string {
+  const url = new URL(posthogHost);
+  const hostname = url.hostname;
+
+  if (hostname === "localhost" || hostname === "127.0.0.1") {
+    return `${url.protocol}//localhost:3308`;
+  }
+
+  // Extract region from hostname (us.posthog.com, eu.posthog.com)
+  // app.posthog.com is legacy US
+  const regionMatch = hostname.match(/^(us|eu)\.posthog\.com$/);
+  const region = regionMatch ? regionMatch[1] : "us";
+
+  return `https://gateway.${region}.posthog.com`;
+}

package/src/worktree-manager.ts

@@ -628,7 +628,9 @@ export class WorktreeManager {
     }
   }
 
-  async createWorktree(): Promise<WorktreeInfo> {
+  async createWorktree(options?: {
+    baseBranch?: string;
+  }): Promise<WorktreeInfo> {
     // Only modify .git/info/exclude when using in-repo storage
     if (!this.usesExternalPath()) {
       await this.ensureArrayDirIgnored();
@@ -644,7 +646,7 @@ export class WorktreeManager {
     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,