AI Code Assistant¶

The AI Code Assistant is a powerful tool enhancing the coding experience in Jupyter Notebooks and Visual Studio Code (in Code Studio).

Enabling AI Code Assistant¶

In order to start using the AI Code Assistant, the feature needs to be enabled by an admin. It also requires a connection to an LLM.

Go to Administration > Settings > OTHER > AI Services
Make sure Enable AI Code Assistant is checked
Select which LLM connection to use as default
Click Save

To enable the feature in Dataiku Cloud you need to be a member of the space_administrators group:

Go to the Launchpad > Extensions > AI Services
Accept the Dataiku AI Services Terms of Use
Make sure Enable AI Code Assistant is checked
Select which LLM connection to use as default
Click Add

Using AI Code Assistant in Jupyter notebooks¶

There is only one step needed to enable AI Code Assistant in Jupyter notebooks:

Copy the following code in a cell and execute it:
```
%load_ext ai_code_assistant
```

Using AI Code Assistant in Visual Studio Code¶

Warning

Using a self-signed certificate may prevent the feature from being fully functional on all browsers.

To use AI Code Assistant in Visual Studio Code, an admin must enable it in the Code Studio Template:

Ensure the Install AI Code Assistant option is enabled in the Visual Code Studio block (it is enabled by default)
Rebuild the template and restart the Code Studio to apply changes

You will find a new side panel in VSCode.

You will also have some new actions in the right-click menu when launched from inside a file (some are only available once you have made a selection in the file).