From c5a70f0ecea5e0df60ce517109115a4b9a8ce956 Mon Sep 17 00:00:00 2001 From: raulraja Date: Thu, 29 Jun 2023 20:15:14 +0200 Subject: [PATCH] Rename to TEXT_EMBEDDING_ADA_002 --- .../functional/xef/llm/models/embeddings/EmbeddingModel.kt | 2 +- .../com/xebia/functional/xef/vectorstores/LocalVectorStore.kt | 2 +- .../main/kotlin/com/xebia/functional/xef/vectorstores/Lucene.kt | 2 +- .../postgresql/src/test/kotlin/xef/PGVectorStoreSpec.kt | 2 +- 4 files changed, 4 insertions(+), 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 index c9f2bdb7f..acfc85cc6 100644 --- 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 index dd5edd0b5..8aa40ec9a 100644 --- 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) { 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 index 0f57617ab..2cc991d88 100644 --- 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) { texts.forEach { diff --git a/integrations/postgresql/src/test/kotlin/xef/PGVectorStoreSpec.kt b/integrations/postgresql/src/test/kotlin/xef/PGVectorStoreSpec.kt index 8e10a9d89..b0a413db0 100644 --- 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 )