Update private_gpt/components/llm/llm_component.py

private_gpt/components/llm/llm_component.py

@@ -77,12 +77,12 @@ class LLMComponent:
         
         if new_model == "gpt-3.5-turbo":
             self.llm = OpenAI(model="gpt-3.5-turbo", api_key=os.environ.get("OPENAI_API_KEY"))
-            logger.info("Initializing the GPT Model in=%s", "gpt-3.5-turbo")
+            logger.info("Initializing the LLM Model in=%s", "gpt-3.5-turbo")
             
         elif new_model == "gpt-4":
             # Initialize with the new model 
             self.llm = OpenAI(model="gpt-4", api_key=os.environ.get("OPENAI_API_KEY"))
-            logger.info("Initializing the GPT Model in=%s", "gpt-4")
+            logger.info("Initializing the LLM Model in=%s", "gpt-4")
 
         
         elif new_model == "mistral-7B":
@@ -102,6 +102,7 @@ class LLMComponent:
                 completion_to_prompt=prompt_style.completion_to_prompt,
                 verbose=True,
             )
+            logger.info("Initializing the LLM Model in=%s", "Mistral-7B")