@posthog/ai 8.0.1 → 8.1.1

package/dist/vercel/index.mjs

@@ -6,6 +6,9 @@ import { uuidv7 } from '@posthog/core';
 const isString = value => {
   return typeof value === 'string';
 };
+const isObject = value => {
+  return value !== null && typeof value === 'object' && !Array.isArray(value);
+};
 
 const DATA_URL_PREFIX_RE = /^data:([^;,\s]+)(?:;[^;,\s]+)*;base64,/i;
 const BASE64_ALPHABET_RE = /^[A-Za-z0-9+/_=-]+$/;
@@ -392,7 +395,7 @@ function sanitizeValues(obj) {
   return jsonSafe;
 }
 
-var version = "8.0.1";
+var version = "8.1.1";
 
 const DEFAULT_MAX_DEPTH = 3;
 const MAX_STACK_LINES = 20;
@@ -800,6 +803,36 @@ const extractProvider = model => {
   return providerName;
 };
 
+/**
+ * Recover the base URL so gateway calls self-identify via `$ai_base_url` (dedup
+ * keys on it). The spec exposes none, so we read the off-spec provider `config`:
+ * `@ai-sdk/anthropic` keeps a `config.baseURL` string; `@ai-sdk/openai`/
+ * `openai-compatible` bury it in a `config.url({ path })` closure. Unknown shapes
+ * degrade to `''` — those providers (or a custom `fetch`) stay invisible to dedup.
+ */
+const extractBaseURL = model => {
+  try {
+    const config = model.config;
+    if (!isObject(config)) {
+      return '';
+    }
+    if (isString(config.baseURL)) {
+      return config.baseURL;
+    }
+    const urlFn = config.url;
+    if (typeof urlFn === 'function') {
+      const url = urlFn({
+        path: '',
+        modelId: model.modelId
+      });
+      return isString(url) ? url : '';
+    }
+  } catch {
+    // Unknown config shape or url() threw.
+  }
+  return '';
+};
+
 // Extract web search count from provider metadata (works for both V2 and V3)
 const extractWebSearchCount = (providerMetadata, usage) => {
   // Try Anthropic-specific extraction
@@ -827,32 +860,22 @@ const extractTokenCount = value => {
   }
   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;
+const extractUsageToken = (usage, topLevelKey, nestedObjectKey, nestedValueKey) => {
+  if (topLevelKey in usage) {
+    return usage[topLevelKey];
   }
-  // V3 style: nested in outputTokens.reasoning
-  if ('outputTokens' in usage && usage.outputTokens && typeof usage.outputTokens === 'object' && 'reasoning' in usage.outputTokens) {
-    return usage.outputTokens.reasoning;
+  const nestedTokens = usage[nestedObjectKey];
+  if (nestedTokens && typeof nestedTokens === 'object' && nestedValueKey in nestedTokens) {
+    return nestedTokens[nestedValueKey];
   }
   return undefined;
 };
 
+// Helper to extract reasoning tokens from V2 (usage.reasoningTokens) or V3 (usage.outputTokens.reasoning)
+const extractReasoningTokens = usage => extractUsageToken(usage, 'reasoningTokens', 'outputTokens', 'reasoning');
+
 // 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;
-};
+const extractCacheReadTokens = usage => extractUsageToken(usage, 'cachedInputTokens', 'inputTokens', 'cacheRead');
 
 // Helper to extract cache write tokens from V3 (usage.inputTokens.cacheWrite). Providers like
 // Amazon Bedrock populate this standardized field instead of providerMetadata.anthropic.
@@ -952,11 +975,11 @@ const wrapVercelLanguageModel = (model, phClient, options) => {
           ...mapVercelParams(params)
         };
         const availableTools = extractAvailableToolCalls('vercel', params);
+        const baseURL = extractBaseURL(model);
         try {
           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
           // result.content is undefined when the model returns only tool calls with no text output
           const content = mapVercelOutput(result.content ?? []);
           const latency = (Date.now() - startTime) / 1000;
@@ -1021,7 +1044,7 @@ const wrapVercelLanguageModel = (model, phClient, options) => {
             input: mergedOptions.posthogPrivacyMode ? '' : mapVercelPrompt(params.prompt),
             output: [],
             latency: 0,
-            baseURL: '',
+            baseURL,
             modelParameters: getModelParams(mergedParams),
             usage: {
               inputTokens: 0,
@@ -1053,7 +1076,7 @@ const wrapVercelLanguageModel = (model, phClient, options) => {
         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
+        const baseURL = extractBaseURL(model);
 
         // Map to track in-progress tool calls
         const toolCallsInProgress = new Map();
@@ -1211,7 +1234,7 @@ const wrapVercelLanguageModel = (model, phClient, options) => {
             input: mergedOptions.posthogPrivacyMode ? '' : mapVercelPrompt(params.prompt),
             output: [],
             latency: 0,
-            baseURL: '',
+            baseURL,
             modelParameters: getModelParams(mergedParams),
             usage: {
               inputTokens: 0,