Version: 0.5.8.dev.260315

♻️ refactor(agent): 拆分 agentsvc，并增强 quicknote/outbox 注释与可维护性 - 📦 将 Agent 服务实现从 `service` 根目录迁移到 `service/agentsvc`，包含 `agent.go`、`agent_quick_note.go` 及相关测试 - 🔌 新增 service 层兼容桥接 `agent_bridge.go`，保持 `service.NewAgentService` 与 `*service.AgentService` 现有调用方式不变 - 📝 为 `quicknote` 补充高密度中文步骤化注释，覆盖 `graph` / `runner` / `nodes` / `tool` / `state` / `prompt`，明确职责边界、分支条件、重试与兜底策略 - 🧭 为 `infra/outbox` 与 service agent 链路补充详细中文注释，覆盖状态机流转、幂等处理、失败回写与异步持久化语义 - ✅ 统一格式化相关文件，并通过全量后端测试：`go test ./...` 📝 chore(docs): 更新 AGENTS.md 注释强制规范 - 📚 追加“注释规范（强制）”与“注释风格示例” - ✍️ 明确复杂逻辑必须使用步骤化注释、跨文件调用需写调用目的、注释需同步维护
2026-03-15 18:08:33 +08:00
parent c689af56c8
commit 7603a7561a
22 changed files with 1009 additions and 429 deletions
--- a/backend/service/agentsvc/agent.go
+++ b/backend/service/agentsvc/agent.go
@@ -0,0 +1,301 @@
+package agentsvc
+
+import (
+	"context"
+	"log"
+	"strings"
+	"time"
+
+	"github.com/LoveLosita/smartflow/backend/agent/chat"
+	"github.com/LoveLosita/smartflow/backend/conv"
+	"github.com/LoveLosita/smartflow/backend/dao"
+	outboxinfra "github.com/LoveLosita/smartflow/backend/infra/outbox"
+	"github.com/LoveLosita/smartflow/backend/inits"
+	"github.com/LoveLosita/smartflow/backend/model"
+	"github.com/LoveLosita/smartflow/backend/pkg"
+	"github.com/cloudwego/eino-ext/components/model/ark"
+	"github.com/cloudwego/eino/schema"
+	"github.com/google/uuid"
+)
+
+type AgentService struct {
+	AIHub         *inits.AIHub
+	repo          *dao.AgentDAO
+	taskRepo      *dao.TaskDAO
+	agentCache    *dao.AgentCache
+	asyncPipeline *outboxinfra.ChatHistoryAsync
+}
+
+// NewAgentService 构造 AgentService。
+// 这里通过依赖注入把“模型、仓储、缓存、异步持久化通道”统一交给服务层管理，
+// 便于后续在单测中替换实现，或在启动流程中按环境切换配置。
+func NewAgentService(aiHub *inits.AIHub, repo *dao.AgentDAO, taskRepo *dao.TaskDAO, agentRedis *dao.AgentCache, asyncPipeline *outboxinfra.ChatHistoryAsync) *AgentService {
+	return &AgentService{
+		AIHub:         aiHub,
+		repo:          repo,
+		taskRepo:      taskRepo,
+		agentCache:    agentRedis,
+		asyncPipeline: asyncPipeline,
+	}
+}
+
+// normalizeConversationID 规范会话 ID。
+// 规则：
+// 1) 去除首尾空白；
+// 2) 若为空则生成 UUID，保证后续缓存/数据库操作始终有合法 chat_id。
+func normalizeConversationID(chatID string) string {
+	trimmed := strings.TrimSpace(chatID)
+	if trimmed == "" {
+		return uuid.NewString()
+	}
+	return trimmed
+}
+
+// pickChatModel 根据请求选择模型。
+// 当前约定：
+// - strategist：策略模型；
+// - 其余值默认 worker（包含空字符串场景）。
+func (s *AgentService) pickChatModel(requestModel string) (*ark.ChatModel, string) {
+	modelName := strings.TrimSpace(requestModel)
+	if strings.EqualFold(modelName, "strategist") {
+		return s.AIHub.Strategist, "strategist"
+	}
+	return s.AIHub.Worker, "worker"
+}
+
+// saveChatHistoryReliable 统一封装“聊天记录持久化入口”：
+// 1) 开启异步链路时，走 outbox + Kafka；
+// 2) 未开启时，直接同步写库。
+func (s *AgentService) saveChatHistoryReliable(ctx context.Context, payload model.ChatHistoryPersistPayload) error {
+	// 1. 未注入异步通道时（例如本地极简环境），直接同步写 DB。
+	//    这样可以保证功能不依赖 Kafka 也能跑通。
+	if s.asyncPipeline == nil {
+		return s.repo.SaveChatHistory(ctx, payload.UserID, payload.ConversationID, payload.Role, payload.Message)
+	}
+	// 2. 已启用异步通道时，只入 outbox，不在请求路径阻塞 Kafka。
+	return s.asyncPipeline.EnqueueChatHistoryPersist(ctx, payload)
+}
+
+// pushErrNonBlocking 向错误通道“尽力投递”错误。
+// 目的：
+// 1) 避免 goroutine 在 errChan 满时被阻塞导致泄漏；
+// 2) 保证主业务协程不因“错误上报拥塞”卡死。
+func pushErrNonBlocking(errChan chan error, err error) {
+	select {
+	case errChan <- err:
+	default:
+		log.Printf("错误通道已满，丢弃错误: %v", err)
+	}
+}
+
+// runNormalChatFlow 执行普通流式聊天链路（非随口记）。
+// 该函数被两处复用：
+// 1) 用户输入本就不是随口记；
+// 2) 开启随口记进度推送后，最终判定“非随口记”时回落到普通聊天。
+func (s *AgentService) runNormalChatFlow(
+	ctx context.Context,
+	selectedModel *ark.ChatModel,
+	resolvedModelName string,
+	userMessage string,
+	ifThinking bool,
+	userID int,
+	chatID string,
+	traceID string,
+	requestStart time.Time,
+	outChan chan<- string,
+	errChan chan error,
+) {
+	// 1. 先尝试从 Redis 读历史，命中可直接进入模型推理，减少 DB 压力。
+	chatHistory, err := s.agentCache.GetHistory(ctx, chatID)
+	if err != nil {
+		pushErrNonBlocking(errChan, err)
+		return
+	}
+
+	cacheMiss := false
+	if chatHistory == nil {
+		// 2. 缓存未命中时回源 DB，并转换为 Eino message 格式。
+		cacheMiss = true
+		histories, hisErr := s.repo.GetUserChatHistories(ctx, userID, pkg.HistoryFetchLimitByModel(resolvedModelName), chatID)
+		if hisErr != nil {
+			pushErrNonBlocking(errChan, hisErr)
+			return
+		}
+		chatHistory = conv.ToEinoMessages(histories)
+	}
+
+	// 3. 计算本次请求可用的历史 token 预算，并执行历史裁剪。
+	//    这样可以在上下文增长时稳定控制模型窗口，避免超长上下文引发报错或高延迟。
+	historyBudget := pkg.HistoryTokenBudgetByModel(resolvedModelName, chat.SystemPrompt, userMessage)
+	trimmedHistory, totalHistoryTokens, keptHistoryTokens, droppedCount := pkg.TrimHistoryByTokenBudget(chatHistory, historyBudget)
+	chatHistory = trimmedHistory
+
+	// 4. 根据裁剪后历史长度更新 Redis 会话窗口配置，并主动执行窗口收敛。
+	targetWindow := pkg.CalcSessionWindowSize(len(chatHistory))
+	if err = s.agentCache.SetSessionWindowSize(ctx, chatID, targetWindow); err != nil {
+		log.Printf("设置历史窗口失败 chat=%s: %v", chatID, err)
+	}
+	if err = s.agentCache.EnforceHistoryWindow(ctx, chatID); err != nil {
+		log.Printf("执行历史窗口裁剪失败 chat=%s: %v", chatID, err)
+	}
+
+	if droppedCount > 0 {
+		log.Printf("历史裁剪: chat=%s total_tokens=%d kept_tokens=%d dropped=%d budget=%d target_window=%d",
+			chatID, totalHistoryTokens, keptHistoryTokens, droppedCount, historyBudget, targetWindow)
+	}
+
+	if cacheMiss {
+		// 5. 回源后把历史回填到 Redis，减少下一次请求的冷启动成本。
+		if err = s.agentCache.BackfillHistory(ctx, chatID, chatHistory); err != nil {
+			pushErrNonBlocking(errChan, err)
+			return
+		}
+	}
+
+	// 6. 执行真正的流式聊天。
+	//    fullText 用于后续写 Redis/持久化，outChan 用于把流片段实时推给前端。
+	fullText, streamErr := chat.StreamChat(ctx, selectedModel, resolvedModelName, userMessage, ifThinking, chatHistory, outChan, traceID, chatID, requestStart)
+	if streamErr != nil {
+		pushErrNonBlocking(errChan, streamErr)
+		return
+	}
+
+	// 7. 后置持久化（用户消息）：
+	//    7.1 先写 Redis，保证“最新会话上下文”可立即用于下一轮推理；
+	//    7.2 再走可靠持久化入口（outbox 或同步 DB）。
+	if err = s.agentCache.PushMessage(ctx, chatID, &schema.Message{Role: schema.User, Content: userMessage}); err != nil {
+		log.Printf("写入用户消息到 Redis 失败: %v", err)
+	}
+
+	if err = s.saveChatHistoryReliable(ctx, model.ChatHistoryPersistPayload{
+		UserID:         userID,
+		ConversationID: chatID,
+		Role:           "user",
+		Message:        userMessage,
+	}); err != nil {
+		pushErrNonBlocking(errChan, err)
+		return
+	}
+
+	// 普通聊天链路也需要把助手回复写入 Redis，
+	// 否则会出现“数据库有助手消息，但 Redis 最新会话只有用户消息”的口径不一致。
+	// 8. 后置持久化（助手消息）：
+	//    8.1 先写 Redis，保证下一轮上下文可见；
+	//    8.2 再异步可靠落库，失败通过 errChan 回传给上层。
+	if err = s.agentCache.PushMessage(context.Background(), chatID, &schema.Message{Role: schema.Assistant, Content: fullText}); err != nil {
+		log.Printf("写入助手消息到 Redis 失败: %v", err)
+	}
+
+	if saveErr := s.saveChatHistoryReliable(context.Background(), model.ChatHistoryPersistPayload{
+		UserID:         userID,
+		ConversationID: chatID,
+		Role:           "assistant",
+		Message:        fullText,
+	}); saveErr != nil {
+		pushErrNonBlocking(errChan, saveErr)
+	}
+}
+
+func (s *AgentService) AgentChat(ctx context.Context, userMessage string, ifThinking bool, modelName string, userID int, chatID string) (<-chan string, <-chan error) {
+	requestStart := time.Now()
+	traceID := uuid.NewString()
+
+	// 1. 每个请求都返回两个通道：
+	//    - outChan：推送流式输出片段；
+	//    - errChan：推送异步阶段错误（非阻塞上报）。
+	outChan := make(chan string, 8)
+	errChan := make(chan error, 1)
+
+	// 1) 规范会话 ID，选择模型。
+	chatID = normalizeConversationID(chatID)
+	selectedModel, resolvedModelName := s.pickChatModel(modelName)
+
+	// 2) 确保会话存在（优先缓存，必要时回源 DB 并创建）。
+	// 2.1 先查 Redis 会话标记，命中则可跳过 DB 存在性校验。
+	result, err := s.agentCache.GetConversationStatus(ctx, chatID)
+	if err != nil {
+		errChan <- err
+		close(outChan)
+		close(errChan)
+		return outChan, errChan
+	}
+	if !result {
+		// 2.2 缓存未命中时回源 DB：确认会话是否存在。
+		innerResult, ifErr := s.repo.IfChatExists(ctx, userID, chatID)
+		if ifErr != nil {
+			errChan <- ifErr
+			close(outChan)
+			close(errChan)
+			return outChan, errChan
+		}
+		if !innerResult {
+			// 2.3 DB 里也不存在则创建新会话。
+			if _, err = s.repo.CreateNewChat(userID, chatID); err != nil {
+				errChan <- err
+				close(outChan)
+				close(errChan)
+				return outChan, errChan
+			}
+		}
+		// 2.4 补写 Redis 会话标记，优化下次访问。
+		if err = s.agentCache.SetConversationStatus(ctx, chatID); err != nil {
+			log.Printf("设置会话状态缓存失败 chat=%s: %v", chatID, err)
+		}
+	}
+
+	// 3) 统一异步分流：
+	// - 先走“模型控制码路由”决定 quick_note / chat；
+	// - 路由命中 quick_note 时推阶段状态并执行 graph；
+	// - 路由命中 chat 时直接普通流式聊天。
+	go func() {
+		defer close(outChan)
+
+		// 3.1 先走轻量路由，判断是否进入“随口记”图。
+		routing := s.decideQuickNoteRouting(ctx, selectedModel, userMessage)
+		if !routing.EnterQuickNote {
+			// 3.2 非随口记：直接走普通聊天主链路。
+			s.runNormalChatFlow(ctx, selectedModel, resolvedModelName, userMessage, ifThinking, userID, chatID, traceID, requestStart, outChan, errChan)
+			return
+		}
+
+		// 3.3 随口记：先发阶段状态，减少用户等待时的“无反馈感”。
+		progress := newQuickNoteProgressEmitter(outChan, resolvedModelName, true)
+		progress.Emit("request.accepted", routing.Detail)
+
+		// 3.4 执行随口记 graph。
+		quickHandled, quickState, quickErr := s.tryHandleQuickNoteWithGraph(
+			ctx,
+			selectedModel,
+			userMessage,
+			userID,
+			chatID,
+			traceID,
+			routing.TrustRoute,
+			progress.Emit,
+		)
+		if quickErr != nil {
+			// graph 出错不直接中断用户请求，而是回退普通聊天，保证可用性优先。
+			log.Printf("随口记 graph 执行失败，回退普通聊天 trace_id=%s chat_id=%s err=%v", traceID, chatID, quickErr)
+		}
+
+		if quickHandled {
+			// 3.5 随口记处理成功：组织最终回复并按 OpenAI 兼容格式输出。
+			progress.Emit("quick_note.reply.polishing", "正在结合你的话题润色回复。")
+			quickReply := buildQuickNoteFinalReply(ctx, selectedModel, userMessage, quickState)
+			if emitErr := emitSingleAssistantCompletion(outChan, resolvedModelName, quickReply); emitErr != nil {
+				pushErrNonBlocking(errChan, emitErr)
+				return
+			}
+
+			// 3.6 对随口记回复执行统一后置持久化（Redis + outbox/DB）。
+			s.persistChatAfterReply(ctx, userID, chatID, userMessage, quickReply, errChan)
+			return
+		}
+
+		// 3.7 路由误判或 graph 判定非随口记时，回落普通聊天，保证“能聊”。
+		progress.Emit("quick_note.fallback", "当前输入不是随口记请求，切换到普通对话。")
+		s.runNormalChatFlow(ctx, selectedModel, resolvedModelName, userMessage, ifThinking, userID, chatID, traceID, requestStart, outChan, errChan)
+	}()
+
+	return outChan, errChan
+}
--- a/backend/service/agentsvc/agent_quick_note.go
+++ b/backend/service/agentsvc/agent_quick_note.go
@@ -0,0 +1,378 @@
+package agentsvc
+
+import (
+	"context"
+	"fmt"
+	"log"
+	"strings"
+	"time"
+
+	"github.com/LoveLosita/smartflow/backend/agent/chat"
+	"github.com/LoveLosita/smartflow/backend/agent/quicknote"
+	"github.com/LoveLosita/smartflow/backend/agent/route"
+	"github.com/LoveLosita/smartflow/backend/model"
+	"github.com/cloudwego/eino-ext/components/model/ark"
+	einoModel "github.com/cloudwego/eino/components/model"
+	"github.com/cloudwego/eino/schema"
+	"github.com/google/uuid"
+	arkModel "github.com/volcengine/volcengine-go-sdk/service/arkruntime/model"
+)
+
+// quickNoteRoutingDecision 只是路由层结果的本地别名。
+// 保留这个别名是为了尽量少改调用侧（agent.go 中的字段访问保持不变）。
+type quickNoteRoutingDecision = route.RoutingDecision
+
+// quickNoteProgressEmitter 负责把“链路阶段状态”伪装成 OpenAI 兼容的 reasoning_content chunk。
+// 设计目标：
+// 1) 不改现有 OpenAI 兼容协议外壳；
+// 2) 让 Apifox 在等待期间也能看到“思考块”，避免用户空等；
+// 3) 该 emitter 只负责状态，不负责最终正文回复和 [DONE] 结束块。
+type quickNoteProgressEmitter struct {
+	outChan    chan<- string
+	modelName  string
+	requestID  string
+	created    int64
+	enablePush bool
+}
+
+// newQuickNoteProgressEmitter 构造“阶段进度推送器”。
+// 该推送器只负责发 reasoning 块，不负责正文回复。
+func newQuickNoteProgressEmitter(outChan chan<- string, modelName string, enable bool) *quickNoteProgressEmitter {
+	// 1. 模型名兜底，避免出现空 model 字段导致客户端兼容性问题。
+	resolvedModel := strings.TrimSpace(modelName)
+	if resolvedModel == "" {
+		resolvedModel = "worker"
+	}
+	// 2. 每次请求生成独立 request_id，方便前端或日志侧关联本次流式输出。
+	return &quickNoteProgressEmitter{
+		outChan:    outChan,
+		modelName:  resolvedModel,
+		requestID:  "chatcmpl-" + uuid.NewString(),
+		created:    time.Now().Unix(),
+		enablePush: enable,
+	}
+}
+
+// Emit 按“阶段 + 说明”输出 reasoning_content。
+// 注意：
+// 1) 这里不输出 role，避免和后续正文 role 块冲突；
+// 2) 即使发送失败，也只记录日志，不影响主流程继续执行。
+func (e *quickNoteProgressEmitter) Emit(stage, detail string) {
+	// 1. 推送器不可用（nil/禁用/无通道）时直接返回，避免 panic。
+	if e == nil || !e.enablePush || e.outChan == nil {
+		return
+	}
+	// 2. 统一清理空白，避免日志和输出里出现异常空字符串。
+	stage = strings.TrimSpace(stage)
+	detail = strings.TrimSpace(detail)
+	if stage == "" && detail == "" {
+		return
+	}
+
+	reasoning := fmt.Sprintf("阶段：%s", stage)
+	if detail != "" {
+		reasoning += "\n" + detail
+	}
+
+	// 3. 复用 OpenAI 兼容封装：把阶段文本伪装成 reasoning_content。
+	chunk, err := chat.ToOpenAIStream(&schema.Message{ReasoningContent: reasoning}, e.requestID, e.modelName, e.created, false)
+	if err != nil {
+		// 3.1 阶段推送失败不应影响主链路，只打日志即可。
+		log.Printf("输出随口记阶段状态失败 stage=%s err=%v", stage, err)
+		return
+	}
+	if chunk != "" {
+		e.outChan <- chunk
+	}
+}
+
+// tryHandleQuickNoteWithGraph 尝试用“随口记 graph”处理本次用户输入。
+// 返回值语义：
+// 1) handled=true：本次请求已在随口记链路处理完成（成功/失败都会返回文案）；
+// 2) handled=false：不是随口记意图，调用方应回落普通聊天链路；
+// 3) state：用于拼接最终“一次性正文回复”。
+func (s *AgentService) tryHandleQuickNoteWithGraph(
+	ctx context.Context,
+	selectedModel *ark.ChatModel,
+	userMessage string,
+	userID int,
+	chatID string,
+	traceID string,
+	trustRoute bool,
+	emitStage func(stage, detail string),
+) (handled bool, state *quicknote.QuickNoteState, err error) {
+	// 1. 依赖预检：taskRepo 或模型未注入时，不做随口记处理，交给上层回落聊天。
+	if s.taskRepo == nil || selectedModel == nil {
+		return false, nil, nil
+	}
+
+	// 2. 初始化随口记状态对象（贯穿 graph 全流程的共享上下文）。
+	state = quicknote.NewQuickNoteState(traceID, userID, chatID, userMessage)
+
+	// 3. 执行 quick note graph。
+	//    本次依赖注入了两个“工具能力”：
+	//    3.1 ResolveUserID：从当前请求上下文确定 user_id；
+	//    3.2 CreateTask：真正执行任务写库。
+	finalState, runErr := quicknote.RunQuickNoteGraph(ctx, quicknote.QuickNoteGraphRunInput{
+		Model: selectedModel,
+		State: state,
+		Deps: quicknote.QuickNoteToolDeps{
+			ResolveUserID: func(ctx context.Context) (int, error) {
+				// 当前链路 userID 已由上层鉴权拿到，这里直接复用。
+				return userID, nil
+			},
+			CreateTask: func(ctx context.Context, req quicknote.QuickNoteCreateTaskRequest) (*quicknote.QuickNoteCreateTaskResult, error) {
+				// 3.2.1 把 quick note 的工具入参映射成项目 Task 模型。
+				taskModel := &model.Task{
+					UserID:      req.UserID,
+					Title:       req.Title,
+					Priority:    req.PriorityGroup,
+					IsCompleted: false,
+					DeadlineAt:  req.DeadlineAt,
+				}
+
+				// 3.2.2 调用 DAO 写库。
+				created, createErr := s.taskRepo.AddTask(taskModel)
+				if createErr != nil {
+					return nil, createErr
+				}
+
+				// 3.2.3 把写库结果回填给 graph 状态，用于后续回复拼装。
+				return &quicknote.QuickNoteCreateTaskResult{
+					TaskID:        created.ID,
+					Title:         created.Title,
+					PriorityGroup: created.Priority,
+					DeadlineAt:    created.DeadlineAt,
+				}, nil
+			},
+		},
+		SkipIntentVerification: trustRoute,
+		EmitStage:              emitStage,
+	})
+	if runErr != nil {
+		// 4. graph 执行失败由上层统一决定是否回退普通聊天。
+		return false, nil, runErr
+	}
+
+	// 5. graph 正常结束但判定“非随口记”时，明确返回 handled=false。
+	if finalState == nil || !finalState.IsQuickNoteIntent {
+		return false, nil, nil
+	}
+	// 6. 走到这里表示随口记链路已完成（含写库成功或业务失败反馈文案）。
+	return true, finalState, nil
+}
+
+// emitSingleAssistantCompletion 将单条完整回复包装成 OpenAI 兼容 chunk 流并写入 outChan。
+// 说明：
+// 1) 保持现有 OpenAI 兼容格式不变；
+// 2) 正文只发一次，不做伪分段。
+func emitSingleAssistantCompletion(outChan chan<- string, modelName, reply string) error {
+	// 1. 模型名兜底，保持 OpenAI 兼容响应字段完整。
+	if strings.TrimSpace(modelName) == "" {
+		modelName = "worker"
+	}
+	requestID := "chatcmpl-" + uuid.NewString()
+	created := time.Now().Unix()
+
+	// 2. 正文 chunk（完整一次性输出，不做人为拆片）。
+	chunk, err := chat.ToOpenAIStream(&schema.Message{Role: schema.Assistant, Content: reply}, requestID, modelName, created, true)
+	if err != nil {
+		return err
+	}
+	if chunk != "" {
+		outChan <- chunk
+	}
+
+	// 3. 按 OpenAI 风格补 finish chunk + [DONE]，确保客户端可正确收尾。
+	finishChunk, err := chat.ToOpenAIFinishStream(requestID, modelName, created)
+	if err != nil {
+		return err
+	}
+	outChan <- finishChunk
+	outChan <- "[DONE]"
+	return nil
+}
+
+// buildQuickNoteFinalReply 生成最终的一次性正文回复。
+// 组合策略：
+// 1) 任务事实（标题/优先级/截止时间）由后端拼接，确保准确；
+// 2) 轻松跟进句交给 AI 生成，贴合用户话题；
+// 3) AI 生成失败时自动降级为固定友好文案，保证稳定可用。
+func buildQuickNoteFinalReply(ctx context.Context, selectedModel *ark.ChatModel, userMessage string, state *quicknote.QuickNoteState) string {
+	// 1. 极端兜底：状态为空时给出稳定失败文案，避免返回空字符串。
+	if state == nil {
+		return "我这次没成功记上，别急，再发我一次我马上补上。"
+	}
+
+	// 仅当“确实拿到了有效 task_id”时才走成功文案，避免出现“回复成功但库里没数据”的错觉。
+	if state.Persisted && state.PersistedTaskID > 0 {
+		// 2. 组装“事实段”：标题 + 优先级 + 截止时间。
+		title := strings.TrimSpace(state.ExtractedTitle)
+		if title == "" {
+			title = "这条任务"
+		}
+
+		priorityText := "已安排优先级"
+		if quicknote.IsValidTaskPriority(state.ExtractedPriority) {
+			priorityText = fmt.Sprintf("优先级：%s", quicknote.PriorityLabelCN(state.ExtractedPriority))
+		}
+
+		deadlineText := ""
+		if state.ExtractedDeadline != nil {
+			deadlineText = fmt.Sprintf("；截止时间 %s", state.ExtractedDeadline.In(time.Local).Format("2006-01-02 15:04"))
+		}
+
+		factLine := fmt.Sprintf("好，给你安排上了：%s（%s%s）。", title, priorityText, deadlineText)
+
+		// 2.1 如果 graph 单次请求已生成 banter，直接使用，避免重复调用模型。
+		if strings.TrimSpace(state.ExtractedBanter) != "" {
+			return factLine + " " + strings.TrimSpace(state.ExtractedBanter)
+		}
+		// 2.2 聚合调用模式下，通常已在主流程完成风格化，给稳定文案即可。
+		if state.PlannedBySingleCall {
+			return factLine + " 已帮你稳稳记下，放心推进。"
+		}
+
+		// 2.3 兜底生成轻松跟进句；失败则降级固定文案，确保体验连续。
+		banter, err := generateQuickNoteBanter(ctx, selectedModel, userMessage, title, priorityText, deadlineText)
+		if err != nil {
+			return factLine + " 这下可以先安心推进，不用等 ddl 来敲门了。"
+		}
+		if strings.TrimSpace(banter) == "" {
+			return factLine + " 这下可以先安心推进，不用等 ddl 来敲门了。"
+		}
+		return factLine + " " + banter
+	}
+
+	// 3. 若时间校验失败，优先返回“可执行的修正引导”。
+	if strings.TrimSpace(state.DeadlineValidationError) != "" {
+		return "我识别到你给了时间，但格式不够明确，暂时不敢乱记。你可以改成比如：2026-03-20 18:30、明天下午3点、下周一上午9点，我立刻帮你安排。"
+	}
+
+	// 4. 若 graph 已给出助手回复（例如非意图/业务失败原因），优先透传。
+	if strings.TrimSpace(state.AssistantReply) != "" {
+		return strings.TrimSpace(state.AssistantReply)
+	}
+	// 5. 最终兜底文案。
+	return "这次没成功写入任务，我没跑路，再给我一次我就把它稳稳记上。"
+}
+
+// generateQuickNoteBanter 让模型根据用户原话生成一条“贴题轻松句”。
+// 约束：
+// 1) 只生成跟进语气，不承担事实表达；
+// 2) 不得改动任务事实；
+// 3) 输出控制在一句，方便直接拼接在事实句后。
+func generateQuickNoteBanter(
+	ctx context.Context,
+	selectedModel *ark.ChatModel,
+	userMessage string,
+	title string,
+	priorityText string,
+	deadlineText string,
+) (string, error) {
+	// 1. 模型防御校验。
+	if selectedModel == nil {
+		return "", fmt.Errorf("model is nil")
+	}
+
+	// 2. 把事实信息显式塞入 prompt，约束模型只能“润色语气”。
+	prompt := fmt.Sprintf(`用户原话：%s
+已确认事实：
+- 任务标题：%s
+- %s
+- %s
+
+请输出一句轻松自然的跟进话术（仅一句）。`,
+		strings.TrimSpace(userMessage),
+		strings.TrimSpace(title),
+		strings.TrimSpace(priorityText),
+		strings.TrimSpace(deadlineText),
+	)
+
+	// 3. 构造消息：
+	//    - system：定义输出边界（一句话、不改事实）；
+	//    - user：提供本次上下文素材。
+	messages := []*schema.Message{
+		schema.SystemMessage(quicknote.QuickNoteReplyBanterPrompt),
+		schema.UserMessage(prompt),
+	}
+
+	// 4. 调用模型生成 banter，并显式关闭 thinking，减少额外延迟。
+	resp, err := selectedModel.Generate(ctx, messages,
+		ark.WithThinking(&arkModel.Thinking{Type: arkModel.ThinkingTypeDisabled}),
+		einoModel.WithTemperature(0.7),
+		einoModel.WithMaxTokens(72),
+	)
+	if err != nil {
+		return "", err
+	}
+	if resp == nil {
+		return "", fmt.Errorf("empty response")
+	}
+
+	// 5. 输出清洗：
+	//    5.1 去首尾空白与引号；
+	//    5.2 若模型多行输出，只取第一行；
+	//    5.3 最终为空则视为失败，让上层走降级文案。
+	text := strings.TrimSpace(resp.Content)
+	text = strings.Trim(text, "\"'“”‘’")
+	if text == "" {
+		return "", fmt.Errorf("empty content")
+	}
+	if idx := strings.Index(text, "\n"); idx >= 0 {
+		text = strings.TrimSpace(text[:idx])
+	}
+	return text, nil
+}
+
+// decideQuickNoteRouting 决定当前输入是否进入“随口记 graph”。
+// 该函数只是服务层薄封装，具体控制码解析逻辑已下沉到 agent/route 包。
+func (s *AgentService) decideQuickNoteRouting(ctx context.Context, selectedModel *ark.ChatModel, userMessage string) quickNoteRoutingDecision {
+	// 这里保留方法是为了让 AgentService 对外语义完整，
+	// 同时避免上层调用方直接依赖 route 包，降低耦合。
+	_ = s
+	return route.DecideQuickNoteRouting(ctx, selectedModel, userMessage)
+}
+
+// persistChatAfterReply 在“随口记 graph”返回后，复用当前项目的后置持久化策略：
+// 1) 用户消息写 Redis + outbox/DB；
+// 2) 助手消息写 Redis + outbox/DB。
+func (s *AgentService) persistChatAfterReply(
+	ctx context.Context,
+	userID int,
+	chatID string,
+	userMessage string,
+	assistantReply string,
+	errChan chan error,
+) {
+	// 1. 先把用户消息写入 Redis，保证会话上下文“马上可见”。
+	if err := s.agentCache.PushMessage(ctx, chatID, &schema.Message{Role: schema.User, Content: userMessage}); err != nil {
+		log.Printf("写入用户消息到 Redis 失败: %v", err)
+	}
+
+	// 2. 再把用户消息写入可靠持久化通道（outbox 或同步 DB）。
+	if err := s.saveChatHistoryReliable(ctx, model.ChatHistoryPersistPayload{
+		UserID:         userID,
+		ConversationID: chatID,
+		Role:           "user",
+		Message:        userMessage,
+	}); err != nil {
+		pushErrNonBlocking(errChan, err)
+		return
+	}
+
+	// 3. 助手消息同样遵循“Redis 先行 + 可靠持久化补齐”策略。
+	if err := s.agentCache.PushMessage(context.Background(), chatID, &schema.Message{Role: schema.Assistant, Content: assistantReply}); err != nil {
+		log.Printf("写入助手消息到 Redis 失败: %v", err)
+	}
+
+	// 4. 助手消息持久化失败不阻断主流程，通过 errChan 异步上报。
+	if err := s.saveChatHistoryReliable(context.Background(), model.ChatHistoryPersistPayload{
+		UserID:         userID,
+		ConversationID: chatID,
+		Role:           "assistant",
+		Message:        assistantReply,
+	}); err != nil {
+		pushErrNonBlocking(errChan, err)
+	}
+}
--- a/backend/service/agentsvc/agent_quick_note_route_test.go
+++ b/backend/service/agentsvc/agent_quick_note_route_test.go
@@ -0,0 +1,73 @@
+package agentsvc
+
+import (
+	"strings"
+	"testing"
+
+	"github.com/LoveLosita/smartflow/backend/agent/quicknote"
+	"github.com/LoveLosita/smartflow/backend/agent/route"
+)
+
+// TestParseQuickNoteRouteControlTag_QuickNote
+// 目的：验证模型控制码在 action=quick_note 时可被稳定解析，
+// 并且会校验 nonce，避免历史脏内容或伪造片段误命中。
+func TestParseQuickNoteRouteControlTag_QuickNote(t *testing.T) {
+	nonce := "abc123nonce"
+	raw := `<SMARTFLOW_ROUTE nonce="abc123nonce" action="quick_note"></SMARTFLOW_ROUTE>
+<SMARTFLOW_REASON>用户明确在请求未来提醒</SMARTFLOW_REASON>`
+
+	decision, err := route.ParseQuickNoteRouteControlTag(raw, nonce)
+	if err != nil {
+		t.Fatalf("解析失败: %v", err)
+	}
+	if decision == nil {
+		t.Fatalf("decision 不应为空")
+	}
+	if decision.Action != route.ActionQuickNote {
+		t.Fatalf("action 解析错误，期望=%s 实际=%s", route.ActionQuickNote, decision.Action)
+	}
+	if strings.TrimSpace(decision.Reason) == "" {
+		t.Fatalf("reason 不应为空")
+	}
+}
+
+// TestParseQuickNoteRouteControlTag_NonceMismatch
+// 目的：确保 nonce 不匹配时直接报错，避免把非本次请求的控制码当作有效路由。
+func TestParseQuickNoteRouteControlTag_NonceMismatch(t *testing.T) {
+	raw := `<SMARTFLOW_ROUTE nonce="wrongnonce" action="chat"></SMARTFLOW_ROUTE>`
+	if _, err := route.ParseQuickNoteRouteControlTag(raw, "expectednonce"); err == nil {
+		t.Fatalf("期望 nonce 不匹配时报错，但未报错")
+	}
+}
+
+// TestBuildQuickNoteFinalReply_NoFalseSuccessWithoutTaskID
+// 目的：即使 state.Persisted 被错误置为 true，只要 task_id 无效，也不能返回“安排成功”文案。
+func TestBuildQuickNoteFinalReply_NoFalseSuccessWithoutTaskID(t *testing.T) {
+	state := &quicknote.QuickNoteState{
+		Persisted:       true,
+		PersistedTaskID: 0,
+		ExtractedTitle:  "去下馆子",
+	}
+
+	reply := buildQuickNoteFinalReply(nil, nil, "我今天晚上6点要去下馆子，记得喊我", state)
+	if strings.Contains(reply, "给你安排上了") || strings.Contains(reply, "已安排") {
+		t.Fatalf("不应返回成功文案，实际回复=%s", reply)
+	}
+}
+
+// TestBuildQuickNoteFinalReply_UseExtractedBanter
+// 目的：当聚合规划阶段已经产出 banter 时，最终回复应直接复用，避免再次调用润色模型。
+func TestBuildQuickNoteFinalReply_UseExtractedBanter(t *testing.T) {
+	state := &quicknote.QuickNoteState{
+		Persisted:         true,
+		PersistedTaskID:   12,
+		ExtractedTitle:    "明天去取快递",
+		ExtractedPriority: 2,
+		ExtractedBanter:   "取件路上注意保暖，别被风吹懵了。",
+	}
+
+	reply := buildQuickNoteFinalReply(nil, nil, "明天上午12点我要去取快递，到时候记得q我", state)
+	if !strings.Contains(reply, "取件路上注意保暖") {
+		t.Fatalf("期望复用 ExtractedBanter，实际回复=%s", reply)
+	}
+}