From 9fdcc528e82d80589636216ccdf614b5348eb427 Mon Sep 17 00:00:00 2001
From: ChenYuZe519 <chenyuewan@icloud.com>
Date: Fri, 12 Apr 2024 23:04:38 +0800
Subject: [PATCH] Added support for "gpt-4-turbo-2024-04-09" and "gpt-4-turbo"
 models. (#572)

* Update chat.ts

Added support for "gpt-4-turbo-2024-04-09" and "gpt-4-turbo" models.

* Update chat.ts

Added support for "gpt-4-turbo-2024-04-09" and "gpt-4-turbo" models.
---
 src/constants/chat.ts | 8 ++++++++
 src/types/chat.ts     | 2 ++
 2 files changed, 10 insertions(+)

diff --git a/src/constants/chat.ts b/src/constants/chat.ts
index 6ec24be82..e8c8fcab5 100644
--- a/src/constants/chat.ts
+++ b/src/constants/chat.ts
@@ -134,6 +134,14 @@ export const modelTypes: ModelTypes = {
   'gpt-4-1106-preview': 'text',
   'gpt-4-0125-preview': 'text',
   'gpt-4-vision-preview': 'image'
+  'gpt-4-turbo': {
+    prompt: { price: 0.01, unit: 1000 },
+    completion: { price: 0.03, unit: 1000 },
+  },
+  'gpt-4-turbo-2024-04-09': {
+    prompt: { price: 0.01, unit: 1000 },
+    completion: { price: 0.03, unit: 1000 },
+  },
 };
 
 export const defaultUserMaxToken = 4000;
diff --git a/src/types/chat.ts b/src/types/chat.ts
index b7da57823..3ffa6fec6 100644
--- a/src/types/chat.ts
+++ b/src/types/chat.ts
@@ -77,6 +77,8 @@ export type ModelOptions =
   | 'gpt-4-32k'
   | 'gpt-4-1106-preview'
   | 'gpt-4-0125-preview'
+  | 'gpt-4-turbo'
+  | 'gpt-4-turbo-2024-04-09'
   | 'gpt-3.5-turbo'
   | 'gpt-3.5-turbo-16k'
   | 'gpt-3.5-turbo-1106'