Update src/llm_boilers.py

src/llm_boilers.py

@@ -14,6 +14,8 @@ class llm_boiler:
     def __init__(self, model_id, openai_key):
         self.model_id = model_id
         self.openai_key = openai_key
+        self.load_fn = None  # Add the load_fn attribute
+        self.run_fn = None  # Add the run_fn attribute
         for f_idx, run_function in enumerate(MODEL_FUNCTIONS):
             if run_function.__name__.lower() in self.model_id:
                 print(
@@ -26,6 +28,8 @@ class llm_boiler:
                     f"Run function recognized for {self.model_id}: {run_function.__name__.lower()}"
                 )
                 self.run_fn = run_function
+        if self.load_fn is None or self.run_fn is None:
+            raise ValueError("Invalid model_id")
         self.model = self.load_fn(self.model_id, self.openai_key)
         self.name = self.run_fn.__name__.lower()
 
@@ -41,6 +45,7 @@ class llm_boiler:
         )
 
 
+
 LOAD_MODEL_FUNCTIONS = []
 MODEL_FUNCTIONS = []