Add gpt-3.5-turbo-16k model option, update gpt-3.5 prompt cost (#351)

* add gpt-3.5-turbo-16k model option. update gpt-3.5 prompt cost * add gpt-4-0613 models
ztjhz · Jun 14, 2023 · f28ff87 · f28ff87
1 parent 6a022d9
commit f28ff87
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 3 deletions.
diff --git a/src/constants/chat.ts b/src/constants/chat.ts
@@ -19,6 +19,7 @@ Respond using Markdown.`;
 
 export const modelOptions: ModelOptions[] = [
   'gpt-3.5-turbo',
+  'gpt-3.5-turbo-16k',
   'gpt-4',
   'gpt-4-32k',
   // 'gpt-3.5-turbo-0301',
@@ -31,21 +32,38 @@ export const defaultModel = 'gpt-3.5-turbo';
 export const modelMaxToken = {
   'gpt-3.5-turbo': 4096,
   'gpt-3.5-turbo-0301': 4096,
+  'gpt-3.5-turbo-0613': 4096,
+  'gpt-3.5-turbo-16k': 16384,
+  'gpt-3.5-turbo-16k-0613': 16384,
   'gpt-4': 8192,
   'gpt-4-0314': 8192,
+  'gpt-4-0613': 8192,
   'gpt-4-32k': 32768,
   'gpt-4-32k-0314': 32768,
+  'gpt-4-32k-0613': 32768,
 };
 
 export const modelCost = {
   'gpt-3.5-turbo': {
-    prompt: { price: 0.002, unit: 1000 },
+    prompt: { price: 0.0015, unit: 1000 },
     completion: { price: 0.002, unit: 1000 },
   },
   'gpt-3.5-turbo-0301': {
-    prompt: { price: 0.002, unit: 1000 },
+    prompt: { price: 0.0015, unit: 1000 },
     completion: { price: 0.002, unit: 1000 },
   },
+  'gpt-3.5-turbo-0613': {
+    prompt: { price: 0.0015, unit: 1000 },
+    completion: { price: 0.002, unit: 1000 },
+  },
+  'gpt-3.5-turbo-16k': {
+    prompt: { price: 0.003, unit: 1000 },
+    completion: { price: 0.004, unit: 1000 },
+  },
+  'gpt-3.5-turbo-16k-0613': {
+    prompt: { price: 0.003, unit: 1000 },
+    completion: { price: 0.004, unit: 1000 },
+  },
   'gpt-4': {
     prompt: { price: 0.03, unit: 1000 },
     completion: { price: 0.06, unit: 1000 },
@@ -54,6 +72,10 @@ export const modelCost = {
     prompt: { price: 0.03, unit: 1000 },
     completion: { price: 0.06, unit: 1000 },
   },
+  'gpt-4-0613': {
+    prompt: { price: 0.03, unit: 1000 },
+    completion: { price: 0.06, unit: 1000 },
+  },
   'gpt-4-32k': {
     prompt: { price: 0.06, unit: 1000 },
     completion: { price: 0.12, unit: 1000 },
@@ -62,6 +84,10 @@ export const modelCost = {
     prompt: { price: 0.06, unit: 1000 },
     completion: { price: 0.12, unit: 1000 },
   },
+  'gpt-4-32k-0613': {
+    prompt: { price: 0.06, unit: 1000 },
+    completion: { price: 0.12, unit: 1000 },
+  },
 };
 
 export const defaultUserMaxToken = 4000;

diff --git a/src/types/chat.ts b/src/types/chat.ts
@@ -49,7 +49,7 @@ export interface Folder {
   color?: string;
 }
 
-export type ModelOptions = 'gpt-4' | 'gpt-4-32k' | 'gpt-3.5-turbo';
+export type ModelOptions = 'gpt-4' | 'gpt-4-32k' | 'gpt-3.5-turbo' | 'gpt-3.5-turbo-16k' ;
 // | 'gpt-3.5-turbo-0301';
 // | 'gpt-4-0314'
 // | 'gpt-4-32k-0314'