diff --git a/src/chat/message_receive/message.py b/src/chat/message_receive/message.py
index fc57b6fc6..a113088c9 100644
--- a/src/chat/message_receive/message.py
+++ b/src/chat/message_receive/message.py
@@ -13,6 +13,7 @@ from src.chat.utils.utils_video import get_video_analyzer, is_video_analysis_ava
 from src.chat.utils.utils_voice import get_voice_text
 from src.common.logger import get_logger
 from src.config.config import global_config
+rom src.chat.message_receive.chat_stream import ChatStream
 
 install(extra_lines=3)