Implemented .env file API key storage

pull/12/head
Christopher Carroll Smith 2023-04-12 13:45:54 -04:00
rodzic 1b9649ed7a
commit e8a893156e
4 zmienionych plików z 11 dodań i 5 usunięć

1
.env.sample 100644
Wyświetl plik

@ -0,0 +1 @@
OPENAI_API_KEY=your_api_key

2
.gitignore vendored
Wyświetl plik

@ -1,2 +1,4 @@
venv venv
openai_key.txt openai_key.txt
.venv
.env

Wyświetl plik

@ -13,8 +13,11 @@ For a quick demonstration see my [demo video on twitter](https://twitter.com/bio
python3 -m venv venv python3 -m venv venv
source venv/bin/activate source venv/bin/activate
pip install -r requirements.txt pip install -r requirements.txt
cp .env.sample .env
Add your openAI api key to `openai_key.txt` - _warning!_ by default this uses GPT-4 and may make many repeated calls to the api. 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 ## Example Usage

Wyświetl plik

@ -5,13 +5,14 @@ import os
import shutil import shutil
import subprocess import subprocess
import sys import sys
import openai import openai
from termcolor import cprint from termcolor import cprint
from dotenv import load_dotenv
# Set up the OpenAI API # Set up the OpenAI API
with open("openai_key.txt") as f: load_dotenv()
openai.api_key = f.read().strip() openai.api_key = os.getenv("OPENAI_API_KEY")
def run_script(script_name, script_args): def run_script(script_name, script_args):
@ -51,7 +52,6 @@ def send_error_to_gpt(file_path, args, error_message, model):
) )
# print(prompt) # print(prompt)
response = openai.ChatCompletion.create( response = openai.ChatCompletion.create(
model=model, model=model,
messages=[ messages=[