From 50b1c277760e244f3b5a8929476315bc28a7d0aa Mon Sep 17 00:00:00 2001 From: i-w-a Date: Sat, 25 Jan 2025 18:53:18 +0900 Subject: [PATCH] fix: _convert_ai_message --- src/ragas/integrations/langgraph.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/ragas/integrations/langgraph.py b/src/ragas/integrations/langgraph.py index 25fa2e31b..ab40114f3 100644 --- a/src/ragas/integrations/langgraph.py +++ b/src/ragas/integrations/langgraph.py @@ -79,7 +79,7 @@ def _extract_tool_calls(message: AIMessage) -> List[r.ToolCall]: for tool_call in tool_calls ] - def _convert_ai_message(message: AIMessage) -> r.AIMessage: + def _convert_ai_message(message: AIMessage, metadata: bool) -> r.AIMessage: tool_calls = _extract_tool_calls(message) if message.additional_kwargs else None if metadata: return r.AIMessage(