From 72b01a990980a752b02735d1a20da30e992ea783 Mon Sep 17 00:00:00 2001
From: Aymeric <aymeric.roucher@gmail.com>
Date: Thu, 16 Jan 2025 11:54:23 +0100
Subject: [PATCH] Restore previous agents.py

---
 src/smolagents/agents.py | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

diff --git a/src/smolagents/agents.py b/src/smolagents/agents.py
index 6a2bcde..cfa8a6f 100644
--- a/src/smolagents/agents.py
+++ b/src/smolagents/agents.py
@@ -75,12 +75,12 @@ class ToolCall:
     id: str
 
 
-class AgentStepLog:
+class AgentStep:
     pass
 
 
 @dataclass
-class ActionStep(AgentStepLog):
+class ActionStep(AgentStep):
     agent_memory: List[Dict[str, str]] | None = None
     tool_calls: List[ToolCall] | None = None
     start_time: float | None = None
@@ -90,24 +90,22 @@ class ActionStep(AgentStepLog):
     duration: float | None = None
     llm_output: str | None = None
     observations: str | None = None
-    observations_images: List[Image]
     action_output: Any = None
 
 
 @dataclass
-class PlanningStep(AgentStepLog):
+class PlanningStep(AgentStep):
     plan: str
     facts: str
 
 
 @dataclass
-class TaskStep(AgentStepLog):
+class TaskStep(AgentStep):
     task: str
-    task_images: List[Image]
 
 
 @dataclass
-class SystemPromptStep(AgentStepLog):
+class SystemPromptStep(AgentStep):
     system_prompt: str