mirror of
https://gitee.com/dify_ai/dify.git
synced 2024-11-30 10:18:13 +08:00
42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
from flask_restful import fields
|
|
|
|
from libs.helper import TimestampField
|
|
|
|
document_fields = {
|
|
"id": fields.String,
|
|
"data_source_type": fields.String,
|
|
"name": fields.String,
|
|
"doc_type": fields.String,
|
|
}
|
|
|
|
segment_fields = {
|
|
"id": fields.String,
|
|
"position": fields.Integer,
|
|
"document_id": fields.String,
|
|
"content": fields.String,
|
|
"answer": fields.String,
|
|
"word_count": fields.Integer,
|
|
"tokens": fields.Integer,
|
|
"keywords": fields.List(fields.String),
|
|
"index_node_id": fields.String,
|
|
"index_node_hash": fields.String,
|
|
"hit_count": fields.Integer,
|
|
"enabled": fields.Boolean,
|
|
"disabled_at": TimestampField,
|
|
"disabled_by": fields.String,
|
|
"status": fields.String,
|
|
"created_by": fields.String,
|
|
"created_at": TimestampField,
|
|
"indexing_at": TimestampField,
|
|
"completed_at": TimestampField,
|
|
"error": fields.String,
|
|
"stopped_at": TimestampField,
|
|
"document": fields.Nested(document_fields),
|
|
}
|
|
|
|
hit_testing_record_fields = {
|
|
"segment": fields.Nested(segment_fields),
|
|
"score": fields.Float,
|
|
"tsne_position": fields.Raw,
|
|
}
|