diff --git a/app/services/llm_service.py b/app/services/llm_service.py index 022c1d0..6bfd73b 100644 --- a/app/services/llm_service.py +++ b/app/services/llm_service.py @@ -97,7 +97,7 @@ class GeminiService(LLMService): """Initialize the Gemini service.""" try: import google.generativeai as genai - genai.configure(api_key=settings.gemini_api_key) + genai.configure(api_key=settings.GEMINI_API_KEY) self.model = genai.GenerativeModel(settings.gemini_model) except (ImportError, AttributeError) as e: logger.error(f"Failed to initialize Gemini service: {e}") @@ -198,9 +198,9 @@ def get_llm_service() -> LLMService: """ llm_provider = settings.LLM_PROVIDER.lower() - if llm_provider == "openai" and settings.openai_api_key: + if llm_provider == "openai" and settings.OPEANAI_API_KEY: return OpenAIService() - elif llm_provider == "gemini" and settings.gemini_api_key: + elif llm_provider == "gemini" and settings.GEMINI_API_KEY: return GeminiService() elif llm_provider == "mock" or settings.environment == "test": # Use mock service for testing or when configured