mirror of
https://gitcode.com/JianFeeeee/TrulyMEM-TrueHumanMEM.git
synced 2026-09-20 08:58:15 +00:00
109 lines
4.3 KiB
Python
109 lines
4.3 KiB
Python
"""消息组件"""
|
||
|
||
from textual.containers import Container, Vertical
|
||
from textual.widgets import Static
|
||
from textual.message import Message
|
||
from textual.css.query import NoMatches
|
||
from ..models.message import Message as MessageModel
|
||
|
||
|
||
class MessageWidget(Container):
|
||
"""单条消息组件"""
|
||
|
||
def __init__(self, message: MessageModel, **kwargs):
|
||
super().__init__(**kwargs)
|
||
self._message = message
|
||
self._show_tool_details = False
|
||
self._content_widget = None # 保存内容组件的引用
|
||
self._tool_details_container = None # 保存工具详情容器引用
|
||
|
||
def compose(self):
|
||
"""构建消息组件"""
|
||
# 消息头
|
||
role_emoji = "🟠" if self._message.role == "user" else "🔵"
|
||
timestamp_str = self._message.timestamp.strftime("%H:%M:%S")
|
||
yield Static(
|
||
f"{role_emoji} {timestamp_str}",
|
||
classes="message-header"
|
||
)
|
||
|
||
# 消息内容 - 保存引用以便后续更新
|
||
self._content_widget = Static(
|
||
self._message.content,
|
||
classes="message-content"
|
||
)
|
||
yield self._content_widget
|
||
|
||
# 工具调用指示器
|
||
if self._message.tool_calls:
|
||
tool_count = len(self._message.tool_calls)
|
||
toggle_hint = "(F3折叠)" if self._show_tool_details else "(F3展开)"
|
||
yield Static(
|
||
f"[工具:{tool_count}次] {toggle_hint}",
|
||
classes="tool-indicator"
|
||
)
|
||
|
||
# 工具调用详情容器 - 始终创建，但根据状态显示/隐藏
|
||
self._tool_details_container = Vertical(classes="tool-details")
|
||
with self._tool_details_container:
|
||
for i, tool_call in enumerate(self._message.tool_calls, 1):
|
||
yield Static(
|
||
f"工具 {i}: {tool_call.name}",
|
||
classes="tool-name"
|
||
)
|
||
yield Static(
|
||
f"参数: {tool_call.arguments}",
|
||
classes="tool-args"
|
||
)
|
||
|
||
# 显示执行结果
|
||
if self._message.tool_results:
|
||
for result in self._message.tool_results:
|
||
if result.tool_call_id == tool_call.id:
|
||
# 显示完整结果，不截断
|
||
result_text = result.result
|
||
# 如果结果太长，只显示前1000字符，但提供完整信息
|
||
if len(result_text) > 1000:
|
||
result_text = result_text[:1000] + f"\n... (共{len(result.result)}字符，按F3查看完整内容)"
|
||
yield Static(
|
||
f"结果: {result_text}",
|
||
classes="tool-result"
|
||
)
|
||
|
||
# 根据状态设置初始显示/隐藏
|
||
if not self._show_tool_details:
|
||
self._tool_details_container.styles.display = "none"
|
||
|
||
def update_content(self, new_content: str) -> None:
|
||
"""更新消息内容"""
|
||
self._message.content = new_content
|
||
if self._content_widget:
|
||
self._content_widget.update(new_content)
|
||
|
||
def toggle_tool_details(self) -> None:
|
||
"""切换工具详情显示状态"""
|
||
if self._message.tool_calls and self._tool_details_container:
|
||
self._show_tool_details = not self._show_tool_details
|
||
|
||
# 切换显示/隐藏
|
||
if self._show_tool_details:
|
||
self._tool_details_container.styles.display = "block"
|
||
else:
|
||
self._tool_details_container.styles.display = "none"
|
||
|
||
# 更新指示器文字
|
||
self._update_indicator()
|
||
|
||
# 刷新布局
|
||
self.refresh(layout=True)
|
||
|
||
def _update_indicator(self) -> None:
|
||
"""更新工具调用指示器文字"""
|
||
try:
|
||
indicator = self.query_one(".tool-indicator", Static)
|
||
tool_count = len(self._message.tool_calls)
|
||
toggle_hint = "(F3折叠)" if self._show_tool_details else "(F3展开)"
|
||
indicator.update(f"[工具:{tool_count}次] {toggle_hint}")
|
||
except NoMatches:
|
||
pass
|