add model: gpt-3.5-turbo-16k

lucoiso · Aug 12, 2023 · a22d7ae · a22d7ae · lucoiso · Aug 13, 2023
1 parent eabe6a7
commit a22d7ae
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/Source/HttpGPTCommonModule/Private/Utils/HttpGPTHelper.cpp b/Source/HttpGPTCommonModule/Private/Utils/HttpGPTHelper.cpp
@@ -22,6 +22,9 @@ const FName UHttpGPTHelper::ModelToName(const EHttpGPTChatModel& Model)
     case EHttpGPTChatModel::gpt35turbo:
         return "gpt-3.5-turbo";
 
+    case EHttpGPTChatModel::gpt35turbo16k:
+        return "gpt-3.5-turbo-16k";
+
     case EHttpGPTChatModel::textdavinci003:
         return "text-davinci-003";
 
@@ -51,6 +54,10 @@ const EHttpGPTChatModel UHttpGPTHelper::NameToModel(const FName Model)
     {
         return EHttpGPTChatModel::gpt35turbo;
     }
+    else if (Model.IsEqual("gpt-3.5-turbo-16k", ENameCase::IgnoreCase))
+    {
+        return EHttpGPTChatModel::gpt35turbo16k;
+    }
     else if (Model.IsEqual("text-davinci-003", ENameCase::IgnoreCase))
     {
         return EHttpGPTChatModel::textdavinci003;

diff --git a/Source/HttpGPTCommonModule/Public/Structures/HttpGPTChatTypes.h b/Source/HttpGPTCommonModule/Public/Structures/HttpGPTChatTypes.h
@@ -107,6 +107,7 @@ enum class EHttpGPTChatModel : uint8
     gpt4			UMETA(DisplayName = "gpt-4"),
     gpt432k			UMETA(DisplayName = "gpt-4-32k"),
     gpt35turbo		UMETA(DisplayName = "gpt-3.5-turbo"),
+    gpt35turbo16k   UMETA(DisplayName = "gpt-3.5-turbo-16k"),
     textdavinci003	UMETA(DisplayName = "text-davinci-003"),
     textdavinci002	UMETA(DisplayName = "text-davinci-002"),
     codedavinci002	UMETA(DisplayName = "code-davinci-002"),