mirror of
https://github.com/langgenius/dify.git
synced 2025-10-31 10:53:02 +00:00
87 lines
2.9 KiB
Python
87 lines
2.9 KiB
Python
from flask_restful import fields # type: ignore
|
|
|
|
from fields.dataset_fields import dataset_fields
|
|
from libs.helper import TimestampField
|
|
|
|
document_metadata_fields = {
|
|
"id": fields.String,
|
|
"name": fields.String,
|
|
"type": fields.String,
|
|
"value": fields.String,
|
|
}
|
|
|
|
document_fields = {
|
|
"id": fields.String,
|
|
"position": fields.Integer,
|
|
"data_source_type": fields.String,
|
|
"data_source_info": fields.Raw(attribute="data_source_info_dict"),
|
|
"data_source_detail_dict": fields.Raw(attribute="data_source_detail_dict"),
|
|
"dataset_process_rule_id": fields.String,
|
|
"name": fields.String,
|
|
"created_from": fields.String,
|
|
"created_by": fields.String,
|
|
"created_at": TimestampField,
|
|
"tokens": fields.Integer,
|
|
"indexing_status": fields.String,
|
|
"error": fields.String,
|
|
"enabled": fields.Boolean,
|
|
"disabled_at": TimestampField,
|
|
"disabled_by": fields.String,
|
|
"archived": fields.Boolean,
|
|
"display_status": fields.String,
|
|
"word_count": fields.Integer,
|
|
"hit_count": fields.Integer,
|
|
"doc_form": fields.String,
|
|
"doc_metadata": fields.List(fields.Nested(document_metadata_fields), attribute="doc_metadata_details"),
|
|
}
|
|
|
|
document_with_segments_fields = {
|
|
"id": fields.String,
|
|
"position": fields.Integer,
|
|
"data_source_type": fields.String,
|
|
"data_source_info": fields.Raw(attribute="data_source_info_dict"),
|
|
"data_source_detail_dict": fields.Raw(attribute="data_source_detail_dict"),
|
|
"dataset_process_rule_id": fields.String,
|
|
"process_rule_dict": fields.Raw(attribute="process_rule_dict"),
|
|
"name": fields.String,
|
|
"created_from": fields.String,
|
|
"created_by": fields.String,
|
|
"created_at": TimestampField,
|
|
"tokens": fields.Integer,
|
|
"indexing_status": fields.String,
|
|
"error": fields.String,
|
|
"enabled": fields.Boolean,
|
|
"disabled_at": TimestampField,
|
|
"disabled_by": fields.String,
|
|
"archived": fields.Boolean,
|
|
"display_status": fields.String,
|
|
"word_count": fields.Integer,
|
|
"hit_count": fields.Integer,
|
|
"completed_segments": fields.Integer,
|
|
"total_segments": fields.Integer,
|
|
"doc_metadata": fields.List(fields.Nested(document_metadata_fields), attribute="doc_metadata_details"),
|
|
}
|
|
|
|
dataset_and_document_fields = {
|
|
"dataset": fields.Nested(dataset_fields),
|
|
"documents": fields.List(fields.Nested(document_fields)),
|
|
"batch": fields.String,
|
|
}
|
|
|
|
document_status_fields = {
|
|
"id": fields.String,
|
|
"indexing_status": fields.String,
|
|
"processing_started_at": TimestampField,
|
|
"parsing_completed_at": TimestampField,
|
|
"cleaning_completed_at": TimestampField,
|
|
"splitting_completed_at": TimestampField,
|
|
"completed_at": TimestampField,
|
|
"paused_at": TimestampField,
|
|
"error": fields.String,
|
|
"stopped_at": TimestampField,
|
|
"completed_segments": fields.Integer,
|
|
"total_segments": fields.Integer,
|
|
}
|
|
|
|
document_status_fields_list = {"data": fields.List(fields.Nested(document_status_fields))}
|