fix: _convert_ai_message

explodinggradients · Jan 25, 2025 · 50b1c27 · 50b1c27
1 parent 17ba21e
commit 50b1c27
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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(