smartmate/backend/service/agentsvc/agent_memory.go

package agentsvc

import (
	"context"
	"fmt"
	"log"
	"strings"
	"time"

	memorymodel "github.com/LoveLosita/smartflow/backend/memory/model"
	newagentmodel "github.com/LoveLosita/smartflow/backend/newAgent/model"
)

const (
	newAgentMemoryBlockKey      = "memory_context"
	newAgentMemoryRetrieveLimit = 5
	newAgentMemoryBlockTitle    = "相关记忆"
	newAgentMemoryIntroLine     = "以下是与当前对话相关的用户记忆，仅在自然且确实有帮助时参考，不要生硬复述。"
)

// MemoryReader 描述 newAgent 主链路读取记忆所需的最小能力。
//
// 职责边界：
// 1. 只负责“按当前输入取回候选记忆”；
// 2. 不负责 prompt 拼装，也不要求调用方感知 memory 模块内部 repo/service 结构；
// 3. 返回值直接复用 memory DTO，避免 service 层再维护一套重复结构。
type MemoryReader interface {
	Retrieve(ctx context.Context, req memorymodel.RetrieveRequest) ([]memorymodel.ItemDTO, error)
}

// SetMemoryReader 注入 newAgent 主链路读取记忆所需的薄接口。
func (s *AgentService) SetMemoryReader(reader MemoryReader) {
	s.memoryReader = reader
}

// injectMemoryContext 在 graph 执行前，把本轮相关记忆写入 ConversationContext 的 pinned block。
//
// 步骤说明：
// 1. 先做前置门控：没有 reader、没有有效用户、或输入属于“确认/应答型短句”时，直接清掉旧 block，避免快照残留污染本轮 prompt。
// 2. 再调用 memory 检索：查询失败只记日志，不中断主链路，保证 newAgent 的可用性优先。
// 3. 检索成功后把结果渲染成稳定的中文文本，并用固定 key 覆盖写入，确保每轮都能刷新而不是越积越多。
func (s *AgentService) injectMemoryContext(
	ctx context.Context,
	conversationContext *newagentmodel.ConversationContext,
	userID int,
	chatID string,
	userMessage string,
) {
	if conversationContext == nil {
		return
	}

	if s.memoryReader == nil || userID <= 0 || !shouldInjectMemoryForInput(userMessage) {
		conversationContext.RemovePinnedBlock(newAgentMemoryBlockKey)
		return
	}

	items, err := s.memoryReader.Retrieve(ctx, memorymodel.RetrieveRequest{
		Query:          strings.TrimSpace(userMessage),
		UserID:         userID,
		ConversationID: strings.TrimSpace(chatID),
		Limit:          newAgentMemoryRetrieveLimit,
		Now:            time.Now(),
	})
	if err != nil {
		conversationContext.RemovePinnedBlock(newAgentMemoryBlockKey)
		log.Printf("读取记忆上下文失败 user=%d chat=%s err=%v", userID, chatID, err)
		return
	}

	content := renderMemoryPinnedContent(items)
	if content == "" {
		conversationContext.RemovePinnedBlock(newAgentMemoryBlockKey)
		return
	}

	conversationContext.UpsertPinnedBlock(newagentmodel.ContextBlock{
		Key:     newAgentMemoryBlockKey,
		Title:   newAgentMemoryBlockTitle,
		Content: content,
	})
}

// shouldInjectMemoryForInput 判断当前输入是否值得触发一次记忆召回。
//
// 步骤说明：
// 1. 空输入直接跳过；
// 2. 对“好/确认/ok”这类弱语义应答做显式拦截，避免 legacy fallback 在无查询价值时注入一批高分但不相关的旧记忆；
// 3. 其余输入一律放行，优先保证 MVP 可用。
func shouldInjectMemoryForInput(userMessage string) bool {
	trimmed := strings.TrimSpace(userMessage)
	if trimmed == "" {
		return false
	}

	switch strings.ToLower(trimmed) {
	case "好", "好的", "嗯", "嗯嗯", "行", "可以", "收到", "明白", "确认", "取消", "是", "不是", "对", "不对", "ok", "okay", "yes", "no":
		return false
	default:
		return true
	}
}

// renderMemoryPinnedContent 把召回结果转成一段稳定、紧凑、适合 prompt 注入的自然语言文本。
func renderMemoryPinnedContent(items []memorymodel.ItemDTO) string {
	if len(items) == 0 {
		return ""
	}

	var sb strings.Builder
	sb.WriteString(newAgentMemoryIntroLine)

	seen := make(map[string]struct{}, len(items))
	written := 0
	for _, item := range items {
		line := buildMemoryPinnedLine(item)
		if line == "" {
			continue
		}
		if _, exists := seen[line]; exists {
			continue
		}
		seen[line] = struct{}{}
		sb.WriteString("\n- ")
		sb.WriteString(line)
		written++
	}

	if written == 0 {
		return ""
	}
	return strings.TrimSpace(sb.String())
}

// buildMemoryPinnedLine 把单条记忆渲染成“[类型] 内容”的简洁格式。
func buildMemoryPinnedLine(item memorymodel.ItemDTO) string {
	text := strings.TrimSpace(item.Content)
	if text == "" {
		text = strings.TrimSpace(item.Title)
	}
	if text == "" {
		return ""
	}
	return fmt.Sprintf("[%s] %s", localizeMemoryType(item.MemoryType), text)
}

// localizeMemoryType 把 memory 类型映射成 prompt 里更自然的中文标签。
func localizeMemoryType(memoryType string) string {
	switch strings.TrimSpace(memoryType) {
	case memorymodel.MemoryTypePreference:
		return "偏好"
	case memorymodel.MemoryTypeConstraint:
		return "约束"
	case memorymodel.MemoryTypeTodoHint:
		return "待办线索"
	case memorymodel.MemoryTypeFact:
		return "事实"
	default:
		return "记忆"
	}
}