Skip to content

Commit fbf9d0d

Browse files
committed
Revise to pass make check/test
1 parent 3acff52 commit fbf9d0d

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

private_gpt/ui/ui.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -98,11 +98,12 @@ def __init__(
9898

9999
self._selected_filename = None
100100

101-
self._default_mode = settings().ui.default_mode
102101
# Initialize system prompt based on default mode
103-
self.mode = self._default_mode
104-
self._system_prompt = self._get_default_system_prompt(self.mode)
105-
102+
default_mode_map = {mode.value: mode for mode in Modes}
103+
self._default_mode = default_mode_map.get(
104+
settings().ui.default_mode, Modes.RAG_MODE
105+
)
106+
self._system_prompt = self._get_default_system_prompt(self._default_mode)
106107

107108
def _chat(
108109
self, message: str, history: list[list[str]], mode: Modes, *_: Any

0 commit comments

Comments
 (0)