fix: 恢复template_info功能

src/chat/message_receive/message.py

@@ -1,12 +1,14 @@
 import time
 from abc import abstractmethod
 from dataclasses import dataclass
-from typing import Optional, Any
+from typing import Optional, Any, TYPE_CHECKING
 
 import urllib3
 
 from src.common.logger_manager import get_logger
-from .chat_stream import ChatStream
+
+if TYPE_CHECKING:
+    from .chat_stream import ChatStream
 from ..utils.utils_image import image_manager
 from maim_message import Seg, UserInfo, BaseMessageInfo, MessageBase
 from rich.traceback import install
@@ -25,7 +27,7 @@ urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
 
 @dataclass
 class Message(MessageBase):
-    chat_stream: ChatStream = None
+    chat_stream: "ChatStream" = None
     reply: Optional["Message"] = None
     detailed_plain_text: str = ""
     processed_plain_text: str = ""
@@ -34,7 +36,7 @@ class Message(MessageBase):
     def __init__(
         self,
         message_id: str,
-        chat_stream: ChatStream,
+        chat_stream: "ChatStream",
         user_info: UserInfo,
         message_segment: Optional[Seg] = None,
         timestamp: Optional[float] = None,
@@ -111,7 +113,7 @@ class MessageRecv(Message):
         self.detailed_plain_text = ""  # 初始化为空字符串
         self.is_emoji = False
 
-    def update_chat_stream(self, chat_stream: ChatStream):
+    def update_chat_stream(self, chat_stream: "ChatStream"):
         self.chat_stream = chat_stream
 
     async def process(self) -> None:
@@ -165,7 +167,7 @@ class MessageProcessBase(Message):
     def __init__(
         self,
         message_id: str,
-        chat_stream: ChatStream,
+        chat_stream: "ChatStream",
         bot_user_info: UserInfo,
         message_segment: Optional[Seg] = None,
         reply: Optional["MessageRecv"] = None,
@@ -241,7 +243,7 @@ class MessageThinking(MessageProcessBase):
     def __init__(
         self,
         message_id: str,
-        chat_stream: ChatStream,
+        chat_stream: "ChatStream",
         bot_user_info: UserInfo,
         reply: Optional["MessageRecv"] = None,
         thinking_start_time: float = 0,
@@ -269,7 +271,7 @@ class MessageSending(MessageProcessBase):
     def __init__(
         self,
         message_id: str,
-        chat_stream: ChatStream,
+        chat_stream: "ChatStream",
         bot_user_info: UserInfo,
         sender_info: UserInfo | None,  # 用来记录发送者信息,用于私聊回复
         message_segment: Seg,
@@ -353,7 +355,7 @@ class MessageSending(MessageProcessBase):
 class MessageSet:
     """消息集合类，可以存储多个发送消息"""
 
-    def __init__(self, chat_stream: ChatStream, message_id: str):
+    def __init__(self, chat_stream: "ChatStream", message_id: str):
         self.chat_stream = chat_stream
         self.message_id = message_id
         self.messages: list[MessageSending] = []