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

Update model names (#16)

Browse Source
This commit is contained in:
Andy Day
2022-03-14 15:10:30 -07:00
committed by GitHub
parent eecd212963
commit 260ef874de
1 changed files with 32 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

64
embeddings.go
View File

@@ -56,41 +56,41 @@ const (
) )
var enumToString = map[EmbeddingModel]string{ var enumToString = map[EmbeddingModel]string{
AdaSimilarity: "ada-similarity", AdaSimilarity: "text-similarity-ada-001",
BabbageSimilarity: "babbage-similarity", BabbageSimilarity: "text-similarity-babbage-001",
CurieSimilarity: "curie-similarity", CurieSimilarity: "text-similarity-curie-001",
DavinciSimilarity: "davinci-similarity", DavinciSimilarity: "text-similarity-davinci-001",
AdaSearchDocument: "ada-search-document", AdaSearchDocument: "text-search-ada-doc-001",
AdaSearchQuery: "ada-search-query", AdaSearchQuery: "text-search-ada-query-001",
BabbageSearchDocument: "babbage-search-document", BabbageSearchDocument: "text-search-babbage-doc-001",
BabbageSearchQuery: "babbage-search-query", BabbageSearchQuery: "text-search-babbage-query-001",
CurieSearchDocument: "curie-search-document", CurieSearchDocument: "text-search-curie-doc-001",
CurieSearchQuery: "curie-search-query", CurieSearchQuery: "text-search-curie-query-001",
DavinciSearchDocument: "davinci-search-document", DavinciSearchDocument: "text-search-davinci-doc-001",
DavinciSearchQuery: "davinci-search-query", DavinciSearchQuery: "text-search-davinci-query-001",
AdaCodeSearchCode: "ada-code-search-code", AdaCodeSearchCode: "code-search-ada-code-001",
AdaCodeSearchText: "ada-code-search-text", AdaCodeSearchText: "code-search-ada-text-001",
BabbageCodeSearchCode: "babbage-code-search-code", BabbageCodeSearchCode: "code-search-babbage-code-001",
BabbageCodeSearchText: "babbage-code-search-text", BabbageCodeSearchText: "code-search-babbage-text-001",
} }
var stringToEnum = map[string]EmbeddingModel{ var stringToEnum = map[string]EmbeddingModel{
"ada-similarity": AdaSimilarity, "text-similarity-ada-001": AdaSimilarity,
"babbage-similarity": BabbageSimilarity, "text-similarity-babbage-001": BabbageSimilarity,
"curie-similarity": CurieSimilarity, "text-similarity-curie-001": CurieSimilarity,
"davinci-similarity": DavinciSimilarity, "text-similarity-davinci-001": DavinciSimilarity,
"ada-search-document": AdaSearchDocument, "text-search-ada-doc-001": AdaSearchDocument,
"ada-search-query": AdaSearchQuery, "text-search-ada-query-001": AdaSearchQuery,
"babbage-search-document": BabbageSearchDocument, "text-search-babbage-doc-001": BabbageSearchDocument,
"babbage-search-query": BabbageSearchQuery, "text-search-babbage-query-001": BabbageSearchQuery,
"curie-search-document": CurieSearchDocument, "text-search-curie-doc-001": CurieSearchDocument,
"curie-search-query": CurieSearchQuery, "text-search-curie-query-001": CurieSearchQuery,
"davinci-search-document": DavinciSearchDocument, "text-search-davinci-doc-001": DavinciSearchDocument,
"davinci-search-query": DavinciSearchQuery, "text-search-davinci-query-001": DavinciSearchQuery,
"ada-code-search-code": AdaCodeSearchCode, "code-search-ada-code-001": AdaCodeSearchCode,
"ada-code-search-text": AdaCodeSearchText, "code-search-ada-text-001": AdaCodeSearchText,
"babbage-code-search-code": BabbageCodeSearchCode, "code-search-babbage-code-001": BabbageCodeSearchCode,
"babbage-code-search-text": BabbageCodeSearchText, "code-search-babbage-text-001": BabbageCodeSearchText,
} }
// Embedding is a special format of data representation that can be easily utilized by machine learning models and algorithms. // Embedding is a special format of data representation that can be easily utilized by machine learning models and algorithms.