Ykeremy/workflow prompt block (#124)

2024-03-25 00:57:37 -07:00
parent 0b5456a4c6
commit c58aaba4bb
7 changed files with 129 additions and 5 deletions
--- a/skyvern/forge/sdk/db/client.py
+++ b/skyvern/forge/sdk/db/client.py
@@ -692,7 +692,9 @@ class AgentDB:
    ) -> Workflow:
        try:
            async with self.Session() as session:
-                if workflow := await session.scalars(select(WorkflowModel).filter_by(workflow_id=workflow_id).first()):
+                if workflow := (
+                    await session.scalars(select(WorkflowModel).filter_by(workflow_id=workflow_id))
+                ).first():
                    if title:
                        workflow.title = title
                    if description: