From 370a680d94c30439168e454d94348c15c3f7fea3 Mon Sep 17 00:00:00 2001
From: heimoshuiyu <heimoshuiyu@gmail.com>
Date: Fri, 19 Jul 2024 09:34:22 +0800
Subject: [PATCH] update model list

- add gpt-4o-mini
- remove EOL models
- set default model to gpt-4o-mini
---
 src/chatgpt.ts |  4 +++-
 src/models.ts  | 38 +++++++++++++++++---------------------
 2 files changed, 20 insertions(+), 22 deletions(-)

diff --git a/src/chatgpt.ts b/src/chatgpt.ts
index a193f4d..3898ad1 100644
--- a/src/chatgpt.ts
+++ b/src/chatgpt.ts
@@ -1,3 +1,5 @@
+import { defaultModel } from "./models";
+
 export interface ImageURL {
   url: string;
   detail: "low" | "high";
@@ -155,7 +157,7 @@ class Chat {
       enable_max_gen_tokens = true,
       tokens_margin = 1024,
       apiEndPoint = "https://api.openai.com/v1/chat/completions",
-      model = "gpt-3.5-turbo",
+      model = defaultModel,
       temperature = 0.7,
       enable_temperature = true,
       top_p = 1,
diff --git a/src/models.ts b/src/models.ts
index 50727df..f852bfa 100644
--- a/src/models.ts
+++ b/src/models.ts
@@ -7,6 +7,22 @@ interface Model {
 }
 
 const models: Record<string, Model> = {
+  "gpt-4o": {
+    maxToken: 128000,
+    price: { prompt: 0.005 / 1000, completion: 0.015 / 1000 },
+  },
+  "gpt-4o-2024-05-13": {
+    maxToken: 128000,
+    price: { prompt: 0.005 / 1000, completion: 0.015 / 1000 },
+  },
+  "gpt-4o-mini": {
+    maxToken: 128000,
+    price: { prompt: 0.15 / 1000 / 1000, completion: 0.6 / 1000 / 1000 },
+  },
+  "gpt-4o-mini-2024-07-18": {
+    maxToken: 128000,
+    price: { prompt: 0.15 / 1000 / 1000, completion: 0.6 / 1000 / 1000 },
+  },
   "gpt-3.5-turbo-0125": {
     maxToken: 16385,
     price: { prompt: 0.0005 / 1000, completion: 0.0015 / 1000 },
@@ -31,18 +47,6 @@ const models: Record<string, Model> = {
     maxToken: 16385,
     price: { prompt: 0.003 / 1000, completion: 0.004 / 1000 },
   },
-  "gpt-3.5-turbo-0301": {
-    maxToken: 4096,
-    price: { prompt: 0.0015 / 1000, completion: 0.002 / 1000 },
-  },
-  "gpt-4o": {
-    maxToken: 128000,
-    price: { prompt: 0.005 / 1000, completion: 0.015 / 1000 },
-  },
-  "gpt-4o-2024-05-13": {
-    maxToken: 128000,
-    price: { prompt: 0.005 / 1000, completion: 0.015 / 1000 },
-  },
   "gpt-4-turbo": {
     maxToken: 128000,
     price: { prompt: 0.01 / 1000, completion: 0.03 / 1000 },
@@ -75,20 +79,12 @@ const models: Record<string, Model> = {
     maxToken: 8192,
     price: { prompt: 0.03 / 1000, completion: 0.06 / 1000 },
   },
-  "gpt-4-0613": {
-    maxToken: 8192,
-    price: { prompt: 0.03 / 1000, completion: 0.06 / 1000 },
-  },
   "gpt-4-32k": {
     maxToken: 8192,
     price: { prompt: 0.06 / 1000, completion: 0.12 / 1000 },
   },
-  "gpt-4-32k-0613": {
-    maxToken: 8192,
-    price: { prompt: 0.06 / 1000, completion: 0.12 / 1000 },
-  },
 };
 
-export const defaultModel = "gpt-3.5-turbo-0125";
+export const defaultModel = "gpt-4o-mini";
 
 export default models;