@posthog/ai 5.2.1 → 5.2.2

package/lib/index.d.ts

@@ -10,10 +10,10 @@ import { GoogleGenAI } from '@google/genai';
 import { BaseCallbackHandler } from '@langchain/core/callbacks/base';
 import { Serialized } from '@langchain/core/load/serializable';
 import { ChainValues } from '@langchain/core/utils/types';
-import { BaseMessage } from '@langchain/core/messages';
 import { LLMResult } from '@langchain/core/outputs';
 import { AgentAction, AgentFinish } from '@langchain/core/agents';
 import { DocumentInterface } from '@langchain/core/documents';
+import { BaseMessage } from '@langchain/core/messages';
 
 interface MonitoringParams {
     posthogDistinctId?: string;
@@ -232,6 +232,7 @@ declare class LangChainCallbackHandler extends BaseCallbackHandler {
     private _captureGeneration;
     private _logDebugEvent;
     private _getLangchainRunName;
+    private _convertLcToolCallsToOai;
     private _convertMessageToDict;
     private _parseUsageModel;
     private parseUsage;

package/lib/index.mjs

@@ -2484,10 +2484,19 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
     }
     return undefined;
   }
+  _convertLcToolCallsToOai(toolCalls) {
+    return toolCalls.map(toolCall => ({
+      type: 'function',
+      id: toolCall.id,
+      function: {
+        name: toolCall.name,
+        arguments: JSON.stringify(toolCall.args)
+      }
+    }));
+  }
   _convertMessageToDict(message) {
     let messageDict = {};
-    // Check the _getType() method or type property instead of instanceof
-    const messageType = message._getType?.() || message.type;
+    const messageType = message.getType();
     switch (messageType) {
       case 'human':
         messageDict = {
@@ -2500,6 +2509,9 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
           role: 'assistant',
           content: message.content
         };
+        if (message.tool_calls) {
+          messageDict.tool_calls = this._convertLcToolCallsToOai(message.tool_calls);
+        }
         break;
       case 'system':
         messageDict = {
@@ -2521,9 +2533,10 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
         break;
       default:
         messageDict = {
-          role: messageType || 'unknown',
+          role: messageType,
           content: String(message.content)
         };
+        break;
     }
     if (message.additional_kwargs) {
       messageDict = {