Add 'Generative AI' submenu

[dlqqq]

Description

• This feature is experimental and may be removed in a future release.
• Adds a "Generative AI" submenu that provides 4 menu items:
  • Explain code
  • Fix code
  • Optimize code
  • Refactor code
• "Fix code" calls /fix, and only works when a code cell with an error output is selected as the active cell.
• "{Explain,Optimize,Refactor} code" simply includes the user's selection with a basic prompt {Explain,Optimize,Refactor} the code below.
  • If a text selection exists, that selection is used to provide the code sent to the language model. Otherwise, if an active cell is selected, then that cell's source is used to provide the code. If neither exist, the menu item is disabled.
  • These commands also work outside of notebooks, given any text selection within an editor.
• When these menu items are invoked, the chat panel is opened automatically if it is not already open.

Demo

Screen.Recording.2024-08-29.at.8.06.48.AM.mov

Future work

• In the future, the submenu should not be titled "Generative AI", but instead be titled according to the persona of the currently selected language model.
  • So by default, it should read "Jupyternaut". However, if a custom persona "Bagel AI" is used, then the submenu should be titled "Bagel AI".
• We are also currently exploring an alternative UX for improving code within editors. Work is still in progress. This should however be a good stopgap solution for allowing users to perform basic AI tasks through the UI.

[srdas]

Tested the code and it looks great! Awesome PR.

One small note: For the ModuleNotFoundError the fix action is not available. Not sure we want to be that inclusive as it is not a typical error. See:

It could add a line of code to install the missing module if fix was possible. See that it works with `/fix' in chat as intended:

Great idea to add it to the right-click menu because the Gen AI submenu can be used to add more functionality as needed in addition to the four actions currently included.

[dlqqq]

@srdas One weird caveat is that to fix/explain/optimize/refactor a cell, you have to left-click on the cell first to make it the active cell, and then right-click to select the AI task you want to perform.

Right-clicking on a cell does not make it the active cell in JupyterLab. This is existing behavior that is also shown by the other context menu options. See the demo video below, where I right click on a cell that is not the active cell, then click delete. Notice that the cell I right-clicked on is not deleted.

Screen.Recording.2024-08-29.at.1.06.41.PM.mov

[srdas]

Right-clicking on a cell does not make it the active cell in JupyterLab. This is existing behavior that is also shown by the other context menu options. See the demo video below, where I right click on a cell that is not the active cell, then click delete. Notice that the cell I right-clicked on is not deleted.

@dlqqq Thanks, I retested it and it is working correctly after the left-click + right-click. Perfect!

[srdas]

All looks good