@@ -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