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()
|