outman/dify
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add external_retrieval_model

Browse Source
This commit is contained in:
jyong 2024-09-24 23:52:01 +08:00
parent a258f8dfdf
commit 02b06c420e
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/fields/dataset_fields.py
View File

@ -38,6 +38,10 @@ dataset_retrieval_model_fields = {
"score_threshold_enabled": fields.Boolean, "score_threshold_enabled": fields.Boolean,
"score_threshold": fields.Float, "score_threshold": fields.Float,
} }
external_retrieval_model_fields = {
"top_k": fields.Integer,
"score_threshold": fields.Float,
}
tag_fields = {"id": fields.String, "name": fields.String, "type": fields.String} tag_fields = {"id": fields.String, "name": fields.String, "type": fields.String}
@ -69,6 +73,7 @@ dataset_detail_fields = {
"retrieval_model_dict": fields.Nested(dataset_retrieval_model_fields), "retrieval_model_dict": fields.Nested(dataset_retrieval_model_fields),
"tags": fields.List(fields.Nested(tag_fields)), "tags": fields.List(fields.Nested(tag_fields)),
"external_knowledge_info": fields.Nested(external_knowledge_info_fields), "external_knowledge_info": fields.Nested(external_knowledge_info_fields),
"external_retrieval_model": fields.Nested(external_retrieval_model_fields, allow_null=True),
} }
dataset_query_detail_fields = { dataset_query_detail_fields = {