Update test_case_generator.py

test_case_generator.py

@@ -46,9 +46,11 @@ class TestCaseGenerator:
     def get_prompt_template(self, question_type):
         """Get the prompt template for the given question type."""
         prompts = {
+            "hallucination": hallucination,
+            "conflicting_instructions": conflicting_instructions,
             "cause_and_effect_reasoning": cause_and_effect_reasoning,
-            "temporal_reasoning": temporal_reasoning,
-            "object_affordance": object_affordance,
+            "factually_incorrect_agreement_sycophancy":factually_incorrect_agreement_sycophancy,
+            "toxicity":toxicity
             # Add other prompts as needed
         }
         return prompts.get(question_type, None)