@posthog/agent 2.3.46 → 2.3.62

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.46",
+  version: "2.3.62",
   repository: "https://github.com/PostHog/code",
   description: "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   exports: {
@@ -1729,6 +1729,14 @@ function isMcpToolReadOnly(toolName) {
   const metadata = mcpToolMetadataCache.get(toolName);
   return metadata?.readOnly === true;
 }
+function getConnectedMcpServerNames() {
+  const names = /* @__PURE__ */ new Set();
+  for (const key of mcpToolMetadataCache.keys()) {
+    const parts = key.split("__");
+    if (parts.length >= 3) names.add(parts[1]);
+  }
+  return [...names];
+}
 
 // src/adapters/claude/conversion/tool-use-to-acp.ts
 var SYSTEM_REMINDER_REGEX = /\s*<system-reminder>[\s\S]*?<\/system-reminder>/g;
@@ -2485,7 +2493,10 @@ function handleToolResultChunk(chunk, ctx) {
     toolCallId: chunk.tool_use_id,
     sessionUpdate: "tool_call_update",
     status: chunk.is_error ? "failed" : "completed",
-    rawOutput: chunk.content,
+    rawOutput: ctx.mcpToolUseResult ? { ...ctx.mcpToolUseResult, isError: chunk.is_error ?? false } : {
+      content: Array.isArray(chunk.content) ? chunk.content : typeof chunk.content === "string" ? [{ type: "text", text: chunk.content }] : [],
+      isError: chunk.is_error ?? false
+    },
     ...toolUpdate
   });
   return updates;
@@ -2539,7 +2550,7 @@ function processContentChunk(chunk, role, ctx) {
       return [];
   }
 }
-function toAcpNotifications(content, role, sessionId, toolUseCache, fileContentCache, client, logger, parentToolCallId, registerHooks, supportsTerminalOutput, cwd) {
+function toAcpNotifications(content, role, sessionId, toolUseCache, fileContentCache, client, logger, parentToolCallId, registerHooks, supportsTerminalOutput, cwd, mcpToolUseResult) {
   if (typeof content === "string") {
     const update = {
       sessionUpdate: messageUpdateType(role),
@@ -2563,7 +2574,8 @@ function toAcpNotifications(content, role, sessionId, toolUseCache, fileContentC
     parentToolCallId,
     registerHooks,
     supportsTerminalOutput,
-    cwd
+    cwd,
+    mcpToolUseResult
   };
   const output = [];
   for (const chunk of content) {
@@ -2822,6 +2834,7 @@ async function handleUserAssistantMessage(message, context) {
   const content = message.message.content;
   const contentToProcess = message.type === "assistant" ? filterMessageContent(content) : content;
   const parentToolCallId = "parent_tool_use_id" in message ? message.parent_tool_use_id ?? void 0 : void 0;
+  const mcpToolUseResult = message.type === "user" && message.tool_use_result != null ? message.tool_use_result : void 0;
   for (const notification of toAcpNotifications(
     contentToProcess,
     message.message.role,
@@ -2833,7 +2846,8 @@ async function handleUserAssistantMessage(message, context) {
     parentToolCallId,
     context.registerHooks,
     context.supportsTerminalOutput,
-    session.cwd
+    session.cwd,
+    mcpToolUseResult
   )) {
     await client.sessionUpdate(notification);
     session.notificationHistory.push(notification);
@@ -4673,11 +4687,17 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
    * Both populate caches used later — neither is needed to return configOptions.
    */
   deferBackgroundFetches(q) {
+    this.logger.info("Starting background fetches (commands + MCP metadata)");
     Promise.all([
       new Promise((resolve4) => setTimeout(resolve4, 10)).then(
         () => this.sendAvailableCommandsUpdate()
       ),
-      fetchMcpToolMetadata(q, this.logger)
+      fetchMcpToolMetadata(q, this.logger).then(() => {
+        const serverNames = getConnectedMcpServerNames();
+        if (serverNames.length > 0) {
+          this.options?.onMcpServersReady?.(serverNames);
+        }
+      })
     ]).catch(
       (err) => this.logger.error("Background fetch failed", { error: err })
     );
@@ -11108,7 +11128,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(
@@ -11155,6 +11175,7 @@ var SessionLogWriter = class _SessionLogWriter {
       const nonChunkAgentText = this.extractAgentMessageText(message);
       if (nonChunkAgentText) {
         session.lastAgentMessage = nonChunkAgentText;
+        session.currentTurnMessages.push(nonChunkAgentText);
       }
       const entry = {
         type: "notification",
@@ -11251,6 +11272,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,
@@ -11276,6 +11298,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;
@@ -11591,6 +11624,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;
@@ -11670,6 +11709,7 @@ var AgentServer = class _AgentServer {
             await this.initializeSession(payload, sseController);
           } else {
             this.session.sseController = sseController;
+            this.replayPendingEvents();
           }
           this.sendSseEvent(sseController, {
             type: "connected",
@@ -11849,6 +11889,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 }],
@@ -11864,7 +11905,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": {
@@ -11887,6 +11945,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();
     }
@@ -12073,6 +12153,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 }]
@@ -12099,7 +12180,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}
@@ -12130,6 +12211,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 }]
@@ -12138,6 +12220,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) {
@@ -12236,6 +12321,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 `
@@ -12338,6 +12443,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;
@@ -12370,7 +12478,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,
@@ -12523,6 +12631,7 @@ Important:
     if (this.session.sseController) {
       this.session.sseController.close();
     }
+    this.pendingEvents = [];
     this.session = null;
   }
   async captureTreeState() {
@@ -12571,6 +12680,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) {