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{
AdaSimilarity: "ada-similarity",
BabbageSimilarity: "babbage-similarity",
CurieSimilarity: "curie-similarity",
DavinciSimilarity: "davinci-similarity",
AdaSearchDocument: "ada-search-document",
AdaSearchQuery: "ada-search-query",
BabbageSearchDocument: "babbage-search-document",
BabbageSearchQuery: "babbage-search-query",
CurieSearchDocument: "curie-search-document",
CurieSearchQuery: "curie-search-query",
DavinciSearchDocument: "davinci-search-document",
DavinciSearchQuery: "davinci-search-query",
AdaCodeSearchCode: "ada-code-search-code",
AdaCodeSearchText: "ada-code-search-text",
BabbageCodeSearchCode: "babbage-code-search-code",
BabbageCodeSearchText: "babbage-code-search-text",
AdaSimilarity: "text-similarity-ada-001",
BabbageSimilarity: "text-similarity-babbage-001",
CurieSimilarity: "text-similarity-curie-001",
DavinciSimilarity: "text-similarity-davinci-001",
AdaSearchDocument: "text-search-ada-doc-001",
AdaSearchQuery: "text-search-ada-query-001",
BabbageSearchDocument: "text-search-babbage-doc-001",
BabbageSearchQuery: "text-search-babbage-query-001",
CurieSearchDocument: "text-search-curie-doc-001",
CurieSearchQuery: "text-search-curie-query-001",
DavinciSearchDocument: "text-search-davinci-doc-001",
DavinciSearchQuery: "text-search-davinci-query-001",
AdaCodeSearchCode: "code-search-ada-code-001",
AdaCodeSearchText: "code-search-ada-text-001",
BabbageCodeSearchCode: "code-search-babbage-code-001",
BabbageCodeSearchText: "code-search-babbage-text-001",
}
var stringToEnum = map[string]EmbeddingModel{
"ada-similarity": AdaSimilarity,
"babbage-similarity": BabbageSimilarity,
"curie-similarity": CurieSimilarity,
"davinci-similarity": DavinciSimilarity,
"ada-search-document": AdaSearchDocument,
"ada-search-query": AdaSearchQuery,
"babbage-search-document": BabbageSearchDocument,
"babbage-search-query": BabbageSearchQuery,
"curie-search-document": CurieSearchDocument,
"curie-search-query": CurieSearchQuery,
"davinci-search-document": DavinciSearchDocument,
"davinci-search-query": DavinciSearchQuery,
"ada-code-search-code": AdaCodeSearchCode,
"ada-code-search-text": AdaCodeSearchText,
"babbage-code-search-code": BabbageCodeSearchCode,
"babbage-code-search-text": BabbageCodeSearchText,
"text-similarity-ada-001": AdaSimilarity,
"text-similarity-babbage-001": BabbageSimilarity,
"text-similarity-curie-001": CurieSimilarity,
"text-similarity-davinci-001": DavinciSimilarity,
"text-search-ada-doc-001": AdaSearchDocument,
"text-search-ada-query-001": AdaSearchQuery,
"text-search-babbage-doc-001": BabbageSearchDocument,
"text-search-babbage-query-001": BabbageSearchQuery,
"text-search-curie-doc-001": CurieSearchDocument,
"text-search-curie-query-001": CurieSearchQuery,
"text-search-davinci-doc-001": DavinciSearchDocument,
"text-search-davinci-query-001": DavinciSearchQuery,
"code-search-ada-code-001": AdaCodeSearchCode,
"code-search-ada-text-001": AdaCodeSearchText,
"code-search-babbage-code-001": BabbageCodeSearchCode,
"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.