2a96f4c6f9
后端： 1.阶段 6 agent / memory 服务化收口 - 新增 cmd/agent 独立进程入口，承载 agent zrpc server、agent outbox relay / consumer 和运行时依赖初始化 - 补齐 services/agent/rpc 的 Chat stream 与 conversation meta/list/timeline、schedule-preview、context-stats、schedule-state unary RPC - 新增 gateway/client/agent 与 shared/contracts/agent，将 /api/v1/agent chat 和非 chat 门面切到 agent zrpc - 收缩 gateway 本地 AgentService 装配，双 RPC 开关开启时不再初始化本地 agent 编排、LLM、RAG 和 memory reader fallback - 将 backend/memory 物理迁入 services/memory，私有实现收入 internal，保留 module/model/observe 作为 memory 服务门面 - 调整 memory outbox、memory reader 和 agent 记忆渲染链路的 import 与服务边界，cmd/memory 独占 memory worker / consumer - 关闭 gateway 侧 agent outbox worker 所有权，agent relay / consumer 由 cmd/agent 独占，gateway 仅保留 HTTP/SSE 门面与迁移期开关回退 - 更新阶段 6 文档，记录 agent / memory 当前切流点、smoke 结果，以及 backend/client 与 gateway/shared 的目录收口口径
82 lines
2.5 KiB
Go
82 lines
2.5 KiB
Go
package utils
|
|
|
|
import (
|
|
"github.com/LoveLosita/smartflow/backend/model"
|
|
memorymodel "github.com/LoveLosita/smartflow/backend/services/memory/model"
|
|
)
|
|
|
|
// EffectiveUserSetting 返回用户记忆设置的生效值。
|
|
//
|
|
// 规则说明：
|
|
// 1. 用户未显式配置时，走系统默认值；
|
|
// 2. 默认允许普通记忆和隐式记忆，但默认关闭敏感记忆；
|
|
// 3. 返回值始终是完整对象，方便调用方直接使用，不再分支判空。
|
|
func EffectiveUserSetting(setting *model.MemoryUserSetting, userID int) model.MemoryUserSetting {
|
|
if setting == nil {
|
|
return model.MemoryUserSetting{
|
|
UserID: userID,
|
|
MemoryEnabled: true,
|
|
ImplicitMemoryEnabled: true,
|
|
SensitiveMemoryEnabled: false,
|
|
}
|
|
}
|
|
return *setting
|
|
}
|
|
|
|
// FilterFactsBySetting 按用户记忆开关过滤候选事实。
|
|
func FilterFactsBySetting(facts []memorymodel.NormalizedFact, setting model.MemoryUserSetting) []memorymodel.NormalizedFact {
|
|
if !setting.MemoryEnabled || len(facts) == 0 {
|
|
return nil
|
|
}
|
|
|
|
result := make([]memorymodel.NormalizedFact, 0, len(facts))
|
|
for _, fact := range facts {
|
|
if !setting.ImplicitMemoryEnabled && !fact.IsExplicit {
|
|
continue
|
|
}
|
|
if !setting.SensitiveMemoryEnabled && fact.SensitivityLevel > 0 {
|
|
continue
|
|
}
|
|
result = append(result, fact)
|
|
}
|
|
return result
|
|
}
|
|
|
|
// FilterItemsBySetting 按用户记忆开关过滤已入库记忆。
|
|
func FilterItemsBySetting(items []model.MemoryItem, setting model.MemoryUserSetting) []model.MemoryItem {
|
|
if !setting.MemoryEnabled || len(items) == 0 {
|
|
return nil
|
|
}
|
|
|
|
result := make([]model.MemoryItem, 0, len(items))
|
|
for _, item := range items {
|
|
if !setting.ImplicitMemoryEnabled && !item.IsExplicit {
|
|
continue
|
|
}
|
|
if !setting.SensitiveMemoryEnabled && item.SensitivityLevel > 0 {
|
|
continue
|
|
}
|
|
result = append(result, item)
|
|
}
|
|
return result
|
|
}
|
|
|
|
// FilterFactsByConfidence 按置信度阈值过滤候选事实。
|
|
//
|
|
// 说明：
|
|
// 1. minConfidence <= 0 时不做过滤，保持向后兼容；
|
|
// 2. 过滤在 FilterFactsBySetting 之后执行，是写入链路的第二道程序化门槛；
|
|
// 3. 阈值由 memory.write.minConfidence 配置控制，默认 0.5。
|
|
func FilterFactsByConfidence(facts []memorymodel.NormalizedFact, minConfidence float64) []memorymodel.NormalizedFact {
|
|
if minConfidence <= 0 || len(facts) == 0 {
|
|
return facts
|
|
}
|
|
result := make([]memorymodel.NormalizedFact, 0, len(facts))
|
|
for _, fact := range facts {
|
|
if fact.Confidence >= minConfidence {
|
|
result = append(result, fact)
|
|
}
|
|
}
|
|
return result
|
|
}
|