diff --git a/src/chatgpt.ts b/src/chatgpt.ts index a193f4d..3898ad1 100644 --- a/src/chatgpt.ts +++ b/src/chatgpt.ts @@ -1,3 +1,5 @@ +import { defaultModel } from "./models"; + export interface ImageURL { url: string; detail: "low" | "high"; @@ -155,7 +157,7 @@ class Chat { enable_max_gen_tokens = true, tokens_margin = 1024, apiEndPoint = "https://api.openai.com/v1/chat/completions", - model = "gpt-3.5-turbo", + model = defaultModel, temperature = 0.7, enable_temperature = true, top_p = 1, diff --git a/src/models.ts b/src/models.ts index 50727df..f852bfa 100644 --- a/src/models.ts +++ b/src/models.ts @@ -7,6 +7,22 @@ interface Model { } const models: Record = { + "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-4o-mini": { + maxToken: 128000, + price: { prompt: 0.15 / 1000 / 1000, completion: 0.6 / 1000 / 1000 }, + }, + "gpt-4o-mini-2024-07-18": { + maxToken: 128000, + price: { prompt: 0.15 / 1000 / 1000, completion: 0.6 / 1000 / 1000 }, + }, "gpt-3.5-turbo-0125": { maxToken: 16385, price: { prompt: 0.0005 / 1000, completion: 0.0015 / 1000 }, @@ -31,18 +47,6 @@ const models: Record = { maxToken: 16385, price: { prompt: 0.003 / 1000, completion: 0.004 / 1000 }, }, - "gpt-3.5-turbo-0301": { - 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-turbo": { maxToken: 128000, price: { prompt: 0.01 / 1000, completion: 0.03 / 1000 }, @@ -75,20 +79,12 @@ const models: Record = { maxToken: 8192, price: { prompt: 0.03 / 1000, completion: 0.06 / 1000 }, }, - "gpt-4-0613": { - maxToken: 8192, - price: { prompt: 0.03 / 1000, completion: 0.06 / 1000 }, - }, "gpt-4-32k": { maxToken: 8192, price: { prompt: 0.06 / 1000, completion: 0.12 / 1000 }, }, - "gpt-4-32k-0613": { - maxToken: 8192, - price: { prompt: 0.06 / 1000, completion: 0.12 / 1000 }, - }, }; -export const defaultModel = "gpt-3.5-turbo-0125"; +export const defaultModel = "gpt-4o-mini"; export default models;