From 1c20931ead68f5d7e7e04747720fac1ebd73d35c Mon Sep 17 00:00:00 2001
From: Matt Trefilek <54865574+mtrefilek@users.noreply.github.com>
Date: Fri, 16 Dec 2022 03:56:10 -0600
Subject: [PATCH] Fix for Ada Model Name (#45)

* Fix for Ada Model Name

* Refactoring ADA embeddings naming
---
 embeddings.go | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/embeddings.go b/embeddings.go
index 291435b..52c5223 100644
--- a/embeddings.go
+++ b/embeddings.go
@@ -52,7 +52,7 @@ const (
 	AdaCodeSearchText
 	BabbageCodeSearchCode
 	BabbageCodeSearchText
-	AdaSimilarityV2
+	AdaEmbeddingV2
 )
 
 var enumToString = map[EmbeddingModel]string{
@@ -72,7 +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",
+	AdaEmbeddingV2:        "text-embedding-ada-002",
 }
 
 var stringToEnum = map[string]EmbeddingModel{
@@ -92,7 +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,
+	"text-embedding-ada-002":        AdaEmbeddingV2,
 }
 
 // Embedding is a special format of data representation that can be easily utilized by machine