@posthog/ai 5.2.3 → 6.0.0

package/dist/index.cjs

@@ -43,30 +43,171 @@ 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) => {
   {
     const messages = params.messages || [];
@@ -96,6 +237,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;
@@ -265,6 +435,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
                 }
               }
               const latency = (Date.now() - startTime) / 1000;
+              const availableTools = extractAvailableToolCalls('openai', openAIParams);
               await sendEventToPosthog({
                 client: this.phClient,
                 distinctId: posthogDistinctId,
@@ -281,6 +452,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
                 params: body,
                 httpStatus: 200,
                 usage,
+                tools: availableTools,
                 captureImmediate: posthogCaptureImmediate
               });
             } catch (error) {
@@ -315,6 +487,7 @@ let WrappedCompletions$1 = class WrappedCompletions 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,
@@ -333,6 +506,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
               reasoningTokens: result.usage?.completion_tokens_details?.reasoning_tokens ?? 0,
               cacheReadInputTokens: result.usage?.prompt_tokens_details?.cached_tokens ?? 0
             },
+            tools: availableTools,
             captureImmediate: posthogCaptureImmediate
           });
         }
@@ -409,6 +583,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
                 }
               }
               const latency = (Date.now() - startTime) / 1000;
+              const availableTools = extractAvailableToolCalls('openai', openAIParams);
               await sendEventToPosthog({
                 client: this.phClient,
                 distinctId: posthogDistinctId,
@@ -423,6 +598,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
                 params: body,
                 httpStatus: 200,
                 usage,
+                tools: availableTools,
                 captureImmediate: posthogCaptureImmediate
               });
             } catch (error) {
@@ -457,6 +633,7 @@ let WrappedResponses$1 = class WrappedResponses 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,
@@ -465,7 +642,9 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
             model: openAIParams.model,
             provider: 'openai',
             input: openAIParams.input,
-            output: result.output,
+            output: formatResponseOpenAI({
+              output: result.output
+            }),
             latency,
             baseURL: this.baseURL ?? '',
             params: body,
@@ -476,6 +655,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
               reasoningTokens: result.usage?.output_tokens_details?.reasoning_tokens ?? 0,
               cacheReadInputTokens: result.usage?.input_tokens_details?.cached_tokens ?? 0
             },
+            tools: availableTools,
             captureImmediate: posthogCaptureImmediate
           });
         }
@@ -955,7 +1135,7 @@ class WrappedResponses extends openai.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,
@@ -963,78 +1143,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
     };
   });
@@ -1066,7 +1235,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;
@@ -1077,8 +1271,8 @@ const mapVercelOutput = result => {
     ...(normalizedResult.object ? {
       object: normalizedResult.object
     } : {}),
-    ...(normalizedResult.reasoning ? {
-      reasoning: normalizedResult.reasoning
+    ...(normalizedResult.reasoningText ? {
+      reasoning: normalizedResult.reasoningText
     } : {}),
     ...(normalizedResult.response ? {
       response: normalizedResult.response
@@ -1137,14 +1331,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 ? {
@@ -1165,19 +1359,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;
@@ -1201,6 +1393,7 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
           },
           isError: true,
           error: truncate(JSON.stringify(error)),
+          tools: availableTools,
           captureImmediate: options.posthogCaptureImmediate
         });
         throw error;
@@ -1212,6 +1405,7 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
     }) => {
       const startTime = Date.now();
       let generatedText = '';
+      let reasoningText = '';
       let usage = {};
       const mergedParams = {
         ...options,
@@ -1219,6 +1413,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 {
@@ -1227,13 +1422,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;
@@ -1252,6 +1451,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,
@@ -1260,7 +1460,7 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
               provider: provider,
               input: options.posthogPrivacyMode ? '' : mapVercelPrompt(params.prompt),
               output: [{
-                content: generatedText,
+                content: outputContent,
                 role: 'assistant'
               }],
               latency,
@@ -1268,6 +1468,7 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
               params: mergedParams,
               httpStatus: 200,
               usage,
+              tools: availableTools,
               captureImmediate: options.posthogCaptureImmediate
             });
           }
@@ -1295,6 +1496,7 @@ const createInstrumentationMiddleware = (phClient, model, options) => {
           },
           isError: true,
           error: truncate(JSON.stringify(error)),
+          tools: availableTools,
           captureImmediate: options.posthogCaptureImmediate
         });
         throw error;
@@ -1310,7 +1512,7 @@ const wrapVercelLanguageModel = (model, phClient, options) => {
     posthogTraceId: traceId,
     posthogDistinctId: options.posthogDistinctId
   });
-  const wrappedModel = ai.experimental_wrapLanguageModel({
+  const wrappedModel = ai.wrapLanguageModel({
     model,
     middleware
   });
@@ -1377,6 +1579,7 @@ class WrappedMessages extends AnthropicOriginal.Messages {
                 }
               }
               const latency = (Date.now() - startTime) / 1000;
+              const availableTools = extractAvailableToolCalls('anthropic', anthropicParams);
               await sendEventToPosthog({
                 client: this.phClient,
                 distinctId: posthogDistinctId,
@@ -1393,6 +1596,7 @@ class WrappedMessages extends AnthropicOriginal.Messages {
                 params: body,
                 httpStatus: 200,
                 usage,
+                tools: availableTools,
                 captureImmediate: posthogCaptureImmediate
               });
             } catch (error) {
@@ -1428,6 +1632,7 @@ 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,
@@ -1446,6 +1651,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
           });
         }
@@ -1508,6 +1714,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,
@@ -1515,7 +1722,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,
@@ -1524,6 +1731,7 @@ class WrappedModels {
           inputTokens: response.usageMetadata?.promptTokenCount ?? 0,
           outputTokens: response.usageMetadata?.candidatesTokenCount ?? 0
         },
+        tools: availableTools,
         captureImmediate: posthogCaptureImmediate
       });
       return response;
@@ -1583,6 +1791,7 @@ class WrappedModels {
         yield chunk;
       }
       const latency = (Date.now() - startTime) / 1000;
+      const availableTools = extractAvailableToolCalls('gemini', geminiParams);
       await sendEventToPosthog({
         client: this.phClient,
         distinctId: posthogDistinctId,
@@ -1599,6 +1808,7 @@ class WrappedModels {
         params: params,
         httpStatus: 200,
         usage,
+        tools: availableTools,
         captureImmediate: posthogCaptureImmediate
       });
     } catch (error) {
@@ -1680,30 +1890,6 @@ 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 [];
-  }
 }
 
 function getDefaultExportFromCjs (x) {
@@ -2400,7 +2586,7 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
     };
     if (extraParams) {
       generation.modelParams = getModelParams(extraParams.invocation_params);
-      if (extraParams.invocation_params.tools) {
+      if (extraParams.invocation_params && extraParams.invocation_params.tools) {
         generation.tools = extraParams.invocation_params.tools;
       }
     }
@@ -2509,7 +2695,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;
@@ -2531,13 +2717,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) {
@@ -2588,6 +2781,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();