@posthog/agent 2.3.126 → 2.3.137

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.126",
+  version: "2.3.137",
   repository: "https://github.com/PostHog/code",
   description: "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   exports: {
@@ -4631,42 +4631,70 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
       isResume ? "Session query initialized, awaiting resumption" : "Session query initialized, awaiting initialization",
       { sessionId, taskId, taskRunId: meta?.taskRunId }
     );
-    try {
-      const result = await withTimeout(
-        q.initializationResult(),
-        SESSION_VALIDATION_TIMEOUT_MS
-      );
-      if (result.result === "timeout") {
-        throw new Error(
-          `Session ${isResume ? forkSession ? "fork" : "resumption" : "initialization"} timed out for sessionId=${sessionId}`
+    if (isResume) {
+      try {
+        const result = await withTimeout(
+          q.initializationResult(),
+          SESSION_VALIDATION_TIMEOUT_MS
         );
+        if (result.result === "timeout") {
+          throw new Error(
+            `Session ${forkSession ? "fork" : "resumption"} timed out for sessionId=${sessionId}`
+          );
+        }
+      } catch (err) {
+        settingsManager.dispose();
+        if (err instanceof Error && err.message === "Query closed before response received") {
+          throw import_sdk2.RequestError.resourceNotFound(sessionId);
+        }
+        this.logger.error(
+          forkSession ? "Session fork failed" : "Session resumption failed",
+          {
+            sessionId,
+            taskId,
+            taskRunId: meta?.taskRunId,
+            error: err instanceof Error ? err.message : String(err)
+          }
+        );
+        throw err;
       }
-    } catch (err) {
-      settingsManager.dispose();
-      if (isResume && err instanceof Error && err.message === "Query closed before response received") {
-        throw import_sdk2.RequestError.resourceNotFound(sessionId);
-      }
-      this.logger.error(
-        isResume ? forkSession ? "Session fork failed" : "Session resumption failed" : "Session initialization failed",
-        {
+    }
+    const initPromise = !isResume ? withTimeout(q.initializationResult(), SESSION_VALIDATION_TIMEOUT_MS) : void 0;
+    const [modelOptions] = await Promise.all([
+      this.getModelConfigOptions(
+        settingsManager.getSettings().model || meta?.model || void 0
+      ),
+      ...meta?.taskRunId ? [
+        this.client.extNotification("_posthog/sdk_session", {
+          taskRunId: meta.taskRunId,
+          sessionId,
+          adapter: "claude"
+        })
+      ] : []
+    ]);
+    if (initPromise) {
+      try {
+        const initResult = await initPromise;
+        if (initResult.result === "timeout") {
+          settingsManager.dispose();
+          throw new Error(
+            `Session initialization timed out for sessionId=${sessionId}`
+          );
+        }
+      } catch (err) {
+        settingsManager.dispose();
+        this.logger.error("Session initialization failed", {
           sessionId,
           taskId,
           taskRunId: meta?.taskRunId,
           error: err instanceof Error ? err.message : String(err)
-        }
-      );
-      throw err;
-    }
-    if (meta?.taskRunId) {
-      await this.client.extNotification("_posthog/sdk_session", {
-        taskRunId: meta.taskRunId,
-        sessionId,
-        adapter: "claude"
-      });
+        });
+        throw err;
+      }
     }
     const settingsModel = settingsManager.getSettings().model;
-    const modelOptions = await this.getModelConfigOptions();
-    const resolvedModelId = settingsModel || modelOptions.currentModelId;
+    const metaModel = meta?.model;
+    const resolvedModelId = settingsModel || metaModel || modelOptions.currentModelId;
     session.modelId = resolvedModelId;
     session.lastContextWindowSize = this.getContextWindowForModel(resolvedModelId);
     const resolvedSdkModel = toSdkModelId(resolvedModelId);
@@ -11908,6 +11936,11 @@ var AgentServer = class _AgentServer {
   // causing a second session to be created and duplicate Slack messages to be sent.
   initializationPromise = null;
   pendingEvents = [];
+  detachSseController(controller) {
+    if (this.session?.sseController === controller) {
+      this.session.sseController = null;
+    }
+  }
   emitConsoleLog = (level, _scope, message, data) => {
     if (!this.session) return;
     const formatted = data !== void 0 ? `${message} ${JSON.stringify(data)}` : message;
@@ -11968,18 +12001,15 @@ var AgentServer = class _AgentServer {
 
 `)
                 );
-              } catch (error) {
-                this.logger.debug(
-                  "SSE send failed (stream may be closed)",
-                  error
-                );
+              } catch {
+                this.detachSseController(sseController);
               }
             },
             close: () => {
               try {
                 controller.close();
-              } catch (error) {
-                this.logger.debug("SSE close failed (already closed)", error);
+              } catch {
+                this.detachSseController(sseController);
               }
             }
           };
@@ -12984,7 +13014,11 @@ Important:
     }
   }
   sendSseEvent(controller, data) {
-    controller.send(data);
+    try {
+      controller.send(data);
+    } catch {
+      this.detachSseController(controller);
+    }
   }
 };