diff --git a/sdk/python/ragflow_sdk/modules/session.py b/sdk/python/ragflow_sdk/modules/session.py
index 9a1a7fd3d..5f5ca6dc3 100644
--- a/sdk/python/ragflow_sdk/modules/session.py
+++ b/sdk/python/ragflow_sdk/modules/session.py
@@ -38,6 +38,7 @@ 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("{"):
@@ -58,8 +59,11 @@ class Session(Base):
                 chunks = reference["chunks"]
                 temp_dict["reference"] = chunks
             message = Message(self.rag, temp_dict)
-            yield message
-
+            if stream:
+                yield message
+        if not stream:
+            return message
+    
     def _ask_chat(self, question: str, stream: bool, **kwargs):
         json_data = {"question": question, "stream": True, "session_id": self.id}
         json_data.update(kwargs)