feat: add LlamaCppEmbeddings to model exports

- Imported LlamaCppEmbeddings in models/__init__.py. - Added LlamaCppEmbeddings to the __all__ exports in both models/__init__.py and embeddings/__init__.py. - Removed duplicate entry of LlamaCppEmbeddings from embeddings/__init__.py exports.
argilla-io · Jan 10, 2025 · a65a268 · a65a268
1 parent ab5e3e7
commit a65a268
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/distilabel/models/__init__.py b/src/distilabel/models/__init__.py
@@ -14,6 +14,7 @@
 
 
 from distilabel.models.embeddings.base import Embeddings
+from distilabel.models.embeddings.llamacpp import LlamaCppEmbeddings
 from distilabel.models.embeddings.sentence_transformers import (
     SentenceTransformerEmbeddings,
 )
@@ -53,6 +54,7 @@
     "HiddenState",
     "InferenceEndpointsLLM",
     "LiteLLM",
+    "LlamaCppEmbeddings",
     "LlamaCppLLM",
     "MistralLLM",
     "MixtureOfAgentsLLM",

diff --git a/src/distilabel/models/embeddings/__init__.py b/src/distilabel/models/embeddings/__init__.py
@@ -22,7 +22,6 @@
 __all__ = [
     "Embeddings",
     "LlamaCppEmbeddings",
-    "LlamaCppEmbeddings",
     "SentenceTransformerEmbeddings",
     "vLLMEmbeddings",
 ]