From 40476dca3b65946ee54a6e9aee82f799a5b8b917 Mon Sep 17 00:00:00 2001
From: Cohee <18619528+Cohee1207@users.noreply.github.com>
Date: Thu, 25 Jan 2024 22:01:02 +0200
Subject: [PATCH] New OAI models

---
 public/index.html        | 21 ++++++++++++---------
 public/scripts/openai.js |  3 +++
 2 files changed, 15 insertions(+), 9 deletions(-)

diff --git a/public/index.html b/public/index.html
index e6b358502..f23ba3c7e 100644
--- a/public/index.html
+++ b/public/index.html
@@ -2109,11 +2109,12 @@
                                 <select id="model_openai_select">
                                     <optgroup label="GPT-3.5 Turbo">
                                         <option value="gpt-3.5-turbo">gpt-3.5-turbo</option>
+                                        <option value="gpt-3.5-turbo-0125">gpt-3.5-turbo-0125 (2024)</option>
+                                        <option value="gpt-3.5-turbo-1106">gpt-3.5-turbo-1106 (2023)</option>
+                                        <option value="gpt-3.5-turbo-0613">gpt-3.5-turbo-0613 (2023)</option>
+                                        <option value="gpt-3.5-turbo-0301">gpt-3.5-turbo-0301 (2023)</option>
                                         <option value="gpt-3.5-turbo-16k">gpt-3.5-turbo-16k</option>
-                                        <option value="gpt-3.5-turbo-16k-0613">gpt-3.5-turbo-16k-0613</option>
-                                        <option value="gpt-3.5-turbo-0613">gpt-3.5-turbo-0613</option>
-                                        <option value="gpt-3.5-turbo-0301">gpt-3.5-turbo-0301</option>
-                                        <option value="gpt-3.5-turbo-1106">gpt-3.5-turbo-1106</option>
+                                        <option value="gpt-3.5-turbo-16k-0613">gpt-3.5-turbo-16k-0613 (2023)</option>
                                     </optgroup>
                                     <optgroup label="GPT-3.5 Turbo Instruct">
                                         <option value="gpt-3.5-turbo-instruct">gpt-3.5-turbo-instruct</option>
@@ -2121,13 +2122,15 @@
                                     </optgroup>
                                     <optgroup label="GPT-4">
                                         <option value="gpt-4">gpt-4</option>
+                                        <option value="gpt-4-turbo-preview">gpt-4-turbo-preview</option>
                                         <option value="gpt-4-vision-preview">gpt-4-vision-preview</option>
-                                        <option value="gpt-4-1106-preview">gpt-4-1106-preview</option>
-                                        <option value="gpt-4-0613">gpt-4-0613</option>
-                                        <option value="gpt-4-0314">gpt-4-0314</option>
+                                        <option value="gpt-4-0125-preview">gpt-4-0125-preview (2024)</option>
+                                        <option value="gpt-4-1106-preview">gpt-4-1106-preview (2023)</option>
+                                        <option value="gpt-4-0613">gpt-4-0613 (2023)</option>
+                                        <option value="gpt-4-0314">gpt-4-0314 (2023)</option>
                                         <option value="gpt-4-32k">gpt-4-32k</option>
-                                        <option value="gpt-4-32k-0314">gpt-4-32k-0314</option>
-                                        <option value="gpt-4-32k-0613">gpt-4-32k-0613</option>
+                                        <option value="gpt-4-32k-0613">gpt-4-32k-0613 (2023)</option>
+                                        <option value="gpt-4-32k-0314">gpt-4-32k-0314 (2023)</option>
                                     </optgroup>
                                     <optgroup label="Other">
                                         <option value="text-davinci-003">text-davinci-003</option>
diff --git a/public/scripts/openai.js b/public/scripts/openai.js
index a5ddd8f9f..f245d5106 100644
--- a/public/scripts/openai.js
+++ b/public/scripts/openai.js
@@ -3122,6 +3122,9 @@ function getMaxContextOpenAI(value) {
     if (oai_settings.max_context_unlocked) {
         return unlocked_max;
     }
+    else if (value.includes('gpt-4-turbo')) {
+        return max_128k;
+    }
     else if (value.includes('gpt-4-1106')) {
         return max_128k;
     }