@posthog/agent 2.3.93 → 2.3.110

package/dist/agent.js

@@ -371,7 +371,7 @@ import { v7 as uuidv7 } from "uuid";
 // package.json
 var package_default = {
   name: "@posthog/agent",
-  version: "2.3.93",
+  version: "2.3.110",
   repository: "https://github.com/PostHog/code",
   description: "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   exports: {
@@ -4530,22 +4530,41 @@ var PostHogAPIClient = class {
     const host = this.config.apiUrl.endsWith("/") ? this.config.apiUrl.slice(0, -1) : this.config.apiUrl;
     return host;
   }
-  get headers() {
-    return {
-      Authorization: `Bearer ${this.config.getApiKey()}`,
-      "Content-Type": "application/json",
-      "User-Agent": this.config.userAgent ?? DEFAULT_USER_AGENT
-    };
+  isAuthFailure(status) {
+    return status === 401 || status === 403;
   }
-  async apiRequest(endpoint, options = {}) {
+  async resolveApiKey(forceRefresh = false) {
+    if (forceRefresh && this.config.refreshApiKey) {
+      return this.config.refreshApiKey();
+    }
+    return this.config.getApiKey();
+  }
+  async buildHeaders(options, forceRefresh = false) {
+    const headers = new Headers(options.headers);
+    headers.set(
+      "Authorization",
+      `Bearer ${await this.resolveApiKey(forceRefresh)}`
+    );
+    headers.set("Content-Type", "application/json");
+    headers.set("User-Agent", this.config.userAgent ?? DEFAULT_USER_AGENT);
+    return headers;
+  }
+  async performRequest(endpoint, options, forceRefresh = false) {
     const url = `${this.baseUrl}${endpoint}`;
-    const response = await fetch(url, {
+    return fetch(url, {
       ...options,
-      headers: {
-        ...this.headers,
-        ...options.headers
-      }
+      headers: await this.buildHeaders(options, forceRefresh)
     });
+  }
+  async performRequestWithRetry(endpoint, options = {}) {
+    let response = await this.performRequest(endpoint, options);
+    if (!response.ok && this.isAuthFailure(response.status)) {
+      response = await this.performRequest(endpoint, options, true);
+    }
+    return response;
+  }
+  async apiRequest(endpoint, options = {}) {
+    const response = await this.performRequestWithRetry(endpoint, options);
     if (!response.ok) {
       let errorMessage;
       try {
@@ -4561,8 +4580,8 @@ var PostHogAPIClient = class {
   getTeamId() {
     return this.config.projectId;
   }
-  getApiKey() {
-    return this.config.getApiKey();
+  async getApiKey(forceRefresh = false) {
+    return this.resolveApiKey(forceRefresh);
   }
   getLlmGatewayUrl() {
     return getLlmGatewayUrl(this.baseUrl);
@@ -4662,11 +4681,9 @@ var PostHogAPIClient = class {
    */
   async fetchTaskRunLogs(taskRun) {
     const teamId = this.getTeamId();
+    const endpoint = `/api/projects/${teamId}/tasks/${taskRun.task}/runs/${taskRun.id}/logs`;
     try {
-      const response = await fetch(
-        `${this.baseUrl}/api/projects/${teamId}/tasks/${taskRun.task}/runs/${taskRun.id}/logs`,
-        { headers: this.headers }
-      );
+      const response = await this.performRequestWithRetry(endpoint);
       if (!response.ok) {
         if (response.status === 404) {
           return [];
@@ -4704,6 +4721,7 @@ var SessionLogWriter = class _SessionLogWriter {
   lastFlushAttemptTime = /* @__PURE__ */ new Map();
   retryCounts = /* @__PURE__ */ new Map();
   sessions = /* @__PURE__ */ new Map();
+  flushQueues = /* @__PURE__ */ new Map();
   logger;
   localCachePath;
   constructor(options = {}) {
@@ -4797,6 +4815,18 @@ var SessionLogWriter = class _SessionLogWriter {
     }
   }
   async flush(sessionId) {
+    const prev = this.flushQueues.get(sessionId) ?? Promise.resolve();
+    const next = prev.catch(() => {
+    }).then(() => this._doFlush(sessionId));
+    this.flushQueues.set(sessionId, next);
+    next.finally(() => {
+      if (this.flushQueues.get(sessionId) === next) {
+        this.flushQueues.delete(sessionId);
+      }
+    });
+    return next;
+  }
+  async _doFlush(sessionId) {
     const session = this.sessions.get(sessionId);
     if (!session) {
       this.logger.warn("flush: no session found", { sessionId });
@@ -5023,13 +5053,13 @@ var Agent = class {
       }
     }
   }
-  _configureLlmGateway(overrideUrl) {
+  async _configureLlmGateway(overrideUrl) {
     if (!this.posthogAPI) {
       return null;
     }
     try {
       const gatewayUrl = overrideUrl ?? this.posthogAPI.getLlmGatewayUrl();
-      const apiKey = this.posthogAPI.getApiKey();
+      const apiKey = await this.posthogAPI.getApiKey();
       process.env.OPENAI_BASE_URL = `${gatewayUrl}/v1`;
       process.env.OPENAI_API_KEY = apiKey;
       process.env.ANTHROPIC_BASE_URL = gatewayUrl;
@@ -5041,7 +5071,7 @@ var Agent = class {
     }
   }
   async run(taskId, taskRunId, options = {}) {
-    const gatewayConfig = this._configureLlmGateway(options.gatewayUrl);
+    const gatewayConfig = await this._configureLlmGateway(options.gatewayUrl);
     this.logger.info("Configured LLM gateway", {
       adapter: options.adapter
     });