Function main

examples/memory-advanced/main.go:196–297

()

}

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

	// 1. 构建 Backend: 默认使用 StateBackend + 本地 ./memories 目录
	stateBackend := backends.NewStateBackend()
	localMemBackend := backends.NewStateBackend()

	memoryBackend := backends.NewCompositeBackend(
		stateBackend,
		[]backends.RouteConfig{
			{
				Prefix:  "/memories/",
				Backend: localMemBackend,
			},
		},
	)

	// 2. 创建 AgentMemoryMiddleware, 以便获取底层 memory_write 工具
	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{memoryMW})
	allTools := stack.Tools()

	var memoryWriteTool tools.Tool
	for _, t := range allTools {
		if t.Name() == "memory_write" {
			memoryWriteTool = t
			break
		}
	}
	if memoryWriteTool == nil {
		log.Fatalf("memory_write 工具未找到, 请检查 AgentMemoryMiddleware 实现")
	}

	// 4. 构造封装工具 (以 user=alice, project=demo, article=abc 为例)
	userID := "alice"
	projectID := "demo"

	userPrefTool := &userPreferenceTool{
		userID:      userID,
		memoryWrite: memoryWriteTool,
	}

	projectFact := &projectFactTool{
		userID:      userID,
		projectID:   projectID,
		shared:      false, // 用户级项目记忆
		memoryWrite: memoryWriteTool,
	}

	resourceNote := &resourceNoteTool{