Add default mode option to settings

This commit is contained in:
Jason 2024-09-09 22:25:26 -04:00
parent 42628596b2
commit a5e796fcea
2 changed files with 8 additions and 2 deletions

View File

@ -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.",

View File

@ -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",