New Ada Similarity (sashabaranov#44)

sfcuboy · Dec 15, 2022 · 9ada7ea · 9ada7ea
1 parent 23925ad
commit 9ada7ea
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/embeddings.go b/embeddings.go
@@ -52,6 +52,7 @@ const (
 	AdaCodeSearchText
 	BabbageCodeSearchCode
 	BabbageCodeSearchText
+	AdaSimilarityV2
 )
 
 var enumToString = map[EmbeddingModel]string{
@@ -71,6 +72,7 @@ var enumToString = map[EmbeddingModel]string{
 	AdaCodeSearchText:     "code-search-ada-text-001",
 	BabbageCodeSearchCode: "code-search-babbage-code-001",
 	BabbageCodeSearchText: "code-search-babbage-text-001",
+	AdaSimilarityV2:       "text-similarity-ada-002",
 }
 
 var stringToEnum = map[string]EmbeddingModel{
@@ -90,6 +92,7 @@ var stringToEnum = map[string]EmbeddingModel{
 	"code-search-ada-text-001":      AdaCodeSearchText,
 	"code-search-babbage-code-001":  BabbageCodeSearchCode,
 	"code-search-babbage-text-001":  BabbageCodeSearchText,
+	"text-similarity-ada-002":       AdaSimilarityV2,
 }
 
 // Embedding is a special format of data representation that can be easily utilized by machine