Sfoglia il codice sorgente

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

VoidIsVoid 4 mesi fa
parent
commit
02a7ae15f9
1 ha cambiato i file con 6 aggiunte e 2 eliminazioni
  1. 6 2
      api/services/dataset_service.py

+ 6 - 2
api/services/dataset_service.py

@@ -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.")