Function TestPromptBuilder_ToolsManualConfig

pkg/agent/prompt_builder_test.go:156–205

(t *testing.T)

}

func TestPromptBuilder_ToolsManualConfig(t *testing.T) {
	deps := setupPromptTestDeps(t)

	// 注册模板，只包含部分工具
	deps.TemplateRegistry.Register(&types.AgentTemplateDefinition{
		ID:           "selective-tools",
		SystemPrompt: "You are a selective assistant.",
		Tools:        []any{"Read", "Write"},
		Runtime: &types.AgentTemplateRuntime{
			ToolsManual: &types.ToolsManualConfig{
				Mode:    "listed",
				Include: []string{"Read"}, // 只包含 Read 工具
			},
		},
	})

	config := &types.AgentConfig{
		TemplateID: "selective-tools",
		ModelConfig: &types.ModelConfig{
			Provider: "anthropic",
			Model:    "claude-sonnet-4-5",
			APIKey:   "test-key",
		},
		Sandbox: &types.SandboxConfig{
			Kind:    types.SandboxKindMock,
			WorkDir: "/tmp/test",
		},
	}

	ag, err := Create(context.Background(), config, deps)
	if err != nil {
		t.Fatalf("Failed to create agent: %v", err)
	}
	defer func() { _ = ag.Close() }()

	// 获取 System Prompt
	systemPrompt := ag.GetSystemPrompt()

	// 验证只包含 Read 工具
	if !strings.Contains(systemPrompt, "`Read`") {
		t.Error("System prompt should contain Read tool")
	}

	// 验证不包含 Write 工具
	if strings.Contains(systemPrompt, "`Write`") {
		t.Error("System prompt should not contain Write tool (excluded by config)")
	}

	t.Logf("Selective Tools System Prompt:\n%s", systemPrompt)
}

// setupPromptTestDeps 创建测试依赖
func setupPromptTestDeps(t *testing.T) *Dependencies {