refactor(embedding): replace STEmbedding with LocalEmbedding

Replace usages of STEmbedding with LocalEmbedding for consistency and clarity.
unit-mesh · Jul 7, 2024 · edcf4ed · edcf4ed
1 parent 3d1f4a8
commit edcf4ed
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 21 deletions.
diff --git a/llm-modules/local-embedding/src/main/kotlin/cc/unitmesh/cf/LocalEmbedding.kt b/llm-modules/local-embedding/src/main/kotlin/cc/unitmesh/cf/LocalEmbedding.kt
@@ -68,7 +68,7 @@ open class LocalEmbedding(
 
     companion object {
         /**
-         * Create a new instance of [STEmbedding] with default model.
+         * Create a new instance of [LocalEmbedding] with default model.
          * We use official model: [all-MiniLM-L6-v2](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2)
          * We can use [optimum](https://github.com/huggingface/optimum) to transform the model to onnx.
          */

diff --git a/llm-modules/local-embedding/src/main/kotlin/cc/unitmesh/cf/STEmbedding.kt b/llm-modules/local-embedding/src/main/kotlin/cc/unitmesh/cf/STEmbedding.kt
diff --git a/server/src/test/kotlin/RagIntegrationTests.kt b/server/src/test/kotlin/RagIntegrationTests.kt
@@ -1,5 +1,4 @@
 import cc.unitmesh.cf.LocalEmbedding
-import cc.unitmesh.cf.STEmbedding
 import cc.unitmesh.cf.infrastructure.llms.embedding.SentenceTransformersEmbedding
 import cc.unitmesh.nlp.embedding.Embedding
 import cc.unitmesh.rag.document.Document