diff --git a/src/chatbot/chatbot.ts b/src/chatbot/chatbot.ts index d08a9a17..ffd20781 100644 --- a/src/chatbot/chatbot.ts +++ b/src/chatbot/chatbot.ts @@ -4,6 +4,12 @@ export enum ChatbotRole { User = 'user', } +export enum GPTVersion { + GPT_3_5_TURBO = 'gpt-3.5-turbo-0125', + GPT_4 = 'gpt-4', + GPT_4_TURBO = 'gpt-4-turbo', +} + export type ChatBotMessage = { role: `${ChatbotRole}` | ChatbotRole; content: string; diff --git a/src/index.ts b/src/index.ts index 8887d82b..8e8eb46b 100644 --- a/src/index.ts +++ b/src/index.ts @@ -279,4 +279,8 @@ export { /** * Chatbot related types */ -export { ChatbotRole, type ChatBotMessage } from './chatbot/chatbot.js'; +export { + ChatbotRole, + type ChatBotMessage, + GPTVersion, +} from './chatbot/chatbot.js';