@posthog/ai 7.16.15 → 7.17.1

package/dist/index.cjs

@@ -9,28 +9,26 @@ var genai = require('@google/genai');
 function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
 
 function _interopNamespace(e) {
-  if (e && e.__esModule) return 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]; }
+    if (e && e.__esModule) return 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]; }
+                });
+            }
         });
-      }
-    });
-  }
-  n.default = e;
-  return Object.freeze(n);
+    }
+    n.default = e;
+    return Object.freeze(n);
 }
 
 var uuid__namespace = /*#__PURE__*/_interopNamespace(uuid);
 var AnthropicOriginal__default = /*#__PURE__*/_interopDefault(AnthropicOriginal);
 
-var version = "7.16.15";
-
 // Type guards for safer type checking
 const isString = value => {
   return typeof value === 'string';
@@ -268,6 +266,13 @@ function getTokensSource(posthogProperties) {
 // limit large outputs by truncating to 200kb (approx 200k bytes)
 const MAX_OUTPUT_SIZE = 200000;
 const STRING_FORMAT = 'utf8';
+// Reused across calls to avoid per-invocation allocation; truncate() runs
+// hundreds of times for prompts with many parts.
+const sharedTextEncoder = new TextEncoder();
+const sharedTextDecoder = new TextDecoder(STRING_FORMAT, {
+  fatal: false
+});
+const utf8ByteLength = str => sharedTextEncoder.encode(str).byteLength;
 /**
  * Safely converts content to a string, preserving structure for objects/arrays.
  * - If content is already a string, returns it as-is
@@ -538,19 +543,15 @@ const truncate = input => {
     return '';
   }
   // Check if we need to truncate and ensure STRING_FORMAT is respected
-  const encoder = new TextEncoder();
-  const buffer = encoder.encode(str);
+  const buffer = sharedTextEncoder.encode(str);
   if (buffer.length <= MAX_OUTPUT_SIZE) {
     // Ensure STRING_FORMAT is respected
-    return new TextDecoder(STRING_FORMAT).decode(buffer);
+    return sharedTextDecoder.decode(buffer);
   }
-  // Truncate the buffer and ensure a valid string is returned
+  // Truncate the buffer and ensure a valid string is returned.
+  // fatal: false means we get U+FFFD at the end if truncation broke the encoding.
   const truncatedBuffer = buffer.slice(0, MAX_OUTPUT_SIZE);
-  // fatal: false means we get U+FFFD at the end if truncation broke the encoding
-  const decoder = new TextDecoder(STRING_FORMAT, {
-    fatal: false
-  });
-  let truncatedStr = decoder.decode(truncatedBuffer);
+  let truncatedStr = sharedTextDecoder.decode(truncatedBuffer);
   if (truncatedStr.endsWith('\uFFFD')) {
     truncatedStr = truncatedStr.slice(0, -1);
   }
@@ -679,11 +680,11 @@ const extractAvailableToolCalls = (provider, params) => {
   }
   return null;
 };
-var AIEvent;
+exports.AIEvent = void 0;
 (function (AIEvent) {
   AIEvent["Generation"] = "$ai_generation";
   AIEvent["Embedding"] = "$ai_embedding";
-})(AIEvent || (AIEvent = {}));
+})(exports.AIEvent || (exports.AIEvent = {}));
 function sanitizeValues(obj) {
   if (obj === undefined || obj === null) {
     return obj;
@@ -734,73 +735,107 @@ function addDefaults(params) {
     traceId: params.traceId ?? uuid.v4()
   };
 }
-const sendEventWithErrorToPosthog = async ({
-  client,
-  traceId,
-  error,
-  ...args
-}) => {
-  const httpStatus = error && typeof error === 'object' && 'status' in error ? error.status ?? 500 : 500;
-  const properties = {
-    client,
-    traceId,
-    httpStatus,
-    error: JSON.stringify(error),
-    ...args
-  };
-  const enrichedError = error;
-  if (client.options?.enableExceptionAutocapture) {
-    // assign a uuid that can be used to link the trace and exception events
-    const exceptionId = core.uuidv7();
-    client.captureException(error, undefined, {
-      $ai_trace_id: traceId
-    }, exceptionId);
-    enrichedError.__posthog_previously_captured_error = true;
-    properties.exceptionId = exceptionId;
-  }
-  await sendEventToPosthog(properties);
-  return enrichedError;
-};
-const sendEventToPosthog = async ({
-  client,
-  eventType = AIEvent.Generation,
-  distinctId,
-  traceId,
-  model,
-  provider,
-  input,
-  output,
-  latency,
-  timeToFirstToken,
-  baseURL,
-  params,
-  httpStatus = 200,
-  usage = {},
-  error,
-  exceptionId,
-  stopReason,
-  tools,
-  captureImmediate = false
-}) => {
-  if (!client.capture) {
-    return Promise.resolve();
+function formatOpenAIResponsesInput(input, instructions) {
+  const messages = [];
+  if (instructions) {
+    messages.push({
+      role: 'system',
+      content: instructions
+    });
+  }
+  if (Array.isArray(input)) {
+    for (const item of input) {
+      if (typeof item === 'string') {
+        messages.push({
+          role: 'user',
+          content: item
+        });
+      } else if (item && typeof item === 'object') {
+        const obj = item;
+        const role = isString(obj.role) ? obj.role : 'user';
+        // Handle content properly - preserve structure for objects/arrays
+        const content = obj.content ?? obj.text ?? item;
+        messages.push({
+          role,
+          content: toContentString(content)
+        });
+      } else {
+        messages.push({
+          role: 'user',
+          content: toContentString(item)
+        });
+      }
+    }
+  } else if (typeof input === 'string') {
+    messages.push({
+      role: 'user',
+      content: input
+    });
+  } else if (input) {
+    messages.push({
+      role: 'user',
+      content: toContentString(input)
+    });
   }
-  // sanitize input and output for UTF-8 validity
-  const safeInput = sanitizeValues(input);
-  const safeOutput = sanitizeValues(output);
-  const safeError = sanitizeValues(error);
+  return messages;
+}
+
+var version = "7.17.1";
+
+/**
+ * Capture an `$ai_generation` (or `$ai_embedding`) event to PostHog.
+ *
+ * This is the canonical primitive that every `@posthog/ai` wrapper
+ * (`withTracing`, `OpenAI`, `Anthropic`, `GoogleGenAI`, …) funnels through, so
+ * external code can use it directly to instrument LLM calls made through
+ * arbitrary clients (Cloudflare Workers AI, custom HTTP, etc.) and get the
+ * same events the SDK wrappers produce.
+ *
+ * When `error` is set, the event is captured as an error. If the error is an
+ * object, it is mutated in place to set `__posthog_previously_captured_error`
+ * so callers can re-throw the original error reference safely.
+ */
+const captureAiGeneration = async (client, options) => {
+  if (!client.capture) {
+    return;
+  }
+  const traceId = options.traceId ?? uuid.v4();
+  const eventType = options.eventType ?? exports.AIEvent.Generation;
+  const privacyMode = options.privacyMode ?? false;
+  const usage = options.usage ?? {};
+  const safeInput = sanitizeValues(options.input);
+  const safeOutput = sanitizeValues(options.output);
+  let httpStatus = options.httpStatus;
   let errorData = {};
-  if (error) {
+  if (options.error) {
+    if (httpStatus === undefined) {
+      if (typeof options.error === 'object' && 'status' in options.error && typeof options.error.status === 'number') {
+        httpStatus = options.error.status;
+      } else {
+        httpStatus = 500;
+      }
+    }
+    let exceptionId;
+    if (client.options?.enableExceptionAutocapture) {
+      exceptionId = core.uuidv7();
+      client.captureException(options.error, undefined, {
+        $ai_trace_id: traceId
+      }, exceptionId);
+      if (typeof options.error === 'object') {
+        options.error.__posthog_previously_captured_error = true;
+      }
+    }
     errorData = {
       $ai_is_error: true,
-      $ai_error: safeError,
+      $ai_error: sanitizeValues(JSON.stringify(options.error)),
       $exception_event_id: exceptionId
     };
   }
+  httpStatus = httpStatus ?? 200;
   let costOverrideData = {};
-  if (params.posthogCostOverride) {
-    const inputCostUSD = (params.posthogCostOverride.inputCost ?? 0) * (usage.inputTokens ?? 0);
-    const outputCostUSD = (params.posthogCostOverride.outputCost ?? 0) * (usage.outputTokens ?? 0);
+  if (options.costOverride) {
+    const inputCostUSD = (options.costOverride.inputCost ?? 0) * (usage.inputTokens ?? 0);
+    const outputCostUSD = (options.costOverride.outputCost ?? 0) * (usage.outputTokens ?? 0);
     costOverrideData = {
       $ai_input_cost_usd: inputCostUSD,
       $ai_output_cost_usd: outputCostUSD,
@@ -827,95 +862,49 @@ 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),
-    $ai_input: withPrivacyMode(client, params.posthogPrivacyMode ?? false, safeInput),
-    $ai_output_choices: withPrivacyMode(client, params.posthogPrivacyMode ?? false, safeOutput),
+    $ai_provider: options.providerOverride ?? options.provider,
+    $ai_model: options.modelOverride ?? options.model,
+    $ai_model_parameters: options.modelParameters ?? {},
+    $ai_input: withPrivacyMode(client, privacyMode, safeInput),
+    $ai_output_choices: withPrivacyMode(client, privacyMode, safeOutput),
     $ai_http_status: httpStatus,
     $ai_input_tokens: usage.inputTokens ?? 0,
     ...(usage.outputTokens !== undefined ? {
       $ai_output_tokens: usage.outputTokens
     } : {}),
     ...additionalTokenValues,
-    $ai_latency: latency,
-    ...(timeToFirstToken !== undefined ? {
-      $ai_time_to_first_token: timeToFirstToken
+    $ai_latency: options.latency ?? 0,
+    ...(options.timeToFirstToken !== undefined ? {
+      $ai_time_to_first_token: options.timeToFirstToken
     } : {}),
     $ai_trace_id: traceId,
-    $ai_base_url: baseURL,
-    ...params.posthogProperties,
-    $ai_tokens_source: getTokensSource(params.posthogProperties),
-    ...(distinctId ? {} : {
+    $ai_base_url: options.baseURL ?? '',
+    ...options.properties,
+    $ai_tokens_source: getTokensSource(options.properties),
+    ...(options.distinctId ? {} : {
       $process_person_profile: false
     }),
-    ...(stopReason ? {
-      $ai_stop_reason: stopReason
+    ...(options.stopReason ? {
+      $ai_stop_reason: options.stopReason
     } : {}),
-    ...(tools ? {
-      $ai_tools: tools
+    ...(options.tools ? {
+      $ai_tools: options.tools
     } : {}),
     ...errorData,
     ...costOverrideData
   };
   const event = {
-    distinctId: distinctId ?? traceId,
+    distinctId: options.distinctId ?? traceId,
     event: eventType,
     properties,
-    groups: params.posthogGroups
+    groups: options.groups
   };
-  if (captureImmediate) {
-    // await capture promise to send single event in serverless environments
+  if (options.captureImmediate) {
     await client.captureImmediate(event);
   } else {
     client.capture(event);
   }
-  return Promise.resolve();
 };
-function formatOpenAIResponsesInput(input, instructions) {
-  const messages = [];
-  if (instructions) {
-    messages.push({
-      role: 'system',
-      content: instructions
-    });
-  }
-  if (Array.isArray(input)) {
-    for (const item of input) {
-      if (typeof item === 'string') {
-        messages.push({
-          role: 'user',
-          content: item
-        });
-      } else if (item && typeof item === 'object') {
-        const obj = item;
-        const role = isString(obj.role) ? obj.role : 'user';
-        // Handle content properly - preserve structure for objects/arrays
-        const content = obj.content ?? obj.text ?? item;
-        messages.push({
-          role,
-          content: toContentString(content)
-        });
-      } else {
-        messages.push({
-          role: 'user',
-          content: toContentString(item)
-        });
-      }
-    }
-  } else if (typeof input === 'string') {
-    messages.push({
-      role: 'user',
-      content: input
-    });
-  } else if (input) {
-    messages.push({
-      role: 'user',
-      content: toContentString(input)
-    });
-  }
-  return messages;
-}
 
 /**
  * Checks if a ResponseStreamEvent chunk represents the first token/content from the model.
@@ -1084,8 +1073,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
               const latency = (Date.now() - startTime) / 1000;
               const timeToFirstToken = firstTokenTime !== undefined ? (firstTokenTime - startTime) / 1000 : undefined;
               const availableTools = extractAvailableToolCalls('openai', openAIParams);
-              await sendEventToPosthog({
-                client: this.phClient,
+              await captureAiGeneration(this.phClient, {
                 ...posthogParams,
                 model: openAIParams.model ?? modelFromResponse,
                 provider: 'openai',
@@ -1094,7 +1082,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
                 latency,
                 timeToFirstToken,
                 baseURL: this.baseURL,
-                params: body,
+                modelParameters: getModelParams(body),
                 httpStatus: 200,
                 usage: {
                   inputTokens: usage.inputTokens,
@@ -1108,8 +1096,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
                 tools: availableTools
               });
             } catch (error) {
-              const enrichedError = await sendEventWithErrorToPosthog({
-                client: this.phClient,
+              await captureAiGeneration(this.phClient, {
                 ...posthogParams,
                 model: openAIParams.model,
                 provider: 'openai',
@@ -1117,14 +1104,14 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
                 output: [],
                 latency: 0,
                 baseURL: this.baseURL,
-                params: body,
+                modelParameters: getModelParams(body),
                 usage: {
                   inputTokens: 0,
                   outputTokens: 0
                 },
                 error
               });
-              throw enrichedError;
+              throw error;
             }
           })();
           // Return the other stream to the user
@@ -1138,8 +1125,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
           const latency = (Date.now() - startTime) / 1000;
           const availableTools = extractAvailableToolCalls('openai', openAIParams);
           const formattedOutput = formatResponseOpenAI(result);
-          await sendEventToPosthog({
-            client: this.phClient,
+          await captureAiGeneration(this.phClient, {
             ...posthogParams,
             model: openAIParams.model ?? result.model,
             provider: 'openai',
@@ -1147,7 +1133,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
             output: formattedOutput,
             latency,
             baseURL: this.baseURL,
-            params: body,
+            modelParameters: getModelParams(body),
             httpStatus: 200,
             usage: {
               inputTokens: result.usage?.prompt_tokens ?? 0,
@@ -1164,8 +1150,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
         return result;
       }, async error => {
         const httpStatus = error && typeof error === 'object' && 'status' in error ? error.status ?? 500 : 500;
-        await sendEventToPosthog({
-          client: this.phClient,
+        await captureAiGeneration(this.phClient, {
           ...posthogParams,
           model: openAIParams.model,
           provider: 'openai',
@@ -1173,13 +1158,13 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
           output: [],
           latency: 0,
           baseURL: this.baseURL,
-          params: body,
+          modelParameters: getModelParams(body),
           httpStatus,
           usage: {
             inputTokens: 0,
             outputTokens: 0
           },
-          error: JSON.stringify(error)
+          error
         });
         throw error;
       });
@@ -1252,8 +1237,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
               const latency = (Date.now() - startTime) / 1000;
               const timeToFirstToken = firstTokenTime !== undefined ? (firstTokenTime - startTime) / 1000 : undefined;
               const availableTools = extractAvailableToolCalls('openai', openAIParams);
-              await sendEventToPosthog({
-                client: this.phClient,
+              await captureAiGeneration(this.phClient, {
                 ...posthogParams,
                 model: openAIParams.model ?? modelFromResponse,
                 provider: 'openai',
@@ -1262,7 +1246,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
                 latency,
                 timeToFirstToken,
                 baseURL: this.baseURL,
-                params: body,
+                modelParameters: getModelParams(body),
                 httpStatus: 200,
                 usage: {
                   inputTokens: usage.inputTokens,
@@ -1276,8 +1260,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
                 tools: availableTools
               });
             } catch (error) {
-              const enrichedError = await sendEventWithErrorToPosthog({
-                client: this.phClient,
+              await captureAiGeneration(this.phClient, {
                 ...posthogParams,
                 model: openAIParams.model,
                 provider: 'openai',
@@ -1285,14 +1268,14 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
                 output: [],
                 latency: 0,
                 baseURL: this.baseURL,
-                params: body,
+                modelParameters: getModelParams(body),
                 usage: {
                   inputTokens: 0,
                   outputTokens: 0
                 },
-                error: error
+                error
               });
-              throw enrichedError;
+              throw error;
             }
           })();
           return stream2;
@@ -1307,8 +1290,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
           const formattedOutput = formatResponseOpenAI({
             output: result.output
           });
-          await sendEventToPosthog({
-            client: this.phClient,
+          await captureAiGeneration(this.phClient, {
             ...posthogParams,
             model: openAIParams.model ?? result.model,
             provider: 'openai',
@@ -1316,7 +1298,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
             output: formattedOutput,
             latency,
             baseURL: this.baseURL,
-            params: body,
+            modelParameters: getModelParams(body),
             httpStatus: 200,
             usage: {
               inputTokens: result.usage?.input_tokens ?? 0,
@@ -1333,8 +1315,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
         return result;
       }, async error => {
         const httpStatus = error && typeof error === 'object' && 'status' in error ? error.status ?? 500 : 500;
-        await sendEventToPosthog({
-          client: this.phClient,
+        await captureAiGeneration(this.phClient, {
           ...posthogParams,
           model: openAIParams.model,
           provider: 'openai',
@@ -1342,13 +1323,13 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
           output: [],
           latency: 0,
           baseURL: this.baseURL,
-          params: body,
+          modelParameters: getModelParams(body),
           httpStatus,
           usage: {
             inputTokens: 0,
             outputTokens: 0
           },
-          error: JSON.stringify(error)
+          error
         });
         throw error;
       });
@@ -1369,8 +1350,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
       const parentPromise = super.parse(openAIParams, options);
       const wrappedPromise = parentPromise.then(async result => {
         const latency = (Date.now() - startTime) / 1000;
-        await sendEventToPosthog({
-          client: this.phClient,
+        await captureAiGeneration(this.phClient, {
           ...posthogParams,
           model: openAIParams.model ?? result.model,
           provider: 'openai',
@@ -1378,7 +1358,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
           output: result.output,
           latency,
           baseURL: this.baseURL,
-          params: body,
+          modelParameters: getModelParams(body),
           httpStatus: 200,
           usage: {
             inputTokens: result.usage?.input_tokens ?? 0,
@@ -1391,8 +1371,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
         });
         return result;
       }, async error => {
-        const enrichedError = await sendEventWithErrorToPosthog({
-          client: this.phClient,
+        await captureAiGeneration(this.phClient, {
           ...posthogParams,
           model: openAIParams.model,
           provider: 'openai',
@@ -1400,14 +1379,14 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
           output: [],
           latency: 0,
           baseURL: this.baseURL,
-          params: body,
+          modelParameters: getModelParams(body),
           usage: {
             inputTokens: 0,
             outputTokens: 0
           },
-          error: JSON.stringify(error)
+          error
         });
-        throw enrichedError;
+        throw error;
       });
       return wrappedPromise;
     } finally {
@@ -1431,10 +1410,9 @@ let WrappedEmbeddings$1 = class WrappedEmbeddings extends Embeddings {
     const parentPromise = super.create(openAIParams, options);
     const wrappedPromise = parentPromise.then(async result => {
       const latency = (Date.now() - startTime) / 1000;
-      await sendEventToPosthog({
-        client: this.phClient,
+      await captureAiGeneration(this.phClient, {
         ...posthogParams,
-        eventType: AIEvent.Embedding,
+        eventType: exports.AIEvent.Embedding,
         model: openAIParams.model,
         provider: 'openai',
         input: withPrivacyMode(this.phClient, posthogParams.privacyMode, openAIParams.input),
@@ -1442,7 +1420,7 @@ let WrappedEmbeddings$1 = class WrappedEmbeddings extends Embeddings {
         // Embeddings don't have output content
         latency,
         baseURL: this.baseURL,
-        params: body,
+        modelParameters: getModelParams(body),
         httpStatus: 200,
         usage: {
           inputTokens: result.usage?.prompt_tokens ?? 0,
@@ -1452,9 +1430,8 @@ let WrappedEmbeddings$1 = class WrappedEmbeddings extends Embeddings {
       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,
+      await captureAiGeneration(this.phClient, {
+        eventType: exports.AIEvent.Embedding,
         ...posthogParams,
         model: openAIParams.model,
         provider: 'openai',
@@ -1463,12 +1440,12 @@ let WrappedEmbeddings$1 = class WrappedEmbeddings extends Embeddings {
         // Embeddings don't have output content
         latency: 0,
         baseURL: this.baseURL,
-        params: body,
+        modelParameters: getModelParams(body),
         httpStatus,
         usage: {
           inputTokens: 0
         },
-        error: JSON.stringify(error)
+        error
       });
       throw error;
     });
@@ -1527,8 +1504,7 @@ class WrappedTranscriptions extends Transcriptions {
               const latency = (Date.now() - startTime) / 1000;
               const timeToFirstToken = firstTokenTime !== undefined ? (firstTokenTime - startTime) / 1000 : undefined;
               const availableTools = extractAvailableToolCalls('openai', openAIParams);
-              await sendEventToPosthog({
-                client: this.phClient,
+              await captureAiGeneration(this.phClient, {
                 ...posthogParams,
                 model: openAIParams.model,
                 provider: 'openai',
@@ -1537,14 +1513,13 @@ class WrappedTranscriptions extends Transcriptions {
                 latency,
                 timeToFirstToken,
                 baseURL: this.baseURL,
-                params: body,
+                modelParameters: getModelParams(body),
                 httpStatus: 200,
                 usage,
                 tools: availableTools
               });
             } catch (error) {
-              const enrichedError = await sendEventWithErrorToPosthog({
-                client: this.phClient,
+              await captureAiGeneration(this.phClient, {
                 ...posthogParams,
                 model: openAIParams.model,
                 provider: 'openai',
@@ -1552,14 +1527,14 @@ class WrappedTranscriptions extends Transcriptions {
                 output: [],
                 latency: 0,
                 baseURL: this.baseURL,
-                params: body,
+                modelParameters: getModelParams(body),
                 usage: {
                   inputTokens: 0,
                   outputTokens: 0
                 },
-                error: error
+                error
               });
-              throw enrichedError;
+              throw error;
             }
           })();
           return stream2;
@@ -1570,8 +1545,7 @@ class WrappedTranscriptions extends Transcriptions {
       const wrappedPromise = parentPromise.then(async result => {
         if ('text' in result) {
           const latency = (Date.now() - startTime) / 1000;
-          await sendEventToPosthog({
-            client: this.phClient,
+          await captureAiGeneration(this.phClient, {
             ...posthogParams,
             model: openAIParams.model,
             provider: 'openai',
@@ -1579,7 +1553,7 @@ class WrappedTranscriptions extends Transcriptions {
             output: result.text,
             latency,
             baseURL: this.baseURL,
-            params: body,
+            modelParameters: getModelParams(body),
             httpStatus: 200,
             usage: {
               inputTokens: result.usage?.type === 'tokens' ? result.usage.input_tokens ?? 0 : 0,
@@ -1590,8 +1564,7 @@ class WrappedTranscriptions extends Transcriptions {
           return result;
         }
       }, async error => {
-        const enrichedError = await sendEventWithErrorToPosthog({
-          client: this.phClient,
+        await captureAiGeneration(this.phClient, {
           ...posthogParams,
           model: openAIParams.model,
           provider: 'openai',
@@ -1599,14 +1572,14 @@ class WrappedTranscriptions extends Transcriptions {
           output: [],
           latency: 0,
           baseURL: this.baseURL,
-          params: body,
+          modelParameters: getModelParams(body),
           usage: {
             inputTokens: 0,
             outputTokens: 0
           },
-          error: error
+          error
         });
-        throw enrichedError;
+        throw error;
       });
       return wrappedPromise;
     }
@@ -1751,8 +1724,7 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
               }];
               const latency = (Date.now() - startTime) / 1000;
               const timeToFirstToken = firstTokenTime !== undefined ? (firstTokenTime - startTime) / 1000 : undefined;
-              await sendEventToPosthog({
-                client: this.phClient,
+              await captureAiGeneration(this.phClient, {
                 ...posthogParams,
                 model: openAIParams.model ?? modelFromResponse,
                 provider: 'azure',
@@ -1761,13 +1733,12 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
                 latency,
                 timeToFirstToken,
                 baseURL: this.baseURL,
-                params: body,
+                modelParameters: getModelParams(body),
                 httpStatus: 200,
                 usage
               });
             } catch (error) {
-              const enrichedError = await sendEventWithErrorToPosthog({
-                client: this.phClient,
+              await captureAiGeneration(this.phClient, {
                 ...posthogParams,
                 model: openAIParams.model,
                 provider: 'azure',
@@ -1775,14 +1746,14 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
                 output: [],
                 latency: 0,
                 baseURL: this.baseURL,
-                params: body,
+                modelParameters: getModelParams(body),
                 usage: {
                   inputTokens: 0,
                   outputTokens: 0
                 },
                 error: error
               });
-              throw enrichedError;
+              throw error;
             }
           })();
           // Return the other stream to the user
@@ -1794,8 +1765,7 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
       const wrappedPromise = parentPromise.then(async result => {
         if ('choices' in result) {
           const latency = (Date.now() - startTime) / 1000;
-          await sendEventToPosthog({
-            client: this.phClient,
+          await captureAiGeneration(this.phClient, {
             ...posthogParams,
             model: openAIParams.model ?? result.model,
             provider: 'azure',
@@ -1803,7 +1773,7 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
             output: formatResponseOpenAI(result),
             latency,
             baseURL: this.baseURL,
-            params: body,
+            modelParameters: getModelParams(body),
             httpStatus: 200,
             usage: {
               inputTokens: result.usage?.prompt_tokens ?? 0,
@@ -1816,8 +1786,7 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
         return result;
       }, async error => {
         const httpStatus = error && typeof error === 'object' && 'status' in error ? error.status ?? 500 : 500;
-        await sendEventToPosthog({
-          client: this.phClient,
+        await captureAiGeneration(this.phClient, {
           ...posthogParams,
           model: openAIParams.model,
           provider: 'azure',
@@ -1825,13 +1794,13 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
           output: [],
           latency: 0,
           baseURL: this.baseURL,
-          params: body,
+          modelParameters: getModelParams(body),
           httpStatus,
           usage: {
             inputTokens: 0,
             outputTokens: 0
           },
-          error: JSON.stringify(error)
+          error
         });
         throw error;
       });
@@ -1891,8 +1860,7 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
               }
               const latency = (Date.now() - startTime) / 1000;
               const timeToFirstToken = firstTokenTime !== undefined ? (firstTokenTime - startTime) / 1000 : undefined;
-              await sendEventToPosthog({
-                client: this.phClient,
+              await captureAiGeneration(this.phClient, {
                 ...posthogParams,
                 model: openAIParams.model ?? modelFromResponse,
                 provider: 'azure',
@@ -1901,13 +1869,12 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
                 latency,
                 timeToFirstToken,
                 baseURL: this.baseURL,
-                params: body,
+                modelParameters: getModelParams(body),
                 httpStatus: 200,
                 usage
               });
             } catch (error) {
-              const enrichedError = await sendEventWithErrorToPosthog({
-                client: this.phClient,
+              await captureAiGeneration(this.phClient, {
                 ...posthogParams,
                 model: openAIParams.model,
                 provider: 'azure',
@@ -1915,14 +1882,14 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
                 output: [],
                 latency: 0,
                 baseURL: this.baseURL,
-                params: body,
+                modelParameters: getModelParams(body),
                 usage: {
                   inputTokens: 0,
                   outputTokens: 0
                 },
                 error: error
               });
-              throw enrichedError;
+              throw error;
             }
           })();
           return stream2;
@@ -1933,8 +1900,7 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
       const wrappedPromise = parentPromise.then(async result => {
         if ('output' in result) {
           const latency = (Date.now() - startTime) / 1000;
-          await sendEventToPosthog({
-            client: this.phClient,
+          await captureAiGeneration(this.phClient, {
             ...posthogParams,
             model: openAIParams.model ?? result.model,
             provider: 'azure',
@@ -1942,7 +1908,7 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
             output: result.output,
             latency,
             baseURL: this.baseURL,
-            params: body,
+            modelParameters: getModelParams(body),
             httpStatus: 200,
             usage: {
               inputTokens: result.usage?.input_tokens ?? 0,
@@ -1955,8 +1921,7 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
         return result;
       }, async error => {
         const httpStatus = error && typeof error === 'object' && 'status' in error ? error.status ?? 500 : 500;
-        await sendEventToPosthog({
-          client: this.phClient,
+        await captureAiGeneration(this.phClient, {
           ...posthogParams,
           model: openAIParams.model,
           provider: 'azure',
@@ -1964,13 +1929,13 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
           output: [],
           latency: 0,
           baseURL: this.baseURL,
-          params: body,
+          modelParameters: getModelParams(body),
           httpStatus,
           usage: {
             inputTokens: 0,
             outputTokens: 0
           },
-          error: JSON.stringify(error)
+          error
         });
         throw error;
       });
@@ -1986,8 +1951,7 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
     const parentPromise = super.parse(openAIParams, options);
     const wrappedPromise = parentPromise.then(async result => {
       const latency = (Date.now() - startTime) / 1000;
-      await sendEventToPosthog({
-        client: this.phClient,
+      await captureAiGeneration(this.phClient, {
         ...posthogParams,
         model: openAIParams.model ?? result.model,
         provider: 'azure',
@@ -1995,7 +1959,7 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
         output: result.output,
         latency,
         baseURL: this.baseURL,
-        params: body,
+        modelParameters: getModelParams(body),
         httpStatus: 200,
         usage: {
           inputTokens: result.usage?.input_tokens ?? 0,
@@ -2006,8 +1970,7 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
       });
       return result;
     }, async error => {
-      await sendEventToPosthog({
-        client: this.phClient,
+      await captureAiGeneration(this.phClient, {
         ...posthogParams,
         model: openAIParams.model,
         provider: 'azure',
@@ -2015,13 +1978,13 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
         output: [],
         latency: 0,
         baseURL: this.baseURL,
-        params: body,
+        modelParameters: getModelParams(body),
         httpStatus: error?.status ? error.status : 500,
         usage: {
           inputTokens: 0,
           outputTokens: 0
         },
-        error: JSON.stringify(error)
+        error
       });
       throw error;
     });
@@ -2043,9 +2006,8 @@ class WrappedEmbeddings extends openai.AzureOpenAI.Embeddings {
     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,
+      await captureAiGeneration(this.phClient, {
+        eventType: exports.AIEvent.Embedding,
         ...posthogParams,
         model: openAIParams.model,
         provider: 'azure',
@@ -2054,7 +2016,7 @@ class WrappedEmbeddings extends openai.AzureOpenAI.Embeddings {
         // Embeddings don't have output content
         latency,
         baseURL: this.baseURL,
-        params: body,
+        modelParameters: getModelParams(body),
         httpStatus: 200,
         usage: {
           inputTokens: result.usage?.prompt_tokens ?? 0
@@ -2063,9 +2025,8 @@ class WrappedEmbeddings extends openai.AzureOpenAI.Embeddings {
       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,
+      await captureAiGeneration(this.phClient, {
+        eventType: exports.AIEvent.Embedding,
         ...posthogParams,
         model: openAIParams.model,
         provider: 'azure',
@@ -2073,12 +2034,12 @@ class WrappedEmbeddings extends openai.AzureOpenAI.Embeddings {
         output: null,
         latency: 0,
         baseURL: this.baseURL,
-        params: body,
+        modelParameters: getModelParams(body),
         httpStatus,
         usage: {
           inputTokens: 0
         },
-        error: JSON.stringify(error)
+        error
       });
       throw error;
     });
@@ -2177,18 +2138,26 @@ const mapVercelPrompt = messages => {
     };
   });
   try {
-    // Trim the inputs array until its JSON size fits within MAX_OUTPUT_SIZE
-    const encoder = new TextEncoder();
-    let serialized = JSON.stringify(inputs);
+    // Trim the inputs array until its serialized JSON size fits within MAX_OUTPUT_SIZE.
+    // Pre-compute each message's byte size once so we can shift by accumulated budget
+    // in a single linear pass, instead of re-stringifying the whole array per iteration.
+    const messageSizes = inputs.map(m => utf8ByteLength(JSON.stringify(m)));
+    // Account for the surrounding `[` `]` plus a comma between each pair of elements.
+    let totalBytes = 2 + Math.max(0, messageSizes.length - 1);
+    for (const size of messageSizes) {
+      totalBytes += size;
+    }
     let removedCount = 0;
-    // We need to keep track of the initial size of the inputs array because we're going to be mutating it
-    const initialSize = inputs.length;
-    for (let i = 0; i < initialSize && encoder.encode(serialized).byteLength > MAX_OUTPUT_SIZE; i++) {
-      inputs.shift();
+    while (totalBytes > MAX_OUTPUT_SIZE && removedCount < messageSizes.length) {
+      totalBytes -= messageSizes[removedCount];
+      // Each removed message past the first also drops the comma that joined it.
+      if (removedCount < messageSizes.length - 1) {
+        totalBytes -= 1;
+      }
       removedCount++;
-      serialized = JSON.stringify(inputs);
     }
     if (removedCount > 0) {
+      inputs.splice(0, removedCount);
       // Add one placeholder to indicate how many were removed
       inputs.unshift({
         role: 'posthog',
@@ -2407,6 +2376,18 @@ const wrapVercelLanguageModel = (model, phClient, options) => {
       $ai_framework_version: model.specificationVersion === 'v3' ? '6' : '5'
     }
   };
+  // Shared `captureAiGeneration` options for every call site in this wrapper.
+  const baseOptions = {
+    distinctId: mergedOptions.posthogDistinctId,
+    traceId,
+    properties: mergedOptions.posthogProperties,
+    groups: mergedOptions.posthogGroups,
+    privacyMode: mergedOptions.posthogPrivacyMode,
+    modelOverride: mergedOptions.posthogModelOverride,
+    providerOverride: mergedOptions.posthogProviderOverride,
+    costOverride: mergedOptions.posthogCostOverride,
+    captureImmediate: mergedOptions.posthogCaptureImmediate
+  };
   // Create wrapped model using Object.create to preserve the prototype chain
   // This automatically inherits all properties (including getters) from the model
   const wrappedModel = Object.create(model, {
@@ -2459,46 +2440,40 @@ const wrapVercelLanguageModel = (model, phClient, options) => {
           // Extract finish reason - V2 returns a string, V3 returns an object with .unified
           const rawFinishReason = result.finishReason;
           const finishReasonStr = typeof rawFinishReason === 'string' ? rawFinishReason : rawFinishReason && typeof rawFinishReason === 'object' && 'unified' in rawFinishReason ? String(rawFinishReason.unified) : undefined;
-          await sendEventToPosthog({
-            client: phClient,
-            distinctId: mergedOptions.posthogDistinctId,
-            traceId: mergedOptions.posthogTraceId ?? uuid.v4(),
+          await captureAiGeneration(phClient, {
+            ...baseOptions,
             model: modelId,
             provider: provider,
             input: mergedOptions.posthogPrivacyMode ? '' : mapVercelPrompt(params.prompt),
             output: content,
             latency,
             baseURL,
-            params: mergedParams,
+            modelParameters: getModelParams(mergedParams),
             httpStatus: 200,
             usage,
             stopReason: finishReasonStr,
-            tools: availableTools,
-            captureImmediate: mergedOptions.posthogCaptureImmediate
+            tools: availableTools
           });
           return result;
         } catch (error) {
           const modelId = model.modelId;
-          const enrichedError = await sendEventWithErrorToPosthog({
-            client: phClient,
-            distinctId: mergedOptions.posthogDistinctId,
-            traceId: mergedOptions.posthogTraceId ?? uuid.v4(),
+          await captureAiGeneration(phClient, {
+            ...baseOptions,
             model: modelId,
             provider: model.provider,
             input: mergedOptions.posthogPrivacyMode ? '' : mapVercelPrompt(params.prompt),
             output: [],
             latency: 0,
             baseURL: '',
-            params: mergedParams,
+            modelParameters: getModelParams(mergedParams),
             usage: {
               inputTokens: 0,
               outputTokens: 0
             },
             error: error,
-            tools: availableTools,
-            captureImmediate: mergedOptions.posthogCaptureImmediate
+            tools: availableTools
           });
-          throw enrichedError;
+          throw error;
         }
       },
       writable: true,
@@ -2644,10 +2619,8 @@ const wrapVercelLanguageModel = (model, phClient, options) => {
                 }
               };
               adjustAnthropicV3CacheTokens(model, modelId, provider, finalUsage);
-              await sendEventToPosthog({
-                client: phClient,
-                distinctId: mergedOptions.posthogDistinctId,
-                traceId: mergedOptions.posthogTraceId ?? uuid.v4(),
+              await captureAiGeneration(phClient, {
+                ...baseOptions,
                 model: modelId,
                 provider: provider,
                 input: mergedOptions.posthogPrivacyMode ? '' : mapVercelPrompt(params.prompt),
@@ -2655,12 +2628,11 @@ const wrapVercelLanguageModel = (model, phClient, options) => {
                 latency,
                 timeToFirstToken,
                 baseURL,
-                params: mergedParams,
+                modelParameters: getModelParams(mergedParams),
                 httpStatus: 200,
                 usage: finalUsage,
                 stopReason,
-                tools: availableTools,
-                captureImmediate: mergedOptions.posthogCaptureImmediate
+                tools: availableTools
               });
             }
           });
@@ -2669,26 +2641,23 @@ const wrapVercelLanguageModel = (model, phClient, options) => {
             ...rest
           };
         } catch (error) {
-          const enrichedError = await sendEventWithErrorToPosthog({
-            client: phClient,
-            distinctId: mergedOptions.posthogDistinctId,
-            traceId: mergedOptions.posthogTraceId ?? uuid.v4(),
+          await captureAiGeneration(phClient, {
+            ...baseOptions,
             model: modelId,
             provider: provider,
             input: mergedOptions.posthogPrivacyMode ? '' : mapVercelPrompt(params.prompt),
             output: [],
             latency: 0,
             baseURL: '',
-            params: mergedParams,
+            modelParameters: getModelParams(mergedParams),
             usage: {
               inputTokens: 0,
               outputTokens: 0
             },
             error: error,
-            tools: availableTools,
-            captureImmediate: mergedOptions.posthogCaptureImmediate
+            tools: availableTools
           });
-          throw enrichedError;
+          throw error;
         }
       },
       writable: true,
@@ -2853,8 +2822,7 @@ class WrappedMessages extends AnthropicOriginal__default.default.Messages {
                   text: accumulatedContent
                 }]
               }];
-              await sendEventToPosthog({
-                client: this.phClient,
+              await captureAiGeneration(this.phClient, {
                 ...posthogParams,
                 model: anthropicParams.model,
                 provider: 'anthropic',
@@ -2863,15 +2831,14 @@ class WrappedMessages extends AnthropicOriginal__default.default.Messages {
                 latency,
                 timeToFirstToken,
                 baseURL: this.baseURL,
-                params: body,
+                modelParameters: getModelParams(body),
                 httpStatus: 200,
                 usage,
                 stopReason,
                 tools: availableTools
               });
             } catch (error) {
-              const enrichedError = await sendEventWithErrorToPosthog({
-                client: this.phClient,
+              await captureAiGeneration(this.phClient, {
                 ...posthogParams,
                 model: anthropicParams.model,
                 provider: 'anthropic',
@@ -2879,14 +2846,14 @@ class WrappedMessages extends AnthropicOriginal__default.default.Messages {
                 output: [],
                 latency: 0,
                 baseURL: this.baseURL,
-                params: body,
+                modelParameters: getModelParams(body),
                 usage: {
                   inputTokens: 0,
                   outputTokens: 0
                 },
                 error: error
               });
-              throw enrichedError;
+              throw error;
             }
           })();
           // Return the other stream to the user
@@ -2899,8 +2866,7 @@ class WrappedMessages extends AnthropicOriginal__default.default.Messages {
         if ('content' in result) {
           const latency = (Date.now() - startTime) / 1000;
           const availableTools = extractAvailableToolCalls('anthropic', anthropicParams);
-          await sendEventToPosthog({
-            client: this.phClient,
+          await captureAiGeneration(this.phClient, {
             ...posthogParams,
             model: anthropicParams.model,
             provider: 'anthropic',
@@ -2908,7 +2874,7 @@ class WrappedMessages extends AnthropicOriginal__default.default.Messages {
             output: formatResponseAnthropic(result),
             latency,
             baseURL: this.baseURL,
-            params: body,
+            modelParameters: getModelParams(body),
             httpStatus: 200,
             usage: {
               inputTokens: result.usage.input_tokens ?? 0,
@@ -2924,8 +2890,7 @@ class WrappedMessages extends AnthropicOriginal__default.default.Messages {
         }
         return result;
       }, async error => {
-        await sendEventToPosthog({
-          client: this.phClient,
+        await captureAiGeneration(this.phClient, {
           ...posthogParams,
           model: anthropicParams.model,
           provider: 'anthropic',
@@ -2933,13 +2898,13 @@ class WrappedMessages extends AnthropicOriginal__default.default.Messages {
           output: [],
           latency: 0,
           baseURL: this.baseURL,
-          params: body,
+          modelParameters: getModelParams(body),
           httpStatus: error?.status ? error.status : 500,
           usage: {
             inputTokens: 0,
             outputTokens: 0
           },
-          error: JSON.stringify(error)
+          error: error
         });
         throw error;
       });
@@ -2976,8 +2941,7 @@ class WrappedModels {
       const availableTools = extractAvailableToolCalls('gemini', geminiParams);
       const metadata = response.usageMetadata;
       const finishReason = response.candidates?.[0]?.finishReason;
-      await sendEventToPosthog({
-        client: this.phClient,
+      await captureAiGeneration(this.phClient, {
         ...posthogParams,
         model: geminiParams.model,
         provider: 'gemini',
@@ -2985,7 +2949,7 @@ class WrappedModels {
         output: formatResponseGemini(response),
         latency,
         baseURL: 'https://generativelanguage.googleapis.com',
-        params: params,
+        modelParameters: getModelParams(params),
         httpStatus: 200,
         usage: {
           inputTokens: metadata?.promptTokenCount ?? 0,
@@ -3001,8 +2965,7 @@ class WrappedModels {
       return response;
     } catch (error) {
       const latency = (Date.now() - startTime) / 1000;
-      const enrichedError = await sendEventWithErrorToPosthog({
-        client: this.phClient,
+      await captureAiGeneration(this.phClient, {
         ...posthogParams,
         model: geminiParams.model,
         provider: 'gemini',
@@ -3010,14 +2973,14 @@ class WrappedModels {
         output: [],
         latency,
         baseURL: 'https://generativelanguage.googleapis.com',
-        params: params,
+        modelParameters: getModelParams(params),
         usage: {
           inputTokens: 0,
           outputTokens: 0
         },
-        error: error
+        error
       });
-      throw enrichedError;
+      throw error;
     }
   }
   async *generateContentStream(params) {
@@ -3116,8 +3079,7 @@ class WrappedModels {
         role: 'assistant',
         content: accumulatedContent
       }] : [];
-      await sendEventToPosthog({
-        client: this.phClient,
+      await captureAiGeneration(this.phClient, {
         ...posthogParams,
         model: geminiParams.model,
         provider: 'gemini',
@@ -3126,7 +3088,7 @@ class WrappedModels {
         latency,
         timeToFirstToken,
         baseURL: 'https://generativelanguage.googleapis.com',
-        params: params,
+        modelParameters: getModelParams(params),
         httpStatus: 200,
         usage: {
           ...usage,
@@ -3138,8 +3100,7 @@ class WrappedModels {
       });
     } catch (error) {
       const latency = (Date.now() - startTime) / 1000;
-      const enrichedError = await sendEventWithErrorToPosthog({
-        client: this.phClient,
+      await captureAiGeneration(this.phClient, {
         ...posthogParams,
         model: geminiParams.model,
         provider: 'gemini',
@@ -3147,14 +3108,14 @@ class WrappedModels {
         output: [],
         latency,
         baseURL: 'https://generativelanguage.googleapis.com',
-        params: params,
+        modelParameters: getModelParams(params),
         usage: {
           inputTokens: 0,
           outputTokens: 0
         },
-        error: error
+        error
       });
-      throw enrichedError;
+      throw error;
     }
   }
   async embedContent(params) {
@@ -3167,17 +3128,16 @@ class WrappedModels {
       const response = await this.client.models.embedContent(geminiParams);
       const latency = (Date.now() - startTime) / 1000;
       const inputTokens = extractEmbeddingTokenCount(response);
-      await sendEventToPosthog({
-        client: this.phClient,
+      await captureAiGeneration(this.phClient, {
         ...posthogParams,
-        eventType: AIEvent.Embedding,
+        eventType: exports.AIEvent.Embedding,
         model: geminiParams.model,
         provider: 'gemini',
         input: withPrivacyMode(this.phClient, posthogParams.privacyMode ?? false, geminiParams.contents),
         output: null,
         latency,
         baseURL: 'https://generativelanguage.googleapis.com',
-        params: params,
+        modelParameters: getModelParams(params),
         httpStatus: 200,
         usage: {
           inputTokens
@@ -3186,23 +3146,22 @@ class WrappedModels {
       return response;
     } catch (error) {
       const latency = (Date.now() - startTime) / 1000;
-      const enrichedError = await sendEventWithErrorToPosthog({
-        client: this.phClient,
+      await captureAiGeneration(this.phClient, {
         ...posthogParams,
-        eventType: AIEvent.Embedding,
+        eventType: exports.AIEvent.Embedding,
         model: geminiParams.model,
         provider: 'gemini',
         input: withPrivacyMode(this.phClient, posthogParams.privacyMode ?? false, geminiParams.contents),
         output: null,
         latency,
         baseURL: 'https://generativelanguage.googleapis.com',
-        params: params,
+        modelParameters: getModelParams(params),
         usage: {
           inputTokens: 0
         },
-        error: error
+        error
       });
-      throw enrichedError;
+      throw error;
     }
   }
   formatPartsAsContentBlocks(parts) {
@@ -4744,5 +4703,6 @@ exports.GoogleGenAI = PostHogGoogleGenAI;
 exports.LangChainCallbackHandler = LangChainCallbackHandler;
 exports.OpenAI = PostHogOpenAI;
 exports.Prompts = Prompts;
+exports.captureAiGeneration = captureAiGeneration;
 exports.withTracing = wrapVercelLanguageModel;
 //# sourceMappingURL=index.cjs.map