diff --git a/docker/base/Dockerfile b/docker/base/Dockerfile
index d555b42b5..0183e7b28 100644
--- a/docker/base/Dockerfile
+++ b/docker/base/Dockerfile
@@ -28,6 +28,7 @@ RUN if [ "$USE_TSINGHUA_UBUNTU" = "true" ]; then \
     rm -rf /var/lib/apt/lists/* && \
     apt-get update && apt-get install -y \
     python${PYTHON_VERSION} \
+    python${PYTHON_VERSION}-dev \
     python${PYTHON_VERSION}-venv \
     python3-pip \
     python3-dev \
diff --git a/packages/dbgpt-app/src/dbgpt_app/scene/base_chat.py b/packages/dbgpt-app/src/dbgpt_app/scene/base_chat.py
index 3bcc3d92b..70b2f61c0 100644
--- a/packages/dbgpt-app/src/dbgpt_app/scene/base_chat.py
+++ b/packages/dbgpt-app/src/dbgpt_app/scene/base_chat.py
@@ -374,7 +374,7 @@ class BaseChat(ABC):
             ai_response_text, view_message = await self._handle_final_output(
                 final_output, incremental=incremental
             )
-            full_text = view_message
+            full_text = view_message if text_output else final_output.text
             # Return the incremental text
             delta_text = full_text[len(previous_text) :]
             result_text = delta_text if incremental else full_text