@posthog/ai 6.3.2 → 6.4.0

package/dist/index.d.ts

@@ -3,8 +3,7 @@ import { PostHog } from 'posthog-node';
 import { Stream } from 'openai/streaming';
 import { ParsedResponse } from 'openai/resources/responses/responses';
 import { LanguageModelV2 } from '@ai-sdk/provider';
-import AnthropicOriginal from '@anthropic-ai/sdk';
-import { RequestOptions as RequestOptions$2, APIPromise as APIPromise$1 } from '@anthropic-ai/sdk/core';
+import AnthropicOriginal, { APIPromise as APIPromise$1 } from '@anthropic-ai/sdk';
 import { Stream as Stream$1 } from '@anthropic-ai/sdk/streaming';
 import { GoogleGenAI, GenerateContentParameters, GenerateContentResponse } from '@google/genai';
 import { BaseCallbackHandler } from '@langchain/core/callbacks/base';
@@ -54,7 +53,7 @@ interface MonitoringOpenAIConfig$1 extends ClientOptions$1 {
     posthog: PostHog;
     baseURL?: string;
 }
-type RequestOptions$1 = Record<string, any>;
+type RequestOptions$2 = Record<string, any>;
 declare class PostHogOpenAI extends OpenAI {
     private readonly phClient;
     chat: WrappedChat$1;
@@ -70,24 +69,24 @@ declare class WrappedCompletions$1 extends Completions {
     private readonly phClient;
     private readonly baseURL;
     constructor(client: OpenAI, phClient: PostHog);
-    create(body: ChatCompletionCreateParamsNonStreaming$1 & MonitoringParams, options?: RequestOptions$1): APIPromise<ChatCompletion$1>;
-    create(body: ChatCompletionCreateParamsStreaming$1 & MonitoringParams, options?: RequestOptions$1): APIPromise<Stream<ChatCompletionChunk$1>>;
-    create(body: ChatCompletionCreateParamsBase$1 & MonitoringParams, options?: RequestOptions$1): APIPromise<ChatCompletion$1 | Stream<ChatCompletionChunk$1>>;
+    create(body: ChatCompletionCreateParamsNonStreaming$1 & MonitoringParams, options?: RequestOptions$2): APIPromise<ChatCompletion$1>;
+    create(body: ChatCompletionCreateParamsStreaming$1 & MonitoringParams, options?: RequestOptions$2): APIPromise<Stream<ChatCompletionChunk$1>>;
+    create(body: ChatCompletionCreateParamsBase$1 & MonitoringParams, options?: RequestOptions$2): APIPromise<ChatCompletion$1 | Stream<ChatCompletionChunk$1>>;
 }
 declare class WrappedResponses extends Responses {
     private readonly phClient;
     private readonly baseURL;
     constructor(client: OpenAI, phClient: PostHog);
-    create(body: ResponsesCreateParamsNonStreaming & MonitoringParams, options?: RequestOptions$1): APIPromise<OpenAI.Responses.Response>;
-    create(body: ResponsesCreateParamsStreaming & MonitoringParams, options?: RequestOptions$1): APIPromise<Stream<OpenAI.Responses.ResponseStreamEvent>>;
-    create(body: ResponsesCreateParamsBase & MonitoringParams, options?: RequestOptions$1): APIPromise<OpenAI.Responses.Response | Stream<OpenAI.Responses.ResponseStreamEvent>>;
-    parse<Params extends ResponsesCreateParamsBase, ParsedT = any>(body: Params & MonitoringParams, options?: RequestOptions$1): APIPromise<ParsedResponse<ParsedT>>;
+    create(body: ResponsesCreateParamsNonStreaming & MonitoringParams, options?: RequestOptions$2): APIPromise<OpenAI.Responses.Response>;
+    create(body: ResponsesCreateParamsStreaming & MonitoringParams, options?: RequestOptions$2): APIPromise<Stream<OpenAI.Responses.ResponseStreamEvent>>;
+    create(body: ResponsesCreateParamsBase & MonitoringParams, options?: RequestOptions$2): APIPromise<OpenAI.Responses.Response | Stream<OpenAI.Responses.ResponseStreamEvent>>;
+    parse<Params extends ResponsesCreateParamsBase, ParsedT = any>(body: Params & MonitoringParams, options?: RequestOptions$2): APIPromise<ParsedResponse<ParsedT>>;
 }
 declare class WrappedEmbeddings$1 extends Embeddings {
     private readonly phClient;
     private readonly baseURL;
     constructor(client: OpenAI, phClient: PostHog);
-    create(body: EmbeddingCreateParams$1 & MonitoringParams, options?: RequestOptions$1): APIPromise<CreateEmbeddingResponse$1>;
+    create(body: EmbeddingCreateParams$1 & MonitoringParams, options?: RequestOptions$2): APIPromise<CreateEmbeddingResponse$1>;
 }
 
 type ChatCompletion = OpenAIOrignal.ChatCompletion;
@@ -102,7 +101,7 @@ interface MonitoringOpenAIConfig {
     posthog: PostHog;
     baseURL?: string;
 }
-type RequestOptions = Record<string, any>;
+type RequestOptions$1 = Record<string, any>;
 declare class PostHogAzureOpenAI extends AzureOpenAI {
     private readonly phClient;
     chat: WrappedChat;
@@ -117,15 +116,15 @@ declare class WrappedCompletions extends AzureOpenAI.Chat.Completions {
     private readonly phClient;
     private readonly baseURL;
     constructor(client: AzureOpenAI, phClient: PostHog);
-    create(body: ChatCompletionCreateParamsNonStreaming & MonitoringParams, options?: RequestOptions): APIPromise<ChatCompletion>;
-    create(body: ChatCompletionCreateParamsStreaming & MonitoringParams, options?: RequestOptions): APIPromise<Stream<ChatCompletionChunk>>;
-    create(body: ChatCompletionCreateParamsBase & MonitoringParams, options?: RequestOptions): APIPromise<ChatCompletion | Stream<ChatCompletionChunk>>;
+    create(body: ChatCompletionCreateParamsNonStreaming & MonitoringParams, options?: RequestOptions$1): APIPromise<ChatCompletion>;
+    create(body: ChatCompletionCreateParamsStreaming & MonitoringParams, options?: RequestOptions$1): APIPromise<Stream<ChatCompletionChunk>>;
+    create(body: ChatCompletionCreateParamsBase & MonitoringParams, options?: RequestOptions$1): APIPromise<ChatCompletion | Stream<ChatCompletionChunk>>;
 }
 declare class WrappedEmbeddings extends AzureOpenAI.Embeddings {
     private readonly phClient;
     private readonly baseURL;
     constructor(client: AzureOpenAI, phClient: PostHog);
-    create(body: EmbeddingCreateParams & MonitoringParams, options?: RequestOptions): APIPromise<CreateEmbeddingResponse>;
+    create(body: EmbeddingCreateParams & MonitoringParams, options?: RequestOptions$1): APIPromise<CreateEmbeddingResponse>;
 }
 
 interface ClientOptions {
@@ -146,6 +145,7 @@ type MessageCreateParamsStreaming = AnthropicOriginal.Messages.MessageCreatePara
 type Message = AnthropicOriginal.Messages.Message;
 type RawMessageStreamEvent = AnthropicOriginal.Messages.RawMessageStreamEvent;
 type MessageCreateParamsBase = AnthropicOriginal.Messages.MessageCreateParams;
+type RequestOptions = AnthropicOriginal.RequestOptions;
 
 interface MonitoringAnthropicConfig {
     apiKey: string;
@@ -161,9 +161,9 @@ declare class WrappedMessages extends AnthropicOriginal.Messages {
     private readonly phClient;
     private readonly baseURL;
     constructor(parentClient: PostHogAnthropic, phClient: PostHog);
-    create(body: MessageCreateParamsNonStreaming, options?: RequestOptions$2): APIPromise$1<Message>;
-    create(body: MessageCreateParamsStreaming & MonitoringParams, options?: RequestOptions$2): APIPromise$1<Stream$1<RawMessageStreamEvent>>;
-    create(body: MessageCreateParamsBase & MonitoringParams, options?: RequestOptions$2): APIPromise$1<Stream$1<RawMessageStreamEvent> | Message>;
+    create(body: MessageCreateParamsNonStreaming, options?: RequestOptions): APIPromise$1<Message>;
+    create(body: MessageCreateParamsStreaming & MonitoringParams, options?: RequestOptions): APIPromise$1<Stream$1<RawMessageStreamEvent>>;
+    create(body: MessageCreateParamsBase & MonitoringParams, options?: RequestOptions): APIPromise$1<Stream$1<RawMessageStreamEvent> | Message>;
 }
 
 interface MonitoringGeminiConfig {

package/dist/index.mjs

@@ -6,7 +6,7 @@ import { wrapLanguageModel } from 'ai';
 import AnthropicOriginal from '@anthropic-ai/sdk';
 import { GoogleGenAI } from '@google/genai';
 
-var version = "6.3.2";
+var version = "6.4.0";
 
 // limit large outputs by truncating to 200kb (approx 200k bytes)
 const MAX_OUTPUT_SIZE = 200000;
@@ -291,6 +291,41 @@ function sanitizeValues(obj) {
   }
   return jsonSafe;
 }
+const POSTHOG_PARAMS_MAP = {
+  posthogDistinctId: 'distinctId',
+  posthogTraceId: 'traceId',
+  posthogProperties: 'properties',
+  posthogPrivacyMode: 'privacyMode',
+  posthogGroups: 'groups',
+  posthogModelOverride: 'modelOverride',
+  posthogProviderOverride: 'providerOverride',
+  posthogCostOverride: 'costOverride',
+  posthogCaptureImmediate: 'captureImmediate'
+};
+function extractPosthogParams(body) {
+  const providerParams = {};
+  const posthogParams = {};
+  for (const [key, value] of Object.entries(body)) {
+    if (POSTHOG_PARAMS_MAP[key]) {
+      posthogParams[POSTHOG_PARAMS_MAP[key]] = value;
+    } else if (key.startsWith('posthog')) {
+      console.warn(`Unknown Posthog parameter ${key}`);
+    } else {
+      providerParams[key] = value;
+    }
+  }
+  return {
+    providerParams: providerParams,
+    posthogParams: addDefaults(posthogParams)
+  };
+}
+function addDefaults(params) {
+  return {
+    ...params,
+    privacyMode: params.privacyMode ?? false,
+    traceId: params.traceId ?? v4()
+  };
+}
 const sendEventToPosthog = async ({
   client,
   eventType = AIEvent.Generation,
@@ -584,42 +619,6 @@ const sanitizeLangChain = data => {
   return processMessages(data, sanitizeLangChainImage);
 };
 
-const POSTHOG_PARAMS_MAP = {
-  posthogDistinctId: 'distinctId',
-  posthogTraceId: 'traceId',
-  posthogProperties: 'properties',
-  posthogPrivacyMode: 'privacyMode',
-  posthogGroups: 'groups',
-  posthogModelOverride: 'modelOverride',
-  posthogProviderOverride: 'providerOverride',
-  posthogCostOverride: 'costOverride',
-  posthogCaptureImmediate: 'captureImmediate'
-};
-function extractPosthogParams(body) {
-  const openAIParams = {};
-  const posthogParams = {};
-  for (const [key, value] of Object.entries(body)) {
-    if (POSTHOG_PARAMS_MAP[key]) {
-      posthogParams[POSTHOG_PARAMS_MAP[key]] = value;
-    } else if (key.startsWith('posthog')) {
-      console.warn(`Unknown Posthog parameter ${key}`);
-    } else {
-      openAIParams[key] = value;
-    }
-  }
-  return {
-    openAIParams: openAIParams,
-    posthogParams: addDefaults(posthogParams)
-  };
-}
-function addDefaults(params) {
-  return {
-    ...params,
-    privacyMode: params.privacyMode ?? false,
-    traceId: params.traceId ?? v4()
-  };
-}
-
 const Chat = OpenAI.Chat;
 const Completions = Chat.Completions;
 const Responses = OpenAI.Responses;
@@ -652,7 +651,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
   // --- Implementation Signature
   create(body, options) {
     const {
-      openAIParams,
+      providerParams: openAIParams,
       posthogParams
     } = extractPosthogParams(body);
     const startTime = Date.now();
@@ -853,7 +852,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
   // --- Implementation Signature
   create(body, options) {
     const {
-      openAIParams,
+      providerParams: openAIParams,
       posthogParams
     } = extractPosthogParams(body);
     const startTime = Date.now();
@@ -983,7 +982,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
   }
   parse(body, options) {
     const {
-      openAIParams,
+      providerParams: openAIParams,
       posthogParams
     } = extractPosthogParams(body);
     const startTime = Date.now();
@@ -1054,7 +1053,7 @@ let WrappedEmbeddings$1 = class WrappedEmbeddings extends Embeddings {
   }
   create(body, options) {
     const {
-      openAIParams,
+      providerParams: openAIParams,
       posthogParams
     } = extractPosthogParams(body);
     const startTime = Date.now();
@@ -1133,7 +1132,7 @@ class WrappedCompletions extends AzureOpenAI.Chat.Completions {
   // --- Implementation Signature
   create(body, options) {
     const {
-      openAIParams,
+      providerParams: openAIParams,
       posthogParams
     } = extractPosthogParams(body);
     const startTime = Date.now();
@@ -1330,7 +1329,7 @@ class WrappedResponses extends AzureOpenAI.Responses {
   // --- Implementation Signature
   create(body, options) {
     const {
-      openAIParams,
+      providerParams: openAIParams,
       posthogParams
     } = extractPosthogParams(body);
     const startTime = Date.now();
@@ -1454,7 +1453,7 @@ class WrappedResponses extends AzureOpenAI.Responses {
   }
   parse(body, options) {
     const {
-      openAIParams,
+      providerParams: openAIParams,
       posthogParams
     } = extractPosthogParams(body);
     const startTime = Date.now();
@@ -1514,7 +1513,7 @@ class WrappedEmbeddings extends AzureOpenAI.Embeddings {
   }
   create(body, options) {
     const {
-      openAIParams,
+      providerParams: openAIParams,
       posthogParams
     } = extractPosthogParams(body);
     const startTime = Date.now();
@@ -2030,12 +2029,9 @@ class WrappedMessages extends AnthropicOriginal.Messages {
   }
   create(body, options) {
     const {
-      posthogDistinctId,
-      posthogTraceId,
-      posthogCaptureImmediate,
-      ...anthropicParams
-    } = body;
-    const traceId = posthogTraceId ?? v4();
+      providerParams: anthropicParams,
+      posthogParams
+    } = extractPosthogParams(body);
     const startTime = Date.now();
     const parentPromise = super.create(anthropicParams, options);
     if (anthropicParams.stream) {
@@ -2145,8 +2141,7 @@ class WrappedMessages extends AnthropicOriginal.Messages {
               }];
               await sendEventToPosthog({
                 client: this.phClient,
-                distinctId: posthogDistinctId,
-                traceId,
+                ...posthogParams,
                 model: anthropicParams.model,
                 provider: 'anthropic',
                 input: sanitizeAnthropic(mergeSystemPrompt(anthropicParams, 'anthropic')),
@@ -2156,15 +2151,13 @@ class WrappedMessages extends AnthropicOriginal.Messages {
                 params: body,
                 httpStatus: 200,
                 usage,
-                tools: availableTools,
-                captureImmediate: posthogCaptureImmediate
+                tools: availableTools
               });
             } catch (error) {
               // error handling
               await sendEventToPosthog({
                 client: this.phClient,
-                distinctId: posthogDistinctId,
-                traceId,
+                ...posthogParams,
                 model: anthropicParams.model,
                 provider: 'anthropic',
                 input: sanitizeAnthropic(mergeSystemPrompt(anthropicParams)),
@@ -2178,8 +2171,7 @@ class WrappedMessages extends AnthropicOriginal.Messages {
                   outputTokens: 0
                 },
                 isError: true,
-                error: JSON.stringify(error),
-                captureImmediate: posthogCaptureImmediate
+                error: JSON.stringify(error)
               });
             }
           })();
@@ -2195,8 +2187,7 @@ class WrappedMessages extends AnthropicOriginal.Messages {
           const availableTools = extractAvailableToolCalls('anthropic', anthropicParams);
           await sendEventToPosthog({
             client: this.phClient,
-            distinctId: posthogDistinctId,
-            traceId,
+            ...posthogParams,
             model: anthropicParams.model,
             provider: 'anthropic',
             input: sanitizeAnthropic(mergeSystemPrompt(anthropicParams)),
@@ -2211,16 +2202,14 @@ class WrappedMessages extends AnthropicOriginal.Messages {
               cacheCreationInputTokens: result.usage.cache_creation_input_tokens ?? 0,
               cacheReadInputTokens: result.usage.cache_read_input_tokens ?? 0
             },
-            tools: availableTools,
-            captureImmediate: posthogCaptureImmediate
+            tools: availableTools
           });
         }
         return result;
       }, async error => {
         await sendEventToPosthog({
           client: this.phClient,
-          distinctId: posthogDistinctId,
-          traceId,
+          ...posthogParams,
           model: anthropicParams.model,
           provider: 'anthropic',
           input: sanitizeAnthropic(mergeSystemPrompt(anthropicParams)),
@@ -2234,8 +2223,7 @@ class WrappedMessages extends AnthropicOriginal.Messages {
             outputTokens: 0
           },
           isError: true,
-          error: JSON.stringify(error),
-          captureImmediate: posthogCaptureImmediate
+          error: JSON.stringify(error)
         });
         throw error;
       });
@@ -2262,12 +2250,9 @@ class WrappedModels {
   }
   async generateContent(params) {
     const {
-      posthogDistinctId,
-      posthogTraceId,
-      posthogCaptureImmediate,
-      ...geminiParams
-    } = params;
-    const traceId = posthogTraceId ?? v4();
+      providerParams: geminiParams,
+      posthogParams
+    } = extractPosthogParams(params);
     const startTime = Date.now();
     try {
       const response = await this.client.models.generateContent(geminiParams);
@@ -2276,8 +2261,7 @@ class WrappedModels {
       const metadata = response.usageMetadata;
       await sendEventToPosthog({
         client: this.phClient,
-        distinctId: posthogDistinctId,
-        traceId,
+        ...posthogParams,
         model: geminiParams.model,
         provider: 'gemini',
         input: this.formatInputForPostHog(geminiParams.contents),
@@ -2292,16 +2276,14 @@ class WrappedModels {
           reasoningTokens: metadata?.thoughtsTokenCount ?? 0,
           cacheReadInputTokens: metadata?.cachedContentTokenCount ?? 0
         },
-        tools: availableTools,
-        captureImmediate: posthogCaptureImmediate
+        tools: availableTools
       });
       return response;
     } catch (error) {
       const latency = (Date.now() - startTime) / 1000;
       await sendEventToPosthog({
         client: this.phClient,
-        distinctId: posthogDistinctId,
-        traceId,
+        ...posthogParams,
         model: geminiParams.model,
         provider: 'gemini',
         input: this.formatInputForPostHog(geminiParams.contents),
@@ -2315,20 +2297,16 @@ class WrappedModels {
           outputTokens: 0
         },
         isError: true,
-        error: JSON.stringify(error),
-        captureImmediate: posthogCaptureImmediate
+        error: JSON.stringify(error)
       });
       throw error;
     }
   }
   async *generateContentStream(params) {
     const {
-      posthogDistinctId,
-      posthogTraceId,
-      posthogCaptureImmediate,
-      ...geminiParams
-    } = params;
-    const traceId = posthogTraceId ?? v4();
+      providerParams: geminiParams,
+      posthogParams
+    } = extractPosthogParams(params);
     const startTime = Date.now();
     const accumulatedContent = [];
     let usage = {
@@ -2400,8 +2378,7 @@ class WrappedModels {
       }] : [];
       await sendEventToPosthog({
         client: this.phClient,
-        distinctId: posthogDistinctId,
-        traceId,
+        ...posthogParams,
         model: geminiParams.model,
         provider: 'gemini',
         input: this.formatInputForPostHog(geminiParams.contents),
@@ -2411,15 +2388,13 @@ class WrappedModels {
         params: params,
         httpStatus: 200,
         usage,
-        tools: availableTools,
-        captureImmediate: posthogCaptureImmediate
+        tools: availableTools
       });
     } catch (error) {
       const latency = (Date.now() - startTime) / 1000;
       await sendEventToPosthog({
         client: this.phClient,
-        distinctId: posthogDistinctId,
-        traceId,
+        ...posthogParams,
         model: geminiParams.model,
         provider: 'gemini',
         input: this.formatInputForPostHog(geminiParams.contents),
@@ -2433,8 +2408,7 @@ class WrappedModels {
           outputTokens: 0
         },
         isError: true,
-        error: JSON.stringify(error),
-        captureImmediate: posthogCaptureImmediate
+        error: JSON.stringify(error)
       });
       throw error;
     }