Add missing openai models

Ironclad · Oct 25, 2023 · d21037b · d21037b
1 parent 3db63a3
commit d21037b
Showing 1 changed file with 32 additions and 2 deletions.
diff --git a/packages/core/src/utils/openai.ts b/packages/core/src/utils/openai.ts
@@ -52,7 +52,14 @@ export const openaiModels = {
     },
     displayName: 'GPT-3.5 Turbo',
   },
-
+  'gpt-3.5-turbo-16k': {
+    maxTokens: 16384,
+    cost: {
+      prompt: 0.003,
+      completion: 0.004,
+    },
+    displayName: 'GPT-3.5 16k',
+  },
   'gpt-3.5-turbo-0613': {
     maxTokens: 16384,
     cost: {
@@ -61,7 +68,6 @@ export const openaiModels = {
     },
     displayName: 'GPT-3.5 (v0613)',
   },
-
   'gpt-3.5-turbo-16k-0613': {
     maxTokens: 16384,
     cost: {
@@ -70,6 +76,30 @@ export const openaiModels = {
     },
     displayName: 'GPT-3.5 16k (v0613)',
   },
+  'gpt-3.5-turbo-0301': {
+    maxTokens: 16384,
+    cost: {
+      prompt: 0.002,
+      completion: 0.002,
+    },
+    displayName: 'GPT-3.5 (v0301)',
+  },
+  'gpt-4-0314': {
+    maxTokens: 8192,
+    cost: {
+      prompt: 0.03,
+      completion: 0.06,
+    },
+    displayName: 'GPT-4 (v0314)',
+  },
+  'gpt-4-32k-0314': {
+    maxTokens: 32768,
+    cost: {
+      prompt: 0.06,
+      completion: 0.12,
+    },
+    displayName: 'GPT-4 32k (v0314)',
+  },
 } satisfies Record<string, OpenAIModel>;
 
 export const openAiModelOptions = Object.entries(openaiModels).map(([id, { displayName }]) => ({