Function Phase4IntegrationExample

examples/phase4_integration.go:14–66

Phase4IntegrationExample 展示如何集成 SummarizationMiddleware 和 AgentMemoryMiddleware

()

// Phase4IntegrationExample 展示如何集成 SummarizationMiddleware 和 AgentMemoryMiddleware
func Phase4IntegrationExample() error {
	_ = context.Background()

	// 1. 创建后端存储
	// 使用 CompositeBackend 支持多路径路由
	filesystemBackend := createFilesystemBackend()
	composite := backends.NewCompositeBackend(filesystemBackend, []backends.RouteConfig{
		{Prefix: "/memories/", Backend: createStoreBackend()},
	})

	// 2. 创建自定义 Summarizer (使用真实 LLM)
	customSummarizer := func(ctx context.Context, messages []types.Message) (string, error) {
		// 这里应该调用真实的 LLM API 生成总结
		// 示例使用 Provider 接口
		return generateSummaryWithLLM(ctx, messages)
	}

	// 3. 创建 SummarizationMiddleware
	summarizationMW, err := middleware.NewSummarizationMiddleware(&middleware.SummarizationMiddlewareConfig{
		Summarizer:             customSummarizer,
		MaxTokensBeforeSummary: 170000, // Claude 3.5 Sonnet 的 200k 上下文的 85%
		MessagesToKeep:         6,      // 保留最近 3 轮对话
		SummaryPrefix:          "## Previous conversation summary:",
	})
	if err != nil {
		return fmt.Errorf("failed to create summarization middleware: %w", err)
	}

	// 4. 创建 AgentMemoryMiddleware
	memoryMW, err := middleware.NewAgentMemoryMiddleware(&middleware.AgentMemoryMiddlewareConfig{
		Backend:    composite,
		MemoryPath: "/memories/",
		// 使用默认模板: <agent_memory>...</agent_memory>
	})
	if err != nil {
		return fmt.Errorf("failed to create agent memory middleware: %w", err)
	}

	// 5. 构建中间件栈
	middlewares := []middleware.Middleware{
		memoryMW,        // 优先级 5: 最早执行,注入记忆
		summarizationMW, // 优先级 40: 在调用模型前检查并总结
		// ... 其他中间件
	}

	// 6. 创建 Agent 并使用中间件
	fmt.Printf("Created agent with %d middlewares\n", len(middlewares))
	fmt.Printf("- SummarizationMiddleware: max_tokens=%d, keep_messages=%d\n",
		170000, 6)
	fmt.Printf("- AgentMemoryMiddleware: memory_path=%s\n", "/memories/")

	return nil
}

// generateSummaryWithLLM 使用 LLM 生成对话总结
func generateSummaryWithLLM(_ context.Context, messages []types.Message) (string, error) {