fix: add gpt enum (#494)

graasp · Apr 23, 2024 · 80b2169 · 80b2169
1 parent 92cb811
commit 80b2169
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git 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
@@ -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';