mirror of
https://gitee.com/dify_ai/dify.git
synced 2024-12-05 04:38:37 +08:00
7753ba2d37
Co-authored-by: Joel <iamjoel007@gmail.com> Co-authored-by: Yeuoly <admin@srmxy.cn> Co-authored-by: JzoNg <jzongcode@gmail.com> Co-authored-by: StyleZhang <jasonapring2015@outlook.com> Co-authored-by: jyong <jyong@dify.ai> Co-authored-by: nite-knite <nkCoding@gmail.com> Co-authored-by: jyong <718720800@qq.com>
103 lines
3.7 KiB
Python
103 lines
3.7 KiB
Python
from flask_restful import fields
|
|
|
|
from fields.end_user_fields import simple_end_user_fields
|
|
from fields.member_fields import simple_account_fields
|
|
from libs.helper import TimestampField
|
|
|
|
workflow_run_for_log_fields = {
|
|
"id": fields.String,
|
|
"version": fields.String,
|
|
"status": fields.String,
|
|
"error": fields.String,
|
|
"elapsed_time": fields.Float,
|
|
"total_tokens": fields.Integer,
|
|
"total_steps": fields.Integer,
|
|
"created_at": TimestampField,
|
|
"finished_at": TimestampField
|
|
}
|
|
|
|
workflow_run_for_list_fields = {
|
|
"id": fields.String,
|
|
"sequence_number": fields.Integer,
|
|
"version": fields.String,
|
|
"status": fields.String,
|
|
"elapsed_time": fields.Float,
|
|
"total_tokens": fields.Integer,
|
|
"total_steps": fields.Integer,
|
|
"created_by_account": fields.Nested(simple_account_fields, attribute='created_by_account', allow_null=True),
|
|
"created_at": TimestampField,
|
|
"finished_at": TimestampField
|
|
}
|
|
|
|
advanced_chat_workflow_run_for_list_fields = {
|
|
"id": fields.String,
|
|
"conversation_id": fields.String,
|
|
"message_id": fields.String,
|
|
"sequence_number": fields.Integer,
|
|
"version": fields.String,
|
|
"status": fields.String,
|
|
"elapsed_time": fields.Float,
|
|
"total_tokens": fields.Integer,
|
|
"total_steps": fields.Integer,
|
|
"created_by_account": fields.Nested(simple_account_fields, attribute='created_by_account', allow_null=True),
|
|
"created_at": TimestampField,
|
|
"finished_at": TimestampField
|
|
}
|
|
|
|
advanced_chat_workflow_run_pagination_fields = {
|
|
'limit': fields.Integer(attribute='limit'),
|
|
'has_more': fields.Boolean(attribute='has_more'),
|
|
'data': fields.List(fields.Nested(advanced_chat_workflow_run_for_list_fields), attribute='data')
|
|
}
|
|
|
|
workflow_run_pagination_fields = {
|
|
'limit': fields.Integer(attribute='limit'),
|
|
'has_more': fields.Boolean(attribute='has_more'),
|
|
'data': fields.List(fields.Nested(workflow_run_for_list_fields), attribute='data')
|
|
}
|
|
|
|
workflow_run_detail_fields = {
|
|
"id": fields.String,
|
|
"sequence_number": fields.Integer,
|
|
"version": fields.String,
|
|
"graph": fields.Raw(attribute='graph_dict'),
|
|
"inputs": fields.Raw(attribute='inputs_dict'),
|
|
"status": fields.String,
|
|
"outputs": fields.Raw(attribute='outputs_dict'),
|
|
"error": fields.String,
|
|
"elapsed_time": fields.Float,
|
|
"total_tokens": fields.Integer,
|
|
"total_steps": fields.Integer,
|
|
"created_by_role": fields.String,
|
|
"created_by_account": fields.Nested(simple_account_fields, attribute='created_by_account', allow_null=True),
|
|
"created_by_end_user": fields.Nested(simple_end_user_fields, attribute='created_by_end_user', allow_null=True),
|
|
"created_at": TimestampField,
|
|
"finished_at": TimestampField
|
|
}
|
|
|
|
workflow_run_node_execution_fields = {
|
|
"id": fields.String,
|
|
"index": fields.Integer,
|
|
"predecessor_node_id": fields.String,
|
|
"node_id": fields.String,
|
|
"node_type": fields.String,
|
|
"title": fields.String,
|
|
"inputs": fields.Raw(attribute='inputs_dict'),
|
|
"process_data": fields.Raw(attribute='process_data_dict'),
|
|
"outputs": fields.Raw(attribute='outputs_dict'),
|
|
"status": fields.String,
|
|
"error": fields.String,
|
|
"elapsed_time": fields.Float,
|
|
"execution_metadata": fields.Raw(attribute='execution_metadata_dict'),
|
|
"extras": fields.Raw,
|
|
"created_at": TimestampField,
|
|
"created_by_role": fields.String,
|
|
"created_by_account": fields.Nested(simple_account_fields, attribute='created_by_account', allow_null=True),
|
|
"created_by_end_user": fields.Nested(simple_end_user_fields, attribute='created_by_end_user', allow_null=True),
|
|
"finished_at": TimestampField
|
|
}
|
|
|
|
workflow_run_node_execution_list_fields = {
|
|
'data': fields.List(fields.Nested(workflow_run_node_execution_fields)),
|
|
}
|