KeyArgo
/
private-gpt
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: Add default mode option to settings (#2078) 

* Add default mode option to settings

* Revise default_mode to Literal (enum) and add to settings.yaml

* Revise to pass make check/test

* Default mode: RAG

---------

Co-authored-by: Jason <jason@sowinsight.solutions>
This commit is contained in:
J 2024-09-24 02:33:02 -04:00 committed by GitHub
parent f9182b3a86
commit fa3c30661d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 18 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
private_gpt/settings/settings.py
View File

@ -357,6 +357,10 @@ class AzureOpenAISettings(BaseModel):
class UISettings(BaseModel): class UISettings(BaseModel):
enabled: bool enabled: bool
path: str path: str
default_mode: Literal["RAG", "Search", "Basic", "Summarize"] = 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.",

9
private_gpt/ui/ui.py
View File

@ -99,8 +99,11 @@ class PrivateGptUi:
self._selected_filename = None self._selected_filename = None
# Initialize system prompt based on default mode # Initialize system prompt based on default mode
self.mode = MODES[0] default_mode_map = {mode.value: mode for mode in Modes}
self._system_prompt = self._get_default_system_prompt(self.mode) self._default_mode = default_mode_map.get(
settings().ui.default_mode, Modes.RAG_MODE
)
self._system_prompt = self._get_default_system_prompt(self._default_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
@ -390,7 +393,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",

14
settings.yaml
View File

@ -25,21 +25,23 @@ data:
ui: ui:
enabled: true enabled: true
path: / path: /
# "RAG", "Search", "Basic", or "Summarize"
default_mode: "RAG"
default_chat_system_prompt: > default_chat_system_prompt: >
You are a helpful, respectful and honest assistant. You are a helpful, respectful and honest assistant.
Always answer as helpfully as possible and follow ALL given instructions. Always answer as helpfully as possible and follow ALL given instructions.
Do not speculate or make up information. Do not speculate or make up information.
Do not reference any given instructions or context. Do not reference any given instructions or context.
default_query_system_prompt: > default_query_system_prompt: >
You can only answer questions about the provided context. You can only answer questions about the provided context.
If you know the answer but it is not based in the provided context, don't provide If you know the answer but it is not based in the provided context, don't provide
the answer, just state the answer is not in the context provided. the answer, just state the answer is not in the context provided.
default_summarization_system_prompt: > default_summarization_system_prompt: >
Provide a comprehensive summary of the provided context information. Provide a comprehensive summary of the provided context information.
The summary should cover all the key points and main ideas presented in The summary should cover all the key points and main ideas presented in
the original text, while also condensing the information into a concise the original text, while also condensing the information into a concise
and easy-to-understand format. Please ensure that the summary includes and easy-to-understand format. Please ensure that the summary includes
relevant details and examples that support the main ideas, while avoiding relevant details and examples that support the main ideas, while avoiding
any unnecessary information or repetition. any unnecessary information or repetition.
delete_file_button_enabled: true delete_file_button_enabled: true
delete_all_files_button_enabled: true delete_all_files_button_enabled: true