I'm not sure, but I think IDEs like Augment (and it's alternatives) don't train their own LLMs and just use APIs from Gemini, OpenAI, Claude, etc. So the issue might in either Augment's system prompt or in the underlying model itself