chore(api): Introduce Ruff Formatter. (#7291)

2024-08-15 12:54:05 +08:00
parent 8f16165f92
commit 3571292fbf
61 changed files with 1315 additions and 1335 deletions
--- a/api/fields/conversation_fields.py
+++ b/api/fields/conversation_fields.py
@@ -6,205 +6,202 @@ from libs.helper import TimestampField

 class MessageTextField(fields.Raw):
    def format(self, value):
-        return value[0]['text'] if value else ''
+        return value[0]["text"] if value else ""


 feedback_fields = {
-    'rating': fields.String,
-    'content': fields.String,
-    'from_source': fields.String,
-    'from_end_user_id': fields.String,
-    'from_account': fields.Nested(simple_account_fields, allow_null=True),
+    "rating": fields.String,
+    "content": fields.String,
+    "from_source": fields.String,
+    "from_end_user_id": fields.String,
+    "from_account": fields.Nested(simple_account_fields, allow_null=True),
 }

 annotation_fields = {
-    'id': fields.String,
-    'question': fields.String,
-    'content': fields.String,
-    'account': fields.Nested(simple_account_fields, allow_null=True),
-    'created_at': TimestampField
+    "id": fields.String,
+    "question": fields.String,
+    "content": fields.String,
+    "account": fields.Nested(simple_account_fields, allow_null=True),
+    "created_at": TimestampField,
 }

 annotation_hit_history_fields = {
-    'annotation_id': fields.String(attribute='id'),
-    'annotation_create_account': fields.Nested(simple_account_fields, allow_null=True),
-    'created_at': TimestampField
+    "annotation_id": fields.String(attribute="id"),
+    "annotation_create_account": fields.Nested(simple_account_fields, allow_null=True),
+    "created_at": TimestampField,
 }

 message_file_fields = {
-    'id': fields.String,
-    'type': fields.String,
-    'url': fields.String,
-    'belongs_to': fields.String(default='user'),
+    "id": fields.String,
+    "type": fields.String,
+    "url": fields.String,
+    "belongs_to": fields.String(default="user"),
 }

 agent_thought_fields = {
-    'id': fields.String,
-    'chain_id': fields.String,
-    'message_id': fields.String,
-    'position': fields.Integer,
-    'thought': fields.String,
-    'tool': fields.String,
-    'tool_labels': fields.Raw,
-    'tool_input': fields.String,
-    'created_at': TimestampField,
-    'observation': fields.String,
-    'files': fields.List(fields.String),
+    "id": fields.String,
+    "chain_id": fields.String,
+    "message_id": fields.String,
+    "position": fields.Integer,
+    "thought": fields.String,
+    "tool": fields.String,
+    "tool_labels": fields.Raw,
+    "tool_input": fields.String,
+    "created_at": TimestampField,
+    "observation": fields.String,
+    "files": fields.List(fields.String),
 }

 message_detail_fields = {
-    'id': fields.String,
-    'conversation_id': fields.String,
-    'inputs': fields.Raw,
-    'query': fields.String,
-    'message': fields.Raw,
-    'message_tokens': fields.Integer,
-    'answer': fields.String(attribute='re_sign_file_url_answer'),
-    'answer_tokens': fields.Integer,
-    'provider_response_latency': fields.Float,
-    'from_source': fields.String,
-    'from_end_user_id': fields.String,
-    'from_account_id': fields.String,
-    'feedbacks': fields.List(fields.Nested(feedback_fields)),
-    'workflow_run_id': fields.String,
-    'annotation': fields.Nested(annotation_fields, allow_null=True),
-    'annotation_hit_history': fields.Nested(annotation_hit_history_fields, allow_null=True),
-    'created_at': TimestampField,
-    'agent_thoughts': fields.List(fields.Nested(agent_thought_fields)),
-    'message_files': fields.List(fields.Nested(message_file_fields), attribute='files'),
-    'metadata': fields.Raw(attribute='message_metadata_dict'),
-    'status': fields.String,
-    'error': fields.String,
+    "id": fields.String,
+    "conversation_id": fields.String,
+    "inputs": fields.Raw,
+    "query": fields.String,
+    "message": fields.Raw,
+    "message_tokens": fields.Integer,
+    "answer": fields.String(attribute="re_sign_file_url_answer"),
+    "answer_tokens": fields.Integer,
+    "provider_response_latency": fields.Float,
+    "from_source": fields.String,
+    "from_end_user_id": fields.String,
+    "from_account_id": fields.String,
+    "feedbacks": fields.List(fields.Nested(feedback_fields)),
+    "workflow_run_id": fields.String,
+    "annotation": fields.Nested(annotation_fields, allow_null=True),
+    "annotation_hit_history": fields.Nested(annotation_hit_history_fields, allow_null=True),
+    "created_at": TimestampField,
+    "agent_thoughts": fields.List(fields.Nested(agent_thought_fields)),
+    "message_files": fields.List(fields.Nested(message_file_fields), attribute="files"),
+    "metadata": fields.Raw(attribute="message_metadata_dict"),
+    "status": fields.String,
+    "error": fields.String,
 }

-feedback_stat_fields = {
-    'like': fields.Integer,
-    'dislike': fields.Integer
-}
+feedback_stat_fields = {"like": fields.Integer, "dislike": fields.Integer}

 model_config_fields = {
-    'opening_statement': fields.String,
-    'suggested_questions': fields.Raw,
-    'model': fields.Raw,
-    'user_input_form': fields.Raw,
-    'pre_prompt': fields.String,
-    'agent_mode': fields.Raw,
+    "opening_statement": fields.String,
+    "suggested_questions": fields.Raw,
+    "model": fields.Raw,
+    "user_input_form": fields.Raw,
+    "pre_prompt": fields.String,
+    "agent_mode": fields.Raw,
 }

 simple_configs_fields = {
-    'prompt_template': fields.String,
+    "prompt_template": fields.String,
 }

 simple_model_config_fields = {
-    'model': fields.Raw(attribute='model_dict'),
-    'pre_prompt': fields.String,
+    "model": fields.Raw(attribute="model_dict"),
+    "pre_prompt": fields.String,
 }

 simple_message_detail_fields = {
-    'inputs': fields.Raw,
-    'query': fields.String,
-    'message': MessageTextField,
-    'answer': fields.String,
+    "inputs": fields.Raw,
+    "query": fields.String,
+    "message": MessageTextField,
+    "answer": fields.String,
 }

 conversation_fields = {
-    'id': fields.String,
-    'status': fields.String,
-    'from_source': fields.String,
-    'from_end_user_id': fields.String,
-    'from_end_user_session_id': fields.String(),
-    'from_account_id': fields.String,
-    'read_at': TimestampField,
-    'created_at': TimestampField,
-    'annotation': fields.Nested(annotation_fields, allow_null=True),
-    'model_config': fields.Nested(simple_model_config_fields),
-    'user_feedback_stats': fields.Nested(feedback_stat_fields),
-    'admin_feedback_stats': fields.Nested(feedback_stat_fields),
-    'message': fields.Nested(simple_message_detail_fields, attribute='first_message')
+    "id": fields.String,
+    "status": fields.String,
+    "from_source": fields.String,
+    "from_end_user_id": fields.String,
+    "from_end_user_session_id": fields.String(),
+    "from_account_id": fields.String,
+    "read_at": TimestampField,
+    "created_at": TimestampField,
+    "annotation": fields.Nested(annotation_fields, allow_null=True),
+    "model_config": fields.Nested(simple_model_config_fields),
+    "user_feedback_stats": fields.Nested(feedback_stat_fields),
+    "admin_feedback_stats": fields.Nested(feedback_stat_fields),
+    "message": fields.Nested(simple_message_detail_fields, attribute="first_message"),
 }

 conversation_pagination_fields = {
-    'page': fields.Integer,
-    'limit': fields.Integer(attribute='per_page'),
-    'total': fields.Integer,
-    'has_more': fields.Boolean(attribute='has_next'),
-    'data': fields.List(fields.Nested(conversation_fields), attribute='items')
+    "page": fields.Integer,
+    "limit": fields.Integer(attribute="per_page"),
+    "total": fields.Integer,
+    "has_more": fields.Boolean(attribute="has_next"),
+    "data": fields.List(fields.Nested(conversation_fields), attribute="items"),
 }

 conversation_message_detail_fields = {
-    'id': fields.String,
-    'status': fields.String,
-    'from_source': fields.String,
-    'from_end_user_id': fields.String,
-    'from_account_id': fields.String,
-    'created_at': TimestampField,
-    'model_config': fields.Nested(model_config_fields),
-    'message': fields.Nested(message_detail_fields, attribute='first_message'),
+    "id": fields.String,
+    "status": fields.String,
+    "from_source": fields.String,
+    "from_end_user_id": fields.String,
+    "from_account_id": fields.String,
+    "created_at": TimestampField,
+    "model_config": fields.Nested(model_config_fields),
+    "message": fields.Nested(message_detail_fields, attribute="first_message"),
 }

 conversation_with_summary_fields = {
-    'id': fields.String,
-    'status': fields.String,
-    'from_source': fields.String,
-    'from_end_user_id': fields.String,
-    'from_end_user_session_id': fields.String,
-    'from_account_id': fields.String,
-    'name': fields.String,
-    'summary': fields.String(attribute='summary_or_query'),
-    'read_at': TimestampField,
-    'created_at': TimestampField,
-    'annotated': fields.Boolean,
-    'model_config': fields.Nested(simple_model_config_fields),
-    'message_count': fields.Integer,
-    'user_feedback_stats': fields.Nested(feedback_stat_fields),
-    'admin_feedback_stats': fields.Nested(feedback_stat_fields)
+    "id": fields.String,
+    "status": fields.String,
+    "from_source": fields.String,
+    "from_end_user_id": fields.String,
+    "from_end_user_session_id": fields.String,
+    "from_account_id": fields.String,
+    "name": fields.String,
+    "summary": fields.String(attribute="summary_or_query"),
+    "read_at": TimestampField,
+    "created_at": TimestampField,
+    "annotated": fields.Boolean,
+    "model_config": fields.Nested(simple_model_config_fields),
+    "message_count": fields.Integer,
+    "user_feedback_stats": fields.Nested(feedback_stat_fields),
+    "admin_feedback_stats": fields.Nested(feedback_stat_fields),
 }

 conversation_with_summary_pagination_fields = {
-    'page': fields.Integer,
-    'limit': fields.Integer(attribute='per_page'),
-    'total': fields.Integer,
-    'has_more': fields.Boolean(attribute='has_next'),
-    'data': fields.List(fields.Nested(conversation_with_summary_fields), attribute='items')
+    "page": fields.Integer,
+    "limit": fields.Integer(attribute="per_page"),
+    "total": fields.Integer,
+    "has_more": fields.Boolean(attribute="has_next"),
+    "data": fields.List(fields.Nested(conversation_with_summary_fields), attribute="items"),
 }

 conversation_detail_fields = {
-    'id': fields.String,
-    'status': fields.String,
-    'from_source': fields.String,
-    'from_end_user_id': fields.String,
-    'from_account_id': fields.String,
-    'created_at': TimestampField,
-    'annotated': fields.Boolean,
-    'introduction': fields.String,
-    'model_config': fields.Nested(model_config_fields),
-    'message_count': fields.Integer,
-    'user_feedback_stats': fields.Nested(feedback_stat_fields),
-    'admin_feedback_stats': fields.Nested(feedback_stat_fields)
+    "id": fields.String,
+    "status": fields.String,
+    "from_source": fields.String,
+    "from_end_user_id": fields.String,
+    "from_account_id": fields.String,
+    "created_at": TimestampField,
+    "annotated": fields.Boolean,
+    "introduction": fields.String,
+    "model_config": fields.Nested(model_config_fields),
+    "message_count": fields.Integer,
+    "user_feedback_stats": fields.Nested(feedback_stat_fields),
+    "admin_feedback_stats": fields.Nested(feedback_stat_fields),
 }

 simple_conversation_fields = {
-    'id': fields.String,
-    'name': fields.String,
-    'inputs': fields.Raw,
-    'status': fields.String,
-    'introduction': fields.String,
-    'created_at': TimestampField
+    "id": fields.String,
+    "name": fields.String,
+    "inputs": fields.Raw,
+    "status": fields.String,
+    "introduction": fields.String,
+    "created_at": TimestampField,
 }

 conversation_infinite_scroll_pagination_fields = {
-    'limit': fields.Integer,
-    'has_more': fields.Boolean,
-    'data': fields.List(fields.Nested(simple_conversation_fields))
+    "limit": fields.Integer,
+    "has_more": fields.Boolean,
+    "data": fields.List(fields.Nested(simple_conversation_fields)),
 }

 conversation_with_model_config_fields = {
    **simple_conversation_fields,
-    'model_config': fields.Raw,
+    "model_config": fields.Raw,
 }

 conversation_with_model_config_infinite_scroll_pagination_fields = {
-    'limit': fields.Integer,
-    'has_more': fields.Boolean,
-    'data': fields.List(fields.Nested(conversation_with_model_config_fields))
+    "limit": fields.Integer,
+    "has_more": fields.Boolean,
+    "data": fields.List(fields.Nested(conversation_with_model_config_fields)),
 }