first_step

Skyvern-AI · Dec 7, 2024 · c263b98 · c263b98
1 parent 457cf51
commit c263b98
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git 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
@@ -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:
-Original file line number
+Diff line change
@@ Expand Up @@
                 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:
@@ Expand Down @@