Add GPT4Turbo and GPT4TurboVision (#125)

softwaremill · Nov 7, 2023 · dfd0826 · dfd0826
1 parent fc224f3
commit dfd0826
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/core/src/main/scala/sttp/openai/requests/completions/chat/ChatRequestBody.scala b/core/src/main/scala/sttp/openai/requests/completions/chat/ChatRequestBody.scala
@@ -100,9 +100,18 @@ object ChatRequestBody {
 
     case object GPT35Turbo0301 extends ChatCompletionModel("gpt-3.5-turbo-0301")
 
+    case object GPT4Turbo extends ChatCompletionModel("gpt-4-1106-preview")
+
+    case object GPT4TurboVision extends ChatCompletionModel("gpt-4-vision-preview")
+
     case class CustomChatCompletionModel(customChatCompletionModel: String) extends ChatCompletionModel(customChatCompletionModel)
 
-    val values: Set[ChatCompletionModel] = Set(GPT4, GPT40314, GPT432k, GPT432k0314, GPT35Turbo, GPT35Turbo0301)
+    val values: Set[ChatCompletionModel] = 
+      Set(
+        GPT4, GPT40314, GPT432k, 
+        GPT432k0314, GPT35Turbo, GPT35Turbo0301, 
+        GPT4Turbo, GPT4TurboVision
+      )
 
     private val byChatModelValue = values.map(model => model.value -> model).toMap
   }