Update private_gpt/components/llm/llm_component.py

private_gpt/components/llm/llm_component.py

@@ -10,6 +10,7 @@ from private_gpt.components.llm.prompt_helper import get_prompt_style
 from private_gpt.paths import models_path
 from private_gpt.settings.settings import Settings
 
+import os
 logger = logging.getLogger(__name__)
 
 @singleton
@@ -51,7 +52,7 @@ class LLMComponent:
 
                 #default startup
                 logger.info("Initializing the GPT Model in=%s", "gpt-3.5-turbo")
-                self.llm = OpenAI(model="gpt-3.5-turbo", api_key=openai_settings)
+                self.llm = OpenAI(model="gpt-3.5-turbo", api_key=os.environ.get("OPENAI_API_KEY"))
   
             case "mock":
                 self.llm = MockLLM()
@@ -61,10 +62,10 @@ class LLMComponent:
         openai_settings = settings.openai.api_key
         if type(self.llm) == OpenAI:
             if new_model == "gpt-3.5-turbo":
-                self.llm = OpenAI(model="gpt-3.5-turbo", api_key=openai_settings)
+                self.llm = OpenAI(model="gpt-3.5-turbo", api_key=os.environ.get("OPENAI_API_KEY"))
             elif new_model == "gpt-4":
                 # Initialize with the new model 
-                self.llm = OpenAI(model="gpt-4", api_key=openai_settings)
+                self.llm = OpenAI(model="gpt-4", api_key=os.environ.get("OPENAI_API_KEY"))
                 logger.info("Initializing the GPT Model in=%s", "gpt-4")