From 1fafdb8471e7398e9b6f95b72562c8625040fc7d Mon Sep 17 00:00:00 2001 From: Valdanito Date: Tue, 24 Sep 2024 12:05:15 +0800 Subject: [PATCH] fix(API): fixed retrieval api parameters matching (#2550) ### What problem does this PR solve? fixed /datasets/retrieval API: KeyError('size') and 'doc_ids': ['Field may not be null.'] ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- api/apps/services/dataset_service.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/apps/services/dataset_service.py b/api/apps/services/dataset_service.py index 09c74a99..d47baa40 100644 --- a/api/apps/services/dataset_service.py +++ b/api/apps/services/dataset_service.py @@ -64,7 +64,7 @@ class RetrievalReq(Schema): question = fields.String(required=True) page = fields.Integer(load_default=1) page_size = fields.Integer(load_default=30) - doc_ids = fields.List(fields.String()) + doc_ids = fields.List(fields.String(), allow_none=True) similarity_threshold = fields.Float(load_default=0.0) vector_similarity_weight = fields.Float(load_default=0.3) top_k = fields.Integer(load_default=1024) @@ -180,7 +180,7 @@ def delete_dataset(tenant_id, kb_id): def retrieval_in_dataset(tenant_id, json_data): page = json_data["page"] - size = json_data["size"] + size = json_data["page_size"] question = json_data["question"] kb_id = json_data["kb_id"] if isinstance(kb_id, str): kb_id = [kb_id]