Spaces:
Sleeping
Sleeping
Update private_gpt/components/llm/llm_component.py
Browse files
private_gpt/components/llm/llm_component.py
CHANGED
@@ -51,8 +51,8 @@ class LLMComponent:
|
|
51 |
openai_settings = settings.openai.api_key
|
52 |
|
53 |
#default startup
|
54 |
-
logger.info("Initializing the GPT Model in=%s", "gpt-
|
55 |
-
self.llm = OpenAI(model="gpt-
|
56 |
|
57 |
case "mock":
|
58 |
self.llm = MockLLM()
|
|
|
51 |
openai_settings = settings.openai.api_key
|
52 |
|
53 |
#default startup
|
54 |
+
logger.info("Initializing the GPT Model in=%s", "gpt-3.5-turbo")
|
55 |
+
self.llm = OpenAI(model="gpt-3.5-turbo", api_key=os.environ.get("OPENAI_API_KEY"))
|
56 |
|
57 |
case "mock":
|
58 |
self.llm = MockLLM()
|