fix(agents-api): Fix update task query

julep-ai · Dec 26, 2024 · e155bb5 · e155bb5
1 parent 6a5fe63
commit e155bb5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/agents-api/agents_api/queries/tasks/patch_task.py b/agents-api/agents_api/queries/tasks/patch_task.py
@@ -191,7 +191,7 @@ async def patch_task(
                         workflow_name,  # $3
                         step_idx,  # $4
                         step["kind_"],  # $5
-                        step[step["kind_"]],  # $6
+                        step,  # $6
                     ]
                 )
 

diff --git a/agents-api/agents_api/queries/tasks/update_task.py b/agents-api/agents_api/queries/tasks/update_task.py
@@ -154,7 +154,7 @@ async def update_task(
                     workflow_name,  # $3
                     step_idx,  # $4
                     step["kind_"],  # $5
-                    step[step["kind_"]],  # $6
+                    step,  # $6
                 ]
             )