diff --git a/src/components/TokenCount/TokenCount.tsx b/src/components/TokenCount/TokenCount.tsx index 5ad12c9a0..7fdcb449b 100644 --- a/src/components/TokenCount/TokenCount.tsx +++ b/src/components/TokenCount/TokenCount.tsx @@ -3,7 +3,7 @@ import useStore from '@store/store'; import { shallow } from 'zustand/shallow'; import countTokens from '@utils/messageUtils'; -import { modelCost } from '@constants/chat'; +import { modelCost, defaultModel } from '@constants/chat'; const TokenCount = React.memo(() => { const [tokenCount, setTokenCount] = useState(0); @@ -17,7 +17,7 @@ const TokenCount = React.memo(() => { const model = useStore((state) => state.chats ? state.chats[state.currentChatIndex].config.model - : 'gpt-3.5-turbo' + : defaultModel ); const cost = useMemo(() => { diff --git a/src/constants/chat.ts b/src/constants/chat.ts index f806ef1b6..4f8433c7f 100644 --- a/src/constants/chat.ts +++ b/src/constants/chat.ts @@ -30,12 +30,11 @@ export const modelOptions: ModelOptions[] = [ 'gpt-4-turbo-2024-04-09', 'gpt-4o', 'gpt-4o-2024-05-13', - // 'gpt-3.5-turbo-0301', - // 'gpt-4-0314', - // 'gpt-4-32k-0314', + 'gpt-4o-mini', + 'gpt-4o-mini-2024-07-18', ]; -export const defaultModel = 'gpt-3.5-turbo'; +export const defaultModel = 'gpt-4o-mini'; export const modelMaxToken = { 'gpt-3.5-turbo': 4096, @@ -57,6 +56,8 @@ export const modelMaxToken = { 'gpt-4-turbo-2024-04-09': 128000, 'gpt-4o': 128000, 'gpt-4o-2024-05-13': 128000, + 'gpt-4o-mini': 128000, + 'gpt-4o-mini-2024-07-18': 128000, }; export const modelCost = { @@ -136,6 +137,14 @@ export const modelCost = { prompt: { price: 0.005, unit: 1000 }, completion: { price: 0.015, unit: 1000 }, }, + 'gpt-4o-mini': { + prompt: { price: 0.00015, unit: 1000 }, + completion: { price: 0.0006, unit: 1000 }, + }, + 'gpt-4o-mini-2024-07-18': { + prompt: { price: 0.00015, unit: 1000 }, + completion: { price: 0.0006, unit: 1000 }, + }, }; export const defaultUserMaxToken = 4000; diff --git a/src/types/chat.ts b/src/types/chat.ts index 5b706952a..194569f5a 100644 --- a/src/types/chat.ts +++ b/src/types/chat.ts @@ -50,6 +50,8 @@ export interface Folder { } export type ModelOptions = + | 'gpt-4o-mini' + | 'gpt-4o-mini-2024-07-18' | 'gpt-4o' | 'gpt-4o-2024-05-13' | 'gpt-4' @@ -62,9 +64,6 @@ export type ModelOptions = | 'gpt-3.5-turbo-16k' | 'gpt-3.5-turbo-1106' | 'gpt-3.5-turbo-0125'; -// | 'gpt-3.5-turbo-0301'; -// | 'gpt-4-0314' -// | 'gpt-4-32k-0314' export type TotalTokenUsed = { [model in ModelOptions]?: {