mirror of
https://github.com/imartinez/privateGPT.git
synced 2025-04-28 03:32:18 +00:00
Add default mode option to settings
This commit is contained in:
parent
42628596b2
commit
a5e796fcea
@ -350,6 +350,10 @@ class AzureOpenAISettings(BaseModel):
|
||||
class UISettings(BaseModel):
|
||||
enabled: bool
|
||||
path: str
|
||||
default_mode: str = Field(
|
||||
"RAG",
|
||||
description="The default mode.",
|
||||
)
|
||||
default_chat_system_prompt: str = Field(
|
||||
None,
|
||||
description="The default system prompt to use for the chat mode.",
|
||||
|
@ -98,10 +98,12 @@ class PrivateGptUi:
|
||||
|
||||
self._selected_filename = None
|
||||
|
||||
self._default_mode = settings().ui.default_mode
|
||||
# Initialize system prompt based on default mode
|
||||
self.mode = MODES[0]
|
||||
self.mode = self._default_mode
|
||||
self._system_prompt = self._get_default_system_prompt(self.mode)
|
||||
|
||||
|
||||
def _chat(
|
||||
self, message: str, history: list[list[str]], mode: Modes, *_: Any
|
||||
) -> Any:
|
||||
@ -390,7 +392,7 @@ class PrivateGptUi:
|
||||
|
||||
with gr.Row(equal_height=False):
|
||||
with gr.Column(scale=3):
|
||||
default_mode = MODES[0]
|
||||
default_mode = self._default_mode
|
||||
mode = gr.Radio(
|
||||
[mode.value for mode in MODES],
|
||||
label="Mode",
|
||||
|
Loading…
Reference in New Issue
Block a user