Function NewAgentMemoryMiddleware

pkg/middleware/agent_memory.go:48–80

NewAgentMemoryMiddleware 创建中间件

(config *AgentMemoryMiddlewareConfig)

// NewAgentMemoryMiddleware 创建中间件
func NewAgentMemoryMiddleware(config *AgentMemoryMiddlewareConfig) (*AgentMemoryMiddleware, error) {
	if config == nil {
		return nil, errors.New("agent memory not configured properly")
	}

	if config.Backend == nil {
		return nil, errors.New("backend is required")
	}

	if config.MemoryPath == "" {
		config.MemoryPath = "/memories/"
	}

	if config.SystemPromptTemplate == "" {
		config.SystemPromptTemplate = "<agent_memory>\n%s\n</agent_memory>"
	}

	m := &AgentMemoryMiddleware{
		BaseMiddleware:       NewBaseMiddleware("agent_memory", 5), // 高优先级,早期执行
		backend:              config.Backend,
		memoryPath:           config.MemoryPath,
		systemPromptTemplate: config.SystemPromptTemplate,
		baseNamespace:        config.BaseNamespace,
		memoryLoaded:         false,
		memoryContent:        "",
	}

	// 创建基于 backend 的高级记忆工具 (搜索 / 写入等)
	m.memoryTools = m.createMemoryTools()

	amLog.Info(context.Background(), "initialized", map[string]any{"memory_path": config.MemoryPath})
	return m, nil
}

// Tools 返回由 AgentMemoryMiddleware 注入的长期记忆相关工具
// 这些工具专门针对 memoryPath 下的记忆文件,提供搜索与写入能力