fix: create_embedding broken response for input type str

abetlen · abetlen · commit 0ce66bc080fe · 2024-02-15T16:09:48.000-05:00
diff --git a/llama_cpp/llama.py b/llama_cpp/llama.py
@@ -720,6 +720,8 @@ def create_embedding(
         assert self._model.model is not None
         model_name: str = model if model is not None else self.model_path
 
+        input = input if isinstance(input, list) else [input]
+
         # get numeric embeddings
         embeds: List[List[float]]
         total_tokens: int
