From 6bc85a41e2782b45a18d74dfcabd13179d08ca07 Mon Sep 17 00:00:00 2001
From: ksfi <kerian.yousfi@hotmail.fr>
Date: Fri, 14 Apr 2023 00:07:57 +0200
Subject: [PATCH 1/2] switch to gpt3 if gpt4 not available

---
 wolverine.py | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/wolverine.py b/wolverine.py
index 42f7ff8..435c4bd 100644
--- a/wolverine.py
+++ b/wolverine.py
@@ -13,7 +13,6 @@ from termcolor import cprint
 with open("openai_key.txt") as f:
     openai.api_key = f.read().strip()
 
-
 def run_script(script_name, script_args):
     script_args = [str(arg) for arg in script_args]
     try:
@@ -113,8 +112,18 @@ def apply_changes(file_path, changes_json):
         else:
             print(line, end="")
 
+def avaibility(openaiObj):
+    models_available = [x['id'] for x in openai.Model.list()['data']]
+    if "gpt-4" not in models_available:
+        model = "gpt-3.5-turbo"
+    else:
+        model = "gpt-4"
+    return model
 
-def main(script_name, *script_args, revert=False, model="gpt-4"):
+
+
+def main(script_name, *script_args, revert=False):
+    model = avaibility(openai)
     if revert:
         backup_file = script_name + ".bak"
         if os.path.exists(backup_file):
@@ -135,10 +144,10 @@ def main(script_name, *script_args, revert=False, model="gpt-4"):
             cprint("Script ran successfully.", "blue")
             print("Output:", output)
             break
+
         else:
             cprint("Script crashed. Trying to fix...", "blue")
             print("Output:", output)
-
             json_response = send_error_to_gpt(
                     file_path=script_name,
                     args=script_args,
@@ -151,3 +160,4 @@ def main(script_name, *script_args, revert=False, model="gpt-4"):
 
 if __name__ == "__main__":
     fire.Fire(main)
+

From 23e53dbb878e82a8ad86cd94163f3076b51a503a Mon Sep 17 00:00:00 2001
From: ksfi <kerian.yousfi@hotmail.fr>
Date: Fri, 14 Apr 2023 01:55:29 +0200
Subject: [PATCH 2/2] let model in default

---
 wolverine.py | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/wolverine.py b/wolverine.py
index 435c4bd..e63d06c 100644
--- a/wolverine.py
+++ b/wolverine.py
@@ -113,17 +113,13 @@ def apply_changes(file_path, changes_json):
             print(line, end="")
 
 def avaibility(openaiObj):
-    models_available = [x['id'] for x in openai.Model.list()['data']]
-    if "gpt-4" not in models_available:
+    if "gpt-4" not in [x['id'] for x in openai.Model.list()['data']]:
         model = "gpt-3.5-turbo"
     else:
         model = "gpt-4"
     return model
 
-
-
-def main(script_name, *script_args, revert=False):
-    model = avaibility(openai)
+def main(script_name, *script_args, revert=False, model=avaibility(openai)):
     if revert:
         backup_file = script_name + ".bak"
         if os.path.exists(backup_file):