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(