github/privateGPT
1
0
Fork 1
mirror of https://github.com/imartinez/privateGPT.git synced 2025-09-05 01:01:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Allow setting OpenAI model in settings (#1386)

Browse Source 
feat(settings): Allow setting openai model to be used. Default to GPT 3.5
This commit is contained in:
3ly-13
2023-12-09 13:13:00 -06:00
committed by GitHub
parent a3ed14c58f
commit a072a40a7c
4 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
private_gpt/components/llm/llm_component.py
View File

@@ -50,7 +50,9 @@ class LLMComponent:
case "openai":
from llama_index.llms import OpenAI
openai_settings = settings.openai.api_key
self.llm = OpenAI(api_key=openai_settings)
openai_settings = settings.openai
self.llm = OpenAI(
api_key=openai_settings.api_key, model=openai_settings.model
)
case "mock":
self.llm = MockLLM()