@posthog/ai 8.9.0 → 8.9.2

package/dist/vercel/index.mjs

@@ -437,7 +437,7 @@ function sanitizeValues(obj) {
   return jsonSafe;
 }
 
-var version = "8.9.0";
+var version = "8.9.2";
 
 const DEFAULT_MAX_DEPTH = 3;
 const MAX_STACK_LINES = 20;
@@ -594,15 +594,22 @@ const captureAiGeneration = async (client, options) => {
       };
     }
     httpStatus = httpStatus ?? 200;
-    let costOverrideData = {};
+
+    // A configured price applies only to a count the provider reported, so a call with no
+    // reported usage sends no cost instead of asserting $0. $ai_total_cost_usd sums the sides
+    // that were priced, which makes it the cost of the known side alone when the other side
+    // went unreported: a lower bound on the true total, not an assertion of it.
+    const costOverrideData = {};
     if (options.costOverride) {
-      const inputCostUSD = (options.costOverride.inputCost ?? 0) * (usage.inputTokens ?? 0);
-      const outputCostUSD = (options.costOverride.outputCost ?? 0) * (usage.outputTokens ?? 0);
-      costOverrideData = {
-        $ai_input_cost_usd: inputCostUSD,
-        $ai_output_cost_usd: outputCostUSD,
-        $ai_total_cost_usd: inputCostUSD + outputCostUSD
-      };
+      if (usage.inputTokens !== undefined) {
+        costOverrideData.$ai_input_cost_usd = (options.costOverride.inputCost ?? 0) * usage.inputTokens;
+      }
+      if (usage.outputTokens !== undefined) {
+        costOverrideData.$ai_output_cost_usd = (options.costOverride.outputCost ?? 0) * usage.outputTokens;
+      }
+      if (Object.keys(costOverrideData).length > 0) {
+        costOverrideData.$ai_total_cost_usd = (costOverrideData.$ai_input_cost_usd ?? 0) + (costOverrideData.$ai_output_cost_usd ?? 0);
+      }
     }
 
     // The caller's own token counts override the SDK-derived ones further down, via the
@@ -646,7 +653,9 @@ const captureAiGeneration = async (client, options) => {
       $ai_input: safeInput,
       $ai_output_choices: safeOutput,
       $ai_http_status: httpStatus,
-      $ai_input_tokens: usage.inputTokens ?? 0,
+      ...(usage.inputTokens !== undefined ? {
+        $ai_input_tokens: usage.inputTokens
+      } : {}),
       ...(usage.outputTokens !== undefined ? {
         $ai_output_tokens: usage.outputTokens
       } : {}),
@@ -1164,13 +1173,10 @@ const wrapVercelLanguageModel = (model, phClient, options) => {
             provider: model.provider,
             input: mergedOptions.posthogPrivacyMode ? '' : mapVercelPrompt(params.prompt, phClient),
             output: [],
-            latency: 0,
+            latency: (Date.now() - startTime) / 1000,
             baseURL,
             modelParameters: getModelParams(mergedParams),
-            usage: {
-              inputTokens: 0,
-              outputTokens: 0
-            },
+            usage: {},
             error: error,
             tools: availableTools
           });
@@ -1395,13 +1401,10 @@ const wrapVercelLanguageModel = (model, phClient, options) => {
             provider: provider,
             input: mergedOptions.posthogPrivacyMode ? '' : mapVercelPrompt(params.prompt, phClient),
             output: [],
-            latency: 0,
+            latency: (Date.now() - startTime) / 1000,
             baseURL,
             modelParameters: getModelParams(mergedParams),
-            usage: {
-              inputTokens: 0,
-              outputTokens: 0
-            },
+            usage: {},
             error: error,
             tools: availableTools
           });