Function main

examples/memory-agent/main.go:21–145

()

)

func main() {
	// 检查 API Key
	apiKey := os.Getenv("ANTHROPIC_API_KEY")
	if apiKey == "" {
		log.Fatal("ANTHROPIC_API_KEY environment variable is required")
	}

	ctx := context.Background()

	// 1. 工具注册表
	toolRegistry := tools.NewRegistry()
	builtin.RegisterAll(toolRegistry)

	// 2. Sandbox
	sandboxFactory := sandbox.NewFactory()

	// 3. Provider 工厂
	providerFactory := &provider.AnthropicFactory{}

	// 4. Store
	jsonStore, err := store.NewJSONStore(".aster-memory-agent")
	if err != nil {
		log.Fatalf("create store failed: %v", err)
	}

	// 5. 模板注册表
	templateRegistry := agent.NewTemplateRegistry()
	templateRegistry.Register(&types.AgentTemplateDefinition{
		ID: "memory-assistant",
		// SystemPrompt 仅描述职责, 具体的长期记忆说明由 AgentMemoryMiddleware 注入
		SystemPrompt: "You are an assistant with file access and long-term memory. " +
			"Always prefer reading from and writing to memory files when users ask to remember or recall information.",
		Model: "claude-sonnet-4-5",
		// 这里只配置基础工具; memory_* 工具由中间件自动注入
		Tools: []any{"Read", "Write", "Bash"},
	})

	deps := &agent.Dependencies{
		Store:            jsonStore,
		SandboxFactory:   sandboxFactory,
		ToolRegistry:     toolRegistry,
		ProviderFactory:  providerFactory,
		TemplateRegistry: templateRegistry,
	}

	// 6. 为 Sandbox 工作目录准备本地路径
	workDir := "./workspace"
	if err := os.MkdirAll(workDir, 0o755); err != nil {
		log.Fatalf("create workspace dir failed: %v", err)
	}

	// 7. 创建 Agent 配置, 启用 filesystem + agent_memory 中间件
	config := &types.AgentConfig{
		TemplateID: "memory-assistant",
		ModelConfig: &types.ModelConfig{
			Provider: "anthropic",
			Model:    "claude-sonnet-4-5",
			APIKey:   apiKey,