From b247ef85bf731c701745e3e61ea41e560db3e819 Mon Sep 17 00:00:00 2001 From: kenwoodjw Date: Wed, 16 Apr 2025 15:50:06 +0800 Subject: [PATCH] fix dataset api retrieval model null handling (#18151) Signed-off-by: kenwoodjw --- api/controllers/service_api/dataset/dataset.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/api/controllers/service_api/dataset/dataset.py b/api/controllers/service_api/dataset/dataset.py index f087243a25..e1e6f3168f 100644 --- a/api/controllers/service_api/dataset/dataset.py +++ b/api/controllers/service_api/dataset/dataset.py @@ -139,7 +139,9 @@ class DatasetListApi(DatasetApiResource): external_knowledge_id=args["external_knowledge_id"], embedding_model_provider=args["embedding_model_provider"], embedding_model_name=args["embedding_model"], - retrieval_model=RetrievalModel(**args["retrieval_model"]), + retrieval_model=RetrievalModel(**args["retrieval_model"]) + if args["retrieval_model"] is not None + else None, ) except services.errors.dataset.DatasetNameDuplicateError: raise DatasetNameDuplicateError()