From ce28f30ca68a2e0469f14beb1b4d65a4633d211e Mon Sep 17 00:00:00 2001 From: yangdx Date: Fri, 7 Nov 2025 21:23:59 +0800 Subject: [PATCH] Add embedding_dim parameter support to embedding functions MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit • Pass embedding_dim to jina_embed call • Pass embedding_dim to openai_embed call --- lightrag/api/lightrag_server.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/lightrag/api/lightrag_server.py b/lightrag/api/lightrag_server.py index f36d88df..60fae05d 100644 --- a/lightrag/api/lightrag_server.py +++ b/lightrag/api/lightrag_server.py @@ -605,7 +605,7 @@ def create_app(args): 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: if binding == "lollms": from lightrag.llm.lollms import lollms_embed @@ -643,12 +643,14 @@ def create_app(args): elif binding == "jina": 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 from lightrag.llm.openai import 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: raise Exception(f"Failed to import {binding} embedding: {e}")