Spaces:
Sleeping
Sleeping
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 =
|
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 |
+
|