Function NewLogicMemoryMiddleware

pkg/middleware/logic_memory.go:87–150

NewLogicMemoryMiddleware 创建 Logic Memory 中间件

(config *LogicMemoryMiddlewareConfig)

// NewLogicMemoryMiddleware 创建 Logic Memory 中间件
func NewLogicMemoryMiddleware(config *LogicMemoryMiddlewareConfig) (*LogicMemoryMiddleware, error) {
	if config == nil {
		return nil, errors.New("logic memory config is required")
	}

	if config.Manager == nil {
		return nil, errors.New("logic memory manager is required")
	}

	// 设置默认值
	if config.MaxMemories <= 0 {
		config.MaxMemories = 5
	}
	if config.MinConfidence <= 0 {
		config.MinConfidence = 0.6
	}
	if config.EventBufferSize <= 0 {
		config.EventBufferSize = 100
	}
	if config.CacheTTL <= 0 {
		config.CacheTTL = 5 * time.Minute
	}
	if config.Priority <= 0 {
		config.Priority = 7
	}
	if config.InjectionPoint == "" {
		config.InjectionPoint = "system_prompt_end"
	}
	if config.SystemPromptTemplate == "" {
		config.SystemPromptTemplate = defaultLogicMemoryPromptTemplate
	}
	if len(config.CaptureChannels) == 0 {
		config.CaptureChannels = []types.AgentChannel{
			types.ChannelControl,
			types.ChannelMonitor,
		}
	}

	// 默认的 namespace 提取器
	namespaceExtractor := config.NamespaceExtractor
	if namespaceExtractor == nil {
		namespaceExtractor = defaultNamespaceExtractor
	}

	m := &LogicMemoryMiddleware{
		BaseMiddleware:     NewBaseMiddleware("logic_memory", config.Priority),
		manager:            config.Manager,
		config:             config,
		namespaceExtractor: namespaceExtractor,
		stopCh:             make(chan struct{}),
	}

	// 创建事件缓冲区（如果启用异步捕获）
	if config.AsyncCapture && config.EnableCapture {
		m.eventBuffer = make(chan *logic.Event, config.EventBufferSize)
	}

	// 创建 Logic Memory 工具