From ddb41739ff62625576e56858134f9067ca2f29e8 Mon Sep 17 00:00:00 2001
From: amrrs <1littlecoder@gmail.com>
Date: Mon, 5 Jun 2023 14:58:03 +0530
Subject: [PATCH 1/2] Update .env.example

---
 .env.example | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/.env.example b/.env.example
index 9847a1d..149d779 100644
--- a/.env.example
+++ b/.env.example
@@ -1 +1,2 @@
-OPENAI_API_KEY=
\ No newline at end of file
+OPENAI_API_KEY=
+MODEL_NAME=gpt-4

From c96658c1b06369563c872a4326b49141e60bdf2c Mon Sep 17 00:00:00 2001
From: amrrs <1littlecoder@gmail.com>
Date: Mon, 5 Jun 2023 15:00:03 +0530
Subject: [PATCH 2/2] Update TermGPT.py

---
 TermGPT.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/TermGPT.py b/TermGPT.py
index 4a432cd..16c5660 100644
--- a/TermGPT.py
+++ b/TermGPT.py
@@ -13,6 +13,7 @@
 load_dotenv()
 
 openai.api_key = os.environ.get("OPENAI_API_KEY")
+model_name = os.environ.get("MODEL_NAME")
 
 DEBUG = False
 logging.basicConfig(level=logging.WARNING)
@@ -20,7 +21,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=model_name, max_retries=15, sleep_time=2):
     global message_history
     retries = 0
     logger = logging.getLogger()
@@ -122,7 +123,7 @@ def extract_paragraphs(url: str):
 
     while not GPT_DONE:
         print(colorama.Fore.GREEN + colorama.Style.DIM + "Querying GPT for next command (these are not running yet)..." + colorama.Style.RESET_ALL)
-        reply_content = gpt_query(model="gpt-4")
+        reply_content = gpt_query(model=model_name)
 
         message_history.append({"role": "assistant", "content": reply_content})
         message_history.append({"role": "user", "content": "NEXT"})