Merge pull request #222 from itmisx/feat/session-topic-hint

✨ session topic

Author: itmisx

agent/llm.go

@@ -44,6 +44,10 @@ const (
 	compactHeadroomFloor = 30_000  // headroom 绝对下限:极小窗口也留够单轮暴涨缓冲
 	compactHeadroomCap   = 180_000 // headroom 绝对上限:大窗口不必按比例留那么多,省得过早压
 
+	// topicHintPct:上下文用量占窗口这个比例以上,主题一旦切换就提醒用户 /new(见 TopicHintTokens)。
+	// 半个窗口已经在讲别的事,旧上下文对新问题就只是噪音了。
+	topicHintPct = 50
+
 	// compactRetryCooldown:轮内压缩**瞬时失败**(超时/网络)后,隔这么多圈工具往返再重试。
 	// 只作用于瞬时失败(轮数不足走永久关,不冷却),故取 4 即可:够稀释反复发压缩请求 / 反复等
 	// 10 分钟超时,又能在长任务里较快恢复压缩。冷却期间 reclaim 不受影响、每圈照常兜底。
@@ -554,11 +558,29 @@ func coreSystemPrompt(workspace, skillCatalog string) string {
 	return base
 }
 
+// topicTrackingPrompt 让主 agent 每轮自报会话主题。**只给主 agent**,不进 coreSystemPrompt ——
+// 那段是主/子 agent 逐字节共用的缓存前缀,写进去子 agent 也会吐标签,而子 agent 的输出是当
+// 工具结果回灌的,标签会混进结果文本里。放在这儿:共用头部不变,主 agent 自己的前缀也仍是
+// 固定文本,缓存两边都不受影响。
+//
+// 只让模型「如实标注」,不让它「据此改变行为」:实测让它自己发现主题变了就改口提醒用户 /new,
+// 换了三种措辞一次都没触发过(它总是优先去帮用户干活);而无条件多吐一个属性稳定生效。
+// 要不要提醒由 tui 侧判(还要叠上模型不知道的上下文用量,见 TopicHintTokens)。
+const topicTrackingPrompt = `
+
+# 会话主题跟踪(格式硬性要求)
+每轮回复的最后一行,固定输出一个主题标签,不要解释它,也不要因为它改变你的正常回答:
+<topic shift="no">一句话概括当前会话在做的事</topic>
+- 本轮和上一轮 <topic> 是同一件事(含它的子问题、追问、返工、换个角度问)→ shift="no",主题沿用或微调措辞。
+- 用户明确转去做另一件不相干的事 → shift="yes",<topic> 写新主题。
+- 会话第一轮 → shift="no"。
+你只负责如实标注,要不要提醒用户新开会话由 DeepX 决定,你自己不要提。`
+
 // BuildSystemPrompt 主 agent 的 system prompt = 共用核心 + 会话摘要尾部。
 // 摘要垫在最后:核心 + skill 那段会话内字节不变,即使摘要每次压缩都变,前缀仍命中,
 // 失效点只从摘要开始(详见前缀缓存优化设计)。
 func BuildSystemPrompt(workspace, skillCatalog, summary string) string {
-	base := coreSystemPrompt(workspace, skillCatalog)
+	base := coreSystemPrompt(workspace, skillCatalog) + topicTrackingPrompt
 	// 持久偏好/项目约定:读会话内冻结的快照(currentPrefs),只在启动 / 压缩时由 RefreshPreferences 刷新,
 	// 中途每轮复用同一份 → 前缀稳定、缓存命中。会话中途写入的 AGENTS.md 下次压缩/重启才生效。
 	if prefs := currentPrefs(); prefs != "" {
@@ -1257,6 +1279,21 @@ func CompactTriggerTokens(ctxWin int) int {
 	return ctxWin - headroom
 }
 
+// TopicHintTokens 返回「上下文用量达到多少 token,主题一旦切换就值得提醒用户 /new」的阈值。
+//
+// 提醒的价值全在「赶在压缩之前」:换会话是零成本的,而压缩是花钱去有损保留一个已经不相关的
+// 话题 —— 真换了话题的会话,那笔压缩钱花得最冤。所以取窗口的 topicHintPct%,再钳到压缩阈值
+// 的 80% 以下:大窗口下 50% 本来就远早于 ~70% 的压缩线,小窗口压缩来得早(64K 时 34K 就压),
+// 靠这个钳位保证任何窗口下提醒都排在压缩前面。
+//
+// 低于这条线不提醒:上下文还小的时候换不换会话都无所谓,提醒纯属噪音。
+func TopicHintTokens(ctxWin int) int {
+	if ctxWin <= 0 {
+		return 0
+	}
+	return min(ctxWin*topicHintPct/100, CompactTriggerTokens(ctxWin)*80/100)
+}
+
 // isRetryableStatus 判断 HTTP 状态码是否值得重试:429 限流、5xx 服务端错误、408/409/425 等瞬时类。
 // 400/401/403/404/422 是请求本身的问题(尤其 400 常是上下文超长),重试无意义 → 直接返回错误。
 func isRetryableStatus(code int) bool {

tui/i18n.go

@@ -265,6 +265,11 @@ var translations = map[string]map[Lang]string{
 		LangZH: "未知沙箱模式:%s(可选 off / native / docker)",
 		LangEN: "Unknown sandbox mode: %s (choose off / native / docker)",
 	},
+	// %d = 当前上下文占窗口的百分比。只有过了 TopicHintTokens 这条线才会发,所以这个数一定不小。
+	"topic.shift": {
+		LangZH: "会话主题已切换,建议 /new 开一段新对话 —— 旧话题的上下文对新问题只是噪音,还占着 %d%% 的窗口。",
+		LangEN: "Topic changed — consider /new for a fresh conversation. The old context is just noise for this question, and it's holding %d%% of the window.",
+	},
 	"session.new": {
 		LangZH: "已开启全新对话。上一段对话已保存,/sessions 可找回。",
 		LangEN: "Started a new conversation. The previous one is saved — see /sessions.",
@@ -485,17 +490,19 @@ var translations = map[string]map[Lang]string{
 	"panel.codegraph": {LangZH: "代码图谱", LangEN: "CodeGraph"},
 	"panel.sandbox":   {LangZH: "沙箱", LangEN: "Sandbox"},
 	"panel.workmode":  {LangZH: "工作模式", LangEN: "Working mode"},
+	"panel.topic":     {LangZH: "会话主题", LangEN: "Topic"},
 	"panel.todo":      {LangZH: "待办", LangEN: "Todo"}, // Todo 工具:主 agent 顺序清单
 	"panel.plan":      {LangZH: "计划", LangEN: "Plan"}, // CreatePlan:并发子 agent DAG
 
 	// === Right panel labels ===
-	"panel.label.used":   {LangZH: "占用", LangEN: "Used"},
-	"panel.label.output": {LangZH: "输出", LangEN: "Output"},
-	"panel.label.cache":  {LangZH: "缓存", LangEN: "Cache"},
+	"panel.label.used":     {LangZH: "占用", LangEN: "Used"},
+	"panel.label.output":   {LangZH: "输出", LangEN: "Output"},
+	"panel.label.cache":    {LangZH: "缓存", LangEN: "Cache"},
 	"panel.label.sbmode":   {LangZH: "隔离", LangEN: "Isolation"},
 	"panel.label.wmode":    {LangZH: "方法", LangEN: "Method"},
 	"panel.label.endpoint": {LangZH: "接口", LangEN: "Endpoint"},
 	"panel.label.balance":  {LangZH: "余额", LangEN: "Balance"},
+	"panel.topic.none":     {LangZH: "(尚未识别)", LangEN: "(not identified yet)"},
 
 	// === Status values ===
 	"status.idle":      {LangZH: "idle", LangEN: "idle"},

tui/model.go

@@ -172,6 +172,11 @@ type model struct {
 	workingMode agent.WorkingMode // 工作模式 kp/openspec/sp(默认 kp);每轮注入对应 skill 引导;按 session 保存/恢复
 	history     []agent.ChatMessage
 
+	// 会话主题:模型每轮结尾输出 <topic shift="yes|no">…</topic>(见 topic.go)。会话级状态,
+	// 跟着 /new、/sessions 走。topicF 逐 chunk 把标签从显示里剥掉并捕获主题。
+	topic  string
+	topicF topicFilter
+
 	streamCh <-chan tea.Msg
 
 	status    string
@@ -773,6 +778,7 @@ func initialModel(models agent.ModelConfig, needsSetup bool, version string, hub
 				gobHistory = gobHistory[1:]
 			}
 			m.history = gobHistory
+			m.topic = lastTopicOf(gobHistory) // 冷启动恢复右栏主题(不额外存盘,从历史里读回)
 			rebuildChatFromHistory(m.chatContent, gobHistory)
 			// 老对话(升级前就有 history、没 conv.json)首次进 /sessions 别显示"(未命名)":
 			// 用第一条用户消息回填标题。session 包自己解码不了 history.gob,放这儿做。
@@ -1180,6 +1186,7 @@ func (m model) submitUserInput(input string) (model, tea.Cmd) {
 	m.streaming = true
 	m.thinking = true
 	m.currentReply.Reset()
+	m.topicF = topicFilter{} // 上一轮若被 ESC 打断,过滤器可能停在标签中间
 
 	// 仪表盘:开计时,快照本轮输入字符数,输出归零
 	m.turnStartedAt = time.Now()
@@ -2599,10 +2606,20 @@ func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 		m.activeTool = ""
 		m.retryNotice = "" // 流式开始 = 重试成功,清掉提示
 		text := string(msg)
-		m.currentReply.WriteString(text)
-		// 上一段若是 tools(刚执行完工具,模型继续说话),切回 assistant 段。
-		m.chatContent.EnsureKind(kindAssistant, "")
-		m.chatContent.Append(text)
+		m.currentReply.WriteString(text) // 落盘/入历史的是原文,只有显示层剥主题标签
+		// 主题标签是给 DeepX 看的元信息,不显示给用户。标签可能被切在 chunk 中间,
+		// 故过滤器是有状态的(见 topic.go),这里逐段喂。
+		visible := m.topicF.feed(text)
+		// 整段都被标签吃掉时什么都不做 —— 开了段就会给"只输出了一行标签"的轮次留个空气泡,
+		// 广播出去也只是条空事件。
+		if visible != "" {
+			// 上一段若是 tools(刚执行完工具,模型继续说话),切回 assistant 段。
+			m.chatContent.EnsureKind(kindAssistant, "")
+			m.chatContent.Append(visible)
+			// token 事件不走 Update 顶部的通用广播(那里拿不到过滤后的文本),在这里显式发,
+			// 免得标签漏到浏览器面板上。
+			m.broadcast(web.Event{Kind: "token", Text: visible})
+		}
 		m.tokens += len([]rune(text))
 		m.turnOutputChars += len([]rune(text))
 		m.refreshViewport()
@@ -2614,6 +2631,11 @@ func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 		}
 		m.status = "tool"
 		m.retryNotice = "" // 工具开始 = 上游请求已成功,清掉重试提示
+		// 助手这段话说完了:把主题过滤器攒着的未决尾巴(可能是 "<topic" 的部分前缀)吐回当前段,
+		// 否则它会跟到下面的工具行后面去。
+		if tail := m.topicF.flush(); tail != "" {
+			m.chatContent.Append(tail)
+		}
 		// Todo 是可见清单的「全量快照更新」,不是一次干活动作:清单本身由下方 live overlay
 		// 实时勾选,每次更新再往 chat 打一行 "📌 Todo" 是纯噪音。所以 Todo 不留 chat 行、
 		// 也不计入"N 次工具调用"。其余工具照旧:紧凑单行 <icon> Name (主参数)。
@@ -2945,6 +2967,17 @@ func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 				_ = m.session.Append("assistant", final)
 			}
 		}
+		// 本轮收尾:先把主题过滤器的未决尾巴吐回对话区,再落地这一轮的主题 / 判是否提醒 /new。
+		// 尾巴只剩空白就丢掉 —— 话都说完了,补几个换行只会在气泡末尾留空行。
+		if tail := strings.TrimRight(m.topicF.flush(), " \t\r\n"); tail != "" {
+			m.chatContent.Append(tail)
+		}
+		ctxWin := m.models.Pro.ContextWindow
+		if ctxWin <= 0 {
+			ctxWin = 65536
+		}
+		m.applyTurnTopic(ctxWin)
+
 		m.status = "idle"
 		m.streaming = false
 		m.thinking = false
@@ -2961,10 +2994,7 @@ func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 		// 这里无需额外动作 — 旧的 trimDisplayTurns 按"10 轮"裁的逻辑已被 chatLog 取代。
 
 		// 检查是否需要触发会话压缩:上下文用量达到 agent.CompactTriggerTokens(动态阈值)就压。
-		ctxWin := m.models.Pro.ContextWindow
-		if ctxWin <= 0 {
-			ctxWin = 65536
-		}
+		// ctxWin 在上面判主题提醒时已经算好。
 		if m.session != nil && m.models.Pro.Model != "" && !m.compacting && m.lastPromptTokens() >= agent.CompactTriggerTokens(ctxWin) {
 			m.compacting = true
 			m.compactingFG = true // 前台阻塞:同手动 /compact,footer 转 spinner + 期间挡输入。子 agent 走 runSubAgent 不经此处,天然例外。
@@ -3374,6 +3404,9 @@ func (m *model) appendChat(role, text string) {
 // chatDisplayText 取一条消息用于"对话区显示"的文本:优先 Content,否则从 ContentParts 拼出文本、图片用 [图片] 表示。
 func chatDisplayText(msg agent.ChatMessage) string {
 	if msg.Content != "" {
+		if msg.Role == "assistant" {
+			return stripTopicTags(msg.Content) // 历史存的是原文,重放到对话区时才剥主题标签
+		}
 		return msg.Content
 	}
 	var sb strings.Builder
@@ -4190,6 +4223,7 @@ func (m *model) startWorkflowTurn(rawInput, name string, args any) tea.Cmd {
 	m.streaming = true
 	m.thinking = true
 	m.currentReply.Reset()
+	m.topicF = topicFilter{}
 	m.turnStartedAt = time.Now()
 	m.turnInputChars = sumHistoryChars(m.history)
 	m.turnOutputChars = 0

tui/session_modal.go

@@ -86,6 +86,9 @@ func (m *model) loadCurrentConversation() {
 	m.plan = nil
 	m.planKind = ""
 	m.pendingUserText = ""
+	// 主题是会话级状态(同 workingMode / modelPin),跟着会话走,不跨会话串。
+	// 不用管 topicF:切会话在流式中被拒,而每轮开头都会把它清零。
+	m.topic = ""
 	// 切会话:影子档位归零;代数 +1 让另一会话在飞的影子结果 gen 失配被丢弃(不串台存错会话)。
 	m.shadowDonePct = 0
 	m.compactGen++
@@ -107,6 +110,7 @@ func (m *model) loadCurrentConversation() {
 				gobHistory = gobHistory[1:]
 			}
 			m.history = gobHistory
+			m.topic = lastTopicOf(gobHistory) // 右栏主题跟着切过去的会话恢复
 			rebuildChatFromHistory(m.chatContent, gobHistory)
 		}
 	}