Added .env.local to git ignore. Updated the model to be an environment variable + updated readme/example.

.env.example

@@ -1 +1,2 @@
-OPENAI_API_KEY=
+OPENAI_API_KEY=
+MODEL="gpt-4"

.gitignore

@@ -1,5 +1,6 @@
 # .env
 .env
+.env.local
 
 # __pycache__
 __pycache__

README.md

@@ -8,7 +8,7 @@ The notebook also breaks down how the script works.
 # Usage
 
 Run with `$ python3 TermGPT.py`
-You will need to create a `.env` file similar to https://github.com/Sentdex/TermGPT/blob/main/.env.example, or set `OPENAI_API_KEY` manually.
+You will need to create a `.env` file similar to https://github.com/Sentdex/TermGPT/blob/main/.env.example, or set `OPENAI_API_KEY` and `MODEL` manually.
 
 From here, you make your programming/development request. The script will run and query GPT-4 for a series of terminal commands to run to achieve this objective. This is including, but not limited to: reading files, writing code, reading websites, running code, running terminal commands...etc. 
 

TermGPT.py

@@ -20,7 +20,7 @@
 READ_RE_PATTERN = r"--r \[(.*?)\]"
 WEB_RE_PATTERN = r"--w \[(.*?)\]"
 
-def gpt_query(model="gpt-4", max_retries=15, sleep_time=2):
+def gpt_query(model=os.environ.get("MODEL"), max_retries=15, sleep_time=2):
     global message_history
     retries = 0
     logger = logging.getLogger()