@posthog/agent 2.3.43 → 2.3.53

package/dist/server/bin.cjs

@@ -904,7 +904,7 @@ var import_hono = require("hono");
 // package.json
 var package_default = {
   name: "@posthog/agent",
-  version: "2.3.43",
+  version: "2.3.53",
   repository: "https://github.com/PostHog/code",
   description: "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   exports: {
@@ -4159,6 +4159,10 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
               (m) => m.contextWindow
             );
             const contextWindowSize = contextWindows.length > 0 ? Math.min(...contextWindows) : getDefaultContextWindow(this.session.modelId ?? "");
+            this.session.contextSize = contextWindowSize;
+            if (lastAssistantTotalUsage !== null) {
+              this.session.contextUsed = lastAssistantTotalUsage;
+            }
             if (lastAssistantTotalUsage !== null) {
               await this.client.sessionUpdate({
                 sessionId: params.sessionId,
@@ -11104,7 +11108,7 @@ var SessionLogWriter = class _SessionLogWriter {
       taskId: context.taskId,
       runId: context.runId
     });
-    this.sessions.set(sessionId, { context });
+    this.sessions.set(sessionId, { context, currentTurnMessages: [] });
     this.lastFlushAttemptTime.set(sessionId, Date.now());
     if (this.localCachePath) {
       const sessionDir = import_node_path7.default.join(
@@ -11151,6 +11155,7 @@ var SessionLogWriter = class _SessionLogWriter {
       const nonChunkAgentText = this.extractAgentMessageText(message);
       if (nonChunkAgentText) {
         session.lastAgentMessage = nonChunkAgentText;
+        session.currentTurnMessages.push(nonChunkAgentText);
       }
       const entry = {
         type: "notification",
@@ -11247,6 +11252,7 @@ var SessionLogWriter = class _SessionLogWriter {
     const { text: text2, firstTimestamp } = session.chunkBuffer;
     session.chunkBuffer = void 0;
     session.lastAgentMessage = text2;
+    session.currentTurnMessages.push(text2);
     const entry = {
       type: "notification",
       timestamp: firstTimestamp,
@@ -11272,6 +11278,17 @@ var SessionLogWriter = class _SessionLogWriter {
   getLastAgentMessage(sessionId) {
     return this.sessions.get(sessionId)?.lastAgentMessage;
   }
+  getFullAgentResponse(sessionId) {
+    const session = this.sessions.get(sessionId);
+    if (!session || session.currentTurnMessages.length === 0) return void 0;
+    return session.currentTurnMessages.join("\n\n");
+  }
+  resetTurnMessages(sessionId) {
+    const session = this.sessions.get(sessionId);
+    if (session) {
+      session.currentTurnMessages = [];
+    }
+  }
   extractAgentMessageText(message) {
     if (message.method !== "session/update") {
       return null;
@@ -11587,6 +11604,12 @@ var AgentServer = class _AgentServer {
   questionRelayedToSlack = false;
   detectedPrUrl = null;
   resumeState = null;
+  // Guards against concurrent session initialization. autoInitializeSession() and
+  // the GET /events SSE handler can both call initializeSession() — the SSE connection
+  // often arrives while newSession() is still awaited (this.session is still null),
+  // causing a second session to be created and duplicate Slack messages to be sent.
+  initializationPromise = null;
+  pendingEvents = [];
   emitConsoleLog = (level, _scope, message, data) => {
     if (!this.session) return;
     const formatted = data !== void 0 ? `${message} ${JSON.stringify(data)}` : message;
@@ -11666,6 +11689,7 @@ var AgentServer = class _AgentServer {
             await this.initializeSession(payload, sseController);
           } else {
             this.session.sseController = sseController;
+            this.replayPendingEvents();
           }
           this.sendSseEvent(sseController, {
             type: "connected",
@@ -11845,6 +11869,7 @@ var AgentServer = class _AgentServer {
         this.logger.info(
           `Processing user message (detectedPrUrl=${this.detectedPrUrl ?? "none"}): ${content.substring(0, 100)}...`
         );
+        this.session.logWriter.resetTurnMessages(this.session.payload.run_id);
         const result = await this.session.clientConnection.prompt({
           sessionId: this.session.acpSessionId,
           prompt: [{ type: "text", text: content }],
@@ -11860,7 +11885,24 @@ You MUST NOT create a new branch, close the existing PR, or create a new PR.`
           }
         });
         this.broadcastTurnComplete(result.stopReason);
-        return { stopReason: result.stopReason };
+        if (result.stopReason === "end_turn") {
+          this.relayAgentResponse(this.session.payload).catch(
+            (err) => this.logger.warn("Failed to relay follow-up response", err)
+          );
+        }
+        let assistantMessage;
+        try {
+          await this.session.logWriter.flush(this.session.payload.run_id);
+          assistantMessage = this.session.logWriter.getFullAgentResponse(
+            this.session.payload.run_id
+          );
+        } catch {
+          this.logger.warn("Failed to extract assistant message from logs");
+        }
+        return {
+          stopReason: result.stopReason,
+          ...assistantMessage && { assistant_message: assistantMessage }
+        };
       }
       case POSTHOG_NOTIFICATIONS.CANCEL:
       case "cancel": {
@@ -11883,6 +11925,28 @@ You MUST NOT create a new branch, close the existing PR, or create a new PR.`
     }
   }
   async initializeSession(payload, sseController) {
+    if (this.initializationPromise) {
+      this.logger.info("Waiting for in-progress initialization", {
+        runId: payload.run_id
+      });
+      await this.initializationPromise;
+      if (this.session && sseController) {
+        this.session.sseController = sseController;
+        this.replayPendingEvents();
+      }
+      return;
+    }
+    this.initializationPromise = this._doInitializeSession(
+      payload,
+      sseController
+    );
+    try {
+      await this.initializationPromise;
+    } finally {
+      this.initializationPromise = null;
+    }
+  }
+  async _doInitializeSession(payload, sseController) {
     if (this.session) {
       await this.cleanupSession();
     }
@@ -12069,6 +12133,7 @@ You MUST NOT create a new branch, close the existing PR, or create a new PR.`
         descriptionLength: initialPrompt.length,
         usedInitialPromptOverride: !!initialPromptOverride
       });
+      this.session.logWriter.resetTurnMessages(payload.run_id);
       const result = await this.session.clientConnection.prompt({
         sessionId: this.session.acpSessionId,
         prompt: [{ type: "text", text: initialPrompt }]
@@ -12095,7 +12160,7 @@ You MUST NOT create a new branch, close the existing PR, or create a new PR.`
         this.resumeState.conversation
       );
       const pendingUserMessage = this.getPendingUserMessage(taskRun);
-      const sandboxContext = this.resumeState.snapshotApplied ? `The sandbox environment (all files, packages, and code changes) has been fully restored from a snapshot.` : `The sandbox could not be restored from a snapshot (it may have expired). You are starting with a fresh environment but have the full conversation history below.`;
+      const sandboxContext = this.resumeState.snapshotApplied ? `The workspace environment (all files, packages, and code changes) has been fully restored from where you left off.` : `The workspace files from the previous session were not restored (the file snapshot may have expired), so you are starting with a fresh environment. Your conversation history is fully preserved below.`;
       let resumePrompt;
       if (pendingUserMessage) {
         resumePrompt = `You are resuming a previous conversation. ${sandboxContext}
@@ -12126,6 +12191,7 @@ Continue from where you left off. The user is waiting for your response.`;
         snapshotApplied: this.resumeState.snapshotApplied
       });
       this.resumeState = null;
+      this.session.logWriter.resetTurnMessages(payload.run_id);
       const result = await this.session.clientConnection.prompt({
         sessionId: this.session.acpSessionId,
         prompt: [{ type: "text", text: resumePrompt }]
@@ -12134,6 +12200,9 @@ Continue from where you left off. The user is waiting for your response.`;
         stopReason: result.stopReason
       });
       this.broadcastTurnComplete(result.stopReason);
+      if (result.stopReason === "end_turn") {
+        await this.relayAgentResponse(payload);
+      }
     } catch (error) {
       this.logger.error("Failed to send resume message", error);
       if (this.session) {
@@ -12232,6 +12301,26 @@ Important:
 - Do NOT create a new branch or a new pull request.
 - Do NOT add "Co-Authored-By" trailers to commit messages.
 - Do NOT add "Generated with [Claude Code]" or similar attribution lines to PR descriptions.
+`;
+    }
+    if (!this.config.repositoryPath) {
+      return `
+# Cloud Task Execution \u2014 No Repository Mode
+
+You are a helpful assistant with access to PostHog via MCP tools. You can help with both code tasks and data/analytics questions.
+
+When the user asks about analytics, data, metrics, events, funnels, dashboards, feature flags, experiments, or anything PostHog-related:
+- Use your PostHog MCP tools to query data, search insights, and provide real answers
+- Do NOT tell the user to check an external analytics platform \u2014 you ARE the analytics platform
+- Use tools like insight-query, query-run, event-definitions-list, and others to answer questions directly
+
+When the user asks for code changes or software engineering tasks:
+- Let them know you can help but don't have a repository connected for this session
+- Offer to write code snippets, scripts, or provide guidance
+
+Important:
+- Do NOT create branches, commits, or pull requests in this mode.
+- Prefer using MCP tools to answer questions with real data over giving generic advice.
 `;
     }
     return `
@@ -12334,6 +12423,9 @@ Important:
           }
         };
       },
+      extNotification: async (method, params) => {
+        this.logger.debug("Extension notification", { method, params });
+      },
       sessionUpdate: async (params) => {
         if (params.update?.sessionUpdate === "tool_call_update") {
           const meta = params.update?._meta?.claudeCode;
@@ -12366,7 +12458,7 @@ Important:
         error
       });
     }
-    const message = this.session.logWriter.getLastAgentMessage(payload.run_id);
+    const message = this.session.logWriter.getFullAgentResponse(payload.run_id);
     if (!message) {
       this.logger.warn("No agent message found for Slack relay", {
         taskId: payload.task_id,
@@ -12519,6 +12611,7 @@ Important:
     if (this.session.sseController) {
       this.session.sseController.close();
     }
+    this.pendingEvents = [];
     this.session = null;
   }
   async captureTreeState() {
@@ -12567,6 +12660,16 @@ Important:
   broadcastEvent(event) {
     if (this.session?.sseController) {
       this.sendSseEvent(this.session.sseController, event);
+    } else if (this.session) {
+      this.pendingEvents.push(event);
+    }
+  }
+  replayPendingEvents() {
+    if (!this.session?.sseController || this.pendingEvents.length === 0) return;
+    const events = this.pendingEvents;
+    this.pendingEvents = [];
+    for (const event of events) {
+      this.sendSseEvent(this.session.sseController, event);
     }
   }
   sendSseEvent(controller, data) {