github/privateGPT
1
0
Fork 1
mirror of https://github.com/imartinez/privateGPT.git synced 2025-06-01 11:46:14 +00:00
Code Issues Packages Projects Releases Wiki 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
Show Stats 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._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 = self._default_mode 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