Fix/agent external knowledge retrieval (#9241)

2024-10-11 19:21:03 +08:00 · 2024-10-11 19:21:03 +08:00 · 42b02b3a5f
commit 42b02b3a5f
parent 44f6a536d2
7 changed files with 225 additions and 92 deletions
--- a/api/services/dataset_service.py
+++ b/api/services/dataset_service.py
@ -234,6 +234,7 @@ class DatasetService:
            dataset.name = data.get("name", dataset.name)
            dataset.description = data.get("description", "")
            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.")