Update utils/utils.py

utils/utils.py

@@ -4,11 +4,24 @@ from docx import Document
 from pptx import Presentation
 from datasets import load_dataset
 
-def get_questions_from_file(file_path, level):
+QUESTION_TYPE_GAIA      = "gaia"
+QUESTION_TYPE_HLE       = "hle"
+
+QUESTION_FILE_PATH_GAIA = "files/gaia_validation.jsonl"
+QUESTION_FILE_PATH_HLE  = "files/hle_validation.jsonl"
+
+def get_questions_from_file(question_type, level):
+    file_path = ""
+
+    if question_type == QUESTION_TYPE_GAIA:
+        file_path = QUESTION_FILE_PATH_GAIA
+    elif question_type == QUESTION_TYPE_HLE:
+        file_path = QUESTION_FILE_PATH_HLE
+    
     df = pd.read_json(file_path, lines=True)
   
     df = df[df["Level"] == level]
-    
+
     result=[]
     
     for _, row in df.iterrows():