diff --git a/api/apps/sdk/chat.py b/api/apps/sdk/chat.py index bfeba2b0f..32b30341c 100644 --- a/api/apps/sdk/chat.py +++ b/api/apps/sdk/chat.py @@ -31,7 +31,7 @@ from api.utils.api_utils import get_result @token_required def create(tenant_id): req = request.json - ids = [i for i in req.get("dataset_ids", []) if i] + ids = [i for i in req.get("dataset_ids", []) if i] for kb_id in ids: kbs = KnowledgebaseService.accessible(kb_id=kb_id, user_id=tenant_id) if not kbs: diff --git a/sdk/python/ragflow_sdk/ragflow.py b/sdk/python/ragflow_sdk/ragflow.py index c0da192cc..a84c3515e 100644 --- a/sdk/python/ragflow_sdk/ragflow.py +++ b/sdk/python/ragflow_sdk/ragflow.py @@ -129,7 +129,7 @@ class RAGFlow: temp_dict = {"name": name, "avatar": avatar, - "dataset_ids": dataset_list, + "dataset_ids": dataset_list if dataset_list else [], "llm": llm.to_json(), "prompt": prompt.to_json()} res = self.post("/chats", temp_dict)