from pathlib import Path
import sys
sys.path.append(str(Path(__file__).parent.parent))
from tools import ToolArchive

def test_sampling():
    archive_path = Path("tool_gen/generated/claude_seed_oss_36b/subagent_tool_archive")
    archive = ToolArchive(output_dir=archive_path)
    for i in range(10):
        sample = archive.sample(3)
        print(sample)
        print("-" * 100)

if __name__ == "__main__":
    test_sampling()