make sure you use ai studio (not the vertex one), not the consumer gemini interface. Seems to work better for code there.