fix: correcting tool_call role (#779)

huggingface · Mar 3, 2025 · 34a3913 · 34a3913
1 parent 1077862
commit 34a3913
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/smolagents/memory.py b/src/smolagents/memory.py
@@ -89,7 +89,7 @@ def to_messages(self, summary_mode: bool = False, show_model_input_messages: boo
         if self.tool_calls is not None:
             messages.append(
                 Message(
-                    role=MessageRole.ASSISTANT,
+                    role=MessageRole.TOOL_CALL,
                     content=[
                         {
                             "type": "text",