diff --git a/docs/references/python_api_reference.md b/docs/references/python_api_reference.md
index 21a47264f..10bd62488 100644
--- a/docs/references/python_api_reference.md
+++ b/docs/references/python_api_reference.md
@@ -1485,8 +1485,8 @@ The parameters in `begin` component.
 from ragflow_sdk import RAGFlow, Agent
 
 rag_object = RAGFlow(api_key="<YOUR_API_KEY>", base_url="http://<YOUR_BASE_URL>:9380")
-AGENT_ID = "AGENT_ID"
-agent = rag_object.list_agents(id = AGENT_id)[0]
+agent_id = "AGENT_ID"
+agent = rag_object.list_agents(id = agent_id)[0]
 session = agent.create_session()
 ```
 
diff --git a/sdk/python/ragflow_sdk/modules/session.py b/sdk/python/ragflow_sdk/modules/session.py
index f30898822..f91405cab 100644
--- a/sdk/python/ragflow_sdk/modules/session.py
+++ b/sdk/python/ragflow_sdk/modules/session.py
@@ -38,17 +38,34 @@ class Session(Base):
             res = self._ask_agent(question, stream)
         elif self.__session_type == "chat":
             res = self._ask_chat(question, stream, **kwargs)
-            
-        for line in res.iter_lines():
-            line = line.decode("utf-8")
-            if line.startswith("{"):
-                json_data = json.loads(line)
-                raise Exception(json_data["message"])
-            if not line.startswith("data:"):
-                continue
-            json_data = json.loads(line[5:])
-            if json_data["data"] is True or json_data["data"].get("running_status"):
-                continue
+
+        if stream:
+            for line in res.iter_lines():
+                line = line.decode("utf-8")
+                if line.startswith("{"):
+                    json_data = json.loads(line)
+                    raise Exception(json_data["message"])
+                if not line.startswith("data:"):
+                    continue
+                json_data = json.loads(line[5:])
+                if json_data["data"] is True or json_data["data"].get("running_status"):
+                    continue
+                answer = json_data["data"]["answer"]
+                reference = json_data["data"].get("reference", {})
+                temp_dict = {
+                    "content": answer,
+                    "role": "assistant"
+                }
+                if reference and "chunks" in reference:
+                    chunks = reference["chunks"]
+                    temp_dict["reference"] = chunks
+                message = Message(self.rag, temp_dict)
+                yield message
+        else:
+            try:
+                json_data = json.loads(res.text)
+            except ValueError:
+                raise Exception(f"Invalid response {res}")
             answer = json_data["data"]["answer"]
             reference = json_data["data"].get("reference", {})
             temp_dict = {
@@ -59,11 +76,8 @@ class Session(Base):
                 chunks = reference["chunks"]
                 temp_dict["reference"] = chunks
             message = Message(self.rag, temp_dict)
-            if stream:
-                yield message
-        if not stream:
             return message
-    
+
     def _ask_chat(self, question: str, stream: bool, **kwargs):
         json_data = {"question": question, "stream": stream, "session_id": self.id}
         json_data.update(kwargs)