yujunjun/LightRAG
1
0
Fork 0
mirror of https://github.com/HKUDS/LightRAG.git synced 2025-09-02 12:54:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix ollama embedding func ruturn data type bugs

Browse Source
This commit is contained in:
yangdx 2025-04-21 00:01:25 +08:00
parent 21f5a3923e
commit 99522a088d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lightrag/llm/ollama.py
View File

@ -129,4 +129,4 @@ async def ollama_embed(texts: list[str], embed_model, **kwargs) -> np.ndarray:
kwargs["headers"] = headers kwargs["headers"] = headers
ollama_client = ollama.Client(**kwargs) ollama_client = ollama.Client(**kwargs)
data = ollama_client.embed(model=embed_model, input=texts) data = ollama_client.embed(model=embed_model, input=texts)
return data["embeddings"] return np.array(data["embeddings"])