Upload folder using huggingface_hub (part 2)

src/lib/embeddings/client_fastembed.py

@@ -0,0 +1,24 @@
+from typing import List
+from src.adapters.interfaces.embeddings import EmbeddingsClient
+
+class FastEmbedEmbeddingsClient(EmbeddingsClient):
+    def __init__(self, model_name: str = "BAAI/bge-small-en-v1.5"):
+        self.model_name = model_name
+        self._model = None
+
+    @property
+    def model(self):
+        if self._model is None:
+            from fastembed import TextEmbedding
+            self._model = TextEmbedding(model_name=self.model_name)
+        return self._model
+
+    def embed_text(self, text: str) -> List[float]:
+        embeddings = list(self.model.embed([text]))
+        return embeddings[0].tolist()
+
+    def embed_texts(self, texts: List[str]) -> List[List[float]]:
+        embeddings = list(self.model.embed(texts))
+        return [e.tolist() for e in embeddings]
+
+_fastembed_client = FastEmbedEmbeddingsClient()