@posthog/ai 8.0.0 → 8.1.0

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.0";
+var version = "8.1.0";
 
 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
@@ -952,11 +985,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 +1054,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 +1086,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 +1244,7 @@ const wrapVercelLanguageModel = (model, phClient, options) => {
             input: mergedOptions.posthogPrivacyMode ? '' : mapVercelPrompt(params.prompt),
             output: [],
             latency: 0,
-            baseURL: '',
+            baseURL,
             modelParameters: getModelParams(mergedParams),
             usage: {
               inputTokens: 0,