We did this by using the ASTs you can go quite far without embeddings and the result is easier to debug and follow what's going on.