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

View File

@ -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