Update services/llama_generator.py

services/llama_generator.py

@@ -80,6 +80,11 @@ class LlamaGenerator(BaseGenerator):
             self.model_config
         )
         
+        # Assign llama model to self.model
+        self.model = self.model_manager.models.get("llama")
+        if not self.model:
+            raise ValueError(f"Failed to load model: {llama_model_name}")
+            
         self.prompt_builder = LlamaPromptTemplate()
         self._init_strategies()