Underground-Digital/Workflow-Engine

1from flask_restful import fields2 3from fields.end_user_fields import simple_end_user_fields4from fields.member_fields import simple_account_fields5from libs.helper import TimestampField6 7workflow_run_for_log_fields = {8    "id": fields.String,9    "version": fields.String,10    "status": fields.String,11    "error": fields.String,12    "elapsed_time": fields.Float,13    "total_tokens": fields.Integer,14    "total_steps": fields.Integer,15    "created_at": TimestampField,16    "finished_at": TimestampField,17}18 19workflow_run_for_list_fields = {20    "id": fields.String,21    "sequence_number": fields.Integer,22    "version": fields.String,23    "status": fields.String,24    "elapsed_time": fields.Float,25    "total_tokens": fields.Integer,26    "total_steps": fields.Integer,27    "created_by_account": fields.Nested(simple_account_fields, attribute="created_by_account", allow_null=True),28    "created_at": TimestampField,29    "finished_at": TimestampField,30}31 32advanced_chat_workflow_run_for_list_fields = {33    "id": fields.String,34    "conversation_id": fields.String,35    "message_id": fields.String,36    "sequence_number": fields.Integer,37    "version": fields.String,38    "status": fields.String,39    "elapsed_time": fields.Float,40    "total_tokens": fields.Integer,41    "total_steps": fields.Integer,42    "created_by_account": fields.Nested(simple_account_fields, attribute="created_by_account", allow_null=True),43    "created_at": TimestampField,44    "finished_at": TimestampField,45}46 47advanced_chat_workflow_run_pagination_fields = {48    "limit": fields.Integer(attribute="limit"),49    "has_more": fields.Boolean(attribute="has_more"),50    "data": fields.List(fields.Nested(advanced_chat_workflow_run_for_list_fields), attribute="data"),51}52 53workflow_run_pagination_fields = {54    "limit": fields.Integer(attribute="limit"),55    "has_more": fields.Boolean(attribute="has_more"),56    "data": fields.List(fields.Nested(workflow_run_for_list_fields), attribute="data"),57}58 59workflow_run_detail_fields = {60    "id": fields.String,61    "sequence_number": fields.Integer,62    "version": fields.String,63    "graph": fields.Raw(attribute="graph_dict"),64    "inputs": fields.Raw(attribute="inputs_dict"),65    "status": fields.String,66    "outputs": fields.Raw(attribute="outputs_dict"),67    "error": fields.String,68    "elapsed_time": fields.Float,69    "total_tokens": fields.Integer,70    "total_steps": fields.Integer,71    "created_by_role": fields.String,72    "created_by_account": fields.Nested(simple_account_fields, attribute="created_by_account", allow_null=True),73    "created_by_end_user": fields.Nested(simple_end_user_fields, attribute="created_by_end_user", allow_null=True),74    "created_at": TimestampField,75    "finished_at": TimestampField,76}77 78workflow_run_node_execution_fields = {79    "id": fields.String,80    "index": fields.Integer,81    "predecessor_node_id": fields.String,82    "node_id": fields.String,83    "node_type": fields.String,84    "title": fields.String,85    "inputs": fields.Raw(attribute="inputs_dict"),86    "process_data": fields.Raw(attribute="process_data_dict"),87    "outputs": fields.Raw(attribute="outputs_dict"),88    "status": fields.String,89    "error": fields.String,90    "elapsed_time": fields.Float,91    "execution_metadata": fields.Raw(attribute="execution_metadata_dict"),92    "extras": fields.Raw,93    "created_at": TimestampField,94    "created_by_role": fields.String,95    "created_by_account": fields.Nested(simple_account_fields, attribute="created_by_account", allow_null=True),96    "created_by_end_user": fields.Nested(simple_end_user_fields, attribute="created_by_end_user", allow_null=True),97    "finished_at": TimestampField,98}99 100workflow_run_node_execution_list_fields = {101    "data": fields.List(fields.Nested(workflow_run_node_execution_fields)),102}103