From 4c41f24a99ad56f707df7c25b8833fb0a374c8c5 Mon Sep 17 00:00:00 2001
From: Daniil <7709243+bazuker@users.noreply.github.com>
Date: Fri, 26 Jan 2024 00:41:48 -0800
Subject: [PATCH] Support January 25, 2024, models update. (#644)

---
 completion.go | 6 +++++-
 embeddings.go | 4 +++-
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/completion.go b/completion.go
index 2709c8b..6326a72 100644
--- a/completion.go
+++ b/completion.go
@@ -22,7 +22,9 @@ const (
 	GPT432K               = "gpt-4-32k"
 	GPT40613              = "gpt-4-0613"
 	GPT40314              = "gpt-4-0314"
-	GPT4TurboPreview      = "gpt-4-1106-preview"
+	GPT4Turbo0125         = "gpt-4-0125-preview"
+	GPT4Turbo1106         = "gpt-4-1106-preview"
+	GPT4TurboPreview      = "gpt-4-turbo-preview"
 	GPT4VisionPreview     = "gpt-4-vision-preview"
 	GPT4                  = "gpt-4"
 	GPT3Dot5Turbo1106     = "gpt-3.5-turbo-1106"
@@ -78,6 +80,8 @@ var disabledModelsForEndpoints = map[string]map[string]bool{
 		GPT4:                 true,
 		GPT4TurboPreview:     true,
 		GPT4VisionPreview:    true,
+		GPT4Turbo1106:        true,
+		GPT4Turbo0125:        true,
 		GPT40314:             true,
 		GPT40613:             true,
 		GPT432K:              true,
diff --git a/embeddings.go b/embeddings.go
index c144119..517027f 100644
--- a/embeddings.go
+++ b/embeddings.go
@@ -34,7 +34,9 @@ const (
 	BabbageCodeSearchCode EmbeddingModel = "code-search-babbage-code-001"
 	BabbageCodeSearchText EmbeddingModel = "code-search-babbage-text-001"
 
-	AdaEmbeddingV2 EmbeddingModel = "text-embedding-ada-002"
+	AdaEmbeddingV2  EmbeddingModel = "text-embedding-ada-002"
+	SmallEmbedding3 EmbeddingModel = "text-embedding-3-small"
+	LargeEmbedding3 EmbeddingModel = "text-embedding-3-large"
 )
 
 // Embedding is a special format of data representation that can be easily utilized by machine