From ee6899c68aff84003d79e3bc611874395190f07a Mon Sep 17 00:00:00 2001 From: SwiftyOS Date: Mon, 14 Aug 2023 13:14:35 +0200 Subject: [PATCH] Updated step handler to mark step as complete --- autogpt/agent.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/autogpt/agent.py b/autogpt/agent.py index 607dedbc..0f59e4b2 100644 --- a/autogpt/agent.py +++ b/autogpt/agent.py @@ -4,6 +4,8 @@ import autogpt.utils class AutoGPT: + + def __init__(self) -> None: pass @@ -14,8 +16,7 @@ class AutoGPT: async def step_handler(self, step: Step) -> Step: print(f"step: {step.input}") - await Agent.db.create_step( - step.task_id, f"Nothing to see here.. {step.name}", is_last=True - ) - step.output = step.input - return step + step = Agent.db.get_step(step.step_id) + updated_step: Step = Agent.db.update_step(step.step_id, status="completed") + updated_step.output = step.input + return updated_step