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

Add Gemini to APIs requiring embedding dimension parameter

Browse source
This commit is contained in:
yangdx 2025-11-08 03:54:50 +08:00
parent de4ed73652
commit a624a9508a
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
lightrag/api/lightrag_server.py
View file

@ -757,12 +757,12 @@ def create_app(args):
has_embedding_dim_param = "embedding_dim" in sig.parameters has_embedding_dim_param = "embedding_dim" in sig.parameters
# Determine send_dimensions value based on binding type # Determine send_dimensions value based on binding type
# Jina REQUIRES dimension parameter (forced to True) # Jina and Gemini REQUIRE dimension parameter (forced to True)
# OpenAI and others: controlled by EMBEDDING_SEND_DIM environment variable # OpenAI and others: controlled by EMBEDDING_SEND_DIM environment variable
if args.embedding_binding == "jina": if args.embedding_binding in ["jina", "gemini"]:
# Jina API requires dimension parameter - always send it # Jina and Gemini APIs require dimension parameter - always send it
send_dimensions = has_embedding_dim_param send_dimensions = has_embedding_dim_param
dimension_control = "forced by Jina API" dimension_control = f"forced by {args.embedding_binding.title()} API"
else: else:
# For OpenAI and other bindings, respect EMBEDDING_SEND_DIM setting # For OpenAI and other bindings, respect EMBEDDING_SEND_DIM setting
send_dimensions = embedding_send_dim and has_embedding_dim_param send_dimensions = embedding_send_dim and has_embedding_dim_param