@posthog/ai 3.0.0 → 3.1.0

package/CHANGELOG.md

@@ -1,4 +1,6 @@
-# Next
+# 3.1.0 - 2025-02-07
+
+- feat: add posthogCostOverride, posthogModelOverride, and posthogProviderOverride to sendEventToPosthog for vercel
 
 # 2.4.0 - 2025-02-03
 

package/lib/index.cjs.js

@@ -111,12 +111,22 @@ const sendEventToPosthog = ({
         $ai_error: error
       };
     }
+    let costOverrideData = {};
+    if (params.posthogCostOverride) {
+      const inputCostUSD = (params.posthogCostOverride.inputCost ?? 0) * (usage.inputTokens ?? 0);
+      const outputCostUSD = (params.posthogCostOverride.outputCost ?? 0) * (usage.outputTokens ?? 0);
+      costOverrideData = {
+        $ai_input_cost_usd: inputCostUSD,
+        $ai_output_cost_usd: outputCostUSD,
+        $ai_total_cost_usd: inputCostUSD + outputCostUSD
+      };
+    }
     client.capture({
       distinctId: distinctId ?? traceId,
       event: '$ai_generation',
       properties: {
-        $ai_provider: provider,
-        $ai_model: model,
+        $ai_provider: params.posthogProviderOverride ?? provider,
+        $ai_model: params.posthogModelOverride ?? model,
         $ai_model_parameters: getModelParams(params),
         $ai_input: withPrivacyMode(client, params.posthogPrivacyMode ?? false, input),
         $ai_output_choices: withPrivacyMode(client, params.posthogPrivacyMode ?? false, output),
@@ -130,7 +140,8 @@ const sendEventToPosthog = ({
         ...(distinctId ? {} : {
           $process_person_profile: false
         }),
-        ...errorData
+        ...errorData,
+        ...costOverrideData
       },
       groups: params.posthogGroups
     });
@@ -512,6 +523,11 @@ 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);
+        }
         sendEventToPosthog({
           client: phClient,
           distinctId: options.posthogDistinctId,
@@ -520,7 +536,7 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
           provider: provider,
           input: options.posthogPrivacyMode ? '' : mapVercelPrompt(params.prompt),
           output: [{
-            content: result.text,
+            content,
             role: 'assistant'
           }],
           latency,
@@ -1283,9 +1299,7 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
     this.debug = options.debug || false;
   }
   // ===== CALLBACK METHODS =====
-  handleChainStart(chain, inputs, runId, parentRunId, tags, metadata,
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  runType, runName) {
+  handleChainStart(chain, inputs, runId, parentRunId, tags, metadata, runType, runName) {
     this._logDebugEvent('on_chain_start', runId, parentRunId, {
       inputs,
       tags
@@ -1483,7 +1497,9 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
     const traceId = this._getTraceId(runId);
     this._popParentOfRun(runId);
     const run = this._popRunMetadata(runId);
-    if (!run) return;
+    if (!run) {
+      return;
+    }
     if ('modelParams' in run) {
       console.warn(`Run ${runId} is a generation, but attempted to be captured as a trace/span.`);
       return;
@@ -1592,10 +1608,14 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
   _getLangchainRunName(serialized, ...args) {
     if (args && args.length > 0) {
       for (const arg of args) {
-        if (arg && typeof arg === 'object' && 'name' in arg) return arg.name;
+        if (arg && typeof arg === 'object' && 'name' in arg) {
+          return arg.name;
+        }
       }
     }
-    if (serialized && serialized.name) return serialized.name;
+    if (serialized && serialized.name) {
+      return serialized.name;
+    }
     if (serialized && serialized.id) {
       return Array.isArray(serialized.id) ? serialized.id[serialized.id.length - 1] : serialized.id;
     }