import pandas as pd def get_questions(level): df = pd.read_json("gaia.jsonl", lines = True) level_df = df[df["Level"] == level] result = [] for index, row in level_df.iterrows(): result.append([row["Level"], row["Question"], row["file_name"], row["Final answer"]]) return result