Skip to content

Commit 5f598cc

Browse files
authored
fix(langchain): capture tool_call_id (#1155)
1 parent d6f2e69 commit 5f598cc

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

langfuse/callback/langchain.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -997,7 +997,11 @@ def _convert_message_to_dict(self, message: BaseMessage) -> Dict[str, Any]:
997997
elif isinstance(message, SystemMessage):
998998
message_dict = {"role": "system", "content": message.content}
999999
elif isinstance(message, ToolMessage):
1000-
message_dict = {"role": "tool", "content": message.content}
1000+
message_dict = {
1001+
"role": "tool",
1002+
"content": message.content,
1003+
"tool_call_id": message.tool_call_id,
1004+
}
10011005
elif isinstance(message, FunctionMessage):
10021006
message_dict = {"role": "function", "content": message.content}
10031007
elif isinstance(message, ChatMessage):

0 commit comments

Comments
 (0)