Fix Access of content Field in ChatMessage Object (#1533)

Co-authored-by: Albert Villanova del Moral <[email protected]>

Author: mdr223

src/smolagents/agents.py

@@ -860,7 +860,7 @@ def __call__(self, task: str, **kwargs):
         if self.provide_run_summary:
             answer += "\n\nFor more detail, find below a summary of this agent's work:\n<summary_of_work>\n"
             for message in self.write_memory_to_messages(summary_mode=True):
-                content = message["content"]
+                content = message.content
                 answer += "\n" + truncate_content(str(content)) + "\n---"
             answer += "\n</summary_of_work>"
         return answer

tests/test_agents.py

@@ -1786,7 +1786,9 @@ def test_call_with_provide_run_summary(self, provide_run_summary):
         assert agent.provide_run_summary is provide_run_summary
         agent.name = "test_agent"
         agent.run = MagicMock(return_value="Test output")
-        agent.write_memory_to_messages = MagicMock(return_value=[{"content": "Test summary"}])
+        agent.write_memory_to_messages = MagicMock(
+            return_value=[ChatMessage(role=MessageRole.ASSISTANT, content="Test summary")]
+        )
 
         result = agent("Test request")
         expected_summary = "Here is the final answer from your managed agent 'test_agent':\nTest output"