@posthog/ai 6.3.1 → 6.3.3

package/dist/index.mjs

@@ -1,12 +1,12 @@
 import { OpenAI, AzureOpenAI } from 'openai';
+import { Buffer } from 'buffer';
 import * as uuid from 'uuid';
 import { v4 } from 'uuid';
-import { Buffer } from 'buffer';
 import { wrapLanguageModel } from 'ai';
 import AnthropicOriginal from '@anthropic-ai/sdk';
 import { GoogleGenAI } from '@google/genai';
 
-var version = "6.3.1";
+var version = "6.3.3";
 
 // limit large outputs by truncating to 200kb (approx 200k bytes)
 const MAX_OUTPUT_SIZE = 200000;
@@ -291,6 +291,41 @@ function sanitizeValues(obj) {
   }
   return jsonSafe;
 }
+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 providerParams = {};
+  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 {
+      providerParams[key] = value;
+    }
+  }
+  return {
+    providerParams: providerParams,
+    posthogParams: addDefaults(posthogParams)
+  };
+}
+function addDefaults(params) {
+  return {
+    ...params,
+    privacyMode: params.privacyMode ?? false,
+    traceId: params.traceId ?? v4()
+  };
+}
 const sendEventToPosthog = async ({
   client,
   eventType = AIEvent.Generation,
@@ -616,12 +651,9 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
   // --- Implementation Signature
   create(body, options) {
     const {
-      posthogDistinctId,
-      posthogTraceId,
-      posthogCaptureImmediate,
-      ...openAIParams
-    } = body;
-    const traceId = posthogTraceId ?? v4();
+      providerParams: openAIParams,
+      posthogParams
+    } = extractPosthogParams(body);
     const startTime = Date.now();
     const parentPromise = super.create(openAIParams, options);
     if (openAIParams.stream) {
@@ -720,8 +752,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),
@@ -731,15 +762,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),
@@ -753,8 +782,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
                   outputTokens: 0
                 },
                 isError: true,
-                error: JSON.stringify(error),
-                captureImmediate: posthogCaptureImmediate
+                error: JSON.stringify(error)
               });
             }
           })();
@@ -770,8 +798,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),
@@ -786,8 +813,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;
@@ -795,8 +821,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),
@@ -810,8 +835,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
             outputTokens: 0
           },
           isError: true,
-          error: JSON.stringify(error),
-          captureImmediate: posthogCaptureImmediate
+          error: JSON.stringify(error)
         });
         throw error;
       });
@@ -828,12 +852,9 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
   // --- Implementation Signature
   create(body, options) {
     const {
-      posthogDistinctId,
-      posthogTraceId,
-      posthogCaptureImmediate,
-      ...openAIParams
-    } = body;
-    const traceId = posthogTraceId ?? v4();
+      providerParams: openAIParams,
+      posthogParams
+    } = extractPosthogParams(body);
     const startTime = Date.now();
     const parentPromise = super.create(openAIParams, options);
     if (openAIParams.stream) {
@@ -864,8 +885,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',
@@ -876,15 +896,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',
@@ -899,8 +917,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
                   outputTokens: 0
                 },
                 isError: true,
-                error: JSON.stringify(error),
-                captureImmediate: posthogCaptureImmediate
+                error: JSON.stringify(error)
               });
             }
           })();
@@ -915,8 +932,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',
@@ -934,8 +950,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;
@@ -943,8 +958,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',
@@ -959,8 +973,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
             outputTokens: 0
           },
           isError: true,
-          error: JSON.stringify(error),
-          captureImmediate: posthogCaptureImmediate
+          error: JSON.stringify(error)
         });
         throw error;
       });
@@ -969,12 +982,9 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
   }
   parse(body, options) {
     const {
-      posthogDistinctId,
-      posthogTraceId,
-      posthogCaptureImmediate,
-      ...openAIParams
-    } = body;
-    const traceId = posthogTraceId ?? v4();
+      providerParams: openAIParams,
+      posthogParams
+    } = extractPosthogParams(body);
     const startTime = Date.now();
     // Create a temporary instance that bypasses our wrapped create method
     const originalCreate = super.create.bind(this);
@@ -987,8 +997,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',
@@ -1003,16 +1012,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',
@@ -1027,8 +1034,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
             outputTokens: 0
           },
           isError: true,
-          error: JSON.stringify(error),
-          captureImmediate: posthogCaptureImmediate
+          error: JSON.stringify(error)
         });
         throw error;
       });
