From 9ada7ea8e1c23a3e67a1e6de03968dc94abff670 Mon Sep 17 00:00:00 2001
From: Matt Trefilek <54865574+mtrefilek@users.noreply.github.com>
Date: Thu, 15 Dec 2022 14:54:42 -0600
Subject: [PATCH] New Ada Similarity (#44)

---
 embeddings.go | 3 +++
 1 file changed, 3 insertions(+)

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