fix dataset api retrieval model null handling (#18151)

Signed-off-by: kenwoodjw <blackxin55+@gmail.com>
12 months ago · b247ef85bf
parent fcdf965037
commit b247ef85bf
1 changed files with 3 additions and 1 deletions
--- 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()