diff --git a/api/apps/sdk/dataset.py b/api/apps/sdk/dataset.py index 4cd48fe2..ff20e317 100644 --- a/api/apps/sdk/dataset.py +++ b/api/apps/sdk/dataset.py @@ -136,7 +136,8 @@ def create(tenant_id): return get_error_data_result( message="Duplicated dataset name in creating dataset." ) - req["tenant_id"] = req["created_by"] = tenant_id + req["tenant_id"] = tenant_id + req["created_by"] = tenant_id if not req.get("embedding_model"): req["embedding_model"] = t.embd_id else: @@ -178,6 +179,10 @@ def create(tenant_id): if old_key in req } req.update(mapped_keys) + flds = list(req.keys()) + for f in flds: + if req[f] == "" and f in ["permission", "chunk_method"]: + del req[f] if not KnowledgebaseService.save(**req): return get_error_data_result(message="Create dataset error.(Database error)") renamed_data = {} @@ -427,7 +432,7 @@ def update(tenant_id, dataset_id): @manager.route("/datasets", methods=["GET"]) # noqa: F821 @token_required -def list(tenant_id): +def list_datasets(tenant_id): """ List datasets. ---