From b0cf8c00dbea80dc46c4f42d000f7de5f2c27447 Mon Sep 17 00:00:00 2001
From: takatost <takatost@gmail.com>
Date: Fri, 15 Mar 2024 22:08:25 +0800
Subject: [PATCH] add created_at return in publish workflow

---
 api/controllers/console/app/workflow.py | 5 +++--
 api/services/workflow_service.py        | 1 +
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/api/controllers/console/app/workflow.py b/api/controllers/console/app/workflow.py
index 4994e464ba..0b6aa64291 100644
--- a/api/controllers/console/app/workflow.py
+++ b/api/controllers/console/app/workflow.py
@@ -206,10 +206,11 @@ class PublishedWorkflowApi(Resource):
         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
index a768a4a55b..336c6c1aa0 100644
--- a/api/services/workflow_service.py
+++ b/api/services/workflow_service.py
@@ -131,6 +131,7 @@ class WorkflowService:
 
         # commit db session changes
         db.session.add(workflow)
+        db.session.flush()
         db.session.commit()
 
         app_model.workflow_id = workflow.id