From e1d0a790f8941a74c4857ac8404c4c1c9e4fb6ed Mon Sep 17 00:00:00 2001 From: Felix Boehme Date: Fri, 14 Apr 2023 17:46:18 -0400 Subject: [PATCH] cleanup --- .env.sample | 2 +- README.md | 6 +++--- wolverine.py | 9 +++------ 3 files changed, 7 insertions(+), 10 deletions(-) diff --git a/.env.sample b/.env.sample index f2177b3..2184c45 100644 --- a/.env.sample +++ b/.env.sample @@ -1,2 +1,2 @@ OPENAI_API_KEY=your_api_key -DEFAULT_MODEL=gpt-3.5-turbo +#DEFAULT_MODEL=gpt-3.5-turbo diff --git a/README.md b/README.md index 401b812..f671b4e 100644 --- a/README.md +++ b/README.md @@ -15,9 +15,9 @@ For a quick demonstration see my [demo video on twitter](https://twitter.com/bio pip install -r requirements.txt cp .env.sample .env -Add your openAI api key to `.env` - _warning!_ by default this uses GPT-4 and may make many repeated calls to the api. +Add your openAI api key to `.env` - OPENAI_API_KEY=sk-abcdefghijklmopqrstuvwxyz +_warning!_ by default this uses GPT-4 and may make many repeated calls to the api. ## Example Usage @@ -29,7 +29,7 @@ You can also run with other models, but be warned they may not adhere to the edi python wolverine.py --model=gpt-3.5-turbo buggy_script.py "subtract" 20 3 -If you want to use GPT-3.5 by default instead of GPT-4 add this to `.env`: +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 diff --git a/wolverine.py b/wolverine.py index 0501f1b..535db10 100644 --- a/wolverine.py +++ b/wolverine.py @@ -1,21 +1,18 @@ import difflib +import fire import json import os import shutil import subprocess import sys +import openai -import fire -from dotenv import load_dotenv from termcolor import cprint from dotenv import load_dotenv load_dotenv() - -import openai - -openai.api_key = os.environ.get("OPENAI_API_KEY") +openai.api_key = os.getenv("OPENAI_API_KEY") DEFAULT_MODEL = os.environ.get("DEFAULT_MODEL", "gpt-4")