fix: fix update external dataset error in dataset list (#11989)

This commit is contained in:
VoidIsVoid 2024-12-23 13:53:05 +08:00 committed by GitHub
parent 74b1b60125
commit 02a7ae15f9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -231,11 +231,15 @@ class DatasetService:
DatasetService.check_dataset_permission(dataset, user)
if dataset.provider == "external":
dataset.retrieval_model = data.get("external_retrieval_model", None)
external_retrieval_model = data.get("external_retrieval_model", None)
if external_retrieval_model:
dataset.retrieval_model = external_retrieval_model
dataset.name = data.get("name", dataset.name)
dataset.description = data.get("description", "")
permission = data.get("permission")
if permission:
dataset.permission = permission
external_knowledge_id = data.get("external_knowledge_id", None)
dataset.permission = data.get("permission")
db.session.add(dataset)
if not external_knowledge_id:
raise ValueError("External knowledge id is required.")