Update private_gpt/components/llm/llm_component.py

private_gpt/components/llm/llm_component.py

@@ -21,11 +21,16 @@ class LLMComponent:
 
     @inject
     def __init__(self, settings: Settings) -> None:
-        llm_mode = settings.llm.mode
+        llm_mode = "local"
         logger.info("Initializing the LLM in mode=%s", llm_mode)
         
+        allowed_modes = ["local", "openai", "sagemaker", "mock"]
+
+        if llm_mode not in allowed_modes:
+            raise ValueError(f"Invalid LLM mode: {llm_mode}")
+        
         
-        match settings.llm.mode:
+        match llm_mode:
             case "local":
                 from llama_index.llms import LlamaCPP
                 prompt_style_cls = get_prompt_style(settings.local.prompt_style)