Function main

examples/memory/main.go:21–102

()

)

func main() {
	ctx := context.Background()

	// 1. 构建 Backend:
	//    - 使用 StateBackend (内存临时文件)
	stateBackend := backends.NewStateBackend()

	// Note: 在实际应用中，可以使用FilesystemBackend或其他持久化Backend
	// memoryBackend := backends.NewCompositeBackend(...)
	memoryBackend := stateBackend

	// 2. 创建 Filesystem + AgentMemory 中间件
	fsMiddleware := middleware.NewFilesystemMiddleware(&middleware.FilesystemMiddlewareConfig{
		Backend: memoryBackend,
	})

	memoryMW, err := middleware.NewAgentMemoryMiddleware(&middleware.AgentMemoryMiddlewareConfig{
		Backend:    memoryBackend,
		MemoryPath: "/memories/",
	})
	if err != nil {
		log.Fatalf("create AgentMemoryMiddleware failed: %v", err)
	}

	// 3. 组装中间件栈并收集工具
	stack := middleware.NewStack([]middleware.Middleware{
		fsMiddleware,
		memoryMW,
	})

	allTools := stack.Tools()
	fmt.Printf("✅ 中间件栈已创建, 工具总数: %d\n\n", len(allTools))

	var memoryWriteTool tools.Tool
	var memorySearchTool tools.Tool

	for _, t := range allTools {
		fmt.Printf("- 工具: %-16s 描述: %s\n", t.Name(), t.Description())
		switch t.Name() {
		case "memory_write":
			memoryWriteTool = t
		case "memory_search":
			memorySearchTool = t
		}
	}
	fmt.Println()

	if memoryWriteTool == nil || memorySearchTool == nil {
		log.Fatalf("memory_write 或 memory_search 工具未找到, 请检查 AgentMemoryMiddleware 初始化是否成功")
	}

	toolCtx := &tools.ToolContext{} // 本例中不需要 Sandbox, 传空即可

	// 4. 使用 memory_write 追加一条长期记忆
	fmt.Println("=== 使用 memory_write 追加记忆 ===")
	writeInput := map[string]any{
		"file":    "user/alice.md",
		"mode":    "append",