Underground-Digital/Workflow-Engine

1from flask_restful import fields2 3from fields.conversation_fields import message_file_fields4from libs.helper import TimestampField5 6from .raws import FilesContainedField7 8feedback_fields = {"rating": fields.String}9 10retriever_resource_fields = {11    "id": fields.String,12    "message_id": fields.String,13    "position": fields.Integer,14    "dataset_id": fields.String,15    "dataset_name": fields.String,16    "document_id": fields.String,17    "document_name": fields.String,18    "data_source_type": fields.String,19    "segment_id": fields.String,20    "score": fields.Float,21    "hit_count": fields.Integer,22    "word_count": fields.Integer,23    "segment_position": fields.Integer,24    "index_node_hash": fields.String,25    "content": fields.String,26    "created_at": TimestampField,27}28 29feedback_fields = {"rating": fields.String}30 31agent_thought_fields = {32    "id": fields.String,33    "chain_id": fields.String,34    "message_id": fields.String,35    "position": fields.Integer,36    "thought": fields.String,37    "tool": fields.String,38    "tool_labels": fields.Raw,39    "tool_input": fields.String,40    "created_at": TimestampField,41    "observation": fields.String,42    "files": fields.List(fields.String),43}44 45retriever_resource_fields = {46    "id": fields.String,47    "message_id": fields.String,48    "position": fields.Integer,49    "dataset_id": fields.String,50    "dataset_name": fields.String,51    "document_id": fields.String,52    "document_name": fields.String,53    "data_source_type": fields.String,54    "segment_id": fields.String,55    "score": fields.Float,56    "hit_count": fields.Integer,57    "word_count": fields.Integer,58    "segment_position": fields.Integer,59    "index_node_hash": fields.String,60    "content": fields.String,61    "created_at": TimestampField,62}63 64message_fields = {65    "id": fields.String,66    "conversation_id": fields.String,67    "parent_message_id": fields.String,68    "inputs": FilesContainedField,69    "query": fields.String,70    "answer": fields.String(attribute="re_sign_file_url_answer"),71    "feedback": fields.Nested(feedback_fields, attribute="user_feedback", allow_null=True),72    "retriever_resources": fields.List(fields.Nested(retriever_resource_fields)),73    "created_at": TimestampField,74    "agent_thoughts": fields.List(fields.Nested(agent_thought_fields)),75    "message_files": fields.List(fields.Nested(message_file_fields)),76    "status": fields.String,77    "error": fields.String,78}79 80message_infinite_scroll_pagination_fields = {81    "limit": fields.Integer,82    "has_more": fields.Boolean,83    "data": fields.List(fields.Nested(message_fields)),84}85