From a624a9508af9a2af164b11ceec397d2f64f4e0b9 Mon Sep 17 00:00:00 2001 From: yangdx Date: Sat, 8 Nov 2025 03:54:50 +0800 Subject: [PATCH] Add Gemini to APIs requiring embedding dimension parameter --- lightrag/api/lightrag_server.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/lightrag/api/lightrag_server.py b/lightrag/api/lightrag_server.py index 7a291018..ded70d67 100644 --- a/lightrag/api/lightrag_server.py +++ b/lightrag/api/lightrag_server.py @@ -757,12 +757,12 @@ def create_app(args): has_embedding_dim_param = "embedding_dim" in sig.parameters # 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 - if args.embedding_binding == "jina": - # Jina API requires dimension parameter - always send it + if args.embedding_binding in ["jina", "gemini"]: + # Jina and Gemini APIs require dimension parameter - always send it send_dimensions = has_embedding_dim_param - dimension_control = "forced by Jina API" + dimension_control = f"forced by {args.embedding_binding.title()} API" else: # For OpenAI and other bindings, respect EMBEDDING_SEND_DIM setting send_dimensions = embedding_send_dim and has_embedding_dim_param