Add argument to only get the 'yaml tree sitter tree

[zkry]

Fixes #46

The issue was caused by the treesit commands using non-yaml trees. This MR ensures that I'm only using the 'yaml tree parse.