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):