Yes we use Fill in the middle models (Mistral and our own trained Qwen). And we feed it with your data context - we have our own SQL parser to feed you with the right data schema context depending on where your cursor is in the query.
I hadn't realized you trained your own model. That's an important differentiator. How do you get training data of schemas in the wild?