@posthog/agent 1.16.1 → 1.16.3

package/src/agent.ts

@@ -31,7 +31,7 @@ export class Agent {
     private canUseTool?: CanUseTool;
     public debug: boolean;
 
-    constructor(config: AgentConfig = {}) {
+    constructor(config: AgentConfig) {
         this.workingDirectory = config.workingDirectory || process.cwd();
         this.onEvent = config.onEvent;
         this.canUseTool = config.canUseTool;
@@ -81,6 +81,7 @@ export class Agent {
             this.posthogAPI = new PostHogAPIClient({
                 apiUrl: config.posthogApiUrl,
                 apiKey: config.posthogApiKey,
+                projectId: config.posthogProjectId,
             });
         }
 
@@ -91,7 +92,7 @@ export class Agent {
             logger: this.logger.child('PromptBuilder')
         });
         this.progressReporter = new TaskProgressReporter(this.posthogAPI, this.logger);
-        this.extractor = new AISDKExtractor(this.logger.child('AISDKExtractor'));
+        this.extractor = new AISDKExtractor({apiKey: config.posthogApiKey, baseURL: config.posthogApiUrl, logger: this.logger.child('AISDKExtractor')});
     }
 
     /**
@@ -434,7 +435,7 @@ Generated by PostHog Agent`;
         }
 
         if (!this.extractor) {
-            this.extractor = new AISDKExtractor(this.logger.child('AISDKExtractor'));
+            this.extractor = new AISDKExtractor({apiKey: resolvedToken || '', baseURL: resolvedBaseUrl || '', logger: this.logger.child('AISDKExtractor')});
         }
     }
 

package/src/posthog-api.ts

@@ -68,20 +68,7 @@ export class PostHogAPIClient {
   }
 
   async getTeamId(): Promise<number> {
-    if (this._teamId !== null) {
-      return this._teamId;
-    }
-
-    // Fetch user info to get team ID (following Array's pattern)
-    const userResponse = await this.apiRequest<any>('/api/users/@me/');
-
-    if (!userResponse.team?.id) {
-      throw new Error('No team found for user');
-    }
-
-    const teamId = Number(userResponse.team.id);
-    this._teamId = teamId;
-    return teamId;
+    return this.config.projectId;
   }
 
   getBaseUrl(): string {

package/src/structured-extraction.ts

@@ -1,7 +1,7 @@
 import { generateObject } from 'ai';
-import { anthropic } from '@ai-sdk/anthropic';
 import { z } from 'zod';
 import { Logger } from './utils/logger.js';
+import { getAnthropicModel } from './utils/ai-sdk.js';
 
 export interface ExtractedQuestion {
   id: string;
@@ -41,27 +41,34 @@ export interface StructuredExtractor {
   extractQuestionsWithAnswers(researchContent: string): Promise<ExtractedQuestionWithAnswer[]>;
 }
 
+export type StructuredExtractorConfig = {
+  apiKey: string;
+  baseURL: string;
+  modelName?: string;
+  logger?: Logger;
+}
+
 export class AISDKExtractor implements StructuredExtractor {
   private logger: Logger;
   private model: any;
 
-  constructor(logger?: Logger) {
-    this.logger = logger || new Logger({ debug: false, prefix: '[AISDKExtractor]' });
-
-    // Determine which provider to use based on environment variables
-    // Priority: Anthropic (if ANTHROPIC_BASE_URL is set) > OpenAI
-    const apiKey = process.env.ANTHROPIC_AUTH_TOKEN
-      || process.env.ANTHROPIC_API_KEY
-      || process.env.OPENAI_API_KEY;
+  constructor(config: StructuredExtractorConfig) {
+    this.logger = config.logger || new Logger({ debug: false, prefix: '[AISDKExtractor]' });
 
-    if (!apiKey) {
+    if (!config.apiKey) {
       throw new Error('Missing API key for structured extraction. Ensure the LLM gateway is configured.');
     }
 
-    const baseURL = process.env.ANTHROPIC_BASE_URL || process.env.OPENAI_BASE_URL;
-    const modelName = 'claude-haiku-4-5';
-    this.model = anthropic(modelName);
-    this.logger.debug('Using Anthropic provider for structured extraction', { modelName, baseURL });
+    this.model = getAnthropicModel({
+      apiKey: config.apiKey,
+      baseURL: config.baseURL,
+      modelName: config.modelName || 'claude-haiku-4-5',
+    });
+
+    this.logger.debug('Using PostHog LLM gateway for structured extraction', {
+      modelName: config.modelName || 'claude-haiku-4-5',
+      baseURL: config.baseURL
+    });
   }
 
   async extractQuestions(researchContent: string): Promise<ExtractedQuestion[]> {

package/src/types.ts

@@ -301,12 +301,14 @@ export interface AgentConfig {
   onEvent?: (event: AgentEvent) => void;
 
   // PostHog API configuration
-  posthogApiUrl?: string;
-  posthogApiKey?: string;
+  posthogApiUrl: string;
+  posthogApiKey: string;
+  posthogProjectId: number;
 
   // PostHog MCP configuration
   posthogMcpUrl?: string;
 
+
   // MCP Server configuration
   // Additional MCP servers (PostHog MCP is always included by default)
   // You can override the PostHog MCP config by providing mcpServers.posthog
@@ -323,6 +325,7 @@ export interface AgentConfig {
 export interface PostHogAPIConfig {
   apiUrl: string;
   apiKey: string;
+  projectId: number;
 }
 
 // URL mention types

package/src/utils/ai-sdk.ts

@@ -0,0 +1,47 @@
+import { createAnthropic } from '@ai-sdk/anthropic';
+
+export interface PostHogGatewayConfig {
+  apiKey: string;
+  baseURL: string;
+  modelName?: string;
+}
+
+/**
+ * Creates an Anthropic model configured for PostHog LLM gateway.
+ * 
+ * Handles two key differences between AI SDK and PostHog gateway:
+ * 1. Appends /v1 to baseURL (gateway expects /v1/messages, SDK appends /messages)
+ * 2. Converts x-api-key header to Authorization Bearer token
+ */
+export function getAnthropicModel(config: PostHogGatewayConfig) {
+  const modelName = config.modelName || 'claude-haiku-4-5';
+  
+  // PostHog gateway expects /v1/messages, but AI SDK appends /messages
+  // So we need to append /v1 to the baseURL
+  const baseURL = config.baseURL ? `${config.baseURL}/v1` : undefined;
+
+  // Custom fetch to convert x-api-key header to Authorization Bearer
+  // PostHog gateway expects Bearer token, but Anthropic SDK sends x-api-key
+  const customFetch = async (url: RequestInfo, init?: RequestInit): Promise<Response> => {
+    const headers = new Headers(init?.headers);
+
+    if (headers.has('x-api-key')) {
+      headers.delete('x-api-key');
+      headers.set('Authorization', `Bearer ${config.apiKey}`);
+    }
+
+    return fetch(url, {
+      ...init,
+      headers,
+    });
+  };
+
+  const anthropic = createAnthropic({
+    apiKey: config.apiKey,
+    baseURL,
+    //@ts-ignore
+    fetch: customFetch,
+  });
+
+  return anthropic(modelName);
+}