Skip to content

Commit 1c20931

Browse files
authored
Fix for Ada Model Name (#45)
* Fix for Ada Model Name * Refactoring ADA embeddings naming
1 parent 9ada7ea commit 1c20931

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

embeddings.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ const (
5252
AdaCodeSearchText
5353
BabbageCodeSearchCode
5454
BabbageCodeSearchText
55-
AdaSimilarityV2
55+
AdaEmbeddingV2
5656
)
5757

5858
var enumToString = map[EmbeddingModel]string{
@@ -72,7 +72,7 @@ var enumToString = map[EmbeddingModel]string{
7272
AdaCodeSearchText: "code-search-ada-text-001",
7373
BabbageCodeSearchCode: "code-search-babbage-code-001",
7474
BabbageCodeSearchText: "code-search-babbage-text-001",
75-
AdaSimilarityV2: "text-similarity-ada-002",
75+
AdaEmbeddingV2: "text-embedding-ada-002",
7676
}
7777

7878
var stringToEnum = map[string]EmbeddingModel{
@@ -92,7 +92,7 @@ var stringToEnum = map[string]EmbeddingModel{
9292
"code-search-ada-text-001": AdaCodeSearchText,
9393
"code-search-babbage-code-001": BabbageCodeSearchCode,
9494
"code-search-babbage-text-001": BabbageCodeSearchText,
95-
"text-similarity-ada-002": AdaSimilarityV2,
95+
"text-embedding-ada-002": AdaEmbeddingV2,
9696
}
9797

9898
// Embedding is a special format of data representation that can be easily utilized by machine

0 commit comments

Comments
 (0)