Support January 25, 2024, models update. (#644)

sashabaranov · Jan 26, 2024 · 4c41f24 · 4c41f24
1 parent eff8dc1
commit 4c41f24
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git 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
@@ -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