refactor(xxm): use [YOU] prefix for TUI context (align with Hermes)

This commit is contained in:
hmo
2026-06-20 01:31:04 +08:00
parent e8fba49cb8
commit 885439f3c4
+14 -6
View File
@@ -393,6 +393,19 @@ class SessionBridge:
except Exception:
return ""
def _format_tui_context(self, raw_ctx: str) -> str:
"""Mark bot's own TUI messages with [YOU] so LLM recognises itself."""
lines = []
for line in raw_ctx.split("\n"):
line = line.strip()
if not line:
continue
if "小小莫" in line:
lines.append(f"[YOU] {line}")
else:
lines.append(line)
return "\n".join(lines)
def _append_to_log(self, role: str, content: str, source: str = "bridge"):
try:
entry = json.dumps(
@@ -693,12 +706,7 @@ class SessionBridge:
from session_router import extract_session_context
tui_ctx = extract_session_context(tui_sid, limit=50)
if tui_ctx:
sys_prompt += (
"\n\n【你的 TUI 工作台最近对话】"
f"\n(老莫在 TUI 和你讨论这些时,转发了群聊消息,"
f"请结合下面的 TUI 对话理解他在群里的意图。"
f"群聊和 TUI 是同一个人,两份上下文关联看)\n{tui_ctx}"
)
sys_prompt += f"\n\n{self._format_tui_context(tui_ctx)}"
except Exception:
pass