@posthog/agent 2.3.316 → 2.3.326

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@posthog/agent",
-  "version": "2.3.316",
+  "version": "2.3.326",
   "repository": "https://github.com/PostHog/code",
   "description": "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   "exports": {

package/src/server/agent-server.ts

@@ -12,7 +12,7 @@ import { type ServerType, serve } from "@hono/node-server";
 import { getCurrentBranch } from "@posthog/git/queries";
 import { Hono } from "hono";
 import packageJson from "../../package.json" with { type: "json" };
-import { POSTHOG_NOTIFICATIONS } from "../acp-extensions";
+import { POSTHOG_METHODS, POSTHOG_NOTIFICATIONS } from "../acp-extensions";
 import {
   createAcpConnection,
   type InProcessAcpConnection,
@@ -644,6 +644,23 @@ export class AgentServer {
         };
       }
 
+      case POSTHOG_METHODS.REFRESH_SESSION:
+      case "posthog/refresh_session":
+      case "refresh_session": {
+        const mcpServers = Array.isArray(params.mcpServers)
+          ? params.mcpServers
+          : [];
+
+        this.logger.info("Refresh session requested", {
+          serverCount: mcpServers.length,
+        });
+
+        return await this.session.clientConnection.extMethod(
+          POSTHOG_METHODS.REFRESH_SESSION,
+          { mcpServers },
+        );
+      }
+
       case POSTHOG_NOTIFICATIONS.PERMISSION_RESPONSE:
       case "permission_response": {
         const requestId = params.requestId as string;

package/src/server/schemas.test.ts

@@ -184,4 +184,44 @@ describe("validateCommandParams", () => {
 
     expect(result.success).toBe(false);
   });
+
+  it("accepts _posthog/refresh_session with mcpServers", () => {
+    const result = validateCommandParams("_posthog/refresh_session", {
+      mcpServers: [
+        { type: "http", name: "mcp", url: "https://mcp.example.com" },
+      ],
+    });
+
+    expect(result.success).toBe(true);
+  });
+
+  it("accepts posthog/refresh_session with empty mcpServers", () => {
+    const result = validateCommandParams("posthog/refresh_session", {
+      mcpServers: [],
+    });
+
+    expect(result.success).toBe(true);
+  });
+
+  it("accepts bare refresh_session", () => {
+    const result = validateCommandParams("refresh_session", {
+      mcpServers: [],
+    });
+
+    expect(result.success).toBe(true);
+  });
+
+  it("rejects refresh_session without mcpServers", () => {
+    const result = validateCommandParams("_posthog/refresh_session", {});
+
+    expect(result.success).toBe(false);
+  });
+
+  it("rejects refresh_session with invalid mcpServers entry", () => {
+    const result = validateCommandParams("_posthog/refresh_session", {
+      mcpServers: [{ type: "stdio", name: "bad", command: "/bin/x" }],
+    });
+
+    expect(result.success).toBe(false);
+  });
 });

package/src/server/schemas.ts

@@ -60,6 +60,10 @@ export const setConfigOptionParamsSchema = z.object({
   value: z.string().min(1, "value is required"),
 });
 
+export const refreshSessionParamsSchema = z.object({
+  mcpServers: mcpServersSchema,
+});
+
 export const commandParamsSchemas = {
   user_message: userMessageParamsSchema,
   "posthog/user_message": userMessageParamsSchema,
@@ -71,6 +75,9 @@ export const commandParamsSchemas = {
   "posthog/permission_response": permissionResponseParamsSchema,
   set_config_option: setConfigOptionParamsSchema,
   "posthog/set_config_option": setConfigOptionParamsSchema,
+  refresh_session: refreshSessionParamsSchema,
+  "posthog/refresh_session": refreshSessionParamsSchema,
+  "_posthog/refresh_session": refreshSessionParamsSchema,
 } as const;
 
 export type CommandMethod = keyof typeof commandParamsSchemas;
@@ -82,7 +89,7 @@ export function validateCommandParams(
   const schema =
     commandParamsSchemas[method as CommandMethod] ??
     commandParamsSchemas[
-      method.replace("posthog/", "") as keyof typeof commandParamsSchemas
+      method.replace(/^_?posthog\//, "") as keyof typeof commandParamsSchemas
     ];
 
   if (!schema) {