KeyArgo
/
smolagents
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Restore previous agents.py

This commit is contained in:
Aymeric 2025-01-16 11:54:23 +01:00
parent 96ebe01dc7
commit 72b01a9909
1 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/smolagents/agents.py
View File

@ -75,12 +75,12 @@ class ToolCall:
id: str id: str
class AgentStepLog: class AgentStep:
pass pass
@dataclass @dataclass
class ActionStep(AgentStepLog): class ActionStep(AgentStep):
agent_memory: List[Dict[str, str]] | None = None agent_memory: List[Dict[str, str]] | None = None
tool_calls: List[ToolCall] | None = None tool_calls: List[ToolCall] | None = None
start_time: float | None = None start_time: float | None = None
@ -90,24 +90,22 @@ class ActionStep(AgentStepLog):
duration: float | None = None duration: float | None = None
llm_output: str | None = None llm_output: str | None = None
observations: str | None = None observations: str | None = None
observations_images: List[Image]
action_output: Any = None action_output: Any = None
@dataclass @dataclass
class PlanningStep(AgentStepLog): class PlanningStep(AgentStep):
plan: str plan: str
facts: str facts: str
@dataclass @dataclass
class TaskStep(AgentStepLog): class TaskStep(AgentStep):
task: str task: str
task_images: List[Image]
@dataclass @dataclass
class SystemPromptStep(AgentStepLog): class SystemPromptStep(AgentStep):
system_prompt: str system_prompt: str