@posthog/ai 6.2.0 → 6.3.0

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.2.0";
+var version = "6.3.0";
 
 // limit large outputs by truncating to 200kb (approx 200k bytes)
 const MAX_OUTPUT_SIZE = 200000;
@@ -214,7 +214,7 @@ const truncate = str => {
     }
     const truncatedBuffer = buffer.slice(0, MAX_OUTPUT_SIZE);
     return `${truncatedBuffer.toString(STRING_FORMAT)}... [truncated]`;
-  } catch (error) {
+  } catch {
     console.error('Error truncating, likely not a string');
     return str;
   }
@@ -247,6 +247,11 @@ const extractAvailableToolCalls = (provider, params) => {
   }
   return null;
 };
+var AIEvent;
+(function (AIEvent) {
+  AIEvent["Generation"] = "$ai_generation";
+  AIEvent["Embedding"] = "$ai_embedding";
+})(AIEvent || (AIEvent = {}));
 function sanitizeValues(obj) {
   if (obj === undefined || obj === null) {
     return obj;
@@ -263,6 +268,7 @@ function sanitizeValues(obj) {
 }
 const sendEventToPosthog = async ({
   client,
+  eventType = AIEvent.Generation,
   distinctId,
   traceId,
   model,
@@ -324,7 +330,9 @@ const sendEventToPosthog = async ({
     $ai_output_choices: withPrivacyMode(client, params.posthogPrivacyMode ?? false, safeOutput),
     $ai_http_status: httpStatus,
     $ai_input_tokens: usage.inputTokens ?? 0,
-    $ai_output_tokens: usage.outputTokens ?? 0,
+    ...(usage.outputTokens !== undefined ? {
+      $ai_output_tokens: usage.outputTokens
+    } : {}),
     ...additionalTokenValues,
     $ai_latency: latency,
     $ai_trace_id: traceId,
@@ -341,7 +349,7 @@ const sendEventToPosthog = async ({
   };
   const event = {
     distinctId: distinctId ?? traceId,
-    event: '$ai_generation',
+    event: eventType,
     properties,
     groups: params.posthogGroups
   };
@@ -554,6 +562,7 @@ const sanitizeLangChain = data => {
 const Chat = OpenAI.Chat;
 const Completions = Chat.Completions;
 const Responses = OpenAI.Responses;
+const Embeddings = OpenAI.Embeddings;
 class PostHogOpenAI extends OpenAI {
   constructor(config) {
     const {
@@ -564,6 +573,7 @@ class PostHogOpenAI extends OpenAI {
     this.phClient = posthog;
     this.chat = new WrappedChat$1(this, this.phClient);
     this.responses = new WrappedResponses$1(this, this.phClient);
+    this.embeddings = new WrappedEmbeddings$1(this, this.phClient);
   }
 }
 let WrappedChat$1 = class WrappedChat extends Chat {
@@ -576,16 +586,13 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
   constructor(client, phClient) {
     super(client);
     this.phClient = phClient;
+    this.baseURL = client.baseURL;
   }
   // --- Implementation Signature
   create(body, options) {
     const {
       posthogDistinctId,
       posthogTraceId,
-      posthogProperties,
-      // eslint-disable-next-line @typescript-eslint/no-unused-vars
-      posthogPrivacyMode = false,
-      posthogGroups,
       posthogCaptureImmediate,
       ...openAIParams
     } = body;
@@ -695,7 +702,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
                 input: sanitizeOpenAI(openAIParams.messages),
                 output: formattedOutput,
                 latency,
-                baseURL: this.baseURL ?? '',
+                baseURL: this.baseURL,
                 params: body,
                 httpStatus: 200,
                 usage,
@@ -713,7 +720,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
                 input: sanitizeOpenAI(openAIParams.messages),
                 output: [],
                 latency: 0,
-                baseURL: this.baseURL ?? '',
+                baseURL: this.baseURL,
                 params: body,
                 httpStatus,
                 usage: {
@@ -745,7 +752,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
             input: sanitizeOpenAI(openAIParams.messages),
             output: formatResponseOpenAI(result),
             latency,
-            baseURL: this.baseURL ?? '',
+            baseURL: this.baseURL,
             params: body,
             httpStatus: 200,
             usage: {
@@ -770,7 +777,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
           input: sanitizeOpenAI(openAIParams.messages),
           output: [],
           latency: 0,
-          baseURL: this.baseURL ?? '',
+          baseURL: this.baseURL,
           params: body,
           httpStatus,
           usage: {
@@ -791,16 +798,13 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
   constructor(client, phClient) {
     super(client);
     this.phClient = phClient;
+    this.baseURL = client.baseURL;
   }
   // --- Implementation Signature
   create(body, options) {
     const {
       posthogDistinctId,
       posthogTraceId,
-      posthogProperties,
-      // eslint-disable-next-line @typescript-eslint/no-unused-vars
-      posthogPrivacyMode = false,
-      posthogGroups,
       posthogCaptureImmediate,
       ...openAIParams
     } = body;
@@ -843,7 +847,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
                 input: sanitizeOpenAIResponse(openAIParams.input),
                 output: finalContent,
                 latency,
-                baseURL: this.baseURL ?? '',
+                baseURL: this.baseURL,
                 params: body,
                 httpStatus: 200,
                 usage,
@@ -862,7 +866,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
                 input: sanitizeOpenAIResponse(openAIParams.input),
                 output: [],
                 latency: 0,
-                baseURL: this.baseURL ?? '',
+                baseURL: this.baseURL,
                 params: body,
                 httpStatus,
                 usage: {
@@ -896,7 +900,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
               output: result.output
             }),
             latency,
-            baseURL: this.baseURL ?? '',
+            baseURL: this.baseURL,
             params: body,
             httpStatus: 200,
             usage: {
@@ -922,7 +926,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
           input: sanitizeOpenAIResponse(openAIParams.input),
           output: [],
           latency: 0,
-          baseURL: this.baseURL ?? '',
+          baseURL: this.baseURL,
           params: body,
           httpStatus,
           usage: {
@@ -942,10 +946,6 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
     const {
       posthogDistinctId,
       posthogTraceId,
-      posthogProperties,
-      // eslint-disable-next-line @typescript-eslint/no-unused-vars
-      posthogPrivacyMode = false,
-      posthogGroups,
       posthogCaptureImmediate,
       ...openAIParams
     } = body;
@@ -970,7 +970,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
           input: sanitizeOpenAIResponse(openAIParams.input),
           output: result.output,
           latency,
-          baseURL: this.baseURL ?? '',
+          baseURL: this.baseURL,
           params: body,
           httpStatus: 200,
           usage: {
@@ -994,7 +994,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
           input: sanitizeOpenAIResponse(openAIParams.input),
           output: [],
           latency: 0,
-          baseURL: this.baseURL ?? '',
+          baseURL: this.baseURL,
           params: body,
           httpStatus,
           usage: {
@@ -1014,6 +1014,73 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
     }
   }
 };
+let WrappedEmbeddings$1 = class WrappedEmbeddings extends Embeddings {
+  constructor(client, phClient) {
+    super(client);
+    this.phClient = phClient;
+    this.baseURL = client.baseURL;
+  }
+  create(body, options) {
+    const {
+      posthogDistinctId,
+      posthogTraceId,
+      posthogPrivacyMode = false,
+      posthogCaptureImmediate,
+      ...openAIParams
+    } = body;
+    const traceId = posthogTraceId ?? v4();
+    const startTime = Date.now();
+    const parentPromise = super.create(openAIParams, options);
+    const wrappedPromise = parentPromise.then(async result => {
+      const latency = (Date.now() - startTime) / 1000;
+      await sendEventToPosthog({
+        client: this.phClient,
+        eventType: AIEvent.Embedding,
+        distinctId: posthogDistinctId,
+        traceId,
+        model: openAIParams.model,
+        provider: 'openai',
+        input: withPrivacyMode(this.phClient, posthogPrivacyMode, openAIParams.input),
+        output: null,
+        // Embeddings don't have output content
+        latency,
+        baseURL: this.baseURL,
+        params: body,
+        httpStatus: 200,
+        usage: {
+          inputTokens: result.usage?.prompt_tokens ?? 0
+        },
+        captureImmediate: posthogCaptureImmediate
+      });
+      return result;
+    }, async error => {
+      const httpStatus = error && typeof error === 'object' && 'status' in error ? error.status ?? 500 : 500;
+      await sendEventToPosthog({
+        client: this.phClient,
+        eventType: AIEvent.Embedding,
+        distinctId: posthogDistinctId,
+        traceId,
+        model: openAIParams.model,
+        provider: 'openai',
+        input: withPrivacyMode(this.phClient, posthogPrivacyMode, openAIParams.input),
+        output: null,
+        // Embeddings don't have output content
+        latency: 0,
+        baseURL: this.baseURL,
+        params: body,
+        httpStatus,
+        usage: {
+          inputTokens: 0
+        },
+        isError: true,
+        error: JSON.stringify(error),
+        captureImmediate: posthogCaptureImmediate
+      });
+      throw error;
+    });
+    return wrappedPromise;
+  }
+};
 
 class PostHogAzureOpenAI extends AzureOpenAI {
   constructor(config) {
@@ -1024,6 +1091,7 @@ class PostHogAzureOpenAI extends AzureOpenAI {
     super(openAIConfig);
     this.phClient = posthog;
     this.chat = new WrappedChat(this, this.phClient);
+    this.embeddings = new WrappedEmbeddings(this, this.phClient);
   }
 }
 class WrappedChat extends AzureOpenAI.Chat {
@@ -1036,16 +1104,13 @@ class WrappedCompletions extends AzureOpenAI.Chat.Completions {
   constructor(client, phClient) {
     super(client);
     this.phClient = phClient;
+    this.baseURL = client.baseURL;
   }
   // --- Implementation Signature
   create(body, options) {
     const {
       posthogDistinctId,
       posthogTraceId,
-      posthogProperties,
-      // eslint-disable-next-line @typescript-eslint/no-unused-vars
-      posthogPrivacyMode = false,
-      posthogGroups,
       posthogCaptureImmediate,
       ...openAIParams
     } = body;
@@ -1154,7 +1219,7 @@ class WrappedCompletions extends AzureOpenAI.Chat.Completions {
                 input: openAIParams.messages,
                 output: formattedOutput,
                 latency,
-                baseURL: this.baseURL ?? '',
+                baseURL: this.baseURL,
                 params: body,
                 httpStatus: 200,
                 usage,
@@ -1171,7 +1236,7 @@ class WrappedCompletions extends AzureOpenAI.Chat.Completions {
                 input: openAIParams.messages,
                 output: [],
                 latency: 0,
-                baseURL: this.baseURL ?? '',
+                baseURL: this.baseURL,
                 params: body,
                 httpStatus,
                 usage: {
@@ -1202,7 +1267,7 @@ class WrappedCompletions extends AzureOpenAI.Chat.Completions {
             input: openAIParams.messages,
             output: formatResponseOpenAI(result),
             latency,
-            baseURL: this.baseURL ?? '',
+            baseURL: this.baseURL,
             params: body,
             httpStatus: 200,
             usage: {
@@ -1226,7 +1291,7 @@ class WrappedCompletions extends AzureOpenAI.Chat.Completions {
           input: openAIParams.messages,
           output: [],
           latency: 0,
-          baseURL: this.baseURL ?? '',
+          baseURL: this.baseURL,
           params: body,
           httpStatus,
           usage: {
@@ -1247,16 +1312,13 @@ class WrappedResponses extends AzureOpenAI.Responses {
   constructor(client, phClient) {
     super(client);
     this.phClient = phClient;
+    this.baseURL = client.baseURL;
   }
   // --- Implementation Signature
   create(body, options) {
     const {
       posthogDistinctId,
       posthogTraceId,
-      posthogProperties,
-      // eslint-disable-next-line @typescript-eslint/no-unused-vars
-      posthogPrivacyMode = false,
-      posthogGroups,
       posthogCaptureImmediate,
       ...openAIParams
     } = body;
@@ -1298,7 +1360,7 @@ class WrappedResponses extends AzureOpenAI.Responses {
                 input: openAIParams.input,
                 output: finalContent,
                 latency,
-                baseURL: this.baseURL ?? '',
+                baseURL: this.baseURL,
                 params: body,
                 httpStatus: 200,
                 usage,
@@ -1316,7 +1378,7 @@ class WrappedResponses extends AzureOpenAI.Responses {
                 input: openAIParams.input,
                 output: [],
                 latency: 0,
-                baseURL: this.baseURL ?? '',
+                baseURL: this.baseURL,
                 params: body,
                 httpStatus,
                 usage: {
@@ -1347,7 +1409,7 @@ class WrappedResponses extends AzureOpenAI.Responses {
             input: openAIParams.input,
             output: result.output,
             latency,
-            baseURL: this.baseURL ?? '',
+            baseURL: this.baseURL,
             params: body,
             httpStatus: 200,
             usage: {
@@ -1372,7 +1434,7 @@ class WrappedResponses extends AzureOpenAI.Responses {
           input: openAIParams.input,
           output: [],
           latency: 0,
-          baseURL: this.baseURL ?? '',
+          baseURL: this.baseURL,
           params: body,
           httpStatus,
           usage: {
@@ -1392,10 +1454,6 @@ class WrappedResponses extends AzureOpenAI.Responses {
     const {
       posthogDistinctId,
       posthogTraceId,
-      posthogProperties,
-      // eslint-disable-next-line @typescript-eslint/no-unused-vars
-      posthogPrivacyMode = false,
-      posthogGroups,
       posthogCaptureImmediate,
       ...openAIParams
     } = body;
@@ -1414,7 +1472,7 @@ class WrappedResponses extends AzureOpenAI.Responses {
         input: openAIParams.input,
         output: result.output,
         latency,
-        baseURL: this.baseURL ?? '',
+        baseURL: this.baseURL,
         params: body,
         httpStatus: 200,
         usage: {
@@ -1437,7 +1495,7 @@ class WrappedResponses extends AzureOpenAI.Responses {
         input: openAIParams.input,
         output: [],
         latency: 0,
-        baseURL: this.baseURL ?? '',
+        baseURL: this.baseURL,
         params: body,
         httpStatus: error?.status ? error.status : 500,
         usage: {
@@ -1453,6 +1511,72 @@ class WrappedResponses extends AzureOpenAI.Responses {
     return wrappedPromise;
   }
 }
+class WrappedEmbeddings extends AzureOpenAI.Embeddings {
+  constructor(client, phClient) {
+    super(client);
+    this.phClient = phClient;
+    this.baseURL = client.baseURL;
+  }
+  create(body, options) {
+    const {
+      posthogDistinctId,
+      posthogTraceId,
+      posthogPrivacyMode = false,
+      posthogCaptureImmediate,
+      ...openAIParams
+    } = body;
+    const traceId = posthogTraceId ?? v4();
+    const startTime = Date.now();
+    const parentPromise = super.create(openAIParams, options);
+    const wrappedPromise = parentPromise.then(async result => {
+      const latency = (Date.now() - startTime) / 1000;
+      await sendEventToPosthog({
+        client: this.phClient,
+        eventType: AIEvent.Embedding,
+        distinctId: posthogDistinctId,
+        traceId,
+        model: openAIParams.model,
+        provider: 'azure',
+        input: withPrivacyMode(this.phClient, posthogPrivacyMode, openAIParams.input),
+        output: null,
+        // Embeddings don't have output content
+        latency,
+        baseURL: this.baseURL,
+        params: body,
+        httpStatus: 200,
+        usage: {
+          inputTokens: result.usage?.prompt_tokens ?? 0
+        },
+        captureImmediate: posthogCaptureImmediate
+      });
+      return result;
+    }, async error => {
+      const httpStatus = error && typeof error === 'object' && 'status' in error ? error.status ?? 500 : 500;
+      await sendEventToPosthog({
+        client: this.phClient,
+        eventType: AIEvent.Embedding,
+        distinctId: posthogDistinctId,
+        traceId,
+        model: openAIParams.model,
+        provider: 'azure',
+        input: withPrivacyMode(this.phClient, posthogPrivacyMode, openAIParams.input),
+        output: null,
+        latency: 0,
+        baseURL: this.baseURL,
+        params: body,
+        httpStatus,
+        usage: {
+          inputTokens: 0
+        },
+        isError: true,
+        error: JSON.stringify(error),
+        captureImmediate: posthogCaptureImmediate
+      });
+      throw error;
+    });
+    return wrappedPromise;
+  }
+}
 
 const mapVercelParams = params => {
   return {
@@ -1640,7 +1764,7 @@ const mapVercelOutput = result => {
       content: truncate(jsonOutput),
       role: 'assistant'
     }];
-  } catch (error) {
+  } catch {
     console.error('Error stringifying output');
     return [];
   }
@@ -1915,15 +2039,12 @@ class WrappedMessages extends AnthropicOriginal.Messages {
   constructor(parentClient, phClient) {
     super(parentClient);
     this.phClient = phClient;
+    this.baseURL = parentClient.baseURL;
   }
   create(body, options) {
     const {
       posthogDistinctId,
       posthogTraceId,
-      posthogProperties,
-      // eslint-disable-next-line @typescript-eslint/no-unused-vars
-      posthogPrivacyMode = false,
-      posthogGroups,
       posthogCaptureImmediate,
       ...anthropicParams
     } = body;
@@ -1975,7 +2096,7 @@ class WrappedMessages extends AnthropicOriginal.Messages {
                 // Handle text delta events
                 if ('delta' in chunk) {
                   if ('text' in chunk.delta) {
-                    const delta = chunk?.delta?.text ?? '';
+                    const delta = chunk.delta.text;
                     accumulatedContent += delta;
                     if (currentTextBlock) {
                       currentTextBlock.text += delta;
@@ -2044,7 +2165,7 @@ class WrappedMessages extends AnthropicOriginal.Messages {
                 input: sanitizeAnthropic(mergeSystemPrompt(anthropicParams, 'anthropic')),
                 output: formattedOutput,
                 latency,
-                baseURL: this.baseURL ?? '',
+                baseURL: this.baseURL,
                 params: body,
                 httpStatus: 200,
                 usage,
@@ -2062,7 +2183,7 @@ class WrappedMessages extends AnthropicOriginal.Messages {
                 input: sanitizeAnthropic(mergeSystemPrompt(anthropicParams)),
                 output: [],
                 latency: 0,
-                baseURL: this.baseURL ?? '',
+                baseURL: this.baseURL,
                 params: body,
                 httpStatus: error?.status ? error.status : 500,
                 usage: {
@@ -2094,7 +2215,7 @@ class WrappedMessages extends AnthropicOriginal.Messages {
             input: sanitizeAnthropic(mergeSystemPrompt(anthropicParams)),
             output: formatResponseAnthropic(result),
             latency,
-            baseURL: this.baseURL ?? '',
+            baseURL: this.baseURL,
             params: body,
             httpStatus: 200,
             usage: {
@@ -2118,7 +2239,7 @@ class WrappedMessages extends AnthropicOriginal.Messages {
           input: sanitizeAnthropic(mergeSystemPrompt(anthropicParams)),
           output: [],
           latency: 0,
-          baseURL: this.baseURL ?? '',
+          baseURL: this.baseURL,
           params: body,
           httpStatus: error?.status ? error.status : 500,
           usage: {
@@ -2156,8 +2277,6 @@ class WrappedModels {
     const {
       posthogDistinctId,
       posthogTraceId,
-      posthogProperties,
-      posthogGroups,
       posthogCaptureImmediate,
       ...geminiParams
     } = params;
@@ -2219,8 +2338,6 @@ class WrappedModels {
     const {
       posthogDistinctId,
       posthogTraceId,
-      posthogProperties,
-      posthogGroups,
       posthogCaptureImmediate,
       ...geminiParams
     } = params;
@@ -2943,7 +3060,7 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
     this.debug = options.debug || false;
   }
   // ===== CALLBACK METHODS =====
-  handleChainStart(chain, inputs, runId, parentRunId, tags, metadata, runType, runName) {
+  handleChainStart(chain, inputs, runId, parentRunId, tags, metadata, _runType, runName) {
     this._logDebugEvent('on_chain_start', runId, parentRunId, {
       inputs,
       tags
@@ -2951,18 +3068,14 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
     this._setParentOfRun(runId, parentRunId);
     this._setTraceOrSpanMetadata(chain, inputs, runId, parentRunId, metadata, tags, runName);
   }
-  handleChainEnd(outputs, runId, parentRunId, tags,
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  kwargs) {
+  handleChainEnd(outputs, runId, parentRunId, tags, _kwargs) {
     this._logDebugEvent('on_chain_end', runId, parentRunId, {
       outputs,
       tags
     });
     this._popRunAndCaptureTraceOrSpan(runId, parentRunId, outputs);
   }
-  handleChainError(error, runId, parentRunId, tags,
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  kwargs) {
+  handleChainError(error, runId, parentRunId, tags, _kwargs) {
     this._logDebugEvent('on_chain_error', runId, parentRunId, {
       error,
       tags
@@ -2987,18 +3100,14 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
     this._setParentOfRun(runId, parentRunId);
     this._setLLMMetadata(serialized, runId, prompts, metadata, extraParams, runName);
   }
-  handleLLMEnd(output, runId, parentRunId, tags,
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  extraParams) {
+  handleLLMEnd(output, runId, parentRunId, tags, _extraParams) {
     this._logDebugEvent('on_llm_end', runId, parentRunId, {
       output,
       tags
     });
     this._popRunAndCaptureGeneration(runId, parentRunId, output);
   }
-  handleLLMError(err, runId, parentRunId, tags,
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  extraParams) {
+  handleLLMError(err, runId, parentRunId, tags, _extraParams) {
     this._logDebugEvent('on_llm_error', runId, parentRunId, {
       err,
       tags
@@ -3133,7 +3242,7 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
   _getTraceId(runId) {
     return this.traceId ? String(this.traceId) : this._findRootRun(runId);
   }
-  _getParentRunId(traceId, runId, parentRunId) {
+  _getParentRunId(traceId, _runId, parentRunId) {
     // Replace the parent-run if not found in our stored parent tree.
     if (parentRunId && !this.parentTree[parentRunId]) {
       return traceId;