@@ -1047,25 +1053,20 @@ let WrappedEmbeddings$1 = class WrappedEmbeddings extends Embeddings {
   }
   create(body, options) {
     const {
-      posthogDistinctId,
-      posthogTraceId,
-      posthogPrivacyMode = false,
-      posthogCaptureImmediate,
-      ...openAIParams
-    } = body;
-    const traceId = posthogTraceId ?? v4();
+      providerParams: 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,
@@ -1074,8 +1075,7 @@ let WrappedEmbeddings$1 = class WrappedEmbeddings extends Embeddings {
         httpStatus: 200,
         usage: {
           inputTokens: result.usage?.prompt_tokens ?? 0
-        },
-        captureImmediate: posthogCaptureImmediate
+        }
       });
       return result;
     }, async error => {
@@ -1083,11 +1083,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,
@@ -1098,8 +1097,7 @@ let WrappedEmbeddings$1 = class WrappedEmbeddings extends Embeddings {
           inputTokens: 0
         },
         isError: true,
-        error: JSON.stringify(error),
-        captureImmediate: posthogCaptureImmediate
+        error: JSON.stringify(error)
       });
       throw error;
     });
@@ -1134,12 +1132,9 @@ class WrappedCompletions extends AzureOpenAI.Chat.Completions {
   // --- Implementation Signature
   create(body, options) {
     const {
-      posthogDistinctId,
-      posthogTraceId,
-      posthogCaptureImmediate,
-      ...openAIParams
-    } = body;
-    const traceId = posthogTraceId ?? v4();
+      providerParams: openAIParams,
+      posthogParams
+    } = extractPosthogParams(body);
     const startTime = Date.now();
     const parentPromise = super.create(openAIParams, options);
     if (openAIParams.stream) {
@@ -1237,8 +1232,7 @@ class WrappedCompletions extends 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,
@@ -1247,15 +1241,13 @@ class WrappedCompletions extends 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,
@@ -1269,8 +1261,7 @@ class WrappedCompletions extends AzureOpenAI.Chat.Completions {
                   outputTokens: 0
                 },
                 isError: true,
-                error: JSON.stringify(error),
-                captureImmediate: posthogCaptureImmediate
+                error: JSON.stringify(error)
               });
             }
           })();
@@ -1285,8 +1276,7 @@ class WrappedCompletions extends 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,
@@ -1300,8 +1290,7 @@ class WrappedCompletions extends 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;
@@ -1309,8 +1298,7 @@ class WrappedCompletions extends 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,
@@ -1324,8 +1312,7 @@ class WrappedCompletions extends AzureOpenAI.Chat.Completions {
             outputTokens: 0
           },
           isError: true,
-          error: JSON.stringify(error),
-          captureImmediate: posthogCaptureImmediate
+          error: JSON.stringify(error)
         });
         throw error;
       });
@@ -1342,12 +1329,9 @@ class WrappedResponses extends AzureOpenAI.Responses {
   // --- Implementation Signature
   create(body, options) {
     const {
-      posthogDistinctId,
-      posthogTraceId,
-      posthogCaptureImmediate,
-      ...openAIParams
-    } = body;
-    const traceId = posthogTraceId ?? v4();
+      providerParams: openAIParams,
+      posthogParams
+    } = extractPosthogParams(body);
     const startTime = Date.now();
     const parentPromise = super.create(openAIParams, options);
     if (openAIParams.stream) {
@@ -1377,8 +1361,7 @@ class WrappedResponses extends 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',
@@ -1388,15 +1371,13 @@ class WrappedResponses extends 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',
@@ -1411,8 +1392,7 @@ class WrappedResponses extends AzureOpenAI.Responses {
                   outputTokens: 0
                 },
                 isError: true,
-                error: JSON.stringify(error),
-                captureImmediate: posthogCaptureImmediate
+                error: JSON.stringify(error)
               });
             }
           })();
@@ -1426,8 +1406,7 @@ class WrappedResponses extends 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',
@@ -1442,8 +1421,7 @@ class WrappedResponses extends 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;
@@ -1451,8 +1429,7 @@ class WrappedResponses extends 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',
@@ -1467,8 +1444,7 @@ class WrappedResponses extends AzureOpenAI.Responses {
             outputTokens: 0
           },
           isError: true,
-          error: JSON.stringify(error),
-          captureImmediate: posthogCaptureImmediate
+          error: JSON.stringify(error)
         });
         throw error;
       });
