Update test_case_generator.py

test_case_generator.py

@@ -5,6 +5,7 @@ from langchain_openai import ChatOpenAI
 from langchain_core.prompts import PromptTemplate
 from langchain_community.document_loaders import PyPDFLoader
 from langchain_text_splitters import RecursiveCharacterTextSplitter
+from prompts import *
 
 class TestCaseGenerator:
     def __init__(self, api_key=None):
@@ -14,13 +15,11 @@ class TestCaseGenerator:
         
         # Predefined question types
         self.available_question_types = [
-            'cause_and_effect_reasoning', 
-            'temporal_reasoning', 
-            'object_affordance',
-            'adversarial_tasks',
-            'common_sense_reasoning',
-            'hallucination',
-            'sycophancy'
+            'hallucination', 
+            'conflicting_instructions', 
+            'cause_and_effect_reasoning',
+            'factually_incorrect_agreement_sycophancy',
+            'toxicity'
         ]
 
     def load_and_split_document(self, doc, chunk_size=1000, chunk_overlap=100):