From 2a21985a179e00b16f28b27c0d5bd8b1f483df5c Mon Sep 17 00:00:00 2001
From: heimoshuiyu <heimoshuiyu@gmail.com>
Date: Tue, 14 May 2024 09:26:07 +0800
Subject: [PATCH] add gpt-4o models

---
 src/models.ts | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/models.ts b/src/models.ts
index ade3572..41c9f28 100644
--- a/src/models.ts
+++ b/src/models.ts
@@ -35,6 +35,14 @@ const models: Record<string, Model> = {
     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-0125-preview": {
     maxToken: 128000,
     price: { prompt: 0.01 / 1000, completion: 0.03 / 1000 },