Generative Model Integration with Code Editing

Aspects of the disclosed technology include computer-implemented systems and methods for integrating machine-learned generative models with code editing tools. A code editor is configured to execute computer-executable code within code cells of a code editor interface including a first interface portion and a second interface portion. The interface portion is configured to receive user input for defining and editing a set of code cells within the first interface portion. Each code cell of the set of code cells is independently executable by the code editor application. The second interface portion is configured to receive user input for defining and submitting user queries to a machine-learned generative model. The code editor is configured to modify at least one code cell of the set of cells based at least in part on an output of the machine-learned generative model in response to a user query.