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

Revise to pass make check/test

Browse source
This commit is contained in:
Jason 2024-09-10 12:36:49 -04:00
parent 3acff52e4c
commit fbf9d0dd03
1 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
private_gpt/ui/ui.py
View file

@ -98,11 +98,12 @@ class PrivateGptUi:
self._selected_filename = None
self._default_mode = settings().ui.default_mode
# Initialize system prompt based on default mode
self.mode = self._default_mode
self._system_prompt = self._get_default_system_prompt(self.mode)
default_mode_map = {mode.value: mode for mode in Modes}
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(
self, message: str, history: list[list[str]], mode: Modes, *_: Any