@posthog/ai 6.3.0 → 6.3.2

package/dist/index.cjs

@@ -1,8 +1,8 @@
 'use strict';
 
 var openai = require('openai');
-var uuid = require('uuid');
 var buffer = require('buffer');
+var uuid = require('uuid');
 var ai = require('ai');
 var AnthropicOriginal = require('@anthropic-ai/sdk');
 var genai = require('@google/genai');
@@ -26,7 +26,7 @@ function _interopNamespaceDefault(e) {
 
 var uuid__namespace = /*#__PURE__*/_interopNamespaceDefault(uuid);
 
-var version = "6.3.0";
+var version = "6.3.2";
 
 // limit large outputs by truncating to 200kb (approx 200k bytes)
 const MAX_OUTPUT_SIZE = 200000;
@@ -226,18 +226,43 @@ const mergeSystemPrompt = (params, provider) => {
 const withPrivacyMode = (client, privacyMode, input) => {
   return client.privacy_mode || privacyMode ? null : input;
 };
-const truncate = str => {
+function toSafeString(input) {
+  if (input === undefined || input === null) {
+    return '';
+  }
+  if (typeof input === 'string') {
+    return input;
+  }
   try {
-    const buffer$1 = buffer.Buffer.from(str, STRING_FORMAT);
-    if (buffer$1.length <= MAX_OUTPUT_SIZE) {
-      return str;
-    }
-    const truncatedBuffer = buffer$1.slice(0, MAX_OUTPUT_SIZE);
-    return `${truncatedBuffer.toString(STRING_FORMAT)}... [truncated]`;
+    return JSON.stringify(input);
   } catch {
-    console.error('Error truncating, likely not a string');
-    return str;
+    console.warn('Failed to stringify input', input);
+    return '';
   }
+}
+const truncate = input => {
+  const str = toSafeString(input);
+  if (str === '') {
+    return '';
+  }
+  // Check if we need to truncate and ensure STRING_FORMAT is respected
+  const encoder = new TextEncoder();
+  const buffer = encoder.encode(str);
+  if (buffer.length <= MAX_OUTPUT_SIZE) {
+    // Ensure STRING_FORMAT is respected
+    return new TextDecoder(STRING_FORMAT).decode(buffer);
+  }
+  // Truncate the buffer and ensure a valid string is returned
+  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);
+  if (truncatedStr.endsWith('\uFFFD')) {
+    truncatedStr = truncatedStr.slice(0, -1);
+  }
+  return `${truncatedStr}... [truncated]`;
 };
 /**
  * Extract available tool calls from the request parameters.
@@ -579,6 +604,42 @@ const sanitizeLangChain = data => {
   return processMessages(data, sanitizeLangChainImage);
 };
 
+const POSTHOG_PARAMS_MAP = {
+  posthogDistinctId: 'distinctId',
+  posthogTraceId: 'traceId',
+  posthogProperties: 'properties',
+  posthogPrivacyMode: 'privacyMode',
+  posthogGroups: 'groups',
+  posthogModelOverride: 'modelOverride',
+  posthogProviderOverride: 'providerOverride',
+  posthogCostOverride: 'costOverride',
+  posthogCaptureImmediate: 'captureImmediate'
+};
+function extractPosthogParams(body) {
+  const openAIParams = {};
+  const posthogParams = {};
+  for (const [key, value] of Object.entries(body)) {
+    if (POSTHOG_PARAMS_MAP[key]) {
+      posthogParams[POSTHOG_PARAMS_MAP[key]] = value;
+    } else if (key.startsWith('posthog')) {
+      console.warn(`Unknown Posthog parameter ${key}`);
+    } else {
+      openAIParams[key] = value;
+    }
+  }
+  return {
+    openAIParams: openAIParams,
+    posthogParams: addDefaults(posthogParams)
+  };
+}
+function addDefaults(params) {
+  return {
+    ...params,
+    privacyMode: params.privacyMode ?? false,
+    traceId: params.traceId ?? uuid.v4()
+  };
+}
+
 const Chat = openai.OpenAI.Chat;
 const Completions = Chat.Completions;
 const Responses = openai.OpenAI.Responses;
@@ -611,12 +672,9 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
   // --- Implementation Signature
   create(body, options) {
     const {
-      posthogDistinctId,
-      posthogTraceId,
-      posthogCaptureImmediate,
-      ...openAIParams
-    } = body;
-    const traceId = posthogTraceId ?? uuid.v4();
+      openAIParams,
+      posthogParams
+    } = extractPosthogParams(body);
     const startTime = Date.now();
     const parentPromise = super.create(openAIParams, options);
     if (openAIParams.stream) {
@@ -715,8 +773,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
               const availableTools = extractAvailableToolCalls('openai', openAIParams);
               await sendEventToPosthog({
                 client: this.phClient,
-                distinctId: posthogDistinctId,
-                traceId,
+                ...posthogParams,
                 model: openAIParams.model,
                 provider: 'openai',
                 input: sanitizeOpenAI(openAIParams.messages),
@@ -726,15 +783,13 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
                 params: body,
                 httpStatus: 200,
                 usage,
-                tools: availableTools,
-                captureImmediate: posthogCaptureImmediate
+                tools: availableTools
               });
             } catch (error) {
               const httpStatus = error && typeof error === 'object' && 'status' in error ? error.status ?? 500 : 500;
               await sendEventToPosthog({
                 client: this.phClient,
-                distinctId: posthogDistinctId,
-                traceId,
+                ...posthogParams,
                 model: openAIParams.model,
                 provider: 'openai',
                 input: sanitizeOpenAI(openAIParams.messages),
@@ -748,8 +803,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
                   outputTokens: 0
                 },
                 isError: true,
-                error: JSON.stringify(error),
-                captureImmediate: posthogCaptureImmediate
+                error: JSON.stringify(error)
               });
             }
           })();
@@ -765,8 +819,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
           const availableTools = extractAvailableToolCalls('openai', openAIParams);
           await sendEventToPosthog({
             client: this.phClient,
-            distinctId: posthogDistinctId,
-            traceId,
+            ...posthogParams,
             model: openAIParams.model,
             provider: 'openai',
             input: sanitizeOpenAI(openAIParams.messages),
@@ -781,8 +834,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
               reasoningTokens: result.usage?.completion_tokens_details?.reasoning_tokens ?? 0,
               cacheReadInputTokens: result.usage?.prompt_tokens_details?.cached_tokens ?? 0
             },
-            tools: availableTools,
-            captureImmediate: posthogCaptureImmediate
+            tools: availableTools
           });
         }
         return result;
@@ -790,8 +842,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
         const httpStatus = error && typeof error === 'object' && 'status' in error ? error.status ?? 500 : 500;
         await sendEventToPosthog({
           client: this.phClient,
-          distinctId: posthogDistinctId,
-          traceId,
+          ...posthogParams,
           model: openAIParams.model,
           provider: 'openai',
           input: sanitizeOpenAI(openAIParams.messages),
@@ -805,8 +856,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
             outputTokens: 0
           },
           isError: true,
-          error: JSON.stringify(error),
-          captureImmediate: posthogCaptureImmediate
+          error: JSON.stringify(error)
         });
         throw error;
       });
@@ -823,12 +873,9 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
   // --- Implementation Signature
   create(body, options) {
     const {
-      posthogDistinctId,
-      posthogTraceId,
-      posthogCaptureImmediate,
-      ...openAIParams
-    } = body;
-    const traceId = posthogTraceId ?? uuid.v4();
+      openAIParams,
+      posthogParams
+    } = extractPosthogParams(body);
     const startTime = Date.now();
     const parentPromise = super.create(openAIParams, options);
     if (openAIParams.stream) {
@@ -859,8 +906,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
               const availableTools = extractAvailableToolCalls('openai', openAIParams);
               await sendEventToPosthog({
                 client: this.phClient,
-                distinctId: posthogDistinctId,
-                traceId,
+                ...posthogParams,
                 //@ts-expect-error
                 model: openAIParams.model,
                 provider: 'openai',
@@ -871,15 +917,13 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
                 params: body,
                 httpStatus: 200,
                 usage,
-                tools: availableTools,
-                captureImmediate: posthogCaptureImmediate
+                tools: availableTools
               });
             } catch (error) {
               const httpStatus = error && typeof error === 'object' && 'status' in error ? error.status ?? 500 : 500;
               await sendEventToPosthog({
                 client: this.phClient,
-                distinctId: posthogDistinctId,
-                traceId,
+                ...posthogParams,
                 //@ts-expect-error
                 model: openAIParams.model,
                 provider: 'openai',
@@ -894,8 +938,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
                   outputTokens: 0
                 },
                 isError: true,
-                error: JSON.stringify(error),
-                captureImmediate: posthogCaptureImmediate
+                error: JSON.stringify(error)
               });
             }
           })();
@@ -910,8 +953,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
           const availableTools = extractAvailableToolCalls('openai', openAIParams);
           await sendEventToPosthog({
             client: this.phClient,
-            distinctId: posthogDistinctId,
-            traceId,
+            ...posthogParams,
             //@ts-expect-error
             model: openAIParams.model,
             provider: 'openai',
@@ -929,8 +971,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
               reasoningTokens: result.usage?.output_tokens_details?.reasoning_tokens ?? 0,
               cacheReadInputTokens: result.usage?.input_tokens_details?.cached_tokens ?? 0
             },
-            tools: availableTools,
-            captureImmediate: posthogCaptureImmediate
+            tools: availableTools
           });
         }
         return result;
@@ -938,8 +979,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
         const httpStatus = error && typeof error === 'object' && 'status' in error ? error.status ?? 500 : 500;
         await sendEventToPosthog({
           client: this.phClient,
-          distinctId: posthogDistinctId,
-          traceId,
+          ...posthogParams,
           //@ts-expect-error
           model: openAIParams.model,
           provider: 'openai',
@@ -954,8 +994,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
             outputTokens: 0
           },
           isError: true,
-          error: JSON.stringify(error),
-          captureImmediate: posthogCaptureImmediate
+          error: JSON.stringify(error)
         });
         throw error;
       });
@@ -964,12 +1003,9 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
   }
   parse(body, options) {
     const {
-      posthogDistinctId,
-      posthogTraceId,
-      posthogCaptureImmediate,
-      ...openAIParams
-    } = body;
-    const traceId = posthogTraceId ?? uuid.v4();
+      openAIParams,
+      posthogParams
+    } = extractPosthogParams(body);
     const startTime = Date.now();
     // Create a temporary instance that bypasses our wrapped create method
     const originalCreate = super.create.bind(this);
@@ -982,8 +1018,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
         const latency = (Date.now() - startTime) / 1000;
         await sendEventToPosthog({
           client: this.phClient,
-          distinctId: posthogDistinctId,
-          traceId,
+          ...posthogParams,
           //@ts-expect-error
           model: openAIParams.model,
           provider: 'openai',
@@ -998,16 +1033,14 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
             outputTokens: result.usage?.output_tokens ?? 0,
             reasoningTokens: result.usage?.output_tokens_details?.reasoning_tokens ?? 0,
             cacheReadInputTokens: result.usage?.input_tokens_details?.cached_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,
-          distinctId: posthogDistinctId,
-          traceId,
+          ...posthogParams,
           //@ts-expect-error
           model: openAIParams.model,
           provider: 'openai',
@@ -1022,8 +1055,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
             outputTokens: 0
           },
           isError: true,
-          error: JSON.stringify(error),
-          captureImmediate: posthogCaptureImmediate
+          error: JSON.stringify(error)
         });
         throw error;
       });
@@ -1042,25 +1074,20 @@ let WrappedEmbeddings$1 = class WrappedEmbeddings extends Embeddings {
   }
   create(body, options) {
     const {
-      posthogDistinctId,
-      posthogTraceId,
-      posthogPrivacyMode = false,
-      posthogCaptureImmediate,
-      ...openAIParams
-    } = body;
-    const traceId = posthogTraceId ?? uuid.v4();
+      openAIParams,
+      posthogParams
+    } = extractPosthogParams(body);
     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,
+        ...posthogParams,
         eventType: AIEvent.Embedding,
-        distinctId: posthogDistinctId,
-        traceId,
         model: openAIParams.model,
         provider: 'openai',
-        input: withPrivacyMode(this.phClient, posthogPrivacyMode, openAIParams.input),
+        input: withPrivacyMode(this.phClient, posthogParams.privacyMode, openAIParams.input),
         output: null,
         // Embeddings don't have output content
         latency,
@@ -1069,8 +1096,7 @@ let WrappedEmbeddings$1 = class WrappedEmbeddings extends Embeddings {
         httpStatus: 200,
         usage: {
           inputTokens: result.usage?.prompt_tokens ?? 0
-        },
-        captureImmediate: posthogCaptureImmediate
+        }
       });
       return result;
     }, async error => {
@@ -1078,11 +1104,10 @@ let WrappedEmbeddings$1 = class WrappedEmbeddings extends Embeddings {
       await sendEventToPosthog({
         client: this.phClient,
         eventType: AIEvent.Embedding,
-        distinctId: posthogDistinctId,
-        traceId,
+        ...posthogParams,
         model: openAIParams.model,
         provider: 'openai',
-        input: withPrivacyMode(this.phClient, posthogPrivacyMode, openAIParams.input),
+        input: withPrivacyMode(this.phClient, posthogParams.privacyMode, openAIParams.input),
         output: null,
         // Embeddings don't have output content
         latency: 0,
@@ -1093,8 +1118,7 @@ let WrappedEmbeddings$1 = class WrappedEmbeddings extends Embeddings {
           inputTokens: 0
         },
         isError: true,
-        error: JSON.stringify(error),
-        captureImmediate: posthogCaptureImmediate
+        error: JSON.stringify(error)
       });
       throw error;
     });
@@ -1129,12 +1153,9 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
   // --- Implementation Signature
   create(body, options) {
     const {
-      posthogDistinctId,
-      posthogTraceId,
-      posthogCaptureImmediate,
-      ...openAIParams
-    } = body;
-    const traceId = posthogTraceId ?? uuid.v4();
+      openAIParams,
+      posthogParams
+    } = extractPosthogParams(body);
     const startTime = Date.now();
     const parentPromise = super.create(openAIParams, options);
     if (openAIParams.stream) {
@@ -1232,8 +1253,7 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
               const latency = (Date.now() - startTime) / 1000;
               await sendEventToPosthog({
                 client: this.phClient,
-                distinctId: posthogDistinctId,
-                traceId,
+                ...posthogParams,
                 model: openAIParams.model,
                 provider: 'azure',
                 input: openAIParams.messages,
@@ -1242,15 +1262,13 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
                 baseURL: this.baseURL,
                 params: body,
                 httpStatus: 200,
-                usage,
-                captureImmediate: posthogCaptureImmediate
+                usage
               });
             } catch (error) {
               const httpStatus = error && typeof error === 'object' && 'status' in error ? error.status ?? 500 : 500;
               await sendEventToPosthog({
                 client: this.phClient,
-                distinctId: posthogDistinctId,
-                traceId,
+                ...posthogParams,
                 model: openAIParams.model,
                 provider: 'azure',
                 input: openAIParams.messages,
@@ -1264,8 +1282,7 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
                   outputTokens: 0
                 },
                 isError: true,
-                error: JSON.stringify(error),
-                captureImmediate: posthogCaptureImmediate
+                error: JSON.stringify(error)
               });
             }
           })();
@@ -1280,8 +1297,7 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
           const latency = (Date.now() - startTime) / 1000;
           await sendEventToPosthog({
             client: this.phClient,
-            distinctId: posthogDistinctId,
-            traceId,
+            ...posthogParams,
             model: openAIParams.model,
             provider: 'azure',
             input: openAIParams.messages,
@@ -1295,8 +1311,7 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
               outputTokens: result.usage?.completion_tokens ?? 0,
               reasoningTokens: result.usage?.completion_tokens_details?.reasoning_tokens ?? 0,
               cacheReadInputTokens: result.usage?.prompt_tokens_details?.cached_tokens ?? 0
-            },
-            captureImmediate: posthogCaptureImmediate
+            }
           });
         }
         return result;
@@ -1304,8 +1319,7 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
         const httpStatus = error && typeof error === 'object' && 'status' in error ? error.status ?? 500 : 500;
         await sendEventToPosthog({
           client: this.phClient,
-          distinctId: posthogDistinctId,
-          traceId,
+          ...posthogParams,
           model: openAIParams.model,
           provider: 'azure',
           input: openAIParams.messages,
@@ -1319,8 +1333,7 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
             outputTokens: 0
           },
           isError: true,
-          error: JSON.stringify(error),
-          captureImmediate: posthogCaptureImmediate
+          error: JSON.stringify(error)
         });
         throw error;
       });
@@ -1337,12 +1350,9 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
   // --- Implementation Signature
   create(body, options) {
     const {
-      posthogDistinctId,
-      posthogTraceId,
-      posthogCaptureImmediate,
-      ...openAIParams
-    } = body;
-    const traceId = posthogTraceId ?? uuid.v4();
+      openAIParams,
+      posthogParams
+    } = extractPosthogParams(body);
     const startTime = Date.now();
     const parentPromise = super.create(openAIParams, options);
     if (openAIParams.stream) {
@@ -1372,8 +1382,7 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
               const latency = (Date.now() - startTime) / 1000;
               await sendEventToPosthog({
                 client: this.phClient,
-                distinctId: posthogDistinctId,
-                traceId,
+                ...posthogParams,
                 //@ts-expect-error
                 model: openAIParams.model,
                 provider: 'azure',
@@ -1383,15 +1392,13 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
                 baseURL: this.baseURL,
                 params: body,
                 httpStatus: 200,
-                usage,
-                captureImmediate: posthogCaptureImmediate
+                usage
               });
             } catch (error) {
               const httpStatus = error && typeof error === 'object' && 'status' in error ? error.status ?? 500 : 500;
               await sendEventToPosthog({
                 client: this.phClient,
-                distinctId: posthogDistinctId,
-                traceId,
+                ...posthogParams,
                 //@ts-expect-error
                 model: openAIParams.model,
                 provider: 'azure',
@@ -1406,8 +1413,7 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
                   outputTokens: 0
                 },
                 isError: true,
-                error: JSON.stringify(error),
-                captureImmediate: posthogCaptureImmediate
+                error: JSON.stringify(error)
               });
             }
           })();
@@ -1421,8 +1427,7 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
           const latency = (Date.now() - startTime) / 1000;
           await sendEventToPosthog({
             client: this.phClient,
-            distinctId: posthogDistinctId,
-            traceId,
+            ...posthogParams,
             //@ts-expect-error
             model: openAIParams.model,
             provider: 'azure',
@@ -1437,8 +1442,7 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
               outputTokens: result.usage?.output_tokens ?? 0,
               reasoningTokens: result.usage?.output_tokens_details?.reasoning_tokens ?? 0,
               cacheReadInputTokens: result.usage?.input_tokens_details?.cached_tokens ?? 0
-            },
-            captureImmediate: posthogCaptureImmediate
+            }
           });
         }
         return result;
@@ -1446,8 +1450,7 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
         const httpStatus = error && typeof error === 'object' && 'status' in error ? error.status ?? 500 : 500;
         await sendEventToPosthog({
           client: this.phClient,
-          distinctId: posthogDistinctId,
-          traceId,
+          ...posthogParams,
           //@ts-expect-error
           model: openAIParams.model,
           provider: 'azure',
@@ -1462,8 +1465,7 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
             outputTokens: 0
           },
           isError: true,
-          error: JSON.stringify(error),
-          captureImmediate: posthogCaptureImmediate
+          error: JSON.stringify(error)
         });
         throw error;
       });
@@ -1472,20 +1474,16 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
   }
   parse(body, options) {
     const {
-      posthogDistinctId,
-      posthogTraceId,
-      posthogCaptureImmediate,
-      ...openAIParams
-    } = body;
-    const traceId = posthogTraceId ?? uuid.v4();
+      openAIParams,
+      posthogParams
+    } = extractPosthogParams(body);
     const startTime = Date.now();
     const parentPromise = super.parse(openAIParams, options);
     const wrappedPromise = parentPromise.then(async result => {
       const latency = (Date.now() - startTime) / 1000;
       await sendEventToPosthog({
         client: this.phClient,
-        distinctId: posthogDistinctId,
-        traceId,
+        ...posthogParams,
         //@ts-expect-error
         model: openAIParams.model,
         provider: 'azure',
@@ -1500,15 +1498,13 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
           outputTokens: result.usage?.output_tokens ?? 0,
           reasoningTokens: result.usage?.output_tokens_details?.reasoning_tokens ?? 0,
           cacheReadInputTokens: result.usage?.input_tokens_details?.cached_tokens ?? 0
-        },
-        captureImmediate: posthogCaptureImmediate
+        }
       });
       return result;
     }, async error => {
       await sendEventToPosthog({
         client: this.phClient,
-        distinctId: posthogDistinctId,
-        traceId,
+        ...posthogParams,
         //@ts-expect-error
         model: openAIParams.model,
         provider: 'azure',
@@ -1523,8 +1519,7 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
           outputTokens: 0
         },
         isError: true,
-        error: JSON.stringify(error),
-        captureImmediate: posthogCaptureImmediate
+        error: JSON.stringify(error)
       });
       throw error;
     });
@@ -1539,13 +1534,9 @@ class WrappedEmbeddings extends openai.AzureOpenAI.Embeddings {
   }
   create(body, options) {
     const {
-      posthogDistinctId,
-      posthogTraceId,
-      posthogPrivacyMode = false,
-      posthogCaptureImmediate,
-      ...openAIParams
-    } = body;
-    const traceId = posthogTraceId ?? uuid.v4();
+      openAIParams,
+      posthogParams
+    } = extractPosthogParams(body);
     const startTime = Date.now();
     const parentPromise = super.create(openAIParams, options);
     const wrappedPromise = parentPromise.then(async result => {
@@ -1553,11 +1544,10 @@ class WrappedEmbeddings extends openai.AzureOpenAI.Embeddings {
       await sendEventToPosthog({
         client: this.phClient,
         eventType: AIEvent.Embedding,
-        distinctId: posthogDistinctId,
-        traceId,
+        ...posthogParams,
         model: openAIParams.model,
         provider: 'azure',
-        input: withPrivacyMode(this.phClient, posthogPrivacyMode, openAIParams.input),
+        input: withPrivacyMode(this.phClient, posthogParams.privacyMode, openAIParams.input),
         output: null,
         // Embeddings don't have output content
         latency,
@@ -1566,8 +1556,7 @@ class WrappedEmbeddings extends openai.AzureOpenAI.Embeddings {
         httpStatus: 200,
         usage: {
           inputTokens: result.usage?.prompt_tokens ?? 0
-        },
-        captureImmediate: posthogCaptureImmediate
+        }
       });
       return result;
     }, async error => {
@@ -1575,11 +1564,10 @@ class WrappedEmbeddings extends openai.AzureOpenAI.Embeddings {
       await sendEventToPosthog({
         client: this.phClient,
         eventType: AIEvent.Embedding,
-        distinctId: posthogDistinctId,
-        traceId,
+        ...posthogParams,
         model: openAIParams.model,
         provider: 'azure',
-        input: withPrivacyMode(this.phClient, posthogPrivacyMode, openAIParams.input),
+        input: withPrivacyMode(this.phClient, posthogParams.privacyMode, openAIParams.input),
         output: null,
         latency: 0,
         baseURL: this.baseURL,
@@ -1589,8 +1577,7 @@ class WrappedEmbeddings extends openai.AzureOpenAI.Embeddings {
           inputTokens: 0
         },
         isError: true,
-        error: JSON.stringify(error),
-        captureImmediate: posthogCaptureImmediate
+        error: JSON.stringify(error)
       });
       throw error;
     });