@posthog/ai 6.5.1 → 7.0.0

package/dist/index.mjs

@@ -6,7 +6,7 @@ import { wrapLanguageModel } from 'ai';
 import AnthropicOriginal from '@anthropic-ai/sdk';
 import { GoogleGenAI } from '@google/genai';
 
-var version = "6.5.1";
+var version = "7.0.0";
 
 // Type guards for safer type checking
 const isString = value => {
@@ -277,6 +277,101 @@ const truncate = input => {
   }
   return `${truncatedStr}... [truncated]`;
 };
+/**
+ * Calculate web search count from raw API response.
+ *
+ * Uses a two-tier detection strategy:
+ * Priority 1 (Exact Count): Count actual web search calls when available
+ * Priority 2 (Binary Detection): Return 1 if web search indicators are present, 0 otherwise
+ *
+ * @param result - Raw API response from any provider (OpenAI, Perplexity, OpenRouter, Gemini, etc.)
+ * @returns Number of web searches performed (exact count or binary 1/0)
+ */
+function calculateWebSearchCount(result) {
+  if (!result || typeof result !== 'object') {
+    return 0;
+  }
+  // Priority 1: Exact Count
+  // Check for OpenAI Responses API web_search_call items
+  if ('output' in result && Array.isArray(result.output)) {
+    let count = 0;
+    for (const item of result.output) {
+      if (typeof item === 'object' && item !== null && 'type' in item && item.type === 'web_search_call') {
+        count++;
+      }
+    }
+    if (count > 0) {
+      return count;
+    }
+  }
+  // Priority 2: Binary Detection (1 or 0)
+  // Check for citations at root level (Perplexity)
+  if ('citations' in result && Array.isArray(result.citations) && result.citations.length > 0) {
+    return 1;
+  }
+  // Check for search_results at root level (Perplexity via OpenRouter)
+  if ('search_results' in result && Array.isArray(result.search_results) && result.search_results.length > 0) {
+    return 1;
+  }
+  // Check for usage.search_context_size (Perplexity via OpenRouter)
+  if ('usage' in result && typeof result.usage === 'object' && result.usage !== null) {
+    if ('search_context_size' in result.usage && result.usage.search_context_size) {
+      return 1;
+    }
+  }
+  // Check for annotations with url_citation in choices[].message or choices[].delta (OpenAI/Perplexity)
+  if ('choices' in result && Array.isArray(result.choices)) {
+    for (const choice of result.choices) {
+      if (typeof choice === 'object' && choice !== null) {
+        // Check both message (non-streaming) and delta (streaming) for annotations
+        const content = ('message' in choice ? choice.message : null) || ('delta' in choice ? choice.delta : null);
+        if (typeof content === 'object' && content !== null && 'annotations' in content) {
+          const annotations = content.annotations;
+          if (Array.isArray(annotations)) {
+            const hasUrlCitation = annotations.some(ann => {
+              return typeof ann === 'object' && ann !== null && 'type' in ann && ann.type === 'url_citation';
+            });
+            if (hasUrlCitation) {
+              return 1;
+            }
+          }
+        }
+      }
+    }
+  }
+  // Check for annotations in output[].content[] (OpenAI Responses API)
+  if ('output' in result && Array.isArray(result.output)) {
+    for (const item of result.output) {
+      if (typeof item === 'object' && item !== null && 'content' in item) {
+        const content = item.content;
+        if (Array.isArray(content)) {
+          for (const contentItem of content) {
+            if (typeof contentItem === 'object' && contentItem !== null && 'annotations' in contentItem) {
+              const annotations = contentItem.annotations;
+              if (Array.isArray(annotations)) {
+                const hasUrlCitation = annotations.some(ann => {
+                  return typeof ann === 'object' && ann !== null && 'type' in ann && ann.type === 'url_citation';
+                });
+                if (hasUrlCitation) {
+                  return 1;
+                }
+              }
+            }
+          }
+        }
+      }
+    }
+  }
+  // Check for grounding_metadata (Gemini)
+  if ('candidates' in result && Array.isArray(result.candidates)) {
+    for (const candidate of result.candidates) {
+      if (typeof candidate === 'object' && candidate !== null && 'grounding_metadata' in candidate && candidate.grounding_metadata) {
+        return 1;
+      }
+    }
+  }
+  return 0;
+}
 /**
  * Extract available tool calls from the request parameters.
  * These are the tools provided to the LLM, not the tool calls in the response.
@@ -411,6 +506,9 @@ const sendEventToPosthog = async ({
     } : {}),
     ...(usage.cacheCreationInputTokens ? {
       $ai_cache_creation_input_tokens: usage.cacheCreationInputTokens
+    } : {}),
+    ...(usage.webSearchCount ? {
+      $ai_web_search_count: usage.webSearchCount
     } : {})
   };
   const properties = {
@@ -721,12 +819,17 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
               let accumulatedContent = '';
               let usage = {
                 inputTokens: 0,
-                outputTokens: 0
+                outputTokens: 0,
+                webSearchCount: 0
               };
               // Map to track in-progress tool calls
               const toolCallsInProgress = new Map();
               for await (const chunk of stream1) {
                 const choice = chunk?.choices?.[0];
+                const chunkWebSearchCount = calculateWebSearchCount(chunk);
+                if (chunkWebSearchCount > 0 && chunkWebSearchCount > (usage.webSearchCount ?? 0)) {
+                  usage.webSearchCount = chunkWebSearchCount;
+                }
                 // Handle text content
                 const deltaContent = choice?.delta?.content;
                 if (deltaContent) {
@@ -765,6 +868,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
                 // Handle usage information
                 if (chunk.usage) {
                   usage = {
+                    ...usage,
                     inputTokens: chunk.usage.prompt_tokens ?? 0,
                     outputTokens: chunk.usage.completion_tokens ?? 0,
                     reasoningTokens: chunk.usage.completion_tokens_details?.reasoning_tokens ?? 0,
@@ -816,7 +920,13 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
                 baseURL: this.baseURL,
                 params: body,
                 httpStatus: 200,
-                usage,
+                usage: {
+                  inputTokens: usage.inputTokens,
+                  outputTokens: usage.outputTokens,
+                  reasoningTokens: usage.reasoningTokens,
+                  cacheReadInputTokens: usage.cacheReadInputTokens,
+                  webSearchCount: usage.webSearchCount
+                },
                 tools: availableTools
               });
             } catch (error) {
@@ -851,13 +961,14 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
         if ('choices' in result) {
           const latency = (Date.now() - startTime) / 1000;
           const availableTools = extractAvailableToolCalls('openai', openAIParams);
+          const formattedOutput = formatResponseOpenAI(result);
           await sendEventToPosthog({
             client: this.phClient,
             ...posthogParams,
             model: openAIParams.model,
             provider: 'openai',
             input: sanitizeOpenAI(openAIParams.messages),
-            output: formatResponseOpenAI(result),
+            output: formattedOutput,
             latency,
             baseURL: this.baseURL,
             params: body,
@@ -866,7 +977,8 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
               inputTokens: result.usage?.prompt_tokens ?? 0,
               outputTokens: result.usage?.completion_tokens ?? 0,
               reasoningTokens: result.usage?.completion_tokens_details?.reasoning_tokens ?? 0,
-              cacheReadInputTokens: result.usage?.prompt_tokens_details?.cached_tokens ?? 0
+              cacheReadInputTokens: result.usage?.prompt_tokens_details?.cached_tokens ?? 0,
+              webSearchCount: calculateWebSearchCount(result)
             },
             tools: availableTools
           });
@@ -921,14 +1033,22 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
               let finalContent = [];
               let usage = {
                 inputTokens: 0,
-                outputTokens: 0
+                outputTokens: 0,
+                webSearchCount: 0
               };
               for await (const chunk of stream1) {
+                if ('response' in chunk && chunk.response) {
+                  const chunkWebSearchCount = calculateWebSearchCount(chunk.response);
+                  if (chunkWebSearchCount > 0 && chunkWebSearchCount > (usage.webSearchCount ?? 0)) {
+                    usage.webSearchCount = chunkWebSearchCount;
+                  }
+                }
                 if (chunk.type === 'response.completed' && 'response' in chunk && chunk.response?.output && chunk.response.output.length > 0) {
                   finalContent = chunk.response.output;
                 }
                 if ('response' in chunk && chunk.response?.usage) {
                   usage = {
+                    ...usage,
                     inputTokens: chunk.response.usage.input_tokens ?? 0,
                     outputTokens: chunk.response.usage.output_tokens ?? 0,
                     reasoningTokens: chunk.response.usage.output_tokens_details?.reasoning_tokens ?? 0,
@@ -950,7 +1070,13 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
                 baseURL: this.baseURL,
                 params: body,
                 httpStatus: 200,
-                usage,
+                usage: {
+                  inputTokens: usage.inputTokens,
+                  outputTokens: usage.outputTokens,
+                  reasoningTokens: usage.reasoningTokens,
+                  cacheReadInputTokens: usage.cacheReadInputTokens,
+                  webSearchCount: usage.webSearchCount
+                },
                 tools: availableTools
               });
             } catch (error) {
@@ -985,6 +1111,9 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
         if ('output' in result) {
           const latency = (Date.now() - startTime) / 1000;
           const availableTools = extractAvailableToolCalls('openai', openAIParams);
+          const formattedOutput = formatResponseOpenAI({
+            output: result.output
+          });
           await sendEventToPosthog({
             client: this.phClient,
             ...posthogParams,
@@ -992,9 +1121,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
             model: openAIParams.model,
             provider: 'openai',
             input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
-            output: formatResponseOpenAI({
-              output: result.output
-            }),
+            output: formattedOutput,
             latency,
             baseURL: this.baseURL,
             params: body,
@@ -1003,7 +1130,8 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
               inputTokens: result.usage?.input_tokens ?? 0,
               outputTokens: result.usage?.output_tokens ?? 0,
               reasoningTokens: result.usage?.output_tokens_details?.reasoning_tokens ?? 0,
-              cacheReadInputTokens: result.usage?.input_tokens_details?.cached_tokens ?? 0
+              cacheReadInputTokens: result.usage?.input_tokens_details?.cached_tokens ?? 0,
+              webSearchCount: calculateWebSearchCount(result)
             },
             tools: availableTools
           });
@@ -1041,9 +1169,9 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
     } = extractPosthogParams(body);
     const startTime = Date.now();
     const originalCreate = super.create.bind(this);
-    const originalSelf = this;
-    const tempCreate = originalSelf.create;
-    originalSelf.create = originalCreate;
+    const originalSelfRecord = this;
+    const tempCreate = originalSelfRecord['create'];
+    originalSelfRecord['create'] = originalCreate;
     try {
       const parentPromise = super.parse(openAIParams, options);
       const wrappedPromise = parentPromise.then(async result => {
@@ -1092,7 +1220,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
       return wrappedPromise;
     } finally {
       // Restore our wrapped create method
-      originalSelf.create = tempCreate;
+      originalSelfRecord['create'] = tempCreate;
     }
   }
 };
@@ -1838,11 +1966,29 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
             cacheCreationInputTokens: providerMetadata.anthropic.cacheCreationInputTokens
           } : {})
         };
+        // Calculate web search count based on provider
+        let webSearchCount = 0;
+        if (providerMetadata?.anthropic && typeof providerMetadata.anthropic === 'object' && 'server_tool_use' in providerMetadata.anthropic) {
+          // Anthropic-specific extraction
+          const serverToolUse = providerMetadata.anthropic.server_tool_use;
+          if (serverToolUse && typeof serverToolUse === 'object' && 'web_search_requests' in serverToolUse && typeof serverToolUse.web_search_requests === 'number') {
+            webSearchCount = serverToolUse.web_search_requests;
+          }
+        } else {
+          // For other providers through Vercel, pass available metadata to helper
+          // Note: Vercel abstracts provider responses, so we may not have access to
+          // raw citations/annotations unless Vercel exposes them in usage/metadata
+          webSearchCount = calculateWebSearchCount({
+            usage: result.usage,
+            providerMetadata: providerMetadata
+          });
+        }
         const usage = {
           inputTokens: result.usage.inputTokens,
           outputTokens: result.usage.outputTokens,
           reasoningTokens: result.usage.reasoningTokens,
           cacheReadInputTokens: result.usage.cachedInputTokens,
+          webSearchCount,
           ...additionalTokenValues
         };
         await sendEventToPosthog({
@@ -1896,6 +2042,7 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
       let generatedText = '';
       let reasoningText = '';
       let usage = {};
+      let providerMetadata = undefined;
       const mergedParams = {
         ...options,
         ...mapVercelParams(params),
@@ -1953,12 +2100,10 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
               });
             }
             if (chunk.type === 'finish') {
-              const providerMetadata = chunk.providerMetadata;
-              const additionalTokenValues = {
-                ...(providerMetadata?.anthropic ? {
-                  cacheCreationInputTokens: providerMetadata.anthropic.cacheCreationInputTokens
-                } : {})
-              };
+              providerMetadata = chunk.providerMetadata;
+              const additionalTokenValues = providerMetadata && typeof providerMetadata === 'object' && 'anthropic' in providerMetadata && providerMetadata.anthropic && typeof providerMetadata.anthropic === 'object' && 'cacheCreationInputTokens' in providerMetadata.anthropic ? {
+                cacheCreationInputTokens: providerMetadata.anthropic.cacheCreationInputTokens
+              } : {};
               usage = {
                 inputTokens: chunk.usage?.inputTokens,
                 outputTokens: chunk.usage?.outputTokens,
@@ -2003,6 +2148,28 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
               role: 'assistant',
               content: content.length === 1 && content[0].type === 'text' ? content[0].text : content
             }] : [];
+            // Calculate web search count based on provider
+            let webSearchCount = 0;
+            if (providerMetadata && typeof providerMetadata === 'object' && 'anthropic' in providerMetadata && providerMetadata.anthropic && typeof providerMetadata.anthropic === 'object' && 'server_tool_use' in providerMetadata.anthropic) {
+              // Anthropic-specific extraction
+              const serverToolUse = providerMetadata.anthropic.server_tool_use;
+              if (serverToolUse && typeof serverToolUse === 'object' && 'web_search_requests' in serverToolUse && typeof serverToolUse.web_search_requests === 'number') {
+                webSearchCount = serverToolUse.web_search_requests;
+              }
+            } else {
+              // For other providers through Vercel, pass available metadata to helper
+              // Note: Vercel abstracts provider responses, so we may not have access to
+              // raw citations/annotations unless Vercel exposes them in usage/metadata
+              webSearchCount = calculateWebSearchCount({
+                usage: usage,
+                providerMetadata: providerMetadata
+              });
+            }
+            // Update usage with web search count
+            const finalUsage = {
+              ...usage,
+              webSearchCount
+            };
             await sendEventToPosthog({
               client: phClient,
               distinctId: options.posthogDistinctId,
@@ -2015,7 +2182,7 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
               baseURL,
               params: mergedParams,
               httpStatus: 200,
-              usage,
+              usage: finalUsage,
               tools: availableTools,
               captureImmediate: options.posthogCaptureImmediate
             });
@@ -2101,7 +2268,8 @@ class WrappedMessages extends AnthropicOriginal.Messages {
           inputTokens: 0,
           outputTokens: 0,
           cacheCreationInputTokens: 0,
-          cacheReadInputTokens: 0
+          cacheReadInputTokens: 0,
+          webSearchCount: 0
         };
         if ('tee' in value) {
           const [stream1, stream2] = value.tee();
@@ -2178,9 +2346,14 @@ class WrappedMessages extends AnthropicOriginal.Messages {
                   usage.inputTokens = chunk.message.usage.input_tokens ?? 0;
                   usage.cacheCreationInputTokens = chunk.message.usage.cache_creation_input_tokens ?? 0;
                   usage.cacheReadInputTokens = chunk.message.usage.cache_read_input_tokens ?? 0;
+                  usage.webSearchCount = chunk.message.usage.server_tool_use?.web_search_requests ?? 0;
                 }
                 if ('usage' in chunk) {
                   usage.outputTokens = chunk.usage.output_tokens ?? 0;
+                  // Update web search count if present in delta
+                  if (chunk.usage.server_tool_use?.web_search_requests !== undefined) {
+                    usage.webSearchCount = chunk.usage.server_tool_use.web_search_requests;
+                  }
                 }
               }
               const latency = (Date.now() - startTime) / 1000;
@@ -2257,7 +2430,8 @@ class WrappedMessages extends AnthropicOriginal.Messages {
               inputTokens: result.usage.input_tokens ?? 0,
               outputTokens: result.usage.output_tokens ?? 0,
               cacheCreationInputTokens: result.usage.cache_creation_input_tokens ?? 0,
-              cacheReadInputTokens: result.usage.cache_read_input_tokens ?? 0
+              cacheReadInputTokens: result.usage.cache_read_input_tokens ?? 0,
+              webSearchCount: result.usage.server_tool_use?.web_search_requests ?? 0
             },
             tools: availableTools
           });
@@ -2331,7 +2505,8 @@ class WrappedModels {
           inputTokens: metadata?.promptTokenCount ?? 0,
           outputTokens: metadata?.candidatesTokenCount ?? 0,
           reasoningTokens: metadata?.thoughtsTokenCount ?? 0,
-          cacheReadInputTokens: metadata?.cachedContentTokenCount ?? 0
+          cacheReadInputTokens: metadata?.cachedContentTokenCount ?? 0,
+          webSearchCount: calculateGoogleWebSearchCount(response)
         },
         tools: availableTools
       });
@@ -2368,11 +2543,16 @@ class WrappedModels {
     const accumulatedContent = [];
     let usage = {
       inputTokens: 0,
-      outputTokens: 0
+      outputTokens: 0,
+      webSearchCount: 0
     };
     try {
       const stream = await this.client.models.generateContentStream(geminiParams);
       for await (const chunk of stream) {
+        const chunkWebSearchCount = calculateGoogleWebSearchCount(chunk);
+        if (chunkWebSearchCount > 0 && chunkWebSearchCount > (usage.webSearchCount ?? 0)) {
+          usage.webSearchCount = chunkWebSearchCount;
+        }
         // Handle text content
         if (chunk.text) {
           // Find if we already have a text item to append to
@@ -2421,7 +2601,8 @@ class WrappedModels {
             inputTokens: metadata.promptTokenCount ?? 0,
             outputTokens: metadata.candidatesTokenCount ?? 0,
             reasoningTokens: metadata.thoughtsTokenCount ?? 0,
-            cacheReadInputTokens: metadata.cachedContentTokenCount ?? 0
+            cacheReadInputTokens: metadata.cachedContentTokenCount ?? 0,
+            webSearchCount: usage.webSearchCount
           };
         }
         yield chunk;
@@ -2444,7 +2625,10 @@ class WrappedModels {
         baseURL: 'https://generativelanguage.googleapis.com',
         params: params,
         httpStatus: 200,
-        usage,
+        usage: {
+          ...usage,
+          webSearchCount: usage.webSearchCount
+        },
         tools: availableTools
       });
     } catch (error) {
@@ -2587,6 +2771,66 @@ class WrappedModels {
     return messages;
   }
 }
+/**
+ * Detect if Google Search grounding was used in the response.
+ * Gemini bills per request that uses grounding, not per individual query.
+ * Returns 1 if grounding was used, 0 otherwise.
+ */
+function calculateGoogleWebSearchCount(response) {
+  if (!response || typeof response !== 'object' || !('candidates' in response)) {
+    return 0;
+  }
+  const candidates = response.candidates;
+  if (!Array.isArray(candidates)) {
+    return 0;
+  }
+  const hasGrounding = candidates.some(candidate => {
+    if (!candidate || typeof candidate !== 'object') {
+      return false;
+    }
+    // Check for grounding metadata
+    if ('groundingMetadata' in candidate && candidate.groundingMetadata) {
+      const metadata = candidate.groundingMetadata;
+      if (typeof metadata === 'object') {
+        // Check if web_search_queries exists and is non-empty
+        if ('webSearchQueries' in metadata && Array.isArray(metadata.webSearchQueries) && metadata.webSearchQueries.length > 0) {
+          return true;
+        }
+        // Check if grounding_chunks exists and is non-empty
+        if ('groundingChunks' in metadata && Array.isArray(metadata.groundingChunks) && metadata.groundingChunks.length > 0) {
+          return true;
+        }
+      }
+    }
+    // Check for google search in function calls
+    if ('content' in candidate && candidate.content && typeof candidate.content === 'object') {
+      const content = candidate.content;
+      if ('parts' in content && Array.isArray(content.parts)) {
+        return content.parts.some(part => {
+          if (!part || typeof part !== 'object' || !('functionCall' in part)) {
+            return false;
+          }
+          const functionCall = part.functionCall;
+          if (functionCall && typeof functionCall === 'object' && 'name' in functionCall && typeof functionCall.name === 'string') {
+            return functionCall.name.includes('google_search') || functionCall.name.includes('grounding');
+          }
+          return false;
+        });
+      }
+    }
+    return false;
+  });
+  return hasGrounding ? 1 : 0;
+}
+
+//#region rolldown:runtime
+var __defProp = Object.defineProperty;
+var __export = (target, all) => {
+	for (var name in all) __defProp(target, name, {
+		get: all[name],
+		enumerable: true
+	});
+};
 
 function getDefaultExportFromCjs (x) {
 	return x && x.__esModule && Object.prototype.hasOwnProperty.call(x, 'default') ? x['default'] : x;
@@ -2740,370 +2984,304 @@ function requireCamelcase () {
 
 requireCamelcase();
 
+//#region src/load/map_keys.ts
 function keyToJson(key, map) {
-    return map?.[key] || snakeCase(key);
+	return map?.[key] || snakeCase(key);
 }
 function mapKeys(fields, mapper, map) {
-    const mapped = {};
-    for (const key in fields) {
-        if (Object.hasOwn(fields, key)) {
-            mapped[mapper(key, map)] = fields[key];
-        }
-    }
-    return mapped;
+	const mapped = {};
+	for (const key in fields) if (Object.hasOwn(fields, key)) mapped[mapper(key, map)] = fields[key];
+	return mapped;
 }
 
+//#region src/load/serializable.ts
+var serializable_exports = {};
+__export(serializable_exports, {
+	Serializable: () => Serializable,
+	get_lc_unique_name: () => get_lc_unique_name
+});
 function shallowCopy(obj) {
-    return Array.isArray(obj) ? [...obj] : { ...obj };
+	return Array.isArray(obj) ? [...obj] : { ...obj };
 }
 function replaceSecrets(root, secretsMap) {
-    const result = shallowCopy(root);
-    for (const [path, secretId] of Object.entries(secretsMap)) {
-        const [last, ...partsReverse] = path.split(".").reverse();
-        // eslint-disable-next-line @typescript-eslint/no-explicit-any
-        let current = result;
-        for (const part of partsReverse.reverse()) {
-            if (current[part] === undefined) {
-                break;
-            }
-            current[part] = shallowCopy(current[part]);
-            current = current[part];
-        }
-        if (current[last] !== undefined) {
-            current[last] = {
-                lc: 1,
-                type: "secret",
-                id: [secretId],
-            };
-        }
-    }
-    return result;
+	const result = shallowCopy(root);
+	for (const [path, secretId] of Object.entries(secretsMap)) {
+		const [last, ...partsReverse] = path.split(".").reverse();
+		let current = result;
+		for (const part of partsReverse.reverse()) {
+			if (current[part] === void 0) break;
+			current[part] = shallowCopy(current[part]);
+			current = current[part];
+		}
+		if (current[last] !== void 0) current[last] = {
+			lc: 1,
+			type: "secret",
+			id: [secretId]
+		};
+	}
+	return result;
 }
 /**
- * Get a unique name for the module, rather than parent class implementations.
- * Should not be subclassed, subclass lc_name above instead.
- */
-function get_lc_unique_name(
-// eslint-disable-next-line @typescript-eslint/no-use-before-define
-serializableClass) {
-    // "super" here would refer to the parent class of Serializable,
-    // when we want the parent class of the module actually calling this method.
-    const parentClass = Object.getPrototypeOf(serializableClass);
-    const lcNameIsSubclassed = typeof serializableClass.lc_name === "function" &&
-        (typeof parentClass.lc_name !== "function" ||
-            serializableClass.lc_name() !== parentClass.lc_name());
-    if (lcNameIsSubclassed) {
-        return serializableClass.lc_name();
-    }
-    else {
-        return serializableClass.name;
-    }
-}
-class Serializable {
-    /**
-     * The name of the serializable. Override to provide an alias or
-     * to preserve the serialized module name in minified environments.
-     *
-     * Implemented as a static method to support loading logic.
-     */
-    static lc_name() {
-        return this.name;
-    }
-    /**
-     * The final serialized identifier for the module.
-     */
-    get lc_id() {
-        return [
-            ...this.lc_namespace,
-            get_lc_unique_name(this.constructor),
-        ];
-    }
-    /**
-     * A map of secrets, which will be omitted from serialization.
-     * Keys are paths to the secret in constructor args, e.g. "foo.bar.baz".
-     * Values are the secret ids, which will be used when deserializing.
-     */
-    get lc_secrets() {
-        return undefined;
-    }
-    /**
-     * A map of additional attributes to merge with constructor args.
-     * Keys are the attribute names, e.g. "foo".
-     * Values are the attribute values, which will be serialized.
-     * These attributes need to be accepted by the constructor as arguments.
-     */
-    get lc_attributes() {
-        return undefined;
-    }
-    /**
-     * A map of aliases for constructor args.
-     * Keys are the attribute names, e.g. "foo".
-     * Values are the alias that will replace the key in serialization.
-     * This is used to eg. make argument names match Python.
-     */
-    get lc_aliases() {
-        return undefined;
-    }
-    /**
-     * A manual list of keys that should be serialized.
-     * If not overridden, all fields passed into the constructor will be serialized.
-     */
-    get lc_serializable_keys() {
-        return undefined;
-    }
-    constructor(kwargs, ..._args) {
-        Object.defineProperty(this, "lc_serializable", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: false
-        });
-        Object.defineProperty(this, "lc_kwargs", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        if (this.lc_serializable_keys !== undefined) {
-            this.lc_kwargs = Object.fromEntries(Object.entries(kwargs || {}).filter(([key]) => this.lc_serializable_keys?.includes(key)));
-        }
-        else {
-            this.lc_kwargs = kwargs ?? {};
-        }
-    }
-    toJSON() {
-        if (!this.lc_serializable) {
-            return this.toJSONNotImplemented();
-        }
-        if (
-        // eslint-disable-next-line no-instanceof/no-instanceof
-        this.lc_kwargs instanceof Serializable ||
-            typeof this.lc_kwargs !== "object" ||
-            Array.isArray(this.lc_kwargs)) {
-            // We do not support serialization of classes with arg not a POJO
-            // I'm aware the check above isn't as strict as it could be
-            return this.toJSONNotImplemented();
-        }
-        const aliases = {};
-        const secrets = {};
-        const kwargs = Object.keys(this.lc_kwargs).reduce((acc, key) => {
-            acc[key] = key in this ? this[key] : this.lc_kwargs[key];
-            return acc;
-        }, {});
-        // get secrets, attributes and aliases from all superclasses
-        for (
-        // eslint-disable-next-line @typescript-eslint/no-this-alias
-        let current = Object.getPrototypeOf(this); current; current = Object.getPrototypeOf(current)) {
-            Object.assign(aliases, Reflect.get(current, "lc_aliases", this));
-            Object.assign(secrets, Reflect.get(current, "lc_secrets", this));
-            Object.assign(kwargs, Reflect.get(current, "lc_attributes", this));
-        }
-        // include all secrets used, even if not in kwargs,
-        // will be replaced with sentinel value in replaceSecrets
-        Object.keys(secrets).forEach((keyPath) => {
-            // eslint-disable-next-line @typescript-eslint/no-this-alias, @typescript-eslint/no-explicit-any
-            let read = this;
-            // eslint-disable-next-line @typescript-eslint/no-explicit-any
-            let write = kwargs;
-            const [last, ...partsReverse] = keyPath.split(".").reverse();
-            for (const key of partsReverse.reverse()) {
-                if (!(key in read) || read[key] === undefined)
-                    return;
-                if (!(key in write) || write[key] === undefined) {
-                    if (typeof read[key] === "object" && read[key] != null) {
-                        write[key] = {};
-                    }
-                    else if (Array.isArray(read[key])) {
-                        write[key] = [];
-                    }
-                }
-                read = read[key];
-                write = write[key];
-            }
-            if (last in read && read[last] !== undefined) {
-                write[last] = write[last] || read[last];
-            }
-        });
-        return {
-            lc: 1,
-            type: "constructor",
-            id: this.lc_id,
-            kwargs: mapKeys(Object.keys(secrets).length ? replaceSecrets(kwargs, secrets) : kwargs, keyToJson, aliases),
-        };
-    }
-    toJSONNotImplemented() {
-        return {
-            lc: 1,
-            type: "not_implemented",
-            id: this.lc_id,
-        };
-    }
+* Get a unique name for the module, rather than parent class implementations.
+* Should not be subclassed, subclass lc_name above instead.
+*/
+function get_lc_unique_name(serializableClass) {
+	const parentClass = Object.getPrototypeOf(serializableClass);
+	const lcNameIsSubclassed = typeof serializableClass.lc_name === "function" && (typeof parentClass.lc_name !== "function" || serializableClass.lc_name() !== parentClass.lc_name());
+	if (lcNameIsSubclassed) return serializableClass.lc_name();
+	else return serializableClass.name;
 }
+var Serializable = class Serializable {
+	lc_serializable = false;
+	lc_kwargs;
+	/**
+	* The name of the serializable. Override to provide an alias or
+	* to preserve the serialized module name in minified environments.
+	*
+	* Implemented as a static method to support loading logic.
+	*/
+	static lc_name() {
+		return this.name;
+	}
+	/**
+	* The final serialized identifier for the module.
+	*/
+	get lc_id() {
+		return [...this.lc_namespace, get_lc_unique_name(this.constructor)];
+	}
+	/**
+	* A map of secrets, which will be omitted from serialization.
+	* Keys are paths to the secret in constructor args, e.g. "foo.bar.baz".
+	* Values are the secret ids, which will be used when deserializing.
+	*/
+	get lc_secrets() {
+		return void 0;
+	}
+	/**
+	* A map of additional attributes to merge with constructor args.
+	* Keys are the attribute names, e.g. "foo".
+	* Values are the attribute values, which will be serialized.
+	* These attributes need to be accepted by the constructor as arguments.
+	*/
+	get lc_attributes() {
+		return void 0;
+	}
+	/**
+	* A map of aliases for constructor args.
+	* Keys are the attribute names, e.g. "foo".
+	* Values are the alias that will replace the key in serialization.
+	* This is used to eg. make argument names match Python.
+	*/
+	get lc_aliases() {
+		return void 0;
+	}
+	/**
+	* A manual list of keys that should be serialized.
+	* If not overridden, all fields passed into the constructor will be serialized.
+	*/
+	get lc_serializable_keys() {
+		return void 0;
+	}
+	constructor(kwargs, ..._args) {
+		if (this.lc_serializable_keys !== void 0) this.lc_kwargs = Object.fromEntries(Object.entries(kwargs || {}).filter(([key]) => this.lc_serializable_keys?.includes(key)));
+		else this.lc_kwargs = kwargs ?? {};
+	}
+	toJSON() {
+		if (!this.lc_serializable) return this.toJSONNotImplemented();
+		if (this.lc_kwargs instanceof Serializable || typeof this.lc_kwargs !== "object" || Array.isArray(this.lc_kwargs)) return this.toJSONNotImplemented();
+		const aliases = {};
+		const secrets = {};
+		const kwargs = Object.keys(this.lc_kwargs).reduce((acc, key) => {
+			acc[key] = key in this ? this[key] : this.lc_kwargs[key];
+			return acc;
+		}, {});
+		for (let current = Object.getPrototypeOf(this); current; current = Object.getPrototypeOf(current)) {
+			Object.assign(aliases, Reflect.get(current, "lc_aliases", this));
+			Object.assign(secrets, Reflect.get(current, "lc_secrets", this));
+			Object.assign(kwargs, Reflect.get(current, "lc_attributes", this));
+		}
+		Object.keys(secrets).forEach((keyPath) => {
+			let read = this;
+			let write = kwargs;
+			const [last, ...partsReverse] = keyPath.split(".").reverse();
+			for (const key of partsReverse.reverse()) {
+				if (!(key in read) || read[key] === void 0) return;
+				if (!(key in write) || write[key] === void 0) {
+					if (typeof read[key] === "object" && read[key] != null) write[key] = {};
+					else if (Array.isArray(read[key])) write[key] = [];
+				}
+				read = read[key];
+				write = write[key];
+			}
+			if (last in read && read[last] !== void 0) write[last] = write[last] || read[last];
+		});
+		return {
+			lc: 1,
+			type: "constructor",
+			id: this.lc_id,
+			kwargs: mapKeys(Object.keys(secrets).length ? replaceSecrets(kwargs, secrets) : kwargs, keyToJson, aliases)
+		};
+	}
+	toJSONNotImplemented() {
+		return {
+			lc: 1,
+			type: "not_implemented",
+			id: this.lc_id
+		};
+	}
+};
 
-// Supabase Edge Function provides a `Deno` global object
-// without `version` property
+//#region src/utils/env.ts
+var env_exports = {};
+__export(env_exports, {
+	getEnv: () => getEnv,
+	getEnvironmentVariable: () => getEnvironmentVariable,
+	getRuntimeEnvironment: () => getRuntimeEnvironment,
+	isBrowser: () => isBrowser,
+	isDeno: () => isDeno,
+	isJsDom: () => isJsDom,
+	isNode: () => isNode,
+	isWebWorker: () => isWebWorker
+});
+const isBrowser = () => typeof window !== "undefined" && typeof window.document !== "undefined";
+const isWebWorker = () => typeof globalThis === "object" && globalThis.constructor && globalThis.constructor.name === "DedicatedWorkerGlobalScope";
+const isJsDom = () => typeof window !== "undefined" && window.name === "nodejs" || typeof navigator !== "undefined" && navigator.userAgent.includes("jsdom");
 const isDeno = () => typeof Deno !== "undefined";
+const isNode = () => typeof process !== "undefined" && typeof process.versions !== "undefined" && typeof process.versions.node !== "undefined" && !isDeno();
+const getEnv = () => {
+	let env;
+	if (isBrowser()) env = "browser";
+	else if (isNode()) env = "node";
+	else if (isWebWorker()) env = "webworker";
+	else if (isJsDom()) env = "jsdom";
+	else if (isDeno()) env = "deno";
+	else env = "other";
+	return env;
+};
+let runtimeEnvironment;
+function getRuntimeEnvironment() {
+	if (runtimeEnvironment === void 0) {
+		const env = getEnv();
+		runtimeEnvironment = {
+			library: "langchain-js",
+			runtime: env
+		};
+	}
+	return runtimeEnvironment;
+}
 function getEnvironmentVariable(name) {
-    // Certain Deno setups will throw an error if you try to access environment variables
-    // https://github.com/langchain-ai/langchainjs/issues/1412
-    try {
-        if (typeof process !== "undefined") {
-            // eslint-disable-next-line no-process-env
-            return process.env?.[name];
-        }
-        else if (isDeno()) {
-            return Deno?.env.get(name);
-        }
-        else {
-            return undefined;
-        }
-    }
-    catch (e) {
-        return undefined;
-    }
+	try {
+		if (typeof process !== "undefined") return process.env?.[name];
+		else if (isDeno()) return Deno?.env.get(name);
+		else return void 0;
+	} catch {
+		return void 0;
+	}
 }
 
+//#region src/callbacks/base.ts
+var base_exports = {};
+__export(base_exports, {
+	BaseCallbackHandler: () => BaseCallbackHandler,
+	callbackHandlerPrefersStreaming: () => callbackHandlerPrefersStreaming,
+	isBaseCallbackHandler: () => isBaseCallbackHandler
+});
 /**
- * Abstract class that provides a set of optional methods that can be
- * overridden in derived classes to handle various events during the
- * execution of a LangChain application.
- */
-class BaseCallbackHandlerMethodsClass {
+* Abstract class that provides a set of optional methods that can be
+* overridden in derived classes to handle various events during the
+* execution of a LangChain application.
+*/
+var BaseCallbackHandlerMethodsClass = class {};
+function callbackHandlerPrefersStreaming(x) {
+	return "lc_prefer_streaming" in x && x.lc_prefer_streaming;
 }
 /**
- * Abstract base class for creating callback handlers in the LangChain
- * framework. It provides a set of optional methods that can be overridden
- * in derived classes to handle various events during the execution of a
- * LangChain application.
- */
-class BaseCallbackHandler extends BaseCallbackHandlerMethodsClass {
-    get lc_namespace() {
-        return ["langchain_core", "callbacks", this.name];
-    }
-    get lc_secrets() {
-        return undefined;
-    }
-    get lc_attributes() {
-        return undefined;
-    }
-    get lc_aliases() {
-        return undefined;
-    }
-    get lc_serializable_keys() {
-        return undefined;
-    }
-    /**
-     * The name of the serializable. Override to provide an alias or
-     * to preserve the serialized module name in minified environments.
-     *
-     * Implemented as a static method to support loading logic.
-     */
-    static lc_name() {
-        return this.name;
-    }
-    /**
-     * The final serialized identifier for the module.
-     */
-    get lc_id() {
-        return [
-            ...this.lc_namespace,
-            get_lc_unique_name(this.constructor),
-        ];
-    }
-    constructor(input) {
-        super();
-        Object.defineProperty(this, "lc_serializable", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: false
-        });
-        Object.defineProperty(this, "lc_kwargs", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "ignoreLLM", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: false
-        });
-        Object.defineProperty(this, "ignoreChain", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: false
-        });
-        Object.defineProperty(this, "ignoreAgent", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: false
-        });
-        Object.defineProperty(this, "ignoreRetriever", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: false
-        });
-        Object.defineProperty(this, "ignoreCustomEvent", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: false
-        });
-        Object.defineProperty(this, "raiseError", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: false
-        });
-        Object.defineProperty(this, "awaitHandlers", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: getEnvironmentVariable("LANGCHAIN_CALLBACKS_BACKGROUND") === "false"
-        });
-        this.lc_kwargs = input || {};
-        if (input) {
-            this.ignoreLLM = input.ignoreLLM ?? this.ignoreLLM;
-            this.ignoreChain = input.ignoreChain ?? this.ignoreChain;
-            this.ignoreAgent = input.ignoreAgent ?? this.ignoreAgent;
-            this.ignoreRetriever = input.ignoreRetriever ?? this.ignoreRetriever;
-            this.ignoreCustomEvent =
-                input.ignoreCustomEvent ?? this.ignoreCustomEvent;
-            this.raiseError = input.raiseError ?? this.raiseError;
-            this.awaitHandlers =
-                this.raiseError || (input._awaitHandler ?? this.awaitHandlers);
-        }
-    }
-    copy() {
-        return new this.constructor(this);
-    }
-    toJSON() {
-        return Serializable.prototype.toJSON.call(this);
-    }
-    toJSONNotImplemented() {
-        return Serializable.prototype.toJSONNotImplemented.call(this);
-    }
-    static fromMethods(methods) {
-        class Handler extends BaseCallbackHandler {
-            constructor() {
-                super();
-                Object.defineProperty(this, "name", {
-                    enumerable: true,
-                    configurable: true,
-                    writable: true,
-                    value: uuid.v4()
-                });
-                Object.assign(this, methods);
-            }
-        }
-        return new Handler();
-    }
-}
+* Abstract base class for creating callback handlers in the LangChain
+* framework. It provides a set of optional methods that can be overridden
+* in derived classes to handle various events during the execution of a
+* LangChain application.
+*/
+var BaseCallbackHandler = class extends BaseCallbackHandlerMethodsClass {
+	lc_serializable = false;
+	get lc_namespace() {
+		return [
+			"langchain_core",
+			"callbacks",
+			this.name
+		];
+	}
+	get lc_secrets() {
+		return void 0;
+	}
+	get lc_attributes() {
+		return void 0;
+	}
+	get lc_aliases() {
+		return void 0;
+	}
+	get lc_serializable_keys() {
+		return void 0;
+	}
+	/**
+	* The name of the serializable. Override to provide an alias or
+	* to preserve the serialized module name in minified environments.
+	*
+	* Implemented as a static method to support loading logic.
+	*/
+	static lc_name() {
+		return this.name;
+	}
+	/**
+	* The final serialized identifier for the module.
+	*/
+	get lc_id() {
+		return [...this.lc_namespace, get_lc_unique_name(this.constructor)];
+	}
+	lc_kwargs;
+	ignoreLLM = false;
+	ignoreChain = false;
+	ignoreAgent = false;
+	ignoreRetriever = false;
+	ignoreCustomEvent = false;
+	raiseError = false;
+	awaitHandlers = getEnvironmentVariable("LANGCHAIN_CALLBACKS_BACKGROUND") === "false";
+	constructor(input) {
+		super();
+		this.lc_kwargs = input || {};
+		if (input) {
+			this.ignoreLLM = input.ignoreLLM ?? this.ignoreLLM;
+			this.ignoreChain = input.ignoreChain ?? this.ignoreChain;
+			this.ignoreAgent = input.ignoreAgent ?? this.ignoreAgent;
+			this.ignoreRetriever = input.ignoreRetriever ?? this.ignoreRetriever;
+			this.ignoreCustomEvent = input.ignoreCustomEvent ?? this.ignoreCustomEvent;
+			this.raiseError = input.raiseError ?? this.raiseError;
+			this.awaitHandlers = this.raiseError || (input._awaitHandler ?? this.awaitHandlers);
+		}
+	}
+	copy() {
+		return new this.constructor(this);
+	}
+	toJSON() {
+		return Serializable.prototype.toJSON.call(this);
+	}
+	toJSONNotImplemented() {
+		return Serializable.prototype.toJSONNotImplemented.call(this);
+	}
+	static fromMethods(methods) {
+		class Handler extends BaseCallbackHandler {
+			name = uuid.v4();
+			constructor() {
+				super();
+				Object.assign(this, methods);
+			}
+		}
+		return new Handler();
+	}
+};
+const isBaseCallbackHandler = (x) => {
+	const callbackHandler = x;
+	return callbackHandler !== void 0 && typeof callbackHandler.copy === "function" && typeof callbackHandler.name === "string" && typeof callbackHandler.awaitHandlers === "boolean";
+};
 
 class LangChainCallbackHandler extends BaseCallbackHandler {
   constructor(options) {
@@ -3407,6 +3585,9 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
       if (additionalTokenData.reasoningTokens) {
         eventProperties['$ai_reasoning_tokens'] = additionalTokenData.reasoningTokens;
       }
+      if (additionalTokenData.webSearchCount !== undefined) {
+        eventProperties['$ai_web_search_count'] = additionalTokenData.webSearchCount;
+      }
       // Handle generations/completions
       let completions;
       if (output.generations && Array.isArray(output.generations)) {
@@ -3572,6 +3753,42 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
     } else if (usage.reasoningTokens != null) {
       additionalTokenData.reasoningTokens = usage.reasoningTokens;
     }
+    // Extract web search counts from various provider formats
+    let webSearchCount;
+    // Priority 1: Exact Count
+    // Check Anthropic format (server_tool_use.web_search_requests)
+    if (usage.server_tool_use?.web_search_requests !== undefined) {
+      webSearchCount = usage.server_tool_use.web_search_requests;
+    }
+    // Priority 2: Binary Detection (1 or 0)
+    // Check for citations array (Perplexity)
+    else if (usage.citations && Array.isArray(usage.citations) && usage.citations.length > 0) {
+      webSearchCount = 1;
+    }
+    // Check for search_results array (Perplexity via OpenRouter)
+    else if (usage.search_results && Array.isArray(usage.search_results) && usage.search_results.length > 0) {
+      webSearchCount = 1;
+    }
+    // Check for search_context_size (Perplexity via OpenRouter)
+    else if (usage.search_context_size) {
+      webSearchCount = 1;
+    }
+    // Check for annotations with url_citation type
+    else if (usage.annotations && Array.isArray(usage.annotations)) {
+      const hasUrlCitation = usage.annotations.some(ann => {
+        return ann && typeof ann === 'object' && 'type' in ann && ann.type === 'url_citation';
+      });
+      if (hasUrlCitation) {
+        webSearchCount = 1;
+      }
+    }
+    // Check Gemini format (grounding metadata - binary 0 or 1)
+    else if (usage.grounding_metadata?.grounding_support !== undefined || usage.grounding_metadata?.web_search_queries !== undefined) {
+      webSearchCount = 1;
+    }
+    if (webSearchCount !== undefined) {
+      additionalTokenData.webSearchCount = webSearchCount;
+    }
     // In LangChain, input_tokens is the sum of input and cache read tokens.
     // Our cost calculation expects them to be separate, for Anthropic.
     if (parsedUsage.input && additionalTokenData.cacheReadInputTokens) {