diff --git a/src/components/Chat/ChatContent/ModelConfigBar.tsx b/src/components/Chat/ChatContent/ModelConfigBar.tsx index cf863ea2..3de69c2c 100644 --- a/src/components/Chat/ChatContent/ModelConfigBar.tsx +++ b/src/components/Chat/ChatContent/ModelConfigBar.tsx @@ -5,11 +5,7 @@ import useStore from '@store/store'; import ConfigMenu from '@components/ConfigMenu'; import { ChatInterface, ConfigInterface } from '@type/chat'; import { ModelChoice } from '@type/chat'; -import { - _defaultChatConfig, - modelMaxToken, - modelOptions, -} from '@constants/chat'; +import { _defaultChatConfig, modelMaxToken } from '@constants/chat'; import { ModelSelect } from '@components/ConfigMenu/ModelSelect'; const ModelConfigBar = React.memo(() => { diff --git a/src/constants/chat.ts b/src/constants/chat.ts index 686a50b7..7026c0d1 100644 --- a/src/constants/chat.ts +++ b/src/constants/chat.ts @@ -25,15 +25,14 @@ export const modelOptions: ModelChoice[] = [ 'gpt-4-32k', 'claude-2', 'claude-instant-1', - // 'gpt-3.5-turbo-0301', - // 'gpt-4-0314', - // 'gpt-4-32k-0314', + 'gpt-3.5-turbo-1106', ]; export const defaultModel = 'gpt-3.5-turbo'; export const modelMaxToken = { 'gpt-3.5-turbo': 4096, + 'gpt-3.5-turbo-1106': 16385, 'gpt-3.5-turbo-0301': 4096, 'gpt-3.5-turbo-0613': 4096, 'gpt-3.5-turbo-16k': 16384, @@ -54,6 +53,10 @@ export const modelCost = { prompt: { price: 0.0015, unit: 1000 }, completion: { price: 0.002, unit: 1000 }, }, + 'gpt-3.5-turbo-1106': { + prompt: { price: 0.001, unit: 1000 }, + completion: { price: 0.0015, unit: 1000 }, + }, 'gpt-3.5-turbo-0301': { prompt: { price: 0.0015, unit: 1000 }, completion: { price: 0.002, unit: 1000 }, diff --git a/src/types/chat.ts b/src/types/chat.ts index 67901dd8..1debaf74 100644 --- a/src/types/chat.ts +++ b/src/types/chat.ts @@ -52,16 +52,21 @@ export interface Folder { } export type ModelChoice = - | 'gpt-4' - | 'gpt-4-32k' - | 'gpt-4-1106-preview' - | 'gpt-3.5-turbo' - | 'gpt-3.5-turbo-16k' - | 'claude-2' - | 'claude-instant-1'; -// | 'gpt-3.5-turbo-0301'; -// | 'gpt-4-0314' -// | 'gpt-4-32k-0314' +| 'gpt-3.5-turbo' +| 'gpt-3.5-turbo-1106' +| 'gpt-3.5-turbo-0301' +| 'gpt-3.5-turbo-0613' +| 'gpt-3.5-turbo-16k' +| 'gpt-3.5-turbo-16k-0613' +| 'gpt-4' +| 'gpt-4-0314' +| 'gpt-4-0613' +| 'gpt-4-1106-preview' +| 'gpt-4-32k' +| 'gpt-4-32k-0314' +| 'gpt-4-32k-0613' +| 'claude-2' +| 'claude-instant-1'; export type TotalTokenUsed = { [model in ModelChoice]?: {