@posthog/ai 6.2.0 → 6.3.0

package/dist/index.cjs

@@ -26,7 +26,7 @@ function _interopNamespaceDefault(e) {
 
 var uuid__namespace = /*#__PURE__*/_interopNamespaceDefault(uuid);
 
-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;
@@ -234,7 +234,7 @@ const truncate = str => {
     }
     const truncatedBuffer = buffer$1.slice(0, MAX_OUTPUT_SIZE);
     return `${truncatedBuffer.toString(STRING_FORMAT)}... [truncated]`;
-  } catch (error) {
+  } catch {
     console.error('Error truncating, likely not a string');
     return str;
   }
@@ -267,6 +267,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;
@@ -283,6 +288,7 @@ function sanitizeValues(obj) {
 }
 const sendEventToPosthog = async ({
   client,
+  eventType = AIEvent.Generation,
   distinctId,
   traceId,
   model,
@@ -344,7 +350,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,
@@ -361,7 +369,7 @@ const sendEventToPosthog = async ({
   };
   const event = {
     distinctId: distinctId ?? traceId,
-    event: '$ai_generation',
+    event: eventType,
     properties,
     groups: params.posthogGroups
   };
@@ -574,6 +582,7 @@ const sanitizeLangChain = data => {
 const Chat = openai.OpenAI.Chat;
 const Completions = Chat.Completions;
 const Responses = openai.OpenAI.Responses;
+const Embeddings = openai.OpenAI.Embeddings;
 class PostHogOpenAI extends openai.OpenAI {
   constructor(config) {
     const {
@@ -584,6 +593,7 @@ class PostHogOpenAI extends openai.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 {
@@ -596,16 +606,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;
@@ -715,7 +722,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,
@@ -733,7 +740,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: {
@@ -765,7 +772,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: {
@@ -790,7 +797,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: {
@@ -811,16 +818,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;
@@ -863,7 +867,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,
@@ -882,7 +886,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: {
@@ -916,7 +920,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
               output: result.output
             }),
             latency,
-            baseURL: this.baseURL ?? '',
+            baseURL: this.baseURL,
             params: body,
             httpStatus: 200,
             usage: {
@@ -942,7 +946,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: {
@@ -962,10 +966,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;
@@ -990,7 +990,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: {
@@ -1014,7 +1014,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: {
@@ -1034,6 +1034,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 ?? uuid.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 openai.AzureOpenAI {
   constructor(config) {
@@ -1044,6 +1111,7 @@ class PostHogAzureOpenAI extends openai.AzureOpenAI {
     super(openAIConfig);
     this.phClient = posthog;
     this.chat = new WrappedChat(this, this.phClient);
+    this.embeddings = new WrappedEmbeddings(this, this.phClient);
   }
 }
 class WrappedChat extends openai.AzureOpenAI.Chat {
@@ -1056,16 +1124,13 @@ class WrappedCompletions extends openai.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;
@@ -1174,7 +1239,7 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
                 input: openAIParams.messages,
                 output: formattedOutput,
                 latency,
-                baseURL: this.baseURL ?? '',
+                baseURL: this.baseURL,
                 params: body,
                 httpStatus: 200,
                 usage,
@@ -1191,7 +1256,7 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
                 input: openAIParams.messages,
                 output: [],
                 latency: 0,
-                baseURL: this.baseURL ?? '',
+                baseURL: this.baseURL,
                 params: body,
                 httpStatus,
                 usage: {
@@ -1222,7 +1287,7 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
             input: openAIParams.messages,
             output: formatResponseOpenAI(result),
             latency,
-            baseURL: this.baseURL ?? '',
+            baseURL: this.baseURL,
             params: body,
             httpStatus: 200,
             usage: {
@@ -1246,7 +1311,7 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
           input: openAIParams.messages,
           output: [],
           latency: 0,
-          baseURL: this.baseURL ?? '',
+          baseURL: this.baseURL,
           params: body,
           httpStatus,
           usage: {
@@ -1267,16 +1332,13 @@ class WrappedResponses extends openai.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;
@@ -1318,7 +1380,7 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
                 input: openAIParams.input,
                 output: finalContent,
                 latency,
-                baseURL: this.baseURL ?? '',
+                baseURL: this.baseURL,
                 params: body,
                 httpStatus: 200,
                 usage,
@@ -1336,7 +1398,7 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
                 input: openAIParams.input,
                 output: [],
                 latency: 0,
-                baseURL: this.baseURL ?? '',
+                baseURL: this.baseURL,
                 params: body,
                 httpStatus,
                 usage: {
@@ -1367,7 +1429,7 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
             input: openAIParams.input,
             output: result.output,
             latency,
-            baseURL: this.baseURL ?? '',
+            baseURL: this.baseURL,
             params: body,
             httpStatus: 200,
             usage: {
@@ -1392,7 +1454,7 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
           input: openAIParams.input,
           output: [],
           latency: 0,
-          baseURL: this.baseURL ?? '',
+          baseURL: this.baseURL,
           params: body,
           httpStatus,
           usage: {
@@ -1412,10 +1474,6 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
     const {
       posthogDistinctId,
       posthogTraceId,
-      posthogProperties,
-      // eslint-disable-next-line @typescript-eslint/no-unused-vars
-      posthogPrivacyMode = false,
-      posthogGroups,
       posthogCaptureImmediate,
       ...openAIParams
     } = body;
@@ -1434,7 +1492,7 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
         input: openAIParams.input,
         output: result.output,
         latency,
-        baseURL: this.baseURL ?? '',
+        baseURL: this.baseURL,
         params: body,
         httpStatus: 200,
         usage: {
@@ -1457,7 +1515,7 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
         input: openAIParams.input,
         output: [],
         latency: 0,
-        baseURL: this.baseURL ?? '',
+        baseURL: this.baseURL,
         params: body,
         httpStatus: error?.status ? error.status : 500,
         usage: {
@@ -1473,6 +1531,72 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
     return wrappedPromise;
   }
 }
+class WrappedEmbeddings extends openai.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 ?? uuid.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 {
@@ -1660,7 +1784,7 @@ const mapVercelOutput = result => {
       content: truncate(jsonOutput),
       role: 'assistant'
     }];
-  } catch (error) {
+  } catch {
     console.error('Error stringifying output');
     return [];
   }
@@ -1935,15 +2059,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;
@@ -1995,7 +2116,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;
@@ -2064,7 +2185,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,
@@ -2082,7 +2203,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: {
@@ -2114,7 +2235,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: {
@@ -2138,7 +2259,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: {
@@ -2176,8 +2297,6 @@ class WrappedModels {
     const {
       posthogDistinctId,
       posthogTraceId,
-      posthogProperties,
-      posthogGroups,
       posthogCaptureImmediate,
       ...geminiParams
     } = params;
@@ -2239,8 +2358,6 @@ class WrappedModels {
     const {
       posthogDistinctId,
       posthogTraceId,
-      posthogProperties,
-      posthogGroups,
       posthogCaptureImmediate,
       ...geminiParams
     } = params;
@@ -2963,7 +3080,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
@@ -2971,18 +3088,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
@@ -3007,18 +3120,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
@@ -3153,7 +3262,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;