@posthog/ai 6.1.2 → 6.3.0

package/dist/index.cjs

@@ -8,24 +8,26 @@ var AnthropicOriginal = require('@anthropic-ai/sdk');
 var genai = require('@google/genai');
 
 function _interopNamespaceDefault(e) {
-    var n = Object.create(null);
-    if (e) {
-        Object.keys(e).forEach(function (k) {
-            if (k !== 'default') {
-                var d = Object.getOwnPropertyDescriptor(e, k);
-                Object.defineProperty(n, k, d.get ? d : {
-                    enumerable: true,
-                    get: function () { return e[k]; }
-                });
-            }
+  var n = Object.create(null);
+  if (e) {
+    Object.keys(e).forEach(function (k) {
+      if (k !== 'default') {
+        var d = Object.getOwnPropertyDescriptor(e, k);
+        Object.defineProperty(n, k, d.get ? d : {
+          enumerable: true,
+          get: function () { return e[k]; }
         });
-    }
-    n.default = e;
-    return Object.freeze(n);
+      }
+    });
+  }
+  n.default = e;
+  return Object.freeze(n);
 }
 
 var uuid__namespace = /*#__PURE__*/_interopNamespaceDefault(uuid);
 
+var version = "6.3.0";
+
 // limit large outputs by truncating to 200kb (approx 200k bytes)
 const MAX_OUTPUT_SIZE = 200000;
 const STRING_FORMAT = 'utf8';
@@ -232,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;
   }
@@ -265,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;
@@ -281,6 +288,7 @@ function sanitizeValues(obj) {
 }
 const sendEventToPosthog = async ({
   client,
+  eventType = AIEvent.Generation,
   distinctId,
   traceId,
   model,
@@ -333,6 +341,8 @@ const sendEventToPosthog = async ({
     } : {})
   };
   const properties = {
+    $ai_lib: 'posthog-ai',
+    $ai_lib_version: version,
     $ai_provider: params.posthogProviderOverride ?? provider,
     $ai_model: params.posthogModelOverride ?? model,
     $ai_model_parameters: getModelParams(params),
@@ -340,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,
@@ -357,7 +369,7 @@ const sendEventToPosthog = async ({
   };
   const event = {
     distinctId: distinctId ?? traceId,
-    event: '$ai_generation',
+    event: eventType,
     properties,
     groups: params.posthogGroups
   };
@@ -570,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 {
@@ -580,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 {
@@ -592,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;
@@ -711,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,
@@ -729,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: {
@@ -761,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: {
@@ -786,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: {
@@ -807,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;
@@ -859,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,
@@ -878,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: {
@@ -912,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: {
@@ -938,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: {
@@ -958,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;
@@ -986,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: {
@@ -1010,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: {
@@ -1030,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) {
@@ -1040,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 {
@@ -1052,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;
@@ -1170,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,
@@ -1187,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: {
@@ -1218,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: {
@@ -1242,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: {
@@ -1263,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;
@@ -1314,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,
@@ -1332,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: {
@@ -1363,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: {
@@ -1388,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: {
@@ -1408,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;
@@ -1430,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: {
@@ -1453,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: {
@@ -1469,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 {
@@ -1656,7 +1784,7 @@ const mapVercelOutput = result => {
       content: truncate(jsonOutput),
       role: 'assistant'
     }];
-  } catch (error) {
+  } catch {
     console.error('Error stringifying output');
     return [];
   }
@@ -1931,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;
@@ -1991,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;
@@ -2060,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,
@@ -2078,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: {
@@ -2110,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: {
@@ -2134,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: {
@@ -2172,8 +2297,6 @@ class WrappedModels {
     const {
       posthogDistinctId,
       posthogTraceId,
-      posthogProperties,
-      posthogGroups,
       posthogCaptureImmediate,
       ...geminiParams
     } = params;
@@ -2235,8 +2358,6 @@ class WrappedModels {
     const {
       posthogDistinctId,
       posthogTraceId,
-      posthogProperties,
-      posthogGroups,
       posthogCaptureImmediate,
       ...geminiParams
     } = params;
@@ -2959,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
@@ -2967,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
@@ -3003,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
@@ -3149,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;
@@ -3174,6 +3287,8 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
     const eventName = parentRunId ? '$ai_span' : '$ai_trace';
     const latency = run.endTime ? (run.endTime - run.startTime) / 1000 : 0;
     const eventProperties = {
+      $ai_lib: 'posthog-ai',
+      $ai_lib_version: version,
       $ai_trace_id: traceId,
       $ai_input_state: withPrivacyMode(this.client, this.privacyMode, run.input),
       $ai_latency: latency,
@@ -3214,6 +3329,8 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
   _captureGeneration(traceId, runId, run, output, parentRunId) {
     const latency = run.endTime ? (run.endTime - run.startTime) / 1000 : 0;
     const eventProperties = {
+      $ai_lib: 'posthog-ai',
+      $ai_lib_version: version,
       $ai_trace_id: traceId,
       $ai_span_id: runId,
       $ai_span_name: run.name,