Fix #16

llama_cpp/llama.py

@@ -262,6 +262,17 @@ class Llama:
             },
         }
 
+    def embed(self, input: str) -> List[float]:
+        """Embed a string.
+
+        Args:
+            input: The utf-8 encoded string to embed.
+
+        Returns:
+            A list of embeddings
+        """
+        return list(map(float, self.create_embedding(input)["data"][0]["embedding"]))
+
     def _create_completion(
         self,
         prompt: str,