@posthog/ai 3.2.0 → 3.3.0

package/lib/index.esm.js

@@ -1,7 +1,7 @@
 import OpenAIOrignal, { AzureOpenAI } from 'openai';
 import * as uuid from 'uuid';
 import { v4 } from 'uuid';
-import { wrapLanguageModel } from 'ai';
+import { experimental_wrapLanguageModel } from 'ai';
 import AnthropicOriginal from '@anthropic-ai/sdk';
 
 const getModelParams = params => {
@@ -73,7 +73,8 @@ const sendEventToPosthog = ({
   httpStatus = 200,
   usage = {},
   isError = false,
-  error
+  error,
+  tools
 }) => {
   if (client.capture) {
     let errorData = {};
@@ -93,6 +94,17 @@ const sendEventToPosthog = ({
         $ai_total_cost_usd: inputCostUSD + outputCostUSD
       };
     }
+    let additionalTokenValues = {
+      ...(usage.reasoningTokens ? {
+        $ai_reasoning_tokens: usage.reasoningTokens
+      } : {}),
+      ...(usage.cacheReadInputTokens ? {
+        $ai_cache_read_input_tokens: usage.cacheReadInputTokens
+      } : {}),
+      ...(usage.cacheCreationInputTokens ? {
+        $ai_cache_creation_input_tokens: usage.cacheCreationInputTokens
+      } : {})
+    };
     client.capture({
       distinctId: distinctId ?? traceId,
       event: '$ai_generation',
@@ -105,6 +117,7 @@ const sendEventToPosthog = ({
         $ai_http_status: httpStatus,
         $ai_input_tokens: usage.inputTokens ?? 0,
         $ai_output_tokens: usage.outputTokens ?? 0,
+        ...additionalTokenValues,
         $ai_latency: latency,
         $ai_trace_id: traceId,
         $ai_base_url: baseURL,
@@ -112,6 +125,9 @@ const sendEventToPosthog = ({
         ...(distinctId ? {} : {
           $process_person_profile: false
         }),
+        ...(tools ? {
+          $ai_tools: tools
+        } : {}),
         ...errorData,
         ...costOverrideData
       },
@@ -173,7 +189,9 @@ class WrappedCompletions$1 extends OpenAIOrignal.Chat.Completions {
                 if (chunk.usage) {
                   usage = {
                     inputTokens: chunk.usage.prompt_tokens ?? 0,
-                    outputTokens: chunk.usage.completion_tokens ?? 0
+                    outputTokens: chunk.usage.completion_tokens ?? 0,
+                    reasoningTokens: chunk.usage.completion_tokens_details?.reasoning_tokens ?? 0,
+                    cacheReadInputTokens: chunk.usage.prompt_tokens_details?.cached_tokens ?? 0
                   };
                 }
               }
@@ -240,7 +258,9 @@ class WrappedCompletions$1 extends OpenAIOrignal.Chat.Completions {
             httpStatus: 200,
             usage: {
               inputTokens: result.usage?.prompt_tokens ?? 0,
-              outputTokens: result.usage?.completion_tokens ?? 0
+              outputTokens: result.usage?.completion_tokens ?? 0,
+              reasoningTokens: result.usage?.completion_tokens_details?.reasoning_tokens ?? 0,
+              cacheReadInputTokens: result.usage?.prompt_tokens_details?.cached_tokens ?? 0
             }
           });
         }
@@ -329,7 +349,9 @@ class WrappedCompletions extends AzureOpenAI.Chat.Completions {
                   }
                   usage = {
                     inputTokens: chunk.usage.prompt_tokens ?? 0,
-                    outputTokens: chunk.usage.completion_tokens ?? 0
+                    outputTokens: chunk.usage.completion_tokens ?? 0,
+                    reasoningTokens: chunk.usage.completion_tokens_details?.reasoning_tokens ?? 0,
+                    cacheReadInputTokens: chunk.usage.prompt_tokens_details?.cached_tokens ?? 0
                   };
                 }
               }
@@ -401,7 +423,9 @@ class WrappedCompletions extends AzureOpenAI.Chat.Completions {
             httpStatus: 200,
             usage: {
               inputTokens: result.usage?.prompt_tokens ?? 0,
-              outputTokens: result.usage?.completion_tokens ?? 0
+              outputTokens: result.usage?.completion_tokens ?? 0,
+              reasoningTokens: result.usage?.completion_tokens_details?.reasoning_tokens ?? 0,
+              cacheReadInputTokens: result.usage?.prompt_tokens_details?.cached_tokens ?? 0
             }
           });
         }
@@ -446,16 +470,60 @@ const mapVercelParams = params => {
 };
 const mapVercelPrompt = prompt => {
   return prompt.map(p => {
-    let content = '';
+    let content = {};
     if (Array.isArray(p.content)) {
       content = p.content.map(c => {
         if (c.type === 'text') {
-          return c.text;
+          return {
+            type: 'text',
+            content: c.text
+          };
+        } else if (c.type === 'image') {
+          return {
+            type: 'image',
+            content: {
+              // if image is a url use it, or use "none supported"
+              image: c.image instanceof URL ? c.image.toString() : 'raw images not supported',
+              mimeType: c.mimeType
+            }
+          };
+        } else if (c.type === 'file') {
+          return {
+            type: 'file',
+            content: {
+              file: c.data instanceof URL ? c.data.toString() : 'raw files not supported',
+              mimeType: c.mimeType
+            }
+          };
+        } else if (c.type === 'tool-call') {
+          return {
+            type: 'tool-call',
+            content: {
+              toolCallId: c.toolCallId,
+              toolName: c.toolName,
+              args: c.args
+            }
+          };
+        } else if (c.type === 'tool-result') {
+          return {
+            type: 'tool-result',
+            content: {
+              toolCallId: c.toolCallId,
+              toolName: c.toolName,
+              result: c.result,
+              isError: c.isError
+            }
+          };
         }
-        return '';
-      }).join('');
+        return {
+          content: ''
+        };
+      });
     } else {
-      content = p.content;
+      content = {
+        type: 'text',
+        text: p.content
+      };
     }
     return {
       role: p.role,
@@ -486,11 +554,21 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
         const modelId = options.posthogModelOverride ?? (result.response?.modelId ? result.response.modelId : model.modelId);
         const provider = options.posthogProviderOverride ?? extractProvider(model);
         const baseURL = ''; // cannot currently get baseURL from vercel
-        let content = result.text;
-        if (!content) {
-          // support generate Object
-          content = result.toolCalls?.[0].args || JSON.stringify(result);
-        }
+        let content = result.text || JSON.stringify(result);
+        // let tools = result.toolCalls
+        let providerMetadata = result.providerMetadata;
+        let additionalTokenValues = {
+          ...(providerMetadata?.openai?.reasoningTokens ? {
+            reasoningTokens: providerMetadata.openai.reasoningTokens
+          } : {}),
+          ...(providerMetadata?.openai?.cachedPromptToken ? {
+            cacheReadInputTokens: providerMetadata.openai.cachedPromptTokens
+          } : {}),
+          ...(providerMetadata?.anthropic ? {
+            cacheReadInputTokens: providerMetadata.anthropic.cacheReadInputTokens,
+            cacheCreationInputTokens: providerMetadata.anthropic.cacheCreationInputTokens
+          } : {})
+        };
         sendEventToPosthog({
           client: phClient,
           distinctId: options.posthogDistinctId,
@@ -508,7 +586,8 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
           httpStatus: 200,
           usage: {
             inputTokens: result.usage.promptTokens,
-            outputTokens: result.usage.completionTokens
+            outputTokens: result.usage.completionTokens,
+            ...additionalTokenValues
           }
         });
         return result;
@@ -565,6 +644,18 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
                 inputTokens: chunk.usage?.promptTokens,
                 outputTokens: chunk.usage?.completionTokens
               };
+              if (chunk.providerMetadata?.openai?.reasoningTokens) {
+                usage.reasoningTokens = chunk.providerMetadata.openai.reasoningTokens;
+              }
+              if (chunk.providerMetadata?.openai?.cachedPromptToken) {
+                usage.cacheReadInputTokens = chunk.providerMetadata.openai.cachedPromptToken;
+              }
+              if (chunk.providerMetadata?.anthropic?.cacheReadInputTokens) {
+                usage.cacheReadInputTokens = chunk.providerMetadata.anthropic.cacheReadInputTokens;
+              }
+              if (chunk.providerMetadata?.anthropic?.cacheCreationInputTokens) {
+                usage.cacheCreationInputTokens = chunk.providerMetadata.anthropic.cacheCreationInputTokens;
+              }
             }
             controller.enqueue(chunk);
           },
@@ -626,7 +717,7 @@ const wrapVercelLanguageModel = (model, phClient, options) => {
     posthogTraceId: traceId,
     posthogDistinctId: options.posthogDistinctId ?? traceId
   });
-  const wrappedModel = wrapLanguageModel({
+  const wrappedModel = experimental_wrapLanguageModel({
     model,
     middleware
   });
@@ -667,7 +758,9 @@ class WrappedMessages extends AnthropicOriginal.Messages {
         let accumulatedContent = '';
         const usage = {
           inputTokens: 0,
-          outputTokens: 0
+          outputTokens: 0,
+          cacheCreationInputTokens: 0,
+          cacheReadInputTokens: 0
         };
         if ('tee' in value) {
           const [stream1, stream2] = value.tee();
@@ -682,6 +775,8 @@ class WrappedMessages extends AnthropicOriginal.Messages {
                 }
                 if (chunk.type == 'message_start') {
                   usage.inputTokens = chunk.message.usage.input_tokens ?? 0;
+                  usage.cacheCreationInputTokens = chunk.message.usage.cache_creation_input_tokens ?? 0;
+                  usage.cacheReadInputTokens = chunk.message.usage.cache_read_input_tokens ?? 0;
                 }
                 if ('usage' in chunk) {
                   usage.outputTokens = chunk.usage.output_tokens ?? 0;
@@ -751,7 +846,9 @@ class WrappedMessages extends AnthropicOriginal.Messages {
             httpStatus: 200,
             usage: {
               inputTokens: result.usage.input_tokens ?? 0,
-              outputTokens: result.usage.output_tokens ?? 0
+              outputTokens: result.usage.output_tokens ?? 0,
+              cacheCreationInputTokens: result.usage.cache_creation_input_tokens ?? 0,
+              cacheReadInputTokens: result.usage.cache_read_input_tokens ?? 0
             }
           });
         }
@@ -1522,6 +1619,9 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
       $ai_latency: latency,
       $ai_base_url: run.baseUrl
     };
+    if (run.tools) {
+      eventProperties['$ai_tools'] = withPrivacyMode(this.client, this.privacyMode, run.tools);
+    }
     if (output instanceof Error) {
       eventProperties['$ai_http_status'] = output.status || 500;
       eventProperties['$ai_error'] = output.toString();