@posthog/ai 5.2.2 → 6.0.0

package/{lib → dist}/index.mjs

@@ -2,7 +2,7 @@ import { OpenAI, AzureOpenAI } from 'openai';
 import * as uuid from 'uuid';
 import { v4 } from 'uuid';
 import { Buffer } from 'buffer';
-import { experimental_wrapLanguageModel } from 'ai';
+import { wrapLanguageModel } from 'ai';
 import AnthropicOriginal from '@anthropic-ai/sdk';
 import { GoogleGenAI } from '@google/genai';
 
@@ -23,32 +23,173 @@ const getModelParams = params => {
   return modelParams;
 };
 const formatResponseAnthropic = response => {
-  // Example approach if "response.content" holds array of text segments, etc.
   const output = [];
+  const content = [];
   for (const choice of response.content ?? []) {
-    if (choice?.text) {
-      output.push({
-        role: 'assistant',
-        content: choice.text
+    if (choice?.type === 'text' && choice?.text) {
+      content.push({
+        type: 'text',
+        text: choice.text
+      });
+    } else if (choice?.type === 'tool_use' && choice?.name && choice?.id) {
+      content.push({
+        type: 'function',
+        id: choice.id,
+        function: {
+          name: choice.name,
+          arguments: choice.input || {}
+        }
       });
     }
   }
+  if (content.length > 0) {
+    output.push({
+      role: 'assistant',
+      content
+    });
+  }
   return output;
 };
 const formatResponseOpenAI = response => {
   const output = [];
-  for (const choice of response.choices ?? []) {
-    if (choice.message?.content) {
+  if (response.choices) {
+    for (const choice of response.choices) {
+      const content = [];
+      let role = 'assistant';
+      if (choice.message) {
+        if (choice.message.role) {
+          role = choice.message.role;
+        }
+        if (choice.message.content) {
+          content.push({
+            type: 'text',
+            text: choice.message.content
+          });
+        }
+        if (choice.message.tool_calls) {
+          for (const toolCall of choice.message.tool_calls) {
+            content.push({
+              type: 'function',
+              id: toolCall.id,
+              function: {
+                name: toolCall.function.name,
+                arguments: toolCall.function.arguments
+              }
+            });
+          }
+        }
+      }
+      if (content.length > 0) {
+        output.push({
+          role,
+          content
+        });
+      }
+    }
+  }
+  // Handle Responses API format
+  if (response.output) {
+    const content = [];
+    let role = 'assistant';
+    for (const item of response.output) {
+      if (item.type === 'message') {
+        role = item.role;
+        if (item.content && Array.isArray(item.content)) {
+          for (const contentItem of item.content) {
+            if (contentItem.type === 'output_text' && contentItem.text) {
+              content.push({
+                type: 'text',
+                text: contentItem.text
+              });
+            } else if (contentItem.text) {
+              content.push({
+                type: 'text',
+                text: contentItem.text
+              });
+            } else if (contentItem.type === 'input_image' && contentItem.image_url) {
+              content.push({
+                type: 'image',
+                image: contentItem.image_url
+              });
+            }
+          }
+        } else if (item.content) {
+          content.push({
+            type: 'text',
+            text: String(item.content)
+          });
+        }
+      } else if (item.type === 'function_call') {
+        content.push({
+          type: 'function',
+          id: item.call_id || item.id || '',
+          function: {
+            name: item.name,
+            arguments: item.arguments || {}
+          }
+        });
+      }
+    }
+    if (content.length > 0) {
       output.push({
-        role: choice.message.role,
-        content: choice.message.content
+        role,
+        content
       });
     }
   }
   return output;
 };
+const formatResponseGemini = response => {
+  const output = [];
+  if (response.candidates && Array.isArray(response.candidates)) {
+    for (const candidate of response.candidates) {
+      if (candidate.content && candidate.content.parts) {
+        const content = [];
+        for (const part of candidate.content.parts) {
+          if (part.text) {
+            content.push({
+              type: 'text',
+              text: part.text
+            });
+          } else if (part.functionCall) {
+            content.push({
+              type: 'function',
+              function: {
+                name: part.functionCall.name,
+                arguments: part.functionCall.args
+              }
+            });
+          }
+        }
+        if (content.length > 0) {
+          output.push({
+            role: 'assistant',
+            content
+          });
+        }
+      } else if (candidate.text) {
+        output.push({
+          role: 'assistant',
+          content: [{
+            type: 'text',
+            text: candidate.text
+          }]
+        });
+      }
+    }
+  } else if (response.text) {
+    output.push({
+      role: 'assistant',
+      content: [{
+        type: 'text',
+        text: response.text
+      }]
+    });
+  }
+  return output;
+};
 const mergeSystemPrompt = (params, provider) => {
-  if (provider == 'anthropic') {
+  {
     const messages = params.messages || [];
     if (!params.system) {
       return messages;
@@ -59,7 +200,6 @@ const mergeSystemPrompt = (params, provider) => {
       content: systemMessage
     }, ...messages];
   }
-  return params.messages;
 };
 const withPrivacyMode = (client, privacyMode, input) => {
   return client.privacy_mode || privacyMode ? null : input;
@@ -77,6 +217,35 @@ const truncate = str => {
     return str;
   }
 };
+/**
+ * Extract available tool calls from the request parameters.
+ * These are the tools provided to the LLM, not the tool calls in the response.
+ */
+const extractAvailableToolCalls = (provider, params) => {
+  if (provider === 'anthropic') {
+    if (params.tools) {
+      return params.tools;
+    }
+    return null;
+  } else if (provider === 'gemini') {
+    if (params.config && params.config.tools) {
+      return params.config.tools;
+    }
+    return null;
+  } else if (provider === 'openai') {
+    if (params.tools) {
+      return params.tools;
+    }
+    return null;
+  } else if (provider === 'vercel') {
+    // Vercel AI SDK stores tools in params.mode.tools when mode type is 'regular'
+    if (params.mode?.type === 'regular' && params.mode.tools) {
+      return params.mode.tools;
+    }
+    return null;
+  }
+  return null;
+};
 function sanitizeValues(obj) {
   if (obj === undefined || obj === null) {
     return obj;
@@ -196,13 +365,13 @@ class PostHogOpenAI extends OpenAI {
     this.responses = new WrappedResponses$1(this, this.phClient);
   }
 }
-class WrappedChat$1 extends Chat {
+let WrappedChat$1 = class WrappedChat extends Chat {
   constructor(parentClient, phClient) {
     super(parentClient);
     this.completions = new WrappedCompletions$1(parentClient, phClient);
   }
-}
-class WrappedCompletions$1 extends Completions {
+};
+let WrappedCompletions$1 = class WrappedCompletions extends Completions {
   constructor(client, phClient) {
     super(client);
     this.phClient = phClient;
@@ -246,6 +415,7 @@ class WrappedCompletions$1 extends Completions {
                 }
               }
               const latency = (Date.now() - startTime) / 1000;
+              const availableTools = extractAvailableToolCalls('openai', openAIParams);
               await sendEventToPosthog({
                 client: this.phClient,
                 distinctId: posthogDistinctId,
@@ -262,6 +432,7 @@ class WrappedCompletions$1 extends Completions {
                 params: body,
                 httpStatus: 200,
                 usage,
+                tools: availableTools,
                 captureImmediate: posthogCaptureImmediate
               });
             } catch (error) {
@@ -296,6 +467,7 @@ class WrappedCompletions$1 extends Completions {
       const wrappedPromise = parentPromise.then(async result => {
         if ('choices' in result) {
           const latency = (Date.now() - startTime) / 1000;
+          const availableTools = extractAvailableToolCalls('openai', openAIParams);
           await sendEventToPosthog({
             client: this.phClient,
             distinctId: posthogDistinctId,
@@ -314,6 +486,7 @@ class WrappedCompletions$1 extends Completions {
               reasoningTokens: result.usage?.completion_tokens_details?.reasoning_tokens ?? 0,
               cacheReadInputTokens: result.usage?.prompt_tokens_details?.cached_tokens ?? 0
             },
+            tools: availableTools,
             captureImmediate: posthogCaptureImmediate
           });
         }
@@ -344,8 +517,8 @@ class WrappedCompletions$1 extends Completions {
       return wrappedPromise;
     }
   }
-}
-class WrappedResponses$1 extends Responses {
+};
+let WrappedResponses$1 = class WrappedResponses extends Responses {
   constructor(client, phClient) {
     super(client);
     this.phClient = phClient;
@@ -390,10 +563,12 @@ class WrappedResponses$1 extends Responses {
                 }
               }
               const latency = (Date.now() - startTime) / 1000;
+              const availableTools = extractAvailableToolCalls('openai', openAIParams);
               await sendEventToPosthog({
                 client: this.phClient,
                 distinctId: posthogDistinctId,
                 traceId,
+                //@ts-expect-error
                 model: openAIParams.model,
                 provider: 'openai',
                 input: openAIParams.input,
@@ -403,6 +578,7 @@ class WrappedResponses$1 extends Responses {
                 params: body,
                 httpStatus: 200,
                 usage,
+                tools: availableTools,
                 captureImmediate: posthogCaptureImmediate
               });
             } catch (error) {
@@ -410,6 +586,7 @@ class WrappedResponses$1 extends Responses {
                 client: this.phClient,
                 distinctId: posthogDistinctId,
                 traceId,
+                //@ts-expect-error
                 model: openAIParams.model,
                 provider: 'openai',
                 input: openAIParams.input,
@@ -436,14 +613,18 @@ class WrappedResponses$1 extends Responses {
       const wrappedPromise = parentPromise.then(async result => {
         if ('output' in result) {
           const latency = (Date.now() - startTime) / 1000;
+          const availableTools = extractAvailableToolCalls('openai', openAIParams);
           await sendEventToPosthog({
             client: this.phClient,
             distinctId: posthogDistinctId,
             traceId,
+            //@ts-expect-error
             model: openAIParams.model,
             provider: 'openai',
             input: openAIParams.input,
-            output: result.output,
+            output: formatResponseOpenAI({
+              output: result.output
+            }),
             latency,
             baseURL: this.baseURL ?? '',
             params: body,
@@ -454,6 +635,7 @@ class WrappedResponses$1 extends Responses {
               reasoningTokens: result.usage?.output_tokens_details?.reasoning_tokens ?? 0,
               cacheReadInputTokens: result.usage?.input_tokens_details?.cached_tokens ?? 0
             },
+            tools: availableTools,
             captureImmediate: posthogCaptureImmediate
           });
         }
@@ -463,6 +645,7 @@ class WrappedResponses$1 extends Responses {
           client: this.phClient,
           distinctId: posthogDistinctId,
           traceId,
+          //@ts-expect-error
           model: openAIParams.model,
           provider: 'openai',
           input: openAIParams.input,
@@ -510,6 +693,7 @@ class WrappedResponses$1 extends Responses {
           client: this.phClient,
           distinctId: posthogDistinctId,
           traceId,
+          //@ts-expect-error
           model: openAIParams.model,
           provider: 'openai',
           input: openAIParams.input,
@@ -532,6 +716,7 @@ class WrappedResponses$1 extends Responses {
           client: this.phClient,
           distinctId: posthogDistinctId,
           traceId,
+          //@ts-expect-error
           model: openAIParams.model,
           provider: 'openai',
           input: openAIParams.input,
@@ -556,7 +741,7 @@ class WrappedResponses$1 extends Responses {
       originalSelf.create = tempCreate;
     }
   }
-}
+};
 
 class PostHogAzureOpenAI extends AzureOpenAI {
   constructor(config) {
@@ -767,6 +952,7 @@ class WrappedResponses extends AzureOpenAI.Responses {
                 client: this.phClient,
                 distinctId: posthogDistinctId,
                 traceId,
+                //@ts-expect-error
                 model: openAIParams.model,
                 provider: 'azure',
                 input: openAIParams.input,
@@ -783,6 +969,7 @@ class WrappedResponses extends AzureOpenAI.Responses {
                 client: this.phClient,
                 distinctId: posthogDistinctId,
                 traceId,
+                //@ts-expect-error
                 model: openAIParams.model,
                 provider: 'azure',
                 input: openAIParams.input,
@@ -813,6 +1000,7 @@ class WrappedResponses extends AzureOpenAI.Responses {
             client: this.phClient,
             distinctId: posthogDistinctId,
             traceId,
+            //@ts-expect-error
             model: openAIParams.model,
             provider: 'azure',
             input: openAIParams.input,
@@ -836,6 +1024,7 @@ class WrappedResponses extends AzureOpenAI.Responses {
           client: this.phClient,
           distinctId: posthogDistinctId,
           traceId,
+          //@ts-expect-error
           model: openAIParams.model,
           provider: 'azure',
           input: openAIParams.input,
@@ -877,6 +1066,7 @@ class WrappedResponses extends AzureOpenAI.Responses {
         client: this.phClient,
         distinctId: posthogDistinctId,
         traceId,
+        //@ts-expect-error
         model: openAIParams.model,
         provider: 'azure',
         input: openAIParams.input,
@@ -899,6 +1089,7 @@ class WrappedResponses extends AzureOpenAI.Responses {
         client: this.phClient,
         distinctId: posthogDistinctId,
         traceId,
+        //@ts-expect-error
         model: openAIParams.model,
         provider: 'azure',
         input: openAIParams.input,
@@ -924,7 +1115,7 @@ class WrappedResponses extends AzureOpenAI.Responses {
 const mapVercelParams = params => {
   return {
     temperature: params.temperature,
-    max_tokens: params.maxTokens,
+    max_output_tokens: params.maxOutputTokens,
     top_p: params.topP,
     frequency_penalty: params.frequencyPenalty,
     presence_penalty: params.presencePenalty,
@@ -932,78 +1123,67 @@ const mapVercelParams = params => {
     stream: params.stream
   };
 };
-const mapVercelPrompt = prompt => {
-  // normalize single inputs into an array of messages
-  let promptsArray;
-  if (typeof prompt === 'string') {
-    promptsArray = [{
-      role: 'user',
-      content: prompt
-    }];
-  } else if (!Array.isArray(prompt)) {
-    promptsArray = [prompt];
-  } else {
-    promptsArray = prompt;
-  }
+const mapVercelPrompt = messages => {
   // Map and truncate individual content
-  const inputs = promptsArray.map(p => {
-    let content = {};
-    if (Array.isArray(p.content)) {
-      content = p.content.map(c => {
-        if (c.type === 'text') {
-          return {
-            type: 'text',
-            content: truncate(c.text)
-          };
-        } else if (c.type === 'image') {
-          return {
-            type: 'image',
-            content: {
-              // if image is a url use it, or use "none supported"
-              image: c.image instanceof URL ? c.image.toString() : 'raw images not supported',
-              mimeType: c.mimeType
-            }
-          };
-        } else if (c.type === 'file') {
-          return {
-            type: 'file',
-            content: {
+  const inputs = messages.map(message => {
+    let content;
+    // Handle system role which has string content
+    if (message.role === 'system') {
+      content = [{
+        type: 'text',
+        text: truncate(String(message.content))
+      }];
+    } else {
+      // Handle other roles which have array content
+      if (Array.isArray(message.content)) {
+        content = message.content.map(c => {
+          if (c.type === 'text') {
+            return {
+              type: 'text',
+              text: truncate(c.text)
+            };
+          } else if (c.type === 'file') {
+            return {
+              type: 'file',
               file: c.data instanceof URL ? c.data.toString() : 'raw files not supported',
-              mimeType: c.mimeType
-            }
-          };
-        } else if (c.type === 'tool-call') {
-          return {
-            type: 'tool-call',
-            content: {
+              mediaType: c.mediaType
+            };
+          } else if (c.type === 'reasoning') {
+            return {
+              type: 'reasoning',
+              text: truncate(c.reasoning)
+            };
+          } else if (c.type === 'tool-call') {
+            return {
+              type: 'tool-call',
               toolCallId: c.toolCallId,
               toolName: c.toolName,
-              args: c.args
-            }
-          };
-        } else if (c.type === 'tool-result') {
-          return {
-            type: 'tool-result',
-            content: {
+              input: c.input
+            };
+          } else if (c.type === 'tool-result') {
+            return {
+              type: 'tool-result',
               toolCallId: c.toolCallId,
               toolName: c.toolName,
-              result: c.result,
+              output: c.output,
               isError: c.isError
-            }
+            };
+          }
+          return {
+            type: 'text',
+            text: ''
           };
-        }
-        return {
-          content: ''
-        };
-      });
-    } else {
-      content = {
-        type: 'text',
-        text: truncate(p.content)
-      };
+        });
+      } else {
+        // Fallback for non-array content
+        content = [{
+          type: 'text',
+          text: truncate(String(message.content))
+        }];
+      }
     }
     return {
-      role: p.role,
+      role: message.role,
       content
     };
   });
@@ -1035,7 +1215,32 @@ const mapVercelPrompt = prompt => {
   return inputs;
 };
 const mapVercelOutput = result => {
-  // normalize string results to object
+  const content = [];
+  if (result.text) {
+    content.push({
+      type: 'text',
+      text: truncate(result.text)
+    });
+  }
+  if (result.toolCalls && Array.isArray(result.toolCalls)) {
+    for (const toolCall of result.toolCalls) {
+      content.push({
+        type: 'function',
+        id: toolCall.toolCallId,
+        function: {
+          name: toolCall.toolName,
+          arguments: typeof toolCall.args === 'string' ? toolCall.args : JSON.stringify(toolCall.args)
+        }
+      });
+    }
+  }
+  if (content.length > 0) {
+    return [{
+      role: 'assistant',
+      content: content.length === 1 && content[0].type === 'text' ? content[0].text : content
+    }];
+  }
+  // Fallback to original behavior for other result types TODO: check if we can remove this
   const normalizedResult = typeof result === 'string' ? {
     text: result
   } : result;
@@ -1046,8 +1251,8 @@ const mapVercelOutput = result => {
     ...(normalizedResult.object ? {
       object: normalizedResult.object
     } : {}),
-    ...(normalizedResult.reasoning ? {
-      reasoning: normalizedResult.reasoning
+    ...(normalizedResult.reasoningText ? {
+      reasoning: normalizedResult.reasoningText
     } : {}),
     ...(normalizedResult.response ? {
       response: normalizedResult.response
@@ -1106,14 +1311,14 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
         ...options,
         ...mapVercelParams(params)
       };
+      const availableTools = extractAvailableToolCalls('vercel', params);
       try {
         const result = await doGenerate();
-        const latency = (Date.now() - startTime) / 1000;
         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
         const content = mapVercelOutput(result);
-        // let tools = result.toolCalls
+        const latency = (Date.now() - startTime) / 1000;
         const providerMetadata = result.providerMetadata;
         const additionalTokenValues = {
           ...(providerMetadata?.openai?.reasoningTokens ? {
@@ -1134,19 +1339,17 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
           model: modelId,
           provider: provider,
           input: options.posthogPrivacyMode ? '' : mapVercelPrompt(params.prompt),
-          output: [{
-            content,
-            role: 'assistant'
-          }],
+          output: content,
           latency,
           baseURL,
           params: mergedParams,
           httpStatus: 200,
           usage: {
-            inputTokens: result.usage.promptTokens,
-            outputTokens: result.usage.completionTokens,
+            inputTokens: result.usage.inputTokens,
+            outputTokens: result.usage.outputTokens,
             ...additionalTokenValues
           },
+          tools: availableTools,
           captureImmediate: options.posthogCaptureImmediate
         });
         return result;
@@ -1170,6 +1373,7 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
           },
           isError: true,
           error: truncate(JSON.stringify(error)),
+          tools: availableTools,
           captureImmediate: options.posthogCaptureImmediate
         });
         throw error;
@@ -1181,6 +1385,7 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
     }) => {
       const startTime = Date.now();
       let generatedText = '';
+      let reasoningText = '';
       let usage = {};
       const mergedParams = {
         ...options,
@@ -1188,6 +1393,7 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
       };
       const modelId = options.posthogModelOverride ?? model.modelId;
       const provider = options.posthogProviderOverride ?? extractProvider(model);
+      const availableTools = extractAvailableToolCalls('vercel', params);
       const baseURL = ''; // cannot currently get baseURL from vercel
       try {
         const {
@@ -1196,13 +1402,17 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
         } = await doStream();
         const transformStream = new TransformStream({
           transform(chunk, controller) {
+            // Handle new v5 streaming patterns
             if (chunk.type === 'text-delta') {
-              generatedText += chunk.textDelta;
+              generatedText += chunk.delta;
+            }
+            if (chunk.type === 'reasoning-delta') {
+              reasoningText += chunk.delta; // New in v5
             }
             if (chunk.type === 'finish') {
               usage = {
-                inputTokens: chunk.usage?.promptTokens,
-                outputTokens: chunk.usage?.completionTokens
+                inputTokens: chunk.usage?.inputTokens,
+                outputTokens: chunk.usage?.outputTokens
               };
               if (chunk.providerMetadata?.openai?.reasoningTokens) {
                 usage.reasoningTokens = chunk.providerMetadata.openai.reasoningTokens;
@@ -1221,6 +1431,7 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
           },
           flush: async () => {
             const latency = (Date.now() - startTime) / 1000;
+            const outputContent = reasoningText ? `${reasoningText}\n\n${generatedText}` : generatedText;
             await sendEventToPosthog({
               client: phClient,
               distinctId: options.posthogDistinctId,
@@ -1229,7 +1440,7 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
               provider: provider,
               input: options.posthogPrivacyMode ? '' : mapVercelPrompt(params.prompt),
               output: [{
-                content: generatedText,
+                content: outputContent,
                 role: 'assistant'
               }],
               latency,
@@ -1237,6 +1448,7 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
               params: mergedParams,
               httpStatus: 200,
               usage,
+              tools: availableTools,
               captureImmediate: options.posthogCaptureImmediate
             });
           }
@@ -1264,6 +1476,7 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
           },
           isError: true,
           error: truncate(JSON.stringify(error)),
+          tools: availableTools,
           captureImmediate: options.posthogCaptureImmediate
         });
         throw error;
@@ -1279,7 +1492,7 @@ const wrapVercelLanguageModel = (model, phClient, options) => {
     posthogTraceId: traceId,
     posthogDistinctId: options.posthogDistinctId
   });
-  const wrappedModel = experimental_wrapLanguageModel({
+  const wrappedModel = wrapLanguageModel({
     model,
     middleware
   });
@@ -1346,6 +1559,7 @@ class WrappedMessages extends AnthropicOriginal.Messages {
                 }
               }
               const latency = (Date.now() - startTime) / 1000;
+              const availableTools = extractAvailableToolCalls('anthropic', anthropicParams);
               await sendEventToPosthog({
                 client: this.phClient,
                 distinctId: posthogDistinctId,
@@ -1362,6 +1576,7 @@ class WrappedMessages extends AnthropicOriginal.Messages {
                 params: body,
                 httpStatus: 200,
                 usage,
+                tools: availableTools,
                 captureImmediate: posthogCaptureImmediate
               });
             } catch (error) {
@@ -1372,7 +1587,7 @@ class WrappedMessages extends AnthropicOriginal.Messages {
                 traceId,
                 model: anthropicParams.model,
                 provider: 'anthropic',
-                input: mergeSystemPrompt(anthropicParams, 'anthropic'),
+                input: mergeSystemPrompt(anthropicParams),
                 output: [],
                 latency: 0,
                 baseURL: this.baseURL ?? '',
@@ -1397,13 +1612,14 @@ class WrappedMessages extends AnthropicOriginal.Messages {
       const wrappedPromise = parentPromise.then(async result => {
         if ('content' in result) {
           const latency = (Date.now() - startTime) / 1000;
+          const availableTools = extractAvailableToolCalls('anthropic', anthropicParams);
           await sendEventToPosthog({
             client: this.phClient,
             distinctId: posthogDistinctId,
             traceId,
             model: anthropicParams.model,
             provider: 'anthropic',
-            input: mergeSystemPrompt(anthropicParams, 'anthropic'),
+            input: mergeSystemPrompt(anthropicParams),
             output: formatResponseAnthropic(result),
             latency,
             baseURL: this.baseURL ?? '',
@@ -1415,6 +1631,7 @@ class WrappedMessages extends AnthropicOriginal.Messages {
               cacheCreationInputTokens: result.usage.cache_creation_input_tokens ?? 0,
               cacheReadInputTokens: result.usage.cache_read_input_tokens ?? 0
             },
+            tools: availableTools,
             captureImmediate: posthogCaptureImmediate
           });
         }
@@ -1426,7 +1643,7 @@ class WrappedMessages extends AnthropicOriginal.Messages {
           traceId,
           model: anthropicParams.model,
           provider: 'anthropic',
-          input: mergeSystemPrompt(anthropicParams, 'anthropic'),
+          input: mergeSystemPrompt(anthropicParams),
           output: [],
           latency: 0,
           baseURL: this.baseURL ?? '',
@@ -1477,6 +1694,7 @@ class WrappedModels {
     try {
       const response = await this.client.models.generateContent(geminiParams);
       const latency = (Date.now() - startTime) / 1000;
+      const availableTools = extractAvailableToolCalls('gemini', geminiParams);
       await sendEventToPosthog({
         client: this.phClient,
         distinctId: posthogDistinctId,
@@ -1484,7 +1702,7 @@ class WrappedModels {
         model: geminiParams.model,
         provider: 'gemini',
         input: this.formatInput(geminiParams.contents),
-        output: this.formatOutput(response),
+        output: formatResponseGemini(response),
         latency,
         baseURL: 'https://generativelanguage.googleapis.com',
         params: params,
@@ -1493,6 +1711,7 @@ class WrappedModels {
           inputTokens: response.usageMetadata?.promptTokenCount ?? 0,
           outputTokens: response.usageMetadata?.candidatesTokenCount ?? 0
         },
+        tools: availableTools,
         captureImmediate: posthogCaptureImmediate
       });
       return response;
@@ -1552,6 +1771,7 @@ class WrappedModels {
         yield chunk;
       }
       const latency = (Date.now() - startTime) / 1000;
+      const availableTools = extractAvailableToolCalls('gemini', geminiParams);
       await sendEventToPosthog({
         client: this.phClient,
         distinctId: posthogDistinctId,
@@ -1568,6 +1788,7 @@ class WrappedModels {
         params: params,
         httpStatus: 200,
         usage,
+        tools: availableTools,
         captureImmediate: posthogCaptureImmediate
       });
     } catch (error) {
@@ -1649,487 +1870,523 @@ class WrappedModels {
       content: String(contents)
     }];
   }
-  formatOutput(response) {
-    if (response.text) {
-      return [{
-        role: 'assistant',
-        content: response.text
-      }];
-    }
-    if (response.candidates && Array.isArray(response.candidates)) {
-      return response.candidates.map(candidate => {
-        if (candidate.content && candidate.content.parts) {
-          const text = candidate.content.parts.filter(part => part.text).map(part => part.text).join('');
-          return {
-            role: 'assistant',
-            content: text
-          };
-        }
-        return {
-          role: 'assistant',
-          content: String(candidate)
-        };
-      });
-    }
-    return [];
-  }
 }
 
-var decamelize = function (str, sep) {
-    if (typeof str !== 'string') {
-        throw new TypeError('Expected a string');
-    }
-    sep = typeof sep === 'undefined' ? '_' : sep;
-    return str
-        .replace(/([a-z\d])([A-Z])/g, '$1' + sep + '$2')
-        .replace(/([A-Z]+)([A-Z][a-z\d]+)/g, '$1' + sep + '$2')
-        .toLowerCase();
-};
+function getDefaultExportFromCjs (x) {
+	return x && x.__esModule && Object.prototype.hasOwnProperty.call(x, 'default') ? x['default'] : x;
+}
+
+var decamelize;
+var hasRequiredDecamelize;
+
+function requireDecamelize () {
+	if (hasRequiredDecamelize) return decamelize;
+	hasRequiredDecamelize = 1;
+	decamelize = function (str, sep) {
+		if (typeof str !== 'string') {
+			throw new TypeError('Expected a string');
+		}
+
+		sep = typeof sep === 'undefined' ? '_' : sep;
+
+		return str
+			.replace(/([a-z\d])([A-Z])/g, '$1' + sep + '$2')
+			.replace(/([A-Z]+)([A-Z][a-z\d]+)/g, '$1' + sep + '$2')
+			.toLowerCase();
+	};
+	return decamelize;
+}
+
+var decamelizeExports = requireDecamelize();
+var snakeCase = /*@__PURE__*/getDefaultExportFromCjs(decamelizeExports);
 
 var camelcase = {exports: {}};
 
-const UPPERCASE = /[\p{Lu}]/u;
-const LOWERCASE = /[\p{Ll}]/u;
-const LEADING_CAPITAL = /^[\p{Lu}](?![\p{Lu}])/gu;
-const IDENTIFIER = /([\p{Alpha}\p{N}_]|$)/u;
-const SEPARATORS = /[_.\- ]+/;
-const LEADING_SEPARATORS = new RegExp('^' + SEPARATORS.source);
-const SEPARATORS_AND_IDENTIFIER = new RegExp(SEPARATORS.source + IDENTIFIER.source, 'gu');
-const NUMBERS_AND_IDENTIFIER = new RegExp('\\d+' + IDENTIFIER.source, 'gu');
-const preserveCamelCase = (string, toLowerCase, toUpperCase) => {
-    let isLastCharLower = false;
-    let isLastCharUpper = false;
-    let isLastLastCharUpper = false;
-    for (let i = 0; i < string.length; i++) {
-        const character = string[i];
-        if (isLastCharLower && UPPERCASE.test(character)) {
-            string = string.slice(0, i) + '-' + string.slice(i);
-            isLastCharLower = false;
-            isLastLastCharUpper = isLastCharUpper;
-            isLastCharUpper = true;
-            i++;
-        }
-        else if (isLastCharUpper && isLastLastCharUpper && LOWERCASE.test(character)) {
-            string = string.slice(0, i - 1) + '-' + string.slice(i - 1);
-            isLastLastCharUpper = isLastCharUpper;
-            isLastCharUpper = false;
-            isLastCharLower = true;
-        }
-        else {
-            isLastCharLower = toLowerCase(character) === character && toUpperCase(character) !== character;
-            isLastLastCharUpper = isLastCharUpper;
-            isLastCharUpper = toUpperCase(character) === character && toLowerCase(character) !== character;
-        }
-    }
-    return string;
-};
-const preserveConsecutiveUppercase = (input, toLowerCase) => {
-    LEADING_CAPITAL.lastIndex = 0;
-    return input.replace(LEADING_CAPITAL, m1 => toLowerCase(m1));
-};
-const postProcess = (input, toUpperCase) => {
-    SEPARATORS_AND_IDENTIFIER.lastIndex = 0;
-    NUMBERS_AND_IDENTIFIER.lastIndex = 0;
-    return input.replace(SEPARATORS_AND_IDENTIFIER, (_, identifier) => toUpperCase(identifier))
-        .replace(NUMBERS_AND_IDENTIFIER, m => toUpperCase(m));
-};
-const camelCase = (input, options) => {
-    if (!(typeof input === 'string' || Array.isArray(input))) {
-        throw new TypeError('Expected the input to be `string | string[]`');
-    }
-    options = {
-        pascalCase: false,
-        preserveConsecutiveUppercase: false,
-        ...options
-    };
-    if (Array.isArray(input)) {
-        input = input.map(x => x.trim())
-            .filter(x => x.length)
-            .join('-');
-    }
-    else {
-        input = input.trim();
-    }
-    if (input.length === 0) {
-        return '';
-    }
-    const toLowerCase = options.locale === false ?
-        string => string.toLowerCase() :
-        string => string.toLocaleLowerCase(options.locale);
-    const toUpperCase = options.locale === false ?
-        string => string.toUpperCase() :
-        string => string.toLocaleUpperCase(options.locale);
-    if (input.length === 1) {
-        return options.pascalCase ? toUpperCase(input) : toLowerCase(input);
-    }
-    const hasUpperCase = input !== toLowerCase(input);
-    if (hasUpperCase) {
-        input = preserveCamelCase(input, toLowerCase, toUpperCase);
-    }
-    input = input.replace(LEADING_SEPARATORS, '');
-    if (options.preserveConsecutiveUppercase) {
-        input = preserveConsecutiveUppercase(input, toLowerCase);
-    }
-    else {
-        input = toLowerCase(input);
-    }
-    if (options.pascalCase) {
-        input = toUpperCase(input.charAt(0)) + input.slice(1);
-    }
-    return postProcess(input, toUpperCase);
-};
-camelcase.exports = camelCase;
-// TODO: Remove this for the next major release
-camelcase.exports.default = camelCase;
+var hasRequiredCamelcase;
+
+function requireCamelcase () {
+	if (hasRequiredCamelcase) return camelcase.exports;
+	hasRequiredCamelcase = 1;
+
+	const UPPERCASE = /[\p{Lu}]/u;
+	const LOWERCASE = /[\p{Ll}]/u;
+	const LEADING_CAPITAL = /^[\p{Lu}](?![\p{Lu}])/gu;
+	const IDENTIFIER = /([\p{Alpha}\p{N}_]|$)/u;
+	const SEPARATORS = /[_.\- ]+/;
+
+	const LEADING_SEPARATORS = new RegExp('^' + SEPARATORS.source);
+	const SEPARATORS_AND_IDENTIFIER = new RegExp(SEPARATORS.source + IDENTIFIER.source, 'gu');
+	const NUMBERS_AND_IDENTIFIER = new RegExp('\\d+' + IDENTIFIER.source, 'gu');
+
+	const preserveCamelCase = (string, toLowerCase, toUpperCase) => {
+		let isLastCharLower = false;
+		let isLastCharUpper = false;
+		let isLastLastCharUpper = false;
+
+		for (let i = 0; i < string.length; i++) {
+			const character = string[i];
+
+			if (isLastCharLower && UPPERCASE.test(character)) {
+				string = string.slice(0, i) + '-' + string.slice(i);
+				isLastCharLower = false;
+				isLastLastCharUpper = isLastCharUpper;
+				isLastCharUpper = true;
+				i++;
+			} else if (isLastCharUpper && isLastLastCharUpper && LOWERCASE.test(character)) {
+				string = string.slice(0, i - 1) + '-' + string.slice(i - 1);
+				isLastLastCharUpper = isLastCharUpper;
+				isLastCharUpper = false;
+				isLastCharLower = true;
+			} else {
+				isLastCharLower = toLowerCase(character) === character && toUpperCase(character) !== character;
+				isLastLastCharUpper = isLastCharUpper;
+				isLastCharUpper = toUpperCase(character) === character && toLowerCase(character) !== character;
+			}
+		}
+
+		return string;
+	};
+
+	const preserveConsecutiveUppercase = (input, toLowerCase) => {
+		LEADING_CAPITAL.lastIndex = 0;
+
+		return input.replace(LEADING_CAPITAL, m1 => toLowerCase(m1));
+	};
+
+	const postProcess = (input, toUpperCase) => {
+		SEPARATORS_AND_IDENTIFIER.lastIndex = 0;
+		NUMBERS_AND_IDENTIFIER.lastIndex = 0;
+
+		return input.replace(SEPARATORS_AND_IDENTIFIER, (_, identifier) => toUpperCase(identifier))
+			.replace(NUMBERS_AND_IDENTIFIER, m => toUpperCase(m));
+	};
+
+	const camelCase = (input, options) => {
+		if (!(typeof input === 'string' || Array.isArray(input))) {
+			throw new TypeError('Expected the input to be `string | string[]`');
+		}
 
-function keyToJson(key, map) {
-    return map?.[key] || decamelize(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;
+		options = {
+			pascalCase: false,
+			preserveConsecutiveUppercase: false,
+			...options
+		};
+
+		if (Array.isArray(input)) {
+			input = input.map(x => x.trim())
+				.filter(x => x.length)
+				.join('-');
+		} else {
+			input = input.trim();
+		}
+
+		if (input.length === 0) {
+			return '';
+		}
+
+		const toLowerCase = options.locale === false ?
+			string => string.toLowerCase() :
+			string => string.toLocaleLowerCase(options.locale);
+		const toUpperCase = options.locale === false ?
+			string => string.toUpperCase() :
+			string => string.toLocaleUpperCase(options.locale);
+
+		if (input.length === 1) {
+			return options.pascalCase ? toUpperCase(input) : toLowerCase(input);
+		}
+
+		const hasUpperCase = input !== toLowerCase(input);
+
+		if (hasUpperCase) {
+			input = preserveCamelCase(input, toLowerCase, toUpperCase);
+		}
+
+		input = input.replace(LEADING_SEPARATORS, '');
+
+		if (options.preserveConsecutiveUppercase) {
+			input = preserveConsecutiveUppercase(input, toLowerCase);
+		} else {
+			input = toLowerCase(input);
+		}
+
+		if (options.pascalCase) {
+			input = toUpperCase(input.charAt(0)) + input.slice(1);
+		}
+
+		return postProcess(input, toUpperCase);
+	};
+
+	camelcase.exports = camelCase;
+	// TODO: Remove this for the next major release
+	camelcase.exports.default = camelCase;
+	return camelcase.exports;
 }
 
-function shallowCopy(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;
-}
-/**
- * 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;
-    }
-    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
-        });
-        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,
-        };
-    }
+requireCamelcase();
+
+function keyToJson(key, map) {
+    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;
 }
 
-// Supabase Edge Function provides a `Deno` global object
-// without `version` property
-const isDeno = () => typeof Deno !== "undefined";
-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;
-    }
+function shallowCopy(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;
+}
+/**
+ * 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,
+        };
+    }
 }
 
-/**
- * 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 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;
-    }
-    /**
-     * 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();
-    }
+// Supabase Edge Function provides a `Deno` global object
+// without `version` property
+const isDeno = () => typeof Deno !== "undefined";
+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;
+    }
+}
+
+/**
+ * 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 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();
+    }
 }
 
 class LangChainCallbackHandler extends BaseCallbackHandler {
@@ -2309,6 +2566,9 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
     };
     if (extraParams) {
       generation.modelParams = getModelParams(extraParams.invocation_params);
+      if (extraParams.invocation_params && extraParams.invocation_params.tools) {
+        generation.tools = extraParams.invocation_params.tools;
+      }
     }
     if (metadata) {
       if (metadata.ls_model_name) {
@@ -2415,7 +2675,7 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
       $ai_base_url: run.baseUrl
     };
     if (run.tools) {
-      eventProperties['$ai_tools'] = withPrivacyMode(this.client, this.privacyMode, run.tools);
+      eventProperties['$ai_tools'] = run.tools;
     }
     if (output instanceof Error) {
       eventProperties['$ai_http_status'] = output.status || 500;
@@ -2437,13 +2697,20 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
       let completions;
       if (output.generations && Array.isArray(output.generations)) {
         const lastGeneration = output.generations[output.generations.length - 1];
-        if (Array.isArray(lastGeneration)) {
-          completions = lastGeneration.map(gen => {
-            return {
-              role: 'assistant',
-              content: gen.text
-            };
-          });
+        if (Array.isArray(lastGeneration) && lastGeneration.length > 0) {
+          // Check if this is a ChatGeneration by looking at the first item
+          const isChatGeneration = 'message' in lastGeneration[0] && lastGeneration[0].message;
+          if (isChatGeneration) {
+            // For ChatGeneration, convert messages to dict format
+            completions = lastGeneration.map(gen => {
+              return this._convertMessageToDict(gen.message);
+            });
+          } else {
+            // For non-ChatGeneration, extract raw response
+            completions = lastGeneration.map(gen => {
+              return this._extractRawResponse(gen);
+            });
+          }
         }
       }
       if (completions) {
@@ -2494,6 +2761,19 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
       }
     }));
   }
+  _extractRawResponse(generation) {
+    // Extract the response from the last response of the LLM call
+    // We return the text of the response if not empty
+    if (generation.text != null && generation.text.trim() !== '') {
+      return generation.text.trim();
+    } else if (generation.message) {
+      // Additional kwargs contains the response in case of tool usage
+      return generation.message.additional_kwargs || generation.message.additionalKwargs || {};
+    } else {
+      // Not tool usage, some LLM responses can be simply empty
+      return '';
+    }
+  }
   _convertMessageToDict(message) {
     let messageDict = {};
     const messageType = message.getType();