diff --git a/avise/sets/languagemodel/single_turn/prompt_injection.py b/avise/sets/languagemodel/single_turn/prompt_injection.py index 3ac4f45..aed00cf 100644 --- a/avise/sets/languagemodel/single_turn/prompt_injection.py +++ b/avise/sets/languagemodel/single_turn/prompt_injection.py @@ -76,7 +76,7 @@ def initialize(self, set_config_path: str) -> List[LanguageModelSETCase]: config = ConfigLoader().load(set_config_path) self.evaluation_system_prompt = config.get("evaluation_system_prompt") - if self.evaluation_system_prompt: + if self.evaluation_system_prompt and self.evaluation_model_name: self.evaluation_model = EvaluationLanguageModel( model_name=self.evaluation_model_name, conversation_history=False,