@posthog/ai 5.2.1 → 5.2.2

package/CHANGELOG.md

@@ -1,3 +1,7 @@
+# 5.2.2
+
+- Add support for parsing tool calls from reasoning models in LangChain by converting the tool call format to the expected shape
+
 # 5.2.1
 
 - Fix crash when importing @posthog/ai with OpenAI SDK v5.x by deferring access to Chat, Completions, and Responses classes until runtime

package/lib/index.cjs

@@ -2510,10 +2510,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 = {
@@ -2526,6 +2535,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 = {
@@ -2547,9 +2559,10 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
         break;
       default:
         messageDict = {
-          role: messageType || 'unknown',
+          role: messageType,
           content: String(message.content)
         };
+        break;
     }
     if (message.additional_kwargs) {
       messageDict = {