Prompt engineering is kind of like alchemy. There's no clear way to predict what will work best. It's all about experimenting until you find the right prompt. gpt-prompt-engineer
is a tool that takes this experimentation to a whole new level.
Simply input a description of your task and some test cases, and the system will generate, test, and rank a multitude of prompts to find the ones that perform the best.
-
Prompt Generation: Using GPT-4 and GPT-3.5-Turbo,
gpt-prompt-engineer
can generate a variety of possible prompts based on a provided use-case and test cases. -
Prompt Testing: The real magic happens after the generation. The system tests each prompt against all the test cases, comparing their performance and ranking them using an ELO rating system.
![Screen Shot 2023-07-04 at 11 41 54 AM](https://private-user-images.githubusercontent.com/41550495/250947004-f8171cff-1703-40ca-b9fd-f0aa24d07110.png?jwt=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJnaXRodWIuY29tIiwiYXVkIjoicmF3LmdpdGh1YnVzZXJjb250ZW50LmNvbSIsImtleSI6ImtleTUiLCJleHAiOjE3MjQyNjMwNjQsIm5iZiI6MTcyNDI2Mjc2NCwicGF0aCI6Ii80MTU1MDQ5NS8yNTA5NDcwMDQtZjgxNzFjZmYtMTcwMy00MGNhLWI5ZmQtZjBhYTI0ZDA3MTEwLnBuZz9YLUFtei1BbGdvcml0aG09QVdTNC1ITUFDLVNIQTI1NiZYLUFtei1DcmVkZW50aWFsPUFLSUFWQ09EWUxTQTUzUFFLNFpBJTJGMjAyNDA4MjElMkZ1cy1lYXN0LTElMkZzMyUyRmF3czRfcmVxdWVzdCZYLUFtei1EYXRlPTIwMjQwODIxVDE3NTI0NFomWC1BbXotRXhwaXJlcz0zMDAmWC1BbXotU2lnbmF0dXJlPWMxYTBhZDkxZTM5NDBjYTM1MzE4MTY1MjcxYzExZjIyMTNlZGM2YmY4MDMxMTU0NjBjNDU2ZWQ4NGJmNTViOTYmWC1BbXotU2lnbmVkSGVhZGVycz1ob3N0JmFjdG9yX2lkPTAma2V5X2lkPTAmcmVwb19pZD0wIn0.R9XqG1GIpBQLxzu-K3T_y78eE5U4RSwabclKMQnZ7FY)
-
ELO Rating System: Each prompt starts with an ELO rating of 1200. As they compete against each other in generating responses to the test cases, their ELO ratings change based on their performance. This way, you can easily see which prompts are the most effective.
-
Weights & Biases Logging: Optional logging to Weights & Biases of your configs such as temperature and max tokens, the system and user prompts for each part, the test cases used and the final ranked ELO rating for each candidate prompt. Set
use_wandb
toTrue
to use.
-
Open the notebook in Google Colab or in a local Jupyter notebook.
-
Add your OpenAI API key to the line
openai.api_key = "ADD YOUR KEY HERE"
. -
If you have GPT-4 access, you're ready to move on. If not, change
CANDIDATE_MODEL='gpt-4'
toCANDIDATE_MODEL='gpt-3.5-turbo'
.
- Define your use-case and test cases. The use-case is a description of what you want the AI to do. Test cases are specific prompts that you would like the AI to respond to. For example:
description = "Given a prompt, generate a landing page headline." # this style of description tends to work well
test_cases = [
{
'prompt': 'Promoting an innovative new fitness app, Smartly',
},
{
'prompt': 'Why a vegan diet is beneficial for your health',
},
{
'prompt': 'Introducing a new online course on digital marketing',
},
{
'prompt': 'Launching a new line of eco-friendly clothing',
},
{
'prompt': 'Promoting a new travel blog focusing on budget travel',
},
{
'prompt': 'Advertising a new software for efficient project management',
},
{
'prompt': 'Introducing a new book on mastering Python programming',
},
{
'prompt': 'Promoting a new online platform for learning languages',
},
{
'prompt': 'Advertising a new service for personalized meal plans',
},
{
'prompt': 'Launching a new app for mental health and mindfulness',
}
]
-
Choose how many prompts to generate. Keep in mind, this can get expensive if you generate many prompts. 10 is a good starting point.
-
Call
generate_optimal_prompt(description, test_cases, number_of_prompts)
to generate a list of potential prompts, and test and rate their performance. -
The final ELO ratings will be printed in a table, sorted in descending order. The higher the rating, the better the prompt.
![Screen Shot 2023-07-04 at 11 48 45 AM](https://private-user-images.githubusercontent.com/41550495/250948262-324f90b8-c0ee-45fd-b219-6c44d9aa281b.png?jwt=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJnaXRodWIuY29tIiwiYXVkIjoicmF3LmdpdGh1YnVzZXJjb250ZW50LmNvbSIsImtleSI6ImtleTUiLCJleHAiOjE3MjQyNjMwNjQsIm5iZiI6MTcyNDI2Mjc2NCwicGF0aCI6Ii80MTU1MDQ5NS8yNTA5NDgyNjItMzI0ZjkwYjgtYzBlZS00NWZkLWIyMTktNmM0NGQ5YWEyODFiLnBuZz9YLUFtei1BbGdvcml0aG09QVdTNC1ITUFDLVNIQTI1NiZYLUFtei1DcmVkZW50aWFsPUFLSUFWQ09EWUxTQTUzUFFLNFpBJTJGMjAyNDA4MjElMkZ1cy1lYXN0LTElMkZzMyUyRmF3czRfcmVxdWVzdCZYLUFtei1EYXRlPTIwMjQwODIxVDE3NTI0NFomWC1BbXotRXhwaXJlcz0zMDAmWC1BbXotU2lnbmF0dXJlPTA1NjBkYWRjZDcyZmY5YTI4YTMxMzJkMmFmZTUxNTlmNjNiNzBhZmVkNTBiYWE0NjI0NDQwYmM0YTdiNDIwYzImWC1BbXotU2lnbmVkSGVhZGVycz1ob3N0JmFjdG9yX2lkPTAma2V5X2lkPTAmcmVwb19pZD0wIn0.Laq7BqpldW6wDFhETcV2Qekn1o8tQoUNeRsZOqtQ6E8)
- have a number of different system prompt generators that create different styles of prompts, to cover more ground (ex. examples, verbose, short, markdown, etc.)
- automatically generate the test cases
This project is MIT licensed.
Matt Shumer - @mattshumer_
Project Link: https://github.com/mshumer/gpt-prompt-engineer