mirror of
https://github.com/langgenius/dify.git
synced 2025-11-10 00:11:30 +00:00
85cda47c70
Signed-off-by: -LAN- <laipz8200@outlook.com> Co-authored-by: twwu <twwu@dify.ai> Co-authored-by: crazywoola <100913391+crazywoola@users.noreply.github.com> Co-authored-by: jyong <718720800@qq.com> Co-authored-by: Wu Tianwei <30284043+WTW0313@users.noreply.github.com> Co-authored-by: QuantumGhost <obelisk.reg+git@gmail.com> Co-authored-by: lyzno1 <yuanyouhuilyz@gmail.com> Co-authored-by: quicksand <quicksandzn@gmail.com> Co-authored-by: Jyong <76649700+JohnJyong@users.noreply.github.com> Co-authored-by: lyzno1 <92089059+lyzno1@users.noreply.github.com> Co-authored-by: zxhlyh <jasonapring2015@outlook.com> Co-authored-by: Yongtao Huang <yongtaoh2022@gmail.com> Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: Joel <iamjoel007@gmail.com> Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> Co-authored-by: nite-knite <nkCoding@gmail.com> Co-authored-by: Hanqing Zhao <sherry9277@gmail.com> Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> Co-authored-by: Harry <xh001x@hotmail.com>
127 lines
4.5 KiB
Python
127 lines
4.5 KiB
Python
from flask_restx import Api, Namespace, 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,
|
|
"exceptions_count": fields.Integer,
|
|
}
|
|
|
|
|
|
def build_workflow_run_for_log_model(api_or_ns: Api | Namespace):
|
|
return api_or_ns.model("WorkflowRunForLog", workflow_run_for_log_fields)
|
|
|
|
|
|
workflow_run_for_list_fields = {
|
|
"id": fields.String,
|
|
"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,
|
|
"exceptions_count": fields.Integer,
|
|
"retry_index": fields.Integer,
|
|
}
|
|
|
|
advanced_chat_workflow_run_for_list_fields = {
|
|
"id": fields.String,
|
|
"conversation_id": fields.String,
|
|
"message_id": fields.String,
|
|
"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,
|
|
"exceptions_count": fields.Integer,
|
|
"retry_index": fields.Integer,
|
|
}
|
|
|
|
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,
|
|
"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,
|
|
"exceptions_count": fields.Integer,
|
|
}
|
|
|
|
retry_event_field = {
|
|
"elapsed_time": fields.Float,
|
|
"status": fields.String,
|
|
"inputs": fields.Raw(attribute="inputs"),
|
|
"process_data": fields.Raw(attribute="process_data"),
|
|
"outputs": fields.Raw(attribute="outputs"),
|
|
"metadata": fields.Raw(attribute="metadata"),
|
|
"llm_usage": fields.Raw(attribute="llm_usage"),
|
|
"error": fields.String,
|
|
"retry_index": fields.Integer,
|
|
}
|
|
|
|
|
|
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,
|
|
"inputs_truncated": fields.Boolean,
|
|
"outputs_truncated": fields.Boolean,
|
|
"process_data_truncated": fields.Boolean,
|
|
}
|
|
|
|
workflow_run_node_execution_list_fields = {
|
|
"data": fields.List(fields.Nested(workflow_run_node_execution_fields)),
|
|
}
|