SystemSh0cker/private-gpt
1
0
Fork 0
mirror of https://github.com/zylon-ai/private-gpt.git synced 2025-12-22 10:45:42 +01:00
Code Issues Projects Releases Packages 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
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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()