From 589105c84c2bf76ac719c48b34910b6eae88ebd9 Mon Sep 17 00:00:00 2001
From: NirantK <nirant.bits@gmail.com>
Date: Wed, 27 Sep 2023 17:27:56 +0530
Subject: [PATCH] * fix(embedding.py): handle single string input in
 `embed_documents` method

---
 fastembed/embedding.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/fastembed/embedding.py b/fastembed/embedding.py
index c2d61a9d..a998067c 100644
--- 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):