From 99522a088d2bc443a1a357674bbebec5a868927e Mon Sep 17 00:00:00 2001 From: yangdx Date: Mon, 21 Apr 2025 00:01:25 +0800 Subject: [PATCH] Fix ollama embedding func ruturn data type bugs --- lightrag/llm/ollama.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lightrag/llm/ollama.py b/lightrag/llm/ollama.py index e6fd5da7..21ae9a67 100644 --- 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"])