Function NewWorkingMemoryMiddleware

pkg/middleware/working_memory.go:44–83

NewWorkingMemoryMiddleware 创建 Working Memory 中间件

(config *WorkingMemoryMiddlewareConfig)

// NewWorkingMemoryMiddleware 创建 Working Memory 中间件
func NewWorkingMemoryMiddleware(config *WorkingMemoryMiddlewareConfig) (*WorkingMemoryMiddleware, error) {
	if config == nil {
		return nil, errors.New("working memory not configured properly")
	}

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

	// 创建 Working Memory 管理器
	manager, err := memory.NewWorkingMemoryManager(&memory.WorkingMemoryConfig{
		Backend:    config.Backend,
		BasePath:   config.BasePath,
		Scope:      config.Scope,
		Schema:     config.Schema,
		Template:   config.Template,
		DefaultTTL: config.TTL,
	})
	if err != nil {
		return nil, fmt.Errorf("create working memory manager: %w", err)
	}

	systemPromptTemplate := config.SystemPromptTemplate
	if systemPromptTemplate == "" {
		systemPromptTemplate = "<working_memory>\n%s\n</working_memory>"
	}

	m := &WorkingMemoryMiddleware{
		BaseMiddleware:       NewBaseMiddleware("working_memory", 6), // 比 agent_memory 稍低优先级
		manager:              manager,
		systemPromptTemplate: systemPromptTemplate,
		experimental:         config.Experimental,
	}

	// 创建 Working Memory 工具
	m.workingMemoryTools = m.createWorkingMemoryTools()

	wmLog.Info(context.Background(), "initialized", map[string]any{"scope": config.Scope, "base_path": config.BasePath})
	return m, nil
}

// Tools 返回 Working Memory 相关工具
func (m *WorkingMemoryMiddleware) Tools() []tools.Tool {