diff --git a/app/services/llm_service.py b/app/services/llm_service.py index 6bfd73b..97f52e4 100644 --- a/app/services/llm_service.py +++ b/app/services/llm_service.py @@ -38,7 +38,7 @@ class OpenAIService(LLMService): try: import openai self.client = openai.AsyncOpenAI(api_key=settings.openai_api_key) - self.model = settings.openai_model + self.model = settings.OPENAI_MODEL except (ImportError, AttributeError) as e: logger.error(f"Failed to initialize OpenAI service: {e}") raise RuntimeError(f"OpenAI service initialization failed: {e}") @@ -98,7 +98,7 @@ class GeminiService(LLMService): try: import google.generativeai as genai genai.configure(api_key=settings.GEMINI_API_KEY) - self.model = genai.GenerativeModel(settings.gemini_model) + self.model = genai.GenerativeModel(settings.GEMINI_MODEL) except (ImportError, AttributeError) as e: logger.error(f"Failed to initialize Gemini service: {e}") raise RuntimeError(f"Gemini service initialization failed: {e}")