A Jupyter/IPython magic command for generating code using GPT
Generating code with GPT by loading the extension and prefixing your line with %gpt
.
Install with pip
pip install gpt-magic
Please note that users will need to provide their own OpenAI API key. Set an environment variable called OPEN_AI_KEY
with your key. Or pass the API key when loading the extension, by running the following line before %load_ext gpt_magic:
import os
os.environ["OPENAI_API_KEY"] = "your_openai_api_key_here"
%load_ext gpt
%gpt "Your request here"
The current default is the gpt-3-turbo model, but can be changed using OPENAI_MODEL
environmental variable.
Using GPT-4 example
import os
os.environ["OPENAI_MODEL"] = "gpt-4"
You can enable memory of all previous requests and responses with the -c
argument. By default it's memory-less.
%gpt -c "Write a function which calculates the average of a numpy array"
You can set the temperature with -t
argument. The default is 0.
%gpt -t 0.5 "Write a function which calculates the average of a numpy array"