From c263b98726de2fb65cb2bd7520f5a7159ec1784a Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Sat, 7 Dec 2024 12:14:34 -0800 Subject: [PATCH] first_step --- skyvern/forge/agent.py | 2 +- skyvern/forge/sdk/db/client.py | 2 -- 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/skyvern/forge/agent.py b/skyvern/forge/agent.py index 95e8eeb01..ee080ec99 100644 --- a/skyvern/forge/agent.py +++ b/skyvern/forge/agent.py @@ -1637,7 +1637,7 @@ async def build_task_response( if first_step: recording_artifact = await app.DATABASE.get_artifact( task_id=task.task_id, - step_id=last_step.step_id, + step_id=first_step.step_id, artifact_type=ArtifactType.RECORDING, organization_id=task.organization_id, ) diff --git a/skyvern/forge/sdk/db/client.py b/skyvern/forge/sdk/db/client.py index cb50a2aa0..4b916f0a3 100644 --- a/skyvern/forge/sdk/db/client.py +++ b/skyvern/forge/sdk/db/client.py @@ -340,8 +340,6 @@ async def get_first_step(self, task_id: str, organization_id: str | None = None) LOG.error("UnexpectedError", exc_info=True) raise - - async def get_latest_step(self, task_id: str, organization_id: str | None = None) -> Step | None: try: async with self.Session() as session: