mirror of
https://github.com/langgenius/dify.git
synced 2024-11-16 19:59:50 +08:00
86 lines
2.3 KiB
Python
86 lines
2.3 KiB
Python
from flask_restful import fields
|
|
|
|
from libs.helper import TimestampField
|
|
|
|
dataset_fields = {
|
|
'id': fields.String,
|
|
'name': fields.String,
|
|
'description': fields.String,
|
|
'permission': fields.String,
|
|
'data_source_type': fields.String,
|
|
'indexing_technique': fields.String,
|
|
'created_by': fields.String,
|
|
'created_at': TimestampField,
|
|
}
|
|
|
|
reranking_model_fields = {
|
|
'reranking_provider_name': fields.String,
|
|
'reranking_model_name': fields.String
|
|
}
|
|
|
|
keyword_setting_fields = {
|
|
'keyword_weight': fields.Float
|
|
}
|
|
|
|
vector_setting_fields = {
|
|
'vector_weight': fields.Float,
|
|
'embedding_model_name': fields.String,
|
|
'embedding_provider_name': fields.String,
|
|
}
|
|
|
|
weighted_score_fields = {
|
|
'keyword_setting': fields.Nested(keyword_setting_fields),
|
|
'vector_setting': fields.Nested(vector_setting_fields),
|
|
}
|
|
|
|
dataset_retrieval_model_fields = {
|
|
'search_method': fields.String,
|
|
'reranking_enable': fields.Boolean,
|
|
'reranking_mode': fields.String,
|
|
'reranking_model': fields.Nested(reranking_model_fields),
|
|
'weights': fields.Nested(weighted_score_fields, allow_null=True),
|
|
'top_k': fields.Integer,
|
|
'score_threshold_enabled': fields.Boolean,
|
|
'score_threshold': fields.Float
|
|
}
|
|
|
|
tag_fields = {
|
|
'id': fields.String,
|
|
'name': fields.String,
|
|
'type': fields.String
|
|
}
|
|
|
|
dataset_detail_fields = {
|
|
'id': fields.String,
|
|
'name': fields.String,
|
|
'description': fields.String,
|
|
'provider': fields.String,
|
|
'permission': fields.String,
|
|
'data_source_type': fields.String,
|
|
'indexing_technique': fields.String,
|
|
'app_count': fields.Integer,
|
|
'document_count': fields.Integer,
|
|
'word_count': fields.Integer,
|
|
'created_by': fields.String,
|
|
'created_at': TimestampField,
|
|
'updated_by': fields.String,
|
|
'updated_at': TimestampField,
|
|
'embedding_model': fields.String,
|
|
'embedding_model_provider': fields.String,
|
|
'embedding_available': fields.Boolean,
|
|
'retrieval_model_dict': fields.Nested(dataset_retrieval_model_fields),
|
|
'tags': fields.List(fields.Nested(tag_fields))
|
|
}
|
|
|
|
dataset_query_detail_fields = {
|
|
"id": fields.String,
|
|
"content": fields.String,
|
|
"source": fields.String,
|
|
"source_app_id": fields.String,
|
|
"created_by_role": fields.String,
|
|
"created_by": fields.String,
|
|
"created_at": TimestampField
|
|
}
|
|
|
|
|