New Ada Similarity (#44)

This commit is contained in:
Matt Trefilek
2022-12-15 14:54:42 -06:00
committed by GitHub
parent 23925ad36e
commit 9ada7ea8e1

View File

@@ -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