github/privateGPT
1
0
Fork 1
mirror of https://github.com/imartinez/privateGPT.git synced 2025-04-29 03:53:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add default mode option to settings

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
private_gpt/settings/settings.py
View File

@ -350,6 +350,10 @@ class AzureOpenAISettings(BaseModel):
class UISettings(BaseModel): class UISettings(BaseModel):
enabled: bool enabled: bool
path: str path: str
default_mode: str = Field(
"RAG",
description="The default mode.",
)
default_chat_system_prompt: str = Field( default_chat_system_prompt: str = Field(
None, None,
description="The default system prompt to use for the chat mode.", description="The default system prompt to use for the chat mode.",

6
private_gpt/ui/ui.py
View File

@ -98,10 +98,12 @@ class PrivateGptUi:
self._selected_filename = None self._selected_filename = None
self._default_mode = settings().ui.default_mode
# Initialize system prompt based on 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) self._system_prompt = self._get_default_system_prompt(self.mode)
def _chat( def _chat(
self, message: str, history: list[list[str]], mode: Modes, *_: Any self, message: str, history: list[list[str]], mode: Modes, *_: Any
) -> Any: ) -> Any:
@ -390,7 +392,7 @@ class PrivateGptUi:
with gr.Row(equal_height=False): with gr.Row(equal_height=False):
with gr.Column(scale=3): with gr.Column(scale=3):
default_mode = MODES[0] default_mode = self._default_mode
mode = gr.Radio( mode = gr.Radio(
[mode.value for mode in MODES], [mode.value for mode in MODES],
label="Mode", label="Mode",