From b12d7f8b630449b8641f3422b4efddb9f497a35c Mon Sep 17 00:00:00 2001 From: Ingrid Stevens Date: Tue, 12 Mar 2024 10:07:48 +0100 Subject: [PATCH] return None rather than raising an error --- private_gpt/ui/ui.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/private_gpt/ui/ui.py b/private_gpt/ui/ui.py index ee521b5b..b685c392 100644 --- a/private_gpt/ui/ui.py +++ b/private_gpt/ui/ui.py @@ -437,12 +437,11 @@ class PrivateGptUi: "ollama": config_settings.ollama.model, } - try: - return model_mapping[llm_mode] - except KeyError: - raise ValueError( - f"Invalid 'llm mode': {llm_mode}" - ) from None + if llm_mode not in model_mapping: + print(f"Invalid 'llm mode': {llm_mode}") + return None + + return model_mapping[llm_mode] with gr.Column(scale=7, elem_id="col"): # Determine the model label based on the value of PGPT_PROFILES