VaalaCat/go-openai
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix for Ada Model Name (#45)

Browse Source 
* Fix for Ada Model Name

* Refactoring ADA embeddings naming
This commit is contained in:
Matt Trefilek
2022-12-16 03:56:10 -06:00
committed by GitHub
parent 9ada7ea8e1
commit 1c20931ead
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
embeddings.go
View File

@@ -52,7 +52,7 @@ const (
AdaCodeSearchText AdaCodeSearchText
BabbageCodeSearchCode BabbageCodeSearchCode
BabbageCodeSearchText BabbageCodeSearchText
AdaSimilarityV2 AdaEmbeddingV2
) )
var enumToString = map[EmbeddingModel]string{ var enumToString = map[EmbeddingModel]string{
@@ -72,7 +72,7 @@ var enumToString = map[EmbeddingModel]string{
AdaCodeSearchText: "code-search-ada-text-001", AdaCodeSearchText: "code-search-ada-text-001",
BabbageCodeSearchCode: "code-search-babbage-code-001", BabbageCodeSearchCode: "code-search-babbage-code-001",
BabbageCodeSearchText: "code-search-babbage-text-001", BabbageCodeSearchText: "code-search-babbage-text-001",
AdaSimilarityV2: "text-similarity-ada-002", AdaEmbeddingV2: "text-embedding-ada-002",
} }
var stringToEnum = map[string]EmbeddingModel{ var stringToEnum = map[string]EmbeddingModel{
@@ -92,7 +92,7 @@ var stringToEnum = map[string]EmbeddingModel{
"code-search-ada-text-001": AdaCodeSearchText, "code-search-ada-text-001": AdaCodeSearchText,
"code-search-babbage-code-001": BabbageCodeSearchCode, "code-search-babbage-code-001": BabbageCodeSearchCode,
"code-search-babbage-text-001": BabbageCodeSearchText, "code-search-babbage-text-001": BabbageCodeSearchText,
"text-similarity-ada-002": AdaSimilarityV2, "text-embedding-ada-002": AdaEmbeddingV2,
} }
// Embedding is a special format of data representation that can be easily utilized by machine // Embedding is a special format of data representation that can be easily utilized by machine