From 0bf77a640c0049c50cb187c87db70c8ebc795755 Mon Sep 17 00:00:00 2001
From: "Alisson R. Perez" <756802+alissonperez@users.noreply.github.com>
Date: Tue, 11 Jun 2024 14:12:44 -0300
Subject: [PATCH] Update default model to gpt-4o

---
 README.md             | 4 ++--
 sgpt/config.py        | 2 +-
 tests/_integration.py | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index 5564997..aef941d 100644
--- a/README.md
+++ b/README.md
@@ -395,7 +395,7 @@ CACHE_PATH=/tmp/shell_gpt/cache
 # Request timeout in seconds.
 REQUEST_TIMEOUT=60
 # Default OpenAI model to use.
-DEFAULT_MODEL=gpt-3.5-turbo
+DEFAULT_MODEL=gpt-4o
 # Default color for shell and code completions.
 DEFAULT_COLOR=magenta
 # When in --shell mode, default to "Y" for no input.
@@ -422,7 +422,7 @@ Possible options for `CODE_THEME`: https://pygments.org/styles/
 │   prompt      [PROMPT]  The prompt to generate completions for.                                          │
 ╰──────────────────────────────────────────────────────────────────────────────────────────────────────────╯
 ╭─ Options ────────────────────────────────────────────────────────────────────────────────────────────────╮
-│ --model            TEXT                       Large language model to use. [default: gpt-4-1106-preview] │
+│ --model            TEXT                       Large language model to use. [default: gpt-4o]             │
 │ --temperature      FLOAT RANGE [0.0<=x<=2.0]  Randomness of generated output. [default: 0.0]             │
 │ --top-p            FLOAT RANGE [0.0<=x<=1.0]  Limits highest probable tokens (words). [default: 1.0]     │
 │ --md             --no-md                      Prettify markdown output. [default: md]                    │
diff --git a/sgpt/config.py b/sgpt/config.py
index 981cba7..5655b03 100644
--- a/sgpt/config.py
+++ b/sgpt/config.py
@@ -22,7 +22,7 @@
     "CHAT_CACHE_LENGTH": int(os.getenv("CHAT_CACHE_LENGTH", "100")),
     "CACHE_LENGTH": int(os.getenv("CHAT_CACHE_LENGTH", "100")),
     "REQUEST_TIMEOUT": int(os.getenv("REQUEST_TIMEOUT", "60")),
-    "DEFAULT_MODEL": os.getenv("DEFAULT_MODEL", "gpt-4-1106-preview"),
+    "DEFAULT_MODEL": os.getenv("DEFAULT_MODEL", "gpt-4o"),
     "DEFAULT_COLOR": os.getenv("DEFAULT_COLOR", "magenta"),
     "ROLE_STORAGE_PATH": os.getenv("ROLE_STORAGE_PATH", str(ROLE_STORAGE_PATH)),
     "DEFAULT_EXECUTE_SHELL_CMD": os.getenv("DEFAULT_EXECUTE_SHELL_CMD", "false"),
diff --git a/tests/_integration.py b/tests/_integration.py
index 513b913..49d4a11 100644
--- a/tests/_integration.py
+++ b/tests/_integration.py
@@ -35,7 +35,7 @@ def setUpClass(cls):
         # Response streaming should be enabled for these tests.
         assert cfg.get("DISABLE_STREAMING") == "false"
         # ShellGPT optimised and tested with gpt-4 turbo.
-        assert cfg.get("DEFAULT_MODEL") == "gpt-4-1106-preview"
+        assert cfg.get("DEFAULT_MODEL") == "gpt-4o"
         # Make sure we will not call any functions.
         assert cfg.get("OPENAI_USE_FUNCTIONS") == "false"