TLDR: Create your own local AI Coding Assistant that integrates with VS Code.
AI Coding Assistants
I didn't jump onboard the AI Coding Assistant train at first.
However, I now open up a chat with ChatGPT as often as I do MDN.
I've found it's often a quicker reference for exactly what I need.
I have considered paying the monthly fee for GitHub Copilot, but what if you could use open source large language models (LLMs) to create your own AI Coding Assistant?
Now you can.
Install Ollama
Start by going to Ollama.com and downloading the version for your operating system.
Note: As I write this, the Windows version is still considered a "preview". After installing on Windows, I had to restart my computer for ollama
to be available at the command line.
Open a terminal window and type ollama --help
to confirm you have Ollama installed and your computer can find it.
Pick an Open Source LLM
Visit the EvalPlus Leaderboard where the performance of many models are compared.
There are currently a couple of options in the EvalPlus Top 5 to consider: DeepSeek-Coder-v2
and CodeQwen1.5
.
Before choosing, go back to Ollama.com and search the models to look at their details.
I personally decided to go with codeqwen. It is a 4.2GB download and deepseek-coder-v2
is 8.9GB.
You can try out several LLMs if you want to.
After choosing at least one, copy the ollama
command and run it in your terminal window to download your LLM of choice.
For example, the command on the codeqwen page is ollama run codeqwen
.
Add the Continue VS Code Extension
Open up VS Code and click the extensions menu icon.
Search for continue
.
You should find Continue
by continue.dev.
Install the extension and you should find it by icon or name in the activity bar afterwards.
Click it to open and you should see a splash screen.
The screen will confirm Ollama is installed and provide other recommendations that you can ignore.
Click the "Local" button and the "next / continue" button.
You should now have a chat screen that opened over your filetree (if your filetree is on the left side).
Configuring Continue
At the bottom of the chat window, select Ollama - codeqwen:latest
from the menu.
Click the settings icon to open up the config.json
file.
Look for the tabAutocompleteModel
setting.
Change both the title
and model
values to codeqwen
.
You will also see a custom commands
setting like this:
"customCommands": [
{
"name": "test",
"prompt": "{{{ input }}}\n\nWrite a comprehensive set of unit tests for the selected code. It should setup, run tests that check for correctness including important edge cases, and teardown. Ensure that the tests are complete and sophisticated. Give the tests just as chat output, don't edit any file.",
"description": "Write unit tests for highlighted code"
}
],
You can add your own custom commands here. In this example, if you highlight a function in your code and type test
in the chat window, it will execute the prompt you see.
Here's an example of a custom command I added:
"customCommands": [
{
"name": "test",
"prompt": "{{{ input }}}\n\nWrite a comprehensive set of unit tests for the selected code. It should setup, run tests that check for correctness including important edge cases, and teardown. Ensure that the tests are complete and sophisticated. Give the tests just as chat output, don't edit any file.",
"description": "Write unit tests for highlighted code"
},
{
"name": "step",
"prompt": "{{{ input }}}\n\nExplain the selected code step by step.",
"description": "Code explanation"
}
],
Getting Started:
Click the help icon at the bottom right of the chat window.
It provides a link to a tutorial, web resources and keyboard shortcuts.
A couple of quick things to try:
-
Select some existing code in your file, then press
Ctrl+L
to start a chat on the selected code. -
Select some existing code in your file, then type
test
in the chat window, and press enter to generate unit tests for that code. -
Create a new empty code file, press
Ctrl+i
and type instructions for code generation. Then watch your AI Code Assistant generate the code. -
Start typing code in a file and look for the autocompletion suggestions. Press tab to use them.