gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

cherry-pick d95efcb9

Browse source
This commit is contained in:
Raphaël MANSUY 2025-12-04 19:19:02 +08:00
parent 8d953340a8
commit b27845ca11
1 changed files with 12 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
lightrag/api/lightrag_server.py
View file

@ -605,7 +605,7 @@ def create_app(args):
Uses lazy imports for all bindings and avoids repeated configuration parsing. Uses lazy imports for all bindings and avoids repeated configuration parsing.
""" """
async def optimized_embedding_function(texts): async def optimized_embedding_function(texts, embedding_dim=None):
try: try:
if binding == "lollms": if binding == "lollms":
from lightrag.llm.lollms import lollms_embed from lightrag.llm.lollms import lollms_embed
@ -643,12 +643,21 @@ def create_app(args):
elif binding == "jina": elif binding == "jina":
from lightrag.llm.jina import jina_embed from lightrag.llm.jina import jina_embed
return await jina_embed(texts, base_url=host, api_key=api_key) return await jina_embed(
texts,
embedding_dim=embedding_dim,
base_url=host,
api_key=api_key,
)
else: # openai and compatible else: # openai and compatible
from lightrag.llm.openai import openai_embed from lightrag.llm.openai import openai_embed
return await openai_embed( return await openai_embed(
texts, model=model, base_url=host, api_key=api_key texts,
model=model,
base_url=host,
api_key=api_key,
embedding_dim=embedding_dim,
) )
except ImportError as e: except ImportError as e:
raise Exception(f"Failed to import {binding} embedding: {e}") raise Exception(f"Failed to import {binding} embedding: {e}")