diff --git a/embeddings.go b/embeddings.go index 765545b..fb6c199 100644 --- a/embeddings.go +++ b/embeddings.go @@ -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.