From 456028e35718fdcf867573c5fc387240760de1a0 Mon Sep 17 00:00:00 2001
From: Hassieb Pakzad <68423100+hassiebp@users.noreply.github.com>
Date: Wed, 2 Apr 2025 16:44:43 +0200
Subject: [PATCH] fix(langchain): capture tool_call_id

---
 langfuse/callback/langchain.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/langfuse/callback/langchain.py b/langfuse/callback/langchain.py
index 2cc316c3b..674ab0b21 100644
--- a/langfuse/callback/langchain.py
+++ b/langfuse/callback/langchain.py
@@ -997,7 +997,11 @@ def _convert_message_to_dict(self, message: BaseMessage) -> Dict[str, Any]:
         elif isinstance(message, SystemMessage):
             message_dict = {"role": "system", "content": message.content}
         elif isinstance(message, ToolMessage):
-            message_dict = {"role": "tool", "content": message.content}
+            message_dict = {
+                "role": "tool",
+                "content": message.content,
+                "tool_call_id": message.tool_call_id,
+            }
         elif isinstance(message, FunctionMessage):
             message_dict = {"role": "function", "content": message.content}
         elif isinstance(message, ChatMessage):