add created_at return in publish workflow

langgenius · Mar 15, 2024 · 50482b0 · 50482b0
1 parent c2edf62
commit 50482b0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/api/controllers/console/app/workflow.py b/api/controllers/console/app/workflow.py
@@ -206,10 +206,11 @@ def post(self, app_model: App):
         Publish workflow
         """
         workflow_service = WorkflowService()
-        workflow_service.publish_workflow(app_model=app_model, account=current_user)
+        workflow = workflow_service.publish_workflow(app_model=app_model, account=current_user)
 
         return {
-            "result": "success"
+            "result": "success",
+            "created_at": TimestampField().format(workflow.created_at)
         }
 
 

diff --git a/api/services/workflow_service.py b/api/services/workflow_service.py
@@ -131,6 +131,7 @@ def publish_workflow(self, app_model: App,
 
         # commit db session changes
         db.session.add(workflow)
+        db.session.flush()
         db.session.commit()
 
         app_model.workflow_id = workflow.id