@posthog/ai 3.3.0 → 3.3.1

package/CHANGELOG.md

@@ -1,3 +1,7 @@
+# 3.3.1 - 2025-03-13
+
+- fix: fix vercel output mapping and token caching
+
 # 3.3.0 - 2025-03-08
 
 - feat: add reasoning and cache tokens to openai and anthropic

package/lib/index.cjs.js

@@ -558,6 +558,45 @@ const mapVercelPrompt = prompt => {
     };
   });
 };
+const mapVercelOutput = result => {
+  let output = {
+    ...(result.text ? {
+      text: result.text
+    } : {}),
+    ...(result.object ? {
+      object: result.object
+    } : {}),
+    ...(result.reasoning ? {
+      reasoning: result.reasoning
+    } : {}),
+    ...(result.response ? {
+      response: result.response
+    } : {}),
+    ...(result.finishReason ? {
+      finishReason: result.finishReason
+    } : {}),
+    ...(result.usage ? {
+      usage: result.usage
+    } : {}),
+    ...(result.warnings ? {
+      warnings: result.warnings
+    } : {}),
+    ...(result.providerMetadata ? {
+      toolCalls: result.providerMetadata
+    } : {})
+  };
+  // if text and no object or reasoning, return text
+  if (output.text && !output.object && !output.reasoning) {
+    return [{
+      content: output.text,
+      role: 'assistant'
+    }];
+  }
+  return [{
+    content: JSON.stringify(output),
+    role: 'assistant'
+  }];
+};
 const extractProvider = model => {
   // vercel provider is in the format of provider.endpoint
   const provider = model.provider.toLowerCase();
@@ -581,14 +620,14 @@ 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 || JSON.stringify(result);
+        let content = mapVercelOutput(result);
         // let tools = result.toolCalls
         let providerMetadata = result.providerMetadata;
         let additionalTokenValues = {
           ...(providerMetadata?.openai?.reasoningTokens ? {
             reasoningTokens: providerMetadata.openai.reasoningTokens
           } : {}),
-          ...(providerMetadata?.openai?.cachedPromptToken ? {
+          ...(providerMetadata?.openai?.cachedPromptTokens ? {
             cacheReadInputTokens: providerMetadata.openai.cachedPromptTokens
           } : {}),
           ...(providerMetadata?.anthropic ? {
@@ -674,8 +713,8 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
               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?.openai?.cachedPromptTokens) {
+                usage.cacheReadInputTokens = chunk.providerMetadata.openai.cachedPromptTokens;
               }
               if (chunk.providerMetadata?.anthropic?.cacheReadInputTokens) {
                 usage.cacheReadInputTokens = chunk.providerMetadata.anthropic.cacheReadInputTokens;