mirror of
https://gitcode.com/JianFeeeee/TrulyMEM-TrueHumanMEM.git
synced 2026-09-21 17:38:18 +00:00
refactor: restructure to core/ + ui/ with multi-threaded backend
This commit is contained in:
57
ui/widgets/message_history.py
Normal file
57
ui/widgets/message_history.py
Normal file
@ -0,0 +1,57 @@
|
||||
"""消息历史组件"""
|
||||
|
||||
from textual.containers import ScrollableContainer
|
||||
from textual.message import Message
|
||||
from .message_widget import MessageWidget
|
||||
from ..models.message import Message as MessageModel
|
||||
|
||||
|
||||
class MessageHistory(ScrollableContainer):
|
||||
"""消息历史区域"""
|
||||
|
||||
def __init__(self, **kwargs):
|
||||
super().__init__(**kwargs)
|
||||
self._messages: list[MessageModel] = []
|
||||
|
||||
def compose(self):
|
||||
"""构建消息历史"""
|
||||
for message in self._messages:
|
||||
yield MessageWidget(message)
|
||||
|
||||
def add_message(self, message: MessageModel) -> None:
|
||||
"""添加新消息"""
|
||||
self._messages.append(message)
|
||||
# 添加新组件
|
||||
message_widget = MessageWidget(message)
|
||||
self.mount(message_widget)
|
||||
# 滚动到最新消息
|
||||
self.scroll_to_widget(message_widget, animate=False)
|
||||
|
||||
def update_latest_message(self, content: str) -> None:
|
||||
"""更新最新消息的内容"""
|
||||
if self.children:
|
||||
latest_widget = self.children[-1]
|
||||
if isinstance(latest_widget, MessageWidget):
|
||||
latest_widget.update_content(content)
|
||||
# 确保滚动到最新消息
|
||||
self.scroll_to_widget(latest_widget, animate=False)
|
||||
|
||||
def clear_messages(self) -> None:
|
||||
"""清空消息历史"""
|
||||
self._messages.clear()
|
||||
# 移除所有子组件
|
||||
for child in self.children:
|
||||
child.remove()
|
||||
|
||||
def get_latest_message(self) -> MessageModel | None:
|
||||
"""获取最新消息"""
|
||||
if self._messages:
|
||||
return self._messages[-1]
|
||||
return None
|
||||
|
||||
def toggle_latest_tool_details(self) -> None:
|
||||
"""切换最新消息的工具详情"""
|
||||
if self.children:
|
||||
latest_widget = self.children[-1]
|
||||
if isinstance(latest_widget, MessageWidget):
|
||||
latest_widget.toggle_tool_details()
|
||||
Reference in New Issue
Block a user