Fix ollama embedding func ruturn data type bugs

2025-04-21 00:01:25 +08:00
parent 21f5a3923e
commit 99522a088d
1 changed files with 1 additions and 1 deletions
--- a/lightrag/llm/ollama.py
+++ b/lightrag/llm/ollama.py
@@ -129,4 +129,4 @@ async def ollama_embed(texts: list[str], embed_model, **kwargs) -> np.ndarray:
    kwargs["headers"] = headers
    ollama_client = ollama.Client(**kwargs)
    data = ollama_client.embed(model=embed_model, input=texts)
-    return data["embeddings"]
+    return np.array(data["embeddings"])