diff --git a/sdk/python/ragflow_sdk/modules/document.py b/sdk/python/ragflow_sdk/modules/document.py index 70c1ac842..c96698079 100644 --- a/sdk/python/ragflow_sdk/modules/document.py +++ b/sdk/python/ragflow_sdk/modules/document.py @@ -69,7 +69,7 @@ class Document(Base): response = res.json() actual_keys = set(response.keys()) if actual_keys == error_keys: - raise Exception(res.get("message")) + raise Exception(response.get("message")) else: return res.content except json.JSONDecodeError: diff --git a/sdk/python/ragflow_sdk/modules/session.py b/sdk/python/ragflow_sdk/modules/session.py index 12141afed..d2bbc76e8 100644 --- a/sdk/python/ragflow_sdk/modules/session.py +++ b/sdk/python/ragflow_sdk/modules/session.py @@ -80,6 +80,7 @@ class Session(Base): def _structure_answer(self, json_data): + answer = "" if self.__session_type == "agent": answer = json_data["data"]["content"] elif self.__session_type == "chat":