Skip to content

Commit 96e8189

Browse files
deersharkhassiebp
andauthored
fix(langchain): Add handling for invalid tool calls in CallbackHandler (#1550)
Co-authored-by: Hassieb Pakzad <68423100+hassiebp@users.noreply.github.com>
1 parent 849db5e commit 96e8189

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

langfuse/langchain/CallbackHandler.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1057,6 +1057,13 @@ def _convert_message_to_dict(self, message: BaseMessage) -> Dict[str, Any]:
10571057
and len(message.tool_calls) > 0
10581058
):
10591059
message_dict["tool_calls"] = message.tool_calls
1060+
1061+
if (
1062+
hasattr(message, "invalid_tool_calls")
1063+
and message.invalid_tool_calls is not None
1064+
and len(message.invalid_tool_calls) > 0
1065+
):
1066+
message_dict["invalid_tool_calls"] = message.invalid_tool_calls
10601067

10611068
elif isinstance(message, SystemMessage):
10621069
message_dict = {"role": "system", "content": message.content}

0 commit comments

Comments
 (0)