Ibraaheem commited on
Commit
1e94414
·
1 Parent(s): 8093abc

Update private_gpt/components/llm/llm_component.py

Browse files
private_gpt/components/llm/llm_component.py CHANGED
@@ -77,6 +77,7 @@ class LLMComponent:
77
 
78
  if new_model == "gpt-3.5-turbo":
79
  self.llm = OpenAI(model="gpt-3.5-turbo", api_key=os.environ.get("OPENAI_API_KEY"))
 
80
 
81
  elif new_model == "gpt-4":
82
  # Initialize with the new model
 
77
 
78
  if new_model == "gpt-3.5-turbo":
79
  self.llm = OpenAI(model="gpt-3.5-turbo", api_key=os.environ.get("OPENAI_API_KEY"))
80
+ logger.info("Initializing the GPT Model in=%s", "gpt-3.5-turbo")
81
 
82
  elif new_model == "gpt-4":
83
  # Initialize with the new model