diff --git a/sdk/python/ragflow_sdk/modules/chat.py b/sdk/python/ragflow_sdk/modules/chat.py index 53fcc1af9..474fa54b8 100644 --- a/sdk/python/ragflow_sdk/modules/chat.py +++ b/sdk/python/ragflow_sdk/modules/chat.py @@ -60,6 +60,12 @@ class Chat(Base): super().__init__(rag, res_dict) def update(self, update_message: dict): + if not isinstance(update_message, dict): + raise Exception("ValueError('`update_message` must be a dict')") + if update_message.get("llm") == {}: + raise Exception("ValueError('`llm` cannot be empty')") + if update_message.get("prompt") == {}: + raise Exception("ValueError('`prompt` cannot be empty')") res = self.put(f"/chats/{self.id}", update_message) res = res.json() if res.get("code") != 0: