New Ada Similarity (#44)

mtrefilek · web-flow · commit 9ada7ea8e1c2 · 2022-12-16T00:54:42.000+04:00
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

Original file line number	Diff line number	Diff line change
`@@ -52,6 +52,7 @@ const (`
`52`	`52`	`AdaCodeSearchText`
`53`	`53`	`BabbageCodeSearchCode`
`54`	`54`	`BabbageCodeSearchText`
	`55`	`+ AdaSimilarityV2`
`55`	`56`	`)`
`56`	`57`
`57`	`58`	`var enumToString = map[EmbeddingModel]string{`
`@@ -71,6 +72,7 @@ var enumToString = map[EmbeddingModel]string{`
`71`	`72`	`AdaCodeSearchText: "code-search-ada-text-001",`
`72`	`73`	`BabbageCodeSearchCode: "code-search-babbage-code-001",`
`73`	`74`	`BabbageCodeSearchText: "code-search-babbage-text-001",`
	`75`	`+ AdaSimilarityV2: "text-similarity-ada-002",`
`74`	`76`	`}`
`75`	`77`
`76`	`78`	`var stringToEnum = map[string]EmbeddingModel{`
`@@ -90,6 +92,7 @@ var stringToEnum = map[string]EmbeddingModel{`
`90`	`92`	`"code-search-ada-text-001": AdaCodeSearchText,`
`91`	`93`	`"code-search-babbage-code-001": BabbageCodeSearchCode,`
`92`	`94`	`"code-search-babbage-text-001": BabbageCodeSearchText,`
	`95`	`+ "text-similarity-ada-002": AdaSimilarityV2,`
`93`	`96`	`}`
`94`	`97`
`95`	`98`	`// Embedding is a special format of data representation that can be easily utilized by machine`