@@ -1477,20 +1453,16 @@ class WrappedResponses extends AzureOpenAI.Responses {
   }
   parse(body, options) {
     const {
-      posthogDistinctId,
-      posthogTraceId,
-      posthogCaptureImmediate,
-      ...openAIParams
-    } = body;
-    const traceId = posthogTraceId ?? v4();
+      providerParams: 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',
@@ -1505,15 +1477,13 @@ class WrappedResponses extends 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',
@@ -1528,8 +1498,7 @@ class WrappedResponses extends AzureOpenAI.Responses {
           outputTokens: 0
         },
         isError: true,
-        error: JSON.stringify(error),
-        captureImmediate: posthogCaptureImmediate
+        error: JSON.stringify(error)
       });
       throw error;
     });
@@ -1544,13 +1513,9 @@ class WrappedEmbeddings extends AzureOpenAI.Embeddings {
   }
   create(body, options) {
     const {
-      posthogDistinctId,
-      posthogTraceId,
-      posthogPrivacyMode = false,
-      posthogCaptureImmediate,
-      ...openAIParams
-    } = body;
-    const traceId = posthogTraceId ?? v4();
+      providerParams: openAIParams,
+      posthogParams
+    } = extractPosthogParams(body);
     const startTime = Date.now();
     const parentPromise = super.create(openAIParams, options);
     const wrappedPromise = parentPromise.then(async result => {
@@ -1558,11 +1523,10 @@ class WrappedEmbeddings extends 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,
@@ -1571,8 +1535,7 @@ class WrappedEmbeddings extends AzureOpenAI.Embeddings {
         httpStatus: 200,
         usage: {
           inputTokens: result.usage?.prompt_tokens ?? 0
-        },
-        captureImmediate: posthogCaptureImmediate
+        }
       });
       return result;
     }, async error => {
@@ -1580,11 +1543,10 @@ class WrappedEmbeddings extends 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,
@@ -1594,8 +1556,7 @@ class WrappedEmbeddings extends AzureOpenAI.Embeddings {
           inputTokens: 0
         },
         isError: true,
-        error: JSON.stringify(error),
-        captureImmediate: posthogCaptureImmediate
+        error: JSON.stringify(error)
       });
       throw error;
     });
@@ -2068,12 +2029,9 @@ class WrappedMessages extends AnthropicOriginal.Messages {
   }
   create(body, options) {
     const {
-      posthogDistinctId,
-      posthogTraceId,
-      posthogCaptureImmediate,
-      ...anthropicParams
-    } = body;
-    const traceId = posthogTraceId ?? v4();
+      providerParams: anthropicParams,
+      posthogParams
+    } = extractPosthogParams(body);
     const startTime = Date.now();
     const parentPromise = super.create(anthropicParams, options);
     if (anthropicParams.stream) {
@@ -2183,8 +2141,7 @@ class WrappedMessages extends AnthropicOriginal.Messages {
               }];
               await sendEventToPosthog({
                 client: this.phClient,
-                distinctId: posthogDistinctId,
-                traceId,
+                ...posthogParams,
                 model: anthropicParams.model,
                 provider: 'anthropic',
                 input: sanitizeAnthropic(mergeSystemPrompt(anthropicParams, 'anthropic')),
@@ -2194,15 +2151,13 @@ class WrappedMessages extends AnthropicOriginal.Messages {
                 params: body,
                 httpStatus: 200,
                 usage,
-                tools: availableTools,
-                captureImmediate: posthogCaptureImmediate
+                tools: availableTools
               });
             } catch (error) {
               // error handling
               await sendEventToPosthog({
                 client: this.phClient,
-                distinctId: posthogDistinctId,
-                traceId,
+                ...posthogParams,
                 model: anthropicParams.model,
                 provider: 'anthropic',
                 input: sanitizeAnthropic(mergeSystemPrompt(anthropicParams)),
@@ -2216,8 +2171,7 @@ class WrappedMessages extends AnthropicOriginal.Messages {
                   outputTokens: 0
                 },
                 isError: true,
-                error: JSON.stringify(error),
-                captureImmediate: posthogCaptureImmediate
+                error: JSON.stringify(error)
               });
             }
           })();
