Skip to content

Commit 7ebdf00

Browse files
committed
fix gemini api
1 parent 6cbbdfb commit 7ebdf00

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

backend/app/main.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -754,6 +754,10 @@ def graphrag_setup_endpoint():
754754
api_key = api_keys.get("geminiKey") or config_manager.get("ai_providers.google_genai.api_key", "")
755755
graphrag_api_keys = {"geminiKey": api_key}
756756
provider = "gemini" # Map to GraphRAG service provider name
757+
elif provider == "gemini":
758+
# Support direct "gemini" provider selection from frontend
759+
api_key = api_keys.get("geminiKey") or config_manager.get("ai_providers.google_genai.api_key", "")
760+
graphrag_api_keys = {"geminiKey": api_key}
757761
elif provider == "anthropic":
758762
api_key = api_keys.get("anthropicKey") or config_manager.get("ai_providers.anthropic.api_key", "")
759763
graphrag_api_keys = {"anthropicKey": api_key}
@@ -881,6 +885,10 @@ def graphrag_change_provider_endpoint():
881885
api_key = api_keys.get("geminiKey") or config_manager.get("ai_providers.google_genai.api_key", "")
882886
graphrag_api_keys = {"geminiKey": api_key}
883887
provider = "gemini" # Map to GraphRAG service provider name
888+
elif provider == "gemini":
889+
# Support direct "gemini" provider selection from frontend
890+
api_key = api_keys.get("geminiKey") or config_manager.get("ai_providers.google_genai.api_key", "")
891+
graphrag_api_keys = {"geminiKey": api_key}
884892
elif provider == "anthropic":
885893
api_key = api_keys.get("anthropicKey") or config_manager.get("ai_providers.anthropic.api_key", "")
886894
graphrag_api_keys = {"anthropicKey": api_key}

0 commit comments

Comments
 (0)