Rename to TEXT_EMBEDDING_ADA_002

xebia-functional · Jun 29, 2023 · c5a70f0 · c5a70f0
1 parent ef1c24c
commit c5a70f0
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/core/src/commonMain/kotlin/com/xebia/functional/xef/llm/models/embeddings/EmbeddingModel.kt b/core/src/commonMain/kotlin/com/xebia/functional/xef/llm/models/embeddings/EmbeddingModel.kt
@@ -1,5 +1,5 @@
 package com.xebia.functional.xef.llm.models.embeddings
 
 enum class EmbeddingModel(val modelName: String) {
-  TextEmbeddingAda002("text-embedding-ada-002")
+  TEXT_EMBEDDING_ADA_002("text-embedding-ada-002")
 }
diff --git a/core/src/commonMain/kotlin/com/xebia/functional/xef/vectorstores/LocalVectorStore.kt b/core/src/commonMain/kotlin/com/xebia/functional/xef/vectorstores/LocalVectorStore.kt
@@ -25,7 +25,7 @@ private constructor(private val embeddings: Embeddings, private val state: Atomi
   constructor(embeddings: Embeddings) : this(embeddings, Atomic(State.empty()))
 
   private val requestConfig =
-    RequestConfig(EmbeddingModel.TextEmbeddingAda002, RequestConfig.Companion.User("user"))
+    RequestConfig(EmbeddingModel.TEXT_EMBEDDING_ADA_002, RequestConfig.Companion.User("user"))
 
   override suspend fun addTexts(texts: List<String>) {
     val embeddingsList =

diff --git a/integrations/lucene/src/main/kotlin/com/xebia/functional/xef/vectorstores/Lucene.kt b/integrations/lucene/src/main/kotlin/com/xebia/functional/xef/vectorstores/Lucene.kt
@@ -29,7 +29,7 @@ open class Lucene(
 ) : VectorStore, AutoCloseable {
 
   private val requestConfig =
-    RequestConfig(EmbeddingModel.TextEmbeddingAda002, RequestConfig.Companion.User("user"))
+    RequestConfig(EmbeddingModel.TEXT_EMBEDDING_ADA_002, RequestConfig.Companion.User("user"))
 
   override suspend fun addTexts(texts: List<String>) {
     texts.forEach {

diff --git a/integrations/postgresql/src/test/kotlin/xef/PGVectorStoreSpec.kt b/integrations/postgresql/src/test/kotlin/xef/PGVectorStoreSpec.kt
@@ -45,7 +45,7 @@ class PGVectorStoreSpec :
         distanceStrategy = PGDistanceStrategy.Euclidean,
         preDeleteCollection = false,
         requestConfig =
-          RequestConfig(EmbeddingModel.TextEmbeddingAda002, RequestConfig.Companion.User("user")),
+          RequestConfig(EmbeddingModel.TEXT_EMBEDDING_ADA_002, RequestConfig.Companion.User("user")),
         chunckSize = null
       )