Function main

examples/agent-working-memory/main.go:29–142

使用场景: - 记住用户偏好和设置 - 跟踪任务进度 - 维护会话状态

()

// - 维护会话状态

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-working-memory")
	if err != nil {
		log.Fatalf("create store failed: %v", err)
	}

	// 5. 模板注册表
	templateRegistry := agent.NewTemplateRegistry()
	templateRegistry.Register(&types.AgentTemplateDefinition{
		ID: "task-assistant",
		SystemPrompt: `You are a helpful task assistant with Working Memory.

Your capabilities:
- Remember user information and preferences
- Track task progress across multiple conversations
- Maintain context between sessions

When you learn something important about the user or their tasks, use the update_working_memory tool to store it.

Remember: Working Memory is automatically loaded at the start of each conversation, so you always have access to it.`,
		Model: "claude-sonnet-4-5",
		// 基础工具，working memory 工具由中间件自动注入
		Tools: []any{"Read", "Write", "Bash"},
	})

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

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