From 80b2169c5a49fbcd3843869ddd53476b29464781 Mon Sep 17 00:00:00 2001
From: Basile Spaenlehauer <spaenleh@gmail.com>
Date: Tue, 23 Apr 2024 08:36:47 +0200
Subject: [PATCH] fix: add gpt enum (#494)

---
 src/chatbot/chatbot.ts | 6 ++++++
 src/index.ts           | 6 +++++-
 2 files changed, 11 insertions(+), 1 deletion(-)

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';