@posthog/ai 7.2.2 → 7.3.1

package/dist/index.cjs

@@ -3,7 +3,6 @@
 var openai = require('openai');
 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 +25,7 @@ function _interopNamespaceDefault(e) {
 
 var uuid__namespace = /*#__PURE__*/_interopNamespaceDefault(uuid);
 
-var version = "7.2.2";
+var version = "7.3.1";
 
 // Type guards for safer type checking
 const isString = value => {
@@ -879,6 +878,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
             try {
               const contentBlocks = [];
               let accumulatedContent = '';
+              let modelFromResponse;
               let usage = {
                 inputTokens: 0,
                 outputTokens: 0,
@@ -887,6 +887,10 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
               // Map to track in-progress tool calls
               const toolCallsInProgress = new Map();
               for await (const chunk of stream1) {
+                // Extract model from chunk (Chat Completions chunks have model field)
+                if (!modelFromResponse && chunk.model) {
+                  modelFromResponse = chunk.model;
+                }
                 const choice = chunk?.choices?.[0];
                 const chunkWebSearchCount = calculateWebSearchCount(chunk);
                 if (chunkWebSearchCount > 0 && chunkWebSearchCount > (usage.webSearchCount ?? 0)) {
@@ -974,7 +978,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
               await sendEventToPosthog({
                 client: this.phClient,
                 ...posthogParams,
-                model: openAIParams.model,
+                model: openAIParams.model ?? modelFromResponse,
                 provider: 'openai',
                 input: sanitizeOpenAI(openAIParams.messages),
                 output: formattedOutput,
@@ -1027,7 +1031,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
           await sendEventToPosthog({
             client: this.phClient,
             ...posthogParams,
-            model: openAIParams.model,
+            model: openAIParams.model ?? result.model,
             provider: 'openai',
             input: sanitizeOpenAI(openAIParams.messages),
             output: formattedOutput,
@@ -1051,7 +1055,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
         await sendEventToPosthog({
           client: this.phClient,
           ...posthogParams,
-          model: String(openAIParams.model ?? ''),
+          model: openAIParams.model,
           provider: 'openai',
           input: sanitizeOpenAI(openAIParams.messages),
           output: [],
@@ -1093,6 +1097,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
           (async () => {
             try {
               let finalContent = [];
+              let modelFromResponse;
               let usage = {
                 inputTokens: 0,
                 outputTokens: 0,
@@ -1100,6 +1105,10 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
               };
               for await (const chunk of stream1) {
                 if ('response' in chunk && chunk.response) {
+                  // Extract model from response object in chunk (for stored prompts)
+                  if (!modelFromResponse && chunk.response.model) {
+                    modelFromResponse = chunk.response.model;
+                  }
                   const chunkWebSearchCount = calculateWebSearchCount(chunk.response);
                   if (chunkWebSearchCount > 0 && chunkWebSearchCount > (usage.webSearchCount ?? 0)) {
                     usage.webSearchCount = chunkWebSearchCount;
@@ -1123,8 +1132,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
               await sendEventToPosthog({
                 client: this.phClient,
                 ...posthogParams,
-                //@ts-expect-error
-                model: openAIParams.model,
+                model: openAIParams.model ?? modelFromResponse,
                 provider: 'openai',
                 input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
                 output: finalContent,
@@ -1146,7 +1154,6 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
               await sendEventToPosthog({
                 client: this.phClient,
                 ...posthogParams,
-                //@ts-expect-error
                 model: openAIParams.model,
                 provider: 'openai',
                 input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
@@ -1179,8 +1186,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
           await sendEventToPosthog({
             client: this.phClient,
             ...posthogParams,
-            //@ts-expect-error
-            model: openAIParams.model,
+            model: openAIParams.model ?? result.model,
             provider: 'openai',
             input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
             output: formattedOutput,
@@ -1204,7 +1210,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
         await sendEventToPosthog({
           client: this.phClient,
           ...posthogParams,
-          model: String(openAIParams.model ?? ''),
+          model: openAIParams.model,
           provider: 'openai',
           input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
           output: [],
@@ -1241,7 +1247,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
         await sendEventToPosthog({
           client: this.phClient,
           ...posthogParams,
-          model: String(openAIParams.model ?? ''),
+          model: openAIParams.model ?? result.model,
           provider: 'openai',
           input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
           output: result.output,
@@ -1262,7 +1268,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
         await sendEventToPosthog({
           client: this.phClient,
           ...posthogParams,
-          model: String(openAIParams.model ?? ''),
+          model: openAIParams.model,
           provider: 'openai',
           input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
           output: [],
@@ -1437,7 +1443,7 @@ class WrappedTranscriptions extends Transcriptions {
           await sendEventToPosthog({
             client: this.phClient,
             ...posthogParams,
-            model: String(openAIParams.model ?? ''),
+            model: openAIParams.model,
             provider: 'openai',
             input: openAIParams.prompt,
             output: result.text,
@@ -1457,7 +1463,7 @@ class WrappedTranscriptions extends Transcriptions {
         await sendEventToPosthog({
           client: this.phClient,
           ...posthogParams,
-          model: String(openAIParams.model ?? ''),
+          model: openAIParams.model,
           provider: 'openai',
           input: openAIParams.prompt,
           output: [],
@@ -1519,6 +1525,7 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
             try {
               const contentBlocks = [];
               let accumulatedContent = '';
+              let modelFromResponse;
               let usage = {
                 inputTokens: 0,
                 outputTokens: 0
@@ -1526,6 +1533,10 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
               // Map to track in-progress tool calls
               const toolCallsInProgress = new Map();
               for await (const chunk of stream1) {
+                // Extract model from response if not in params
+                if (!modelFromResponse && chunk.model) {
+                  modelFromResponse = chunk.model;
+                }
                 const choice = chunk?.choices?.[0];
                 // Handle text content
                 const deltaContent = choice?.delta?.content;
@@ -1607,7 +1618,7 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
               await sendEventToPosthog({
                 client: this.phClient,
                 ...posthogParams,
-                model: openAIParams.model,
+                model: openAIParams.model ?? modelFromResponse,
                 provider: 'azure',
                 input: sanitizeOpenAI(openAIParams.messages),
                 output: formattedOutput,
@@ -1651,7 +1662,7 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
           await sendEventToPosthog({
             client: this.phClient,
             ...posthogParams,
-            model: openAIParams.model,
+            model: openAIParams.model ?? result.model,
             provider: 'azure',
             input: openAIParams.messages,
             output: formatResponseOpenAI(result),
@@ -1715,11 +1726,18 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
           (async () => {
             try {
               let finalContent = [];
+              let modelFromResponse;
               let usage = {
                 inputTokens: 0,
                 outputTokens: 0
               };
               for await (const chunk of stream1) {
+                if ('response' in chunk && chunk.response) {
+                  // Extract model from response if not in params (for stored prompts)
+                  if (!modelFromResponse && chunk.response.model) {
+                    modelFromResponse = chunk.response.model;
+                  }
+                }
                 if (chunk.type === 'response.completed' && 'response' in chunk && chunk.response?.output && chunk.response.output.length > 0) {
                   finalContent = chunk.response.output;
                 }
@@ -1736,10 +1754,9 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
               await sendEventToPosthog({
                 client: this.phClient,
                 ...posthogParams,
-                //@ts-expect-error
-                model: openAIParams.model,
+                model: openAIParams.model ?? modelFromResponse,
                 provider: 'azure',
-                input: openAIParams.input,
+                input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
                 output: finalContent,
                 latency,
                 baseURL: this.baseURL,
@@ -1752,10 +1769,9 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
               await sendEventToPosthog({
                 client: this.phClient,
                 ...posthogParams,
-                //@ts-expect-error
                 model: openAIParams.model,
                 provider: 'azure',
-                input: openAIParams.input,
+                input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
                 output: [],
                 latency: 0,
                 baseURL: this.baseURL,
@@ -1781,10 +1797,9 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
           await sendEventToPosthog({
             client: this.phClient,
             ...posthogParams,
-            //@ts-expect-error
-            model: openAIParams.model,
+            model: openAIParams.model ?? result.model,
             provider: 'azure',
-            input: openAIParams.input,
+            input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
             output: result.output,
             latency,
             baseURL: this.baseURL,
@@ -1804,10 +1819,9 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
         await sendEventToPosthog({
           client: this.phClient,
           ...posthogParams,
-          //@ts-expect-error
           model: openAIParams.model,
           provider: 'azure',
-          input: openAIParams.input,
+          input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
           output: [],
           latency: 0,
           baseURL: this.baseURL,
@@ -1837,9 +1851,9 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
       await sendEventToPosthog({
         client: this.phClient,
         ...posthogParams,
-        model: String(openAIParams.model ?? ''),
+        model: openAIParams.model ?? result.model,
         provider: 'azure',
-        input: openAIParams.input,
+        input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
         output: result.output,
         latency,
         baseURL: this.baseURL,
@@ -1857,9 +1871,9 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
       await sendEventToPosthog({
         client: this.phClient,
         ...posthogParams,
-        model: String(openAIParams.model ?? ''),
+        model: openAIParams.model,
         provider: 'azure',
-        input: openAIParams.input,
+        input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
         output: [],
         latency: 0,
         baseURL: this.baseURL,
@@ -2132,67 +2146,117 @@ const extractProvider = model => {
   const providerName = provider.split('.')[0];
   return providerName;
 };
-const createInstrumentationMiddleware = (phClient, model, options) => {
-  const middleware = {
-    wrapGenerate: async ({
-      doGenerate,
-      params
-    }) => {
+// Extract web search count from provider metadata (works for both V2 and V3)
+const extractWebSearchCount = (providerMetadata, usage) => {
+  // Try Anthropic-specific extraction
+  if (providerMetadata && typeof providerMetadata === 'object' && 'anthropic' in providerMetadata && providerMetadata.anthropic && typeof providerMetadata.anthropic === 'object' && 'server_tool_use' in providerMetadata.anthropic) {
+    const serverToolUse = providerMetadata.anthropic.server_tool_use;
+    if (serverToolUse && typeof serverToolUse === 'object' && 'web_search_requests' in serverToolUse && typeof serverToolUse.web_search_requests === 'number') {
+      return serverToolUse.web_search_requests;
+    }
+  }
+  // Fall back to generic calculation
+  return calculateWebSearchCount({
+    usage,
+    providerMetadata
+  });
+};
+// Extract additional token values from provider metadata
+const extractAdditionalTokenValues = providerMetadata => {
+  if (providerMetadata && typeof providerMetadata === 'object' && 'anthropic' in providerMetadata && providerMetadata.anthropic && typeof providerMetadata.anthropic === 'object' && 'cacheCreationInputTokens' in providerMetadata.anthropic) {
+    return {
+      cacheCreationInputTokens: providerMetadata.anthropic.cacheCreationInputTokens
+    };
+  }
+  return {};
+};
+// Helper to extract numeric token value from V2 (number) or V3 (object with .total) usage formats
+const extractTokenCount = value => {
+  if (typeof value === 'number') {
+    return value;
+  }
+  if (value && typeof value === 'object' && 'total' in value && typeof value.total === 'number') {
+    return value.total;
+  }
+  return undefined;
+};
+// Helper to extract reasoning tokens from V2 (usage.reasoningTokens) or V3 (usage.outputTokens.reasoning)
+const extractReasoningTokens = usage => {
+  // V2 style: top-level reasoningTokens
+  if ('reasoningTokens' in usage) {
+    return usage.reasoningTokens;
+  }
+  // V3 style: nested in outputTokens.reasoning
+  if ('outputTokens' in usage && usage.outputTokens && typeof usage.outputTokens === 'object' && 'reasoning' in usage.outputTokens) {
+    return usage.outputTokens.reasoning;
+  }
+  return undefined;
+};
+// Helper to extract cached input tokens from V2 (usage.cachedInputTokens) or V3 (usage.inputTokens.cacheRead)
+const extractCacheReadTokens = usage => {
+  // V2 style: top-level cachedInputTokens
+  if ('cachedInputTokens' in usage) {
+    return usage.cachedInputTokens;
+  }
+  // V3 style: nested in inputTokens.cacheRead
+  if ('inputTokens' in usage && usage.inputTokens && typeof usage.inputTokens === 'object' && 'cacheRead' in usage.inputTokens) {
+    return usage.inputTokens.cacheRead;
+  }
+  return undefined;
+};
+/**
+ * Wraps a Vercel AI SDK language model (V2 or V3) with PostHog tracing.
+ * Automatically detects the model version and applies appropriate instrumentation.
+ */
+const wrapVercelLanguageModel = (model, phClient, options) => {
+  const traceId = options.posthogTraceId ?? uuid.v4();
+  const mergedOptions = {
+    ...options,
+    posthogTraceId: traceId,
+    posthogDistinctId: options.posthogDistinctId,
+    posthogProperties: {
+      ...options.posthogProperties,
+      $ai_framework: 'vercel',
+      $ai_framework_version: model.specificationVersion === 'v3' ? '6' : '5'
+    }
+  };
+  // Create wrapped model that preserves the original type
+  const wrappedModel = {
+    ...model,
+    doGenerate: async params => {
       const startTime = Date.now();
       const mergedParams = {
-        ...options,
-        ...mapVercelParams(params),
-        posthogProperties: {
-          ...options.posthogProperties,
-          $ai_framework: 'vercel'
-        }
+        ...mergedOptions,
+        ...mapVercelParams(params)
       };
       const availableTools = extractAvailableToolCalls('vercel', params);
       try {
-        const result = await doGenerate();
-        const modelId = options.posthogModelOverride ?? (result.response?.modelId ? result.response.modelId : model.modelId);
-        const provider = options.posthogProviderOverride ?? extractProvider(model);
+        const result = await model.doGenerate(params);
+        const modelId = mergedOptions.posthogModelOverride ?? (result.response?.modelId ? result.response.modelId : model.modelId);
+        const provider = mergedOptions.posthogProviderOverride ?? extractProvider(model);
         const baseURL = ''; // cannot currently get baseURL from vercel
         const content = mapVercelOutput(result.content);
         const latency = (Date.now() - startTime) / 1000;
         const providerMetadata = result.providerMetadata;
-        const additionalTokenValues = {
-          ...(providerMetadata?.anthropic ? {
-            cacheCreationInputTokens: providerMetadata.anthropic.cacheCreationInputTokens
-          } : {})
-        };
-        // Calculate web search count based on provider
-        let webSearchCount = 0;
-        if (providerMetadata?.anthropic && typeof providerMetadata.anthropic === 'object' && 'server_tool_use' in providerMetadata.anthropic) {
-          // Anthropic-specific extraction
-          const serverToolUse = providerMetadata.anthropic.server_tool_use;
-          if (serverToolUse && typeof serverToolUse === 'object' && 'web_search_requests' in serverToolUse && typeof serverToolUse.web_search_requests === 'number') {
-            webSearchCount = serverToolUse.web_search_requests;
-          }
-        } else {
-          // For other providers through Vercel, pass available metadata to helper
-          // Note: Vercel abstracts provider responses, so we may not have access to
-          // raw citations/annotations unless Vercel exposes them in usage/metadata
-          webSearchCount = calculateWebSearchCount({
-            usage: result.usage,
-            providerMetadata: providerMetadata
-          });
-        }
+        const additionalTokenValues = extractAdditionalTokenValues(providerMetadata);
+        const webSearchCount = extractWebSearchCount(providerMetadata, result.usage);
+        // V2 usage has simple numbers, V3 has objects with .total - normalize both
+        const usageObj = result.usage;
         const usage = {
-          inputTokens: result.usage.inputTokens,
-          outputTokens: result.usage.outputTokens,
-          reasoningTokens: result.usage.reasoningTokens,
-          cacheReadInputTokens: result.usage.cachedInputTokens,
+          inputTokens: extractTokenCount(result.usage.inputTokens),
+          outputTokens: extractTokenCount(result.usage.outputTokens),
+          reasoningTokens: extractReasoningTokens(usageObj),
+          cacheReadInputTokens: extractCacheReadTokens(usageObj),
           webSearchCount,
           ...additionalTokenValues
         };
         await sendEventToPosthog({
           client: phClient,
-          distinctId: options.posthogDistinctId,
-          traceId: options.posthogTraceId ?? uuid.v4(),
+          distinctId: mergedOptions.posthogDistinctId,
+          traceId: mergedOptions.posthogTraceId ?? uuid.v4(),
           model: modelId,
           provider: provider,
-          input: options.posthogPrivacyMode ? '' : mapVercelPrompt(params.prompt),
+          input: mergedOptions.posthogPrivacyMode ? '' : mapVercelPrompt(params.prompt),
           output: content,
           latency,
           baseURL,
@@ -2200,18 +2264,18 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
           httpStatus: 200,
           usage,
           tools: availableTools,
-          captureImmediate: options.posthogCaptureImmediate
+          captureImmediate: mergedOptions.posthogCaptureImmediate
         });
         return result;
       } catch (error) {
         const modelId = model.modelId;
         await sendEventToPosthog({
           client: phClient,
-          distinctId: options.posthogDistinctId,
-          traceId: options.posthogTraceId ?? uuid.v4(),
+          distinctId: mergedOptions.posthogDistinctId,
+          traceId: mergedOptions.posthogTraceId ?? uuid.v4(),
           model: modelId,
           provider: model.provider,
-          input: options.posthogPrivacyMode ? '' : mapVercelPrompt(params.prompt),
+          input: mergedOptions.posthogPrivacyMode ? '' : mapVercelPrompt(params.prompt),
           output: [],
           latency: 0,
           baseURL: '',
@@ -2224,30 +2288,23 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
           isError: true,
           error: truncate(JSON.stringify(error)),
           tools: availableTools,
-          captureImmediate: options.posthogCaptureImmediate
+          captureImmediate: mergedOptions.posthogCaptureImmediate
         });
         throw error;
       }
     },
-    wrapStream: async ({
-      doStream,
-      params
-    }) => {
+    doStream: async params => {
       const startTime = Date.now();
       let generatedText = '';
       let reasoningText = '';
       let usage = {};
       let providerMetadata = undefined;
       const mergedParams = {
-        ...options,
-        ...mapVercelParams(params),
-        posthogProperties: {
-          ...options.posthogProperties,
-          $ai_framework: 'vercel'
-        }
+        ...mergedOptions,
+        ...mapVercelParams(params)
       };
-      const modelId = options.posthogModelOverride ?? model.modelId;
-      const provider = options.posthogProviderOverride ?? extractProvider(model);
+      const modelId = mergedOptions.posthogModelOverride ?? model.modelId;
+      const provider = mergedOptions.posthogProviderOverride ?? extractProvider(model);
       const availableTools = extractAvailableToolCalls('vercel', params);
       const baseURL = ''; // cannot currently get baseURL from vercel
       // Map to track in-progress tool calls
@@ -2256,15 +2313,15 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
         const {
           stream,
           ...rest
-        } = await doStream();
+        } = await model.doStream(params);
         const transformStream = new TransformStream({
           transform(chunk, controller) {
-            // Handle new v5 streaming patterns
+            // Handle streaming patterns - compatible with both V2 and V3
             if (chunk.type === 'text-delta') {
               generatedText += chunk.delta;
             }
             if (chunk.type === 'reasoning-delta') {
-              reasoningText += chunk.delta; // New in v5
+              reasoningText += chunk.delta;
             }
             // Handle tool call chunks
             if (chunk.type === 'tool-input-start') {
@@ -2284,7 +2341,6 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
             }
             if (chunk.type === 'tool-input-end') {
               // Tool call is complete, keep it in the map for final processing
-              // Nothing specific to do here, the tool call is already complete
             }
             if (chunk.type === 'tool-call') {
               // Direct tool call chunk (complete tool call)
@@ -2296,14 +2352,13 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
             }
             if (chunk.type === 'finish') {
               providerMetadata = chunk.providerMetadata;
-              const additionalTokenValues = providerMetadata && typeof providerMetadata === 'object' && 'anthropic' in providerMetadata && providerMetadata.anthropic && typeof providerMetadata.anthropic === 'object' && 'cacheCreationInputTokens' in providerMetadata.anthropic ? {
-                cacheCreationInputTokens: providerMetadata.anthropic.cacheCreationInputTokens
-              } : {};
+              const additionalTokenValues = extractAdditionalTokenValues(providerMetadata);
+              const chunkUsage = chunk.usage || {};
               usage = {
-                inputTokens: chunk.usage?.inputTokens,
-                outputTokens: chunk.usage?.outputTokens,
-                reasoningTokens: chunk.usage?.reasoningTokens,
-                cacheReadInputTokens: chunk.usage?.cachedInputTokens,
+                inputTokens: extractTokenCount(chunk.usage?.inputTokens),
+                outputTokens: extractTokenCount(chunk.usage?.outputTokens),
+                reasoningTokens: extractReasoningTokens(chunkUsage),
+                cacheReadInputTokens: extractCacheReadTokens(chunkUsage),
                 ...additionalTokenValues
               };
             }
@@ -2343,23 +2398,7 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
               role: 'assistant',
               content: content.length === 1 && content[0].type === 'text' ? content[0].text : content
             }] : [];
-            // Calculate web search count based on provider
-            let webSearchCount = 0;
-            if (providerMetadata && typeof providerMetadata === 'object' && 'anthropic' in providerMetadata && providerMetadata.anthropic && typeof providerMetadata.anthropic === 'object' && 'server_tool_use' in providerMetadata.anthropic) {
-              // Anthropic-specific extraction
-              const serverToolUse = providerMetadata.anthropic.server_tool_use;
-              if (serverToolUse && typeof serverToolUse === 'object' && 'web_search_requests' in serverToolUse && typeof serverToolUse.web_search_requests === 'number') {
-                webSearchCount = serverToolUse.web_search_requests;
-              }
-            } else {
-              // For other providers through Vercel, pass available metadata to helper
-              // Note: Vercel abstracts provider responses, so we may not have access to
-              // raw citations/annotations unless Vercel exposes them in usage/metadata
-              webSearchCount = calculateWebSearchCount({
-                usage: usage,
-                providerMetadata: providerMetadata
-              });
-            }
+            const webSearchCount = extractWebSearchCount(providerMetadata, usage);
             // Update usage with web search count
             const finalUsage = {
               ...usage,
@@ -2367,11 +2406,11 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
             };
             await sendEventToPosthog({
               client: phClient,
-              distinctId: options.posthogDistinctId,
-              traceId: options.posthogTraceId ?? uuid.v4(),
+              distinctId: mergedOptions.posthogDistinctId,
+              traceId: mergedOptions.posthogTraceId ?? uuid.v4(),
               model: modelId,
               provider: provider,
-              input: options.posthogPrivacyMode ? '' : mapVercelPrompt(params.prompt),
+              input: mergedOptions.posthogPrivacyMode ? '' : mapVercelPrompt(params.prompt),
               output: output,
               latency,
               baseURL,
@@ -2379,7 +2418,7 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
               httpStatus: 200,
               usage: finalUsage,
               tools: availableTools,
-              captureImmediate: options.posthogCaptureImmediate
+              captureImmediate: mergedOptions.posthogCaptureImmediate
             });
           }
         });
@@ -2390,11 +2429,11 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
       } catch (error) {
         await sendEventToPosthog({
           client: phClient,
-          distinctId: options.posthogDistinctId,
-          traceId: options.posthogTraceId ?? uuid.v4(),
+          distinctId: mergedOptions.posthogDistinctId,
+          traceId: mergedOptions.posthogTraceId ?? uuid.v4(),
           model: modelId,
           provider: provider,
-          input: options.posthogPrivacyMode ? '' : mapVercelPrompt(params.prompt),
+          input: mergedOptions.posthogPrivacyMode ? '' : mapVercelPrompt(params.prompt),
           output: [],
           latency: 0,
           baseURL: '',
@@ -2407,25 +2446,12 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
           isError: true,
           error: truncate(JSON.stringify(error)),
           tools: availableTools,
-          captureImmediate: options.posthogCaptureImmediate
+          captureImmediate: mergedOptions.posthogCaptureImmediate
         });
         throw error;
       }
     }
   };
-  return middleware;
-};
-const wrapVercelLanguageModel = (model, phClient, options) => {
-  const traceId = options.posthogTraceId ?? uuid.v4();
-  const middleware = createInstrumentationMiddleware(phClient, model, {
-    ...options,
-    posthogTraceId: traceId,
-    posthogDistinctId: options.posthogDistinctId
-  });
-  const wrappedModel = ai.wrapLanguageModel({
-    model,
-    middleware
-  });
   return wrappedModel;
 };