fix: workflow api (#6810)

This commit is contained in:
crazywoola 2024-07-30 23:51:48 +08:00 committed by GitHub
parent 9ce5cea911
commit 3c371a6cb0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -29,9 +29,7 @@ from services.app_generate_service import AppGenerateService
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
workflow_run_fields = {
class WorkflowRunApi(Resource):
workflow_run_fields = {
'id': fields.String, 'id': fields.String,
'workflow_id': fields.String, 'workflow_id': fields.String,
'status': fields.String, 'status': fields.String,
@ -43,8 +41,9 @@ class WorkflowRunApi(Resource):
'created_at': fields.DateTime, 'created_at': fields.DateTime,
'finished_at': fields.DateTime, 'finished_at': fields.DateTime,
'elapsed_time': fields.Float, 'elapsed_time': fields.Float,
} }
class WorkflowRunDetailApi(Resource):
@validate_app_token @validate_app_token
@marshal_with(workflow_run_fields) @marshal_with(workflow_run_fields)
def get(self, app_model: App, workflow_id: str): def get(self, app_model: App, workflow_id: str):
@ -57,7 +56,7 @@ class WorkflowRunApi(Resource):
workflow_run = db.session.query(WorkflowRun).filter(WorkflowRun.id == workflow_id).first() workflow_run = db.session.query(WorkflowRun).filter(WorkflowRun.id == workflow_id).first()
return workflow_run return workflow_run
class WorkflowRunApi(Resource):
@validate_app_token(fetch_user_arg=FetchUserArg(fetch_from=WhereisUserArg.JSON, required=True)) @validate_app_token(fetch_user_arg=FetchUserArg(fetch_from=WhereisUserArg.JSON, required=True))
def post(self, app_model: App, end_user: EndUser): def post(self, app_model: App, end_user: EndUser):
""" """
@ -117,5 +116,6 @@ class WorkflowTaskStopApi(Resource):
} }
api.add_resource(WorkflowRunApi, '/workflows/run/<string:workflow_id>', '/workflows/run') api.add_resource(WorkflowRunApi, '/workflows/run')
api.add_resource(WorkflowRunDetailApi, '/workflows/run/<string:workflow_id>')
api.add_resource(WorkflowTaskStopApi, '/workflows/tasks/<string:task_id>/stop') api.add_resource(WorkflowTaskStopApi, '/workflows/tasks/<string:task_id>/stop')