feat: add gpt-4o-mini support (#450)

lunary-ai · Jul 29, 2024 · 1b9e11a · 1b9e11a
1 parent 95b5fef
commit 1b9e11a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/packages/db/0028.sql b/packages/db/0028.sql
@@ -0,0 +1,3 @@
+insert into model_mapping (name, pattern, unit, input_cost, output_cost, tokenizer, start_date) values 
+('gpt-4o-mini', '^(gpt-4o)$', 'TOKENS', 0.15, 0.6, 'openai', null),
+('gpt-4o-mini-2024-07-18', '^(gpt-4o-mini-2024-07-18)$', 'TOKENS', 0.15, 0.6, 'openai', null);
diff --git a/packages/shared/models.ts b/packages/shared/models.ts
@@ -4,6 +4,11 @@ export const MODELS = [
     name: "gpt-4o",
     provider: "openai",
   },
+  {
+    id: "gpt-4o-mini",
+    name: "gpt-4o-mini",
+    provider: "openai",
+  },
   {
     id: "gpt-4-turbo",
     name: "gpt-4-turbo",