KMS/ragflow
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix(API): fixed retrieval api parameters matching (#2550)

Browse Source 
### 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)
This commit is contained in:
Valdanito 2024-09-24 12:05:15 +08:00 committed by GitHub
parent 5110a3ba90
commit 1fafdb8471
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/apps/services/dataset_service.py
View File

@ -64,7 +64,7 @@ class RetrievalReq(Schema):
question = fields.String(required=True) question = fields.String(required=True)
page = fields.Integer(load_default=1) page = fields.Integer(load_default=1)
page_size = fields.Integer(load_default=30) 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) similarity_threshold = fields.Float(load_default=0.0)
vector_similarity_weight = fields.Float(load_default=0.3) vector_similarity_weight = fields.Float(load_default=0.3)
top_k = fields.Integer(load_default=1024) 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): def retrieval_in_dataset(tenant_id, json_data):
page = json_data["page"] page = json_data["page"]
size = json_data["size"] size = json_data["page_size"]
question = json_data["question"] question = json_data["question"]
kb_id = json_data["kb_id"] kb_id = json_data["kb_id"]
if isinstance(kb_id, str): kb_id = [kb_id] if isinstance(kb_id, str): kb_id = [kb_id]