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

Add embedding_dim parameter support to embedding functions

Browse source 
• Pass embedding_dim to jina_embed call
• Pass embedding_dim to openai_embed call
This commit is contained in:
yangdx 2025-11-07 21:23:59 +08:00
parent c14f25b7f8
commit ce28f30ca6
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
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,14 @@ 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}")