@@ -2233,8 +2187,7 @@ class WrappedMessages extends AnthropicOriginal.Messages {
           const availableTools = extractAvailableToolCalls('anthropic', anthropicParams);
           await sendEventToPosthog({
             client: this.phClient,
-            distinctId: posthogDistinctId,
-            traceId,
+            ...posthogParams,
             model: anthropicParams.model,
             provider: 'anthropic',
             input: sanitizeAnthropic(mergeSystemPrompt(anthropicParams)),
@@ -2249,16 +2202,14 @@ class WrappedMessages extends AnthropicOriginal.Messages {
               cacheCreationInputTokens: result.usage.cache_creation_input_tokens ?? 0,
               cacheReadInputTokens: result.usage.cache_read_input_tokens ?? 0
             },
-            tools: availableTools,
-            captureImmediate: posthogCaptureImmediate
+            tools: availableTools
           });
         }
         return result;
       }, async error => {
         await sendEventToPosthog({
           client: this.phClient,
-          distinctId: posthogDistinctId,
-          traceId,
+          ...posthogParams,
           model: anthropicParams.model,
           provider: 'anthropic',
           input: sanitizeAnthropic(mergeSystemPrompt(anthropicParams)),
@@ -2272,8 +2223,7 @@ class WrappedMessages extends AnthropicOriginal.Messages {
             outputTokens: 0
           },
           isError: true,
-          error: JSON.stringify(error),
-          captureImmediate: posthogCaptureImmediate
+          error: JSON.stringify(error)
         });
         throw error;
       });
@@ -2300,12 +2250,9 @@ class WrappedModels {
   }
   async generateContent(params) {
     const {
-      posthogDistinctId,
-      posthogTraceId,
-      posthogCaptureImmediate,
-      ...geminiParams
-    } = params;
-    const traceId = posthogTraceId ?? v4();
+      providerParams: geminiParams,
+      posthogParams
+    } = extractPosthogParams(params);
     const startTime = Date.now();
     try {
       const response = await this.client.models.generateContent(geminiParams);
@@ -2314,8 +2261,7 @@ class WrappedModels {
       const metadata = response.usageMetadata;
       await sendEventToPosthog({
         client: this.phClient,
-        distinctId: posthogDistinctId,
-        traceId,
+        ...posthogParams,
         model: geminiParams.model,
         provider: 'gemini',
         input: this.formatInputForPostHog(geminiParams.contents),
@@ -2330,16 +2276,14 @@ class WrappedModels {
           reasoningTokens: metadata?.thoughtsTokenCount ?? 0,
           cacheReadInputTokens: metadata?.cachedContentTokenCount ?? 0
         },
-        tools: availableTools,
-        captureImmediate: posthogCaptureImmediate
+        tools: availableTools
       });
       return response;
     } catch (error) {
       const latency = (Date.now() - startTime) / 1000;
       await sendEventToPosthog({
         client: this.phClient,
-        distinctId: posthogDistinctId,
-        traceId,
+        ...posthogParams,
         model: geminiParams.model,
         provider: 'gemini',
         input: this.formatInputForPostHog(geminiParams.contents),
@@ -2353,20 +2297,16 @@ class WrappedModels {
           outputTokens: 0
         },
         isError: true,
-        error: JSON.stringify(error),
-        captureImmediate: posthogCaptureImmediate
+        error: JSON.stringify(error)
       });
       throw error;
     }
   }
   async *generateContentStream(params) {
     const {
-      posthogDistinctId,
-      posthogTraceId,
-      posthogCaptureImmediate,
-      ...geminiParams
-    } = params;
-    const traceId = posthogTraceId ?? v4();
+      providerParams: geminiParams,
+      posthogParams
+    } = extractPosthogParams(params);
     const startTime = Date.now();
     const accumulatedContent = [];
     let usage = {
@@ -2438,8 +2378,7 @@ class WrappedModels {
       }] : [];
       await sendEventToPosthog({
         client: this.phClient,
-        distinctId: posthogDistinctId,
-        traceId,
+        ...posthogParams,
         model: geminiParams.model,
         provider: 'gemini',
         input: this.formatInputForPostHog(geminiParams.contents),
@@ -2449,15 +2388,13 @@ class WrappedModels {
         params: params,
         httpStatus: 200,
         usage,
-        tools: availableTools,
-        captureImmediate: posthogCaptureImmediate
+        tools: availableTools
       });
     } catch (error) {
       const latency = (Date.now() - startTime) / 1000;
       await sendEventToPosthog({
         client: this.phClient,
-        distinctId: posthogDistinctId,
-        traceId,
+        ...posthogParams,
         model: geminiParams.model,
         provider: 'gemini',
         input: this.formatInputForPostHog(geminiParams.contents),
@@ -2471,8 +2408,7 @@ class WrappedModels {
           outputTokens: 0
         },
         isError: true,
-        error: JSON.stringify(error),
-        captureImmediate: posthogCaptureImmediate
+        error: JSON.stringify(error)
       });
       throw error;
     }