Ibraaheem commited on
Commit
7344079
·
1 Parent(s): 0436e19

Update private_gpt/components/llm/llm_component.py

Browse files
private_gpt/components/llm/llm_component.py CHANGED
@@ -8,6 +8,8 @@ from private_gpt.components.llm.prompt_helper import get_prompt_style
8
  from private_gpt.paths import models_path
9
  from private_gpt.settings.settings import Settings
10
 
 
 
11
  logger = logging.getLogger(__name__)
12
 
13
 
@@ -53,7 +55,9 @@ class LLMComponent:
53
  case "openai":
54
  from llama_index.llms import OpenAI
55
 
56
- openai_settings = settings.openai.api_key
57
  self.llm = OpenAI(model="gpt-4",api_key=openai_settings)
58
  case "mock":
59
  self.llm = MockLLM()
 
 
 
8
  from private_gpt.paths import models_path
9
  from private_gpt.settings.settings import Settings
10
 
11
+ import os
12
+
13
  logger = logging.getLogger(__name__)
14
 
15
 
 
55
  case "openai":
56
  from llama_index.llms import OpenAI
57
 
58
+ openai_settings = os.environ.get("OPENAI_API_KEY")
59
  self.llm = OpenAI(model="gpt-4",api_key=openai_settings)
60
  case "mock":
61
  self.llm = MockLLM()
62
+
63
+