diff --git a/embeddings.go b/embeddings.go index 9fa05d4..291435b 100644 --- a/embeddings.go +++ b/embeddings.go @@ -52,6 +52,7 @@ const ( AdaCodeSearchText BabbageCodeSearchCode BabbageCodeSearchText + AdaSimilarityV2 ) var enumToString = map[EmbeddingModel]string{ @@ -71,6 +72,7 @@ var enumToString = map[EmbeddingModel]string{ AdaCodeSearchText: "code-search-ada-text-001", BabbageCodeSearchCode: "code-search-babbage-code-001", BabbageCodeSearchText: "code-search-babbage-text-001", + AdaSimilarityV2: "text-similarity-ada-002", } var stringToEnum = map[string]EmbeddingModel{ @@ -90,6 +92,7 @@ var stringToEnum = map[string]EmbeddingModel{ "code-search-ada-text-001": AdaCodeSearchText, "code-search-babbage-code-001": BabbageCodeSearchCode, "code-search-babbage-text-001": BabbageCodeSearchText, + "text-similarity-ada-002": AdaSimilarityV2, } // Embedding is a special format of data representation that can be easily utilized by machine