gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #375 from partoneplay/main

Browse Source 
Ollama `embedding` deprecated in favor of `embed`
This commit is contained in:
zrguo
2024-12-04 11:09:44 +08:00
committed by GitHub
parent 87b82746c4 18e86a1825
commit ae920fe43b
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
lightrag/llm.py
View File

@@ -745,6 +745,9 @@ async def hf_embedding(texts: list[str], tokenizer, embed_model) -> np.ndarray:
async def ollama_embedding(texts: list[str], embed_model, **kwargs) -> np.ndarray: async def ollama_embedding(texts: list[str], embed_model, **kwargs) -> np.ndarray:
"""
Deprecated in favor of `embed`.
"""
embed_text = [] embed_text = []
ollama_client = ollama.Client(**kwargs) ollama_client = ollama.Client(**kwargs)
for text in texts: for text in texts:
@@ -754,6 +757,12 @@ async def ollama_embedding(texts: list[str], embed_model, **kwargs) -> np.ndarra
return embed_text return embed_text
async def ollama_embed(texts: list[str], embed_model, **kwargs) -> np.ndarray:
ollama_client = ollama.Client(**kwargs)
data = ollama_client.embed(model=embed_model, input=texts)
return data["embeddings"]
class Model(BaseModel): class Model(BaseModel):
""" """
This is a Pydantic model class named 'Model' that is used to define a custom language model. This is a Pydantic model class named 'Model' that is used to define a custom language model.