* fix(embedding.py): handle single string input in embed_documents …

…method
qdrant · Sep 27, 2023 · 589105c · 589105c
1 parent ac6b8c9
commit 589105c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/fastembed/embedding.py b/fastembed/embedding.py
@@ -267,6 +267,8 @@ def embed(self, documents: List[str], batch_size: int = 256) -> Iterable[np.ndar
         Returns:
             List of embeddings, one per document
         """
+        if type(documents) == str:
+            documents = [documents]
         # TODO: Replace loop with parallelized batching
         if len(documents) >= batch_size:
             for i in range(0, len(documents), batch_size):