Give your python scripts regenerative healing abilities!
Go to file
chlorine 6552cc4dd3
Merge 288af47ca3 into 9418f0fefb
2024-03-08 07:23:41 -07:00
.github/workflows clean requirements 2023-04-26 17:42:47 -07:00
examples
tests
wolverine Go support functional MVP 2023-05-04 14:34:05 +03:00
.env.sample
.gitignore Go support functional MVP 2023-05-04 14:34:05 +03:00
LICENSE
README.md Merge 288af47ca3 into 9418f0fefb 2024-03-08 07:23:41 -07:00
prompt.txt
requirements.txt clean requirements 2023-04-26 17:42:47 -07:00

README.md

DEPRECATED: Try Mentat instead! https://github.com/AbanteAI/mentat

Wolverine

About

Give your python scripts regenerative healing abilities!

Run your scripts with Wolverine and when they crash, GPT-4 edits them and explains what went wrong. Even if you have many bugs it will repeatedly rerun until it's fixed.

For a quick demonstration see my demo video on twitter.

Setup

python3 -m venv venv
source venv/bin/activate
pip install -r requirements.txt
cp .env.sample .env

Add your openAI api key to .env

warning! By default wolverine uses GPT-4 and may make many repeated calls to the api.

Example Usage

To run with gpt-4 (the default, tested option):

python -m wolverine examples/buggy_script.py "subtract" 20 3

You can also run with other models, but be warned they may not adhere to the edit format as well:

python -m wolverine --model=gpt-3.5-turbo examples/buggy_script.py "subtract" 20 3

If you want to use GPT-3.5 by default instead of GPT-4 uncomment the default model line in .env:

DEFAULT_MODEL=gpt-3.5-turbo

You can also use flag --confirm=True which will ask you yes or no before making changes to the file. If flag is not used then it will apply the changes to the file

python -m wolverine examples/buggy_script.py "subtract" 20 3 --confirm=True

Environment variables

env name description default value
OPENAI_API_KEY OpenAI API key None
DEFAULT_MODEL GPT model to use "gpt-4"
VALIDATE_JSON_RETRY Number of retries when requesting OpenAI API (-1 means unlimites) -1

Future Plans

This is just a quick prototype I threw together in a few hours. There are many possible extensions and contributions are welcome:

  • add flags to customize usage, such as asking for user confirmation before running changed code
  • further iterations on the edit format that GPT responds in. Currently it struggles a bit with indentation, but I'm sure that can be improved
  • a suite of example buggy files that we can test prompts on to ensure reliability and measure improvement
  • multiple files / codebases: send GPT everything that appears in the stacktrace
  • graceful handling of large files - should we just send GPT relevant classes / functions?
  • extension to languages other than python

Star History

Star History Chart

Go Usage:

The Wolverine supports .go files to be executed now!

Setup step by step:

1) git clone  https://....<the_link>
2) cd wolverine/wolverine/GoLang/internal/main
3) go mod install
4) go build -o ../../../../../your_wish_folder/wolverine.exe
5) cd ../../../../../your_wish_folder
6) > .env
7) fill the .env file like in the example below, but with your personal data

Example .env:

OPENAI_API_KEY=gorinfwe:mfwbevnmowemo9fn20f439vn03v4
GPT_MODEL=text-davinci-003
ATTEMPTS_TO_TRY=15

Remember that each attempt is a request to GPT so think twice about <ATTEMPTS_TO_TRY> value

GPT 4 is the most preferable model here, but you can try to use any model

Example Usage

./wolverine.exe main

main here is your filename WITHOUT .go extension

If you see the "Success" message, then you must have obtained a file enterFilename+"__fixed".go>, and it's free of any compile errors. So you can freely run it

go run main__fixed.go