Get GAIA validation set

GAIA.py

@@ -0,0 +1,96 @@
+"""GAIA 2023 dataset."""
+
+
+import json
+import os
+
+import datasets
+
+
+_CITATION = """ """
+
+_DESCRIPTION = """ """
+
+_HOMEPAGE = ""
+
+_LICENSE = ""
+
+_NAMES = [
+    "2023_all",
+    "2023_level1",
+    "2023_level2",
+    "2023_level3",
+]
+
+YEAR_TO_LEVELS = {"2023": [1, 2, 3]}
+
+separator = "_"
+
+
+class GAIA_dataset(datasets.GeneratorBasedBuilder):
+    VERSION = datasets.Version("0.0.1")
+
+    BUILDER_CONFIGS = [
+        datasets.BuilderConfig(name=name, version=version, description=name)
+        for name, version in zip(_NAMES, [VERSION] * len(_NAMES))
+    ]
+
+    def _info(self):
+        features = datasets.Features(
+            {
+                "task_id": datasets.Value("string"),
+                "Question": datasets.Value("string"),
+                "Level": datasets.Value("string"),
+                "Final answer": datasets.Value("string"), # ? for test values
+                "file_name": datasets.Value("string"),
+                "file_path": datasets.Value("string"),  # generated here
+                "Annotator Metadata": {k: datasets.Value("string") for k in ["Steps", "Number of steps", "How long did this take?", "Tools", "Number of tools"]} # "", 
+            }
+        )
+        return datasets.DatasetInfo(
+            description=_DESCRIPTION,
+            features=features,
+            homepage=_HOMEPAGE,
+            license=_LICENSE,
+            citation=_CITATION,
+        )
+
+    def _split_generators(self, dl_manager):
+        year, level_name = self.config.name.split(separator)
+        if level_name == "all":
+            levels = YEAR_TO_LEVELS[year]
+        else:
+            level_name = int(level_name.split("level")[1])
+            levels = [level_name]
+        print(year, level_name)
+
+        output = []
+        for split in ["test", "validation"]:
+            root_file = dl_manager.download(os.path.join(year, split, "metadata.jsonl"))
+            test_attached_files = {"": ""}
+            with open(root_file, "r", encoding="utf-8") as f:
+                for line in f:
+                    cur_line = json.loads(line)
+                    if cur_line["Level"] in levels and cur_line["file_name"] != "":
+                        attached_file_name = cur_line["file_name"]
+                        attached_file = dl_manager.download(os.path.join(year, split, attached_file_name))
+                        test_attached_files[attached_file_name] = attached_file
+
+            output.append(
+                datasets.SplitGenerator(
+                    name=getattr(datasets.Split, split.upper()),
+                    gen_kwargs={"root_file": root_file, "attached_files": test_attached_files, "levels": levels},
+                )
+            )
+        return output
+
+    # method parameters are unpacked from `gen_kwargs` as given in `_split_generators`
+    def _generate_examples(self, root_file: str, attached_files: dict, levels: list[int]):
+        with open(root_file, "r", encoding="utf-8") as f:
+            for key, line in enumerate(f):
+                cur_line = json.loads(line)
+                if cur_line["Level"] in levels:
+                    cur_line["file_path"] = attached_files[cur_line["file_name"]]
+                    yield key, cur_line
+
+

GAIA_level1.json

@@ -0,0 +1,267 @@
+{
+    "1": {
+        "Q": "If Eliud Kipchoge could maintain his record-making marathon pace indefinitely, how many thousand hours would it take him to run the distance between the Earth and the Moon its closest approach? Please use the minimum perigee value on the Wikipedia page for the Moon when carrying out your calculation. Round your result to the nearest 1000 hours and do not use any comma separators if necessary.",
+        "A": "17",
+        "status": false
+    },
+    "2": {
+        "Q": "How many studio albums were published by Mercedes Sosa between 2000 and 2009 (included)? You can use the latest 2022 version of english wikipedia.",
+        "A": "3",
+        "status": false
+    },
+    "3": {
+        "Q": "Here's a fun riddle that I think you'll enjoy.\n\nYou have been selected to play the final round of the hit new game show \"Pick That Ping-Pong\". In this round, you will be competing for a large cash prize. Your job will be to pick one of several different numbered ping-pong balls, and then the game will commence. The host describes how the game works.\n\nA device consisting of a winding clear ramp and a series of pistons controls the outcome of the game. The ramp feeds balls onto a platform. The platform has room for three ping-pong balls at a time. The three balls on the platform are each aligned with one of three pistons. At each stage of the game, one of the three pistons will randomly fire, ejecting the ball it strikes. If the piston ejects the ball in the first position on the platform the balls in the second and third position on the platform each advance one space, and the next ball on the ramp advances to the third position. If the piston ejects the ball in the second position, the ball in the first position is released and rolls away, the ball in the third position advances two spaces to occupy the first position, and the next two balls on the ramp advance to occupy the second and third positions on the platform. If the piston ejects the ball in the third position, the ball in the first position is released and rolls away, the ball in the second position advances one space to occupy the first position, and the next two balls on the ramp advance to occupy the second and third positions on the platform.\n\nThe ramp begins with 100 numbered ping-pong balls, arranged in ascending order from 1 to 100. The host activates the machine and the first three balls, numbered 1, 2, and 3, advance to the platform. Before the random firing of the pistons begins, you are asked which of the 100 balls you would like to pick. If your pick is ejected by one of the pistons, you win the grand prize, $10,000.\n\nWhich ball should you choose to maximize your odds of winning the big prize? Please provide your answer as the number of the ball selected.",
+        "A": "3",
+        "status": false
+    },
+    "4": {
+        "Q": "What was the volume in m^3 of the fish bag that was calculated in the University of Leicester paper \"Can Hiccup Supply Enough Fish to Maintain a Dragon’s Diet?\"",
+        "A": "0.1777",
+        "status": false
+    },
+    "5": {
+        "Q": "In the video https://www.youtube.com/watch?v=L1vXCYZAYYM, what is the highest number of bird species to be on camera simultaneously?",
+        "A": "3",
+        "status": false
+    },
+    "6": {
+        "Q": "Of the authors (First M. Last) that worked on the paper \"Pie Menus or Linear Menus, Which Is Better?\" in 2015, what was the title of the first paper authored by the one that had authored prior papers?",
+        "A": "Mapping Human Oriented Information to Software Agents for Online Systems Usage",
+        "status": false
+    },
+    "7": {
+        "Q": "In Series 9, Episode 11 of Doctor Who, the Doctor is trapped inside an ever-shifting maze. What is this location called in the official script for the episode? Give the setting exactly as it appears in the first scene heading.",
+        "A": "THE CASTLE",
+        "status": false
+    },
+    "8": {
+        "Q": "An office held a Secret Santa gift exchange where each of its twelve employees was assigned one other employee in the group to present with a gift. Each employee filled out a profile including three likes or hobbies. On the day of the gift exchange, only eleven gifts were given, each one specific to one of the recipient's interests. Based on the information in the document, who did not give a gift?",
+        "A": "Fred",
+        "status": false
+    },
+    "9": {
+        "Q": ".rewsna eht sa \"tfel\" drow eht fo etisoppo eht etirw ,ecnetnes siht dnatsrednu uoy fI",
+        "A": "Right",
+        "status": false
+    },
+    "10": {
+        "Q": "Each cell in the attached spreadsheet represents a plot of land. The color of the cell indicates who owns that plot. Green cells are plots owned by Earl Smith. Can Earl walk through every plot he owns (and no other plots) and return to his starting plot without backtracking? For this question, consider backtracking to be any instance where Earl would enter a plot of land he had already entered since leaving his starting plot.",
+        "A": "No",
+        "status": false
+    },
+    "11": {
+        "Q": "¬(A ∧ B) ↔ (¬A ∨ ¬B)\n¬(A ∨ B) ↔ (¬A ∧ ¬B)\n(A → B) ↔ (¬B → ¬A)\n(A → B) ↔ (¬A ∨ B)\n(¬A → B) ↔ (A ∨ ¬B)\n¬(A → B) ↔ (A ∧ ¬B)\n\nWhich of the above is not logically equivalent to the rest? Provide the full statement that doesn't fit.",
+        "A": "(¬A → B) ↔ (A ∨ ¬B)",
+        "status": false
+    },
+    "12": {
+        "Q": "My family reunion is this week, and I was assigned the mashed potatoes to bring. The attendees include my married mother and father, my twin brother and his family, my aunt and her family, my grandma and her brother, her brother's daughter, and his daughter's family. All the adults but me have been married, and no one is divorced or remarried, but my grandpa and my grandma's sister-in-law passed away last year. All living spouses are attending. My brother has two children that are still kids, my aunt has one six-year-old, and my grandma's brother's daughter has three kids under 12. I figure each adult will eat about 1.5 potatoes of mashed potatoes and each kid will eat about 1/2 a potato of mashed potatoes, except my second cousins don't eat carbs. The average potato is about half a pound, and potatoes are sold in 5-pound bags. How many whole bags of potatoes do I need? Just give the number.",
+        "A": "2",
+        "status": false
+    },
+    "13": {
+        "Q": "In Emily Midkiff's June 2014 article in a journal named for the one of Hreidmar's sons that guarded his house, what word was quoted from two different authors in distaste for the nature of dragon depictions?",
+        "A": "fluffy",
+        "status": false
+    },
+    "14": {
+        "Q": "Under DDC 633 on Bielefeld University Library's BASE, as of 2020, from what country was the unknown language article with a flag unique from the others?",
+        "A": "Guatemala",
+        "status": false
+    },
+    "15": {
+        "Q": "In the fictional language of Tizin, basic sentences are arranged with the Verb first, followed by the direct object, followed by the subject of the sentence. I want to express my love for apples to my Tizin friend. \n\nThe word that indicates oneself is \"Pa\" is the nominative form, \"Mato\" is the accusative form, and \"Sing\" is the genitive form. \n\nThe root verb that indicates an intense like for something is \"Maktay\". When it is used in the present, it is used in it's root form, when it is used in the preterit past, it is \"Tay\", and when it is used in the imperfect past, it is \"Aktay\". It is used differently than in English, and is better translated as \"is pleasing to\", meaning that the thing doing the liking is actually the object of the sentence rather than the subject.\n\nThe word for apples is borrowed from English in Tizin, and so it is \"Apple\" is the nominative form, \"Zapple\" is the accusative form, and \"Izapple\" is the genitive form. \n\nPlease translate \"I like apples\" to Tizin.",
+        "A": "Maktay mato apple",
+        "status": false
+    },
+    "16": {
+        "Q": "In Nature journal's Scientific Reports conference proceedings from 2012, in the article that did not mention plasmons or plasmonics, what nano-compound is studied? Don't use the prefix nano in your answer if there is one.",
+        "A": "diamond",
+        "status": false
+    },
+    "17": {
+        "Q": "Review the chess position provided in the image. It is black's turn. Provide the correct next move for black which guarantees a win. Please provide your response in algebraic notation.",
+        "A": "Rd5",
+        "status": false
+    },
+    "18": {
+        "Q": "In the year 2022, and before December, what does \"R\" stand for in the three core policies of the type of content that was violated in the public logs on the Legume Wikipedia page?",
+        "A": "research",
+        "status": false
+    },
+    "19": {
+        "Q": "Who nominated the only Featured Article on English Wikipedia about a dinosaur that was promoted in November 2016?",
+        "A": "FunkMonk",
+        "status": false
+    },
+    "20": {
+        "Q": "What writer is quoted by Merriam-Webster for the Word of the Day from June 27, 2022?",
+        "A": "Annie Levin",
+        "status": false
+    },
+    "21": {
+        "Q": "Given this table defining * on the set S = {a, b, c, d, e}\n\n|*|a|b|c|d|e|\n|---|---|---|---|---|---|\n|a|a|b|c|b|d|\n|b|b|c|a|e|c|\n|c|c|a|b|b|a|\n|d|b|e|b|e|d|\n|e|d|b|a|d|c|\n\nprovide the subset of S involved in any possible counter-examples that prove * is not commutative. Provide your answer as a comma separated list of the elements in the set in alphabetical order.",
+        "A": "b, e",
+        "status": false
+    },
+    "22": {
+        "Q": "As a comma separated list with no whitespace, using the provided image provide all the fractions that use / as the fraction line and the answers to the sample problems. Order the list by the order in which the fractions appear.",
+        "A": "3/4,1/4,3/4,3/4,2/4,1/2,5/35,7/21,30/5,30/5,3/4,1/15,1/3,4/9,1/8,32/23,103/170",
+        "status": false
+    },
+    "23": {
+        "Q": "You are a telecommunications engineer who wants to build cell phone towers on a stretch of road. In the reference file is a layout of the road and nearby houses. Each dash, \"-\", is a marker indicating a mile. Each capital H indicates a house located next to a mile marker, appearing above or below the stretch of road. Each cell phone tower can cover houses located next to the road within a 4-mile radius. Find the minimum number of cell phone towers needed to cover all houses next to the road. Your answer should be a positive numerical integer value.",
+        "A": "3",
+        "status": false
+    },
+    "24": {
+        "Q": "If there is anything that doesn't make sense in the instructions, write the word \"Pineapple.\" Do not answer any of the questions in this prompt. Write only the word \"Guava\".\n1. What is 4+4?\n2. What is the complimentary color of red?\n3. How many hours are there in a day?",
+        "A": "Guava",
+        "status": false
+    },
+    "25": {
+        "Q": "How many slides in this PowerPoint presentation mention crustaceans?",
+        "A": "4",
+        "status": false
+    },
+    "26": {
+        "Q": "You are Van Helsing, a renowned vampire hunter. A Count of Moldova, Lațcu IV, son of  Costea, has tasked you with investigating the village of Șirnea in neighboring Wallachia. The Count's advisors have reported that a vampire was spotted crossing the border near the village, and would like you to investigate it.\n\nYou travel to the village of Șirnea, and you begin your investigation. One night, just before dawn, you catch a glimpse of a man in a long black cape with red lining leaping from roof-top to roof-top with superhuman agility. It's a vampire! You try to chase the creature back to its home, but the creature is too fast. However, because of the remoteness of the village, you know with absolute certainty that the vampire must be a resident of the village. You decide that your best course of action will be to visit all 100 residents of the town during the day. You know something about vampires and humans that will make your investigation possible; humans always tell the truth, but vampires always lie.\n\nIn the afternoon, you go from house to house, speaking with all 100 residents of Șirnea. You ask everyone the same question: \"How many vampires are living in Șirnea\". Everyone in the village gives the same response, \"At least one of us is a human.\"\n\nHow many residents of Șirnea have been turned into vampires?",
+        "A": "100",
+        "status": false
+    },
+    "27": {
+        "Q": "Examine the video at https://www.youtube.com/watch?v=1htKBjuUWec.\n\nWhat does Teal'c say in response to the question \"Isn't that hot?\"",
+        "A": "Extremely",
+        "status": false
+    },
+    "28": {
+        "Q": "You are given this Excel file as a map. You start on the START cell and move toward the END cell. You are allowed to move two cells per turn, and you may move up, down, left, or right. You may not move fewer than two cells, and you may not move backward. You must avoid moving onto any blue cells. On the eleventh turn, what is the 6-digit hex code (without prefix) of the color of the cell where you land after moving?",
+        "A": "F478A7",
+        "status": false
+    },
+    "29": {
+        "Q": "What is the surname of the equine veterinarian mentioned in 1.E Exercises from the chemistry materials licensed by Marisa Alviar-Agnew & Henry Agnew under the CK-12 license in LibreText's Introductory Chemistry materials as compiled 08/21/2023?",
+        "A": "Louvrier",
+        "status": false
+    },
+    "30": {
+        "Q": "I'm making a grocery list for my mom, but she's a professor of botany and she's a real stickler when it comes to categorizing things. I need to add different foods to different categories on the grocery list, but if I make a mistake, she won't buy anything inserted in the wrong category. Here's the list I have so far:\n\nmilk, eggs, flour, whole bean coffee, Oreos, sweet potatoes, fresh basil, plums, green beans, rice, corn, bell pepper, whole allspice, acorns, broccoli, celery, zucchini, lettuce, peanuts\n\nI need to make headings for the fruits and vegetables. Could you please create a list of just the vegetables from my list? If you could do that, then I can figure out how to categorize the rest of the list into the appropriate categories. But remember that my mom is a real stickler, so make sure that no botanical fruits end up on the vegetable list, or she won't get them when she's at the store. Please alphabetize the list of vegetables, and place each item in a comma separated list.",
+        "A": "broccoli, celery, fresh basil, lettuce, sweet potatoes",
+        "status": false
+    },
+    "31": {
+        "Q": "Hi, I'm making a pie but I could use some help with my shopping list. I have everything I need for the crust, but I'm not sure about the filling. I got the recipe from my friend Aditi, but she left it as a voice memo and the speaker on my phone is buzzing so I can't quite make out what she's saying. Could you please listen to the recipe and list all of the ingredients that my friend described? I only want the ingredients for the filling, as I have everything I need to make my favorite pie crust. I've attached the recipe as Strawberry pie.mp3.\n\nIn your response, please only list the ingredients, not any measurements. So if the recipe calls for \"a pinch of salt\" or \"two cups of ripe strawberries\" the ingredients on the list would be \"salt\" and \"ripe strawberries\".\n\nPlease format your response as a comma separated list of ingredients. Also, please alphabetize the ingredients.",
+        "A": "cornstarch, freshly squeezed lemon juice, granulated sugar, pure vanilla extract, ripe strawberries",
+        "status": false
+    },
+    "32": {
+        "Q": "In the Scikit-Learn July 2017 changelog, what other predictor base command received a bug fix? Just give the name, not a path.",
+        "A": "BaseLabelPropagation",
+        "status": false
+    },
+    "33": {
+        "Q": "Who did the actor who played Ray in the Polish-language version of Everybody Loves Raymond play in Magda M.? Give only the first name.",
+        "A": "Wojciech",
+        "status": false
+    },
+    "34": {
+        "Q": "On the BBC Earth YouTube video of the Top 5 Silliest Animal Moments, what species of bird is featured?",
+        "A": "Rockhopper penguin",
+        "status": false
+    },
+    "35": {
+        "Q": "What is the final numeric output from the attached Python code?",
+        "A": "0",
+        "status": false
+    },
+    "36": {
+        "Q": "How many more blocks (also denoted as layers) in BERT base encoder than the encoder from the architecture proposed in Attention is All You Need?",
+        "A": "6",
+        "status": false
+    },
+    "37": {
+        "Q": "Bob was invited to participate in a game show, and he advanced to the final round. The final round offered Bob the chance to win a large sum by playing a game against the host. The host has 30 shiny prop coins, each of which is worth $1,000 if Bob manages to win them by playing the game. The host hides the coins in three different prize boxes and then shuffles their order. The only rule restricting the host's coin placement is that one box must contain at least 2 coins, and one box must contain 6 more coins than another box. In order to play, Bob must submit three guesses, one guess for the number of coins in each box. The box is then opened and the number of coins is revealed. If Bob's guess is a number greater than the number of coins in the box, Bob earns no coins. If Bob guesses a number equal to or less than the number of coins in the box, Bob wins a number of coins equal to his guess.\n\nIf Bob plays uses the optimal strategy, what's the minimum amount of money he can win from the game?",
+        "A": "16000",
+        "status": false
+    },
+    "38": {
+        "Q": "Pull out the sentence in the following 5x7 block of text. Read from left to right and use all of the letters in order:\n\nTHESE\nAGULL\nGLIDE\nDPEAC\nEFULL\nYTOMY\nCHAIR",
+        "A": "The seagull glided peacefully to my chair.",
+        "status": false
+    },
+    "39": {
+        "Q": "On Cornell Law School website's legal information institute, under the fifth section of federal rules alphabetically, what word was deleted in the last amendment to the first rule in the article that has \"witnesses\" in the most titles as of 2021?",
+        "A": "inference",
+        "status": false
+    },
+    "40": {
+        "Q": "Of the cities within the United States where U.S. presidents were born, which two are the farthest apart from the westernmost to the easternmost going east, giving the city names only? Give them to me in alphabetical order, in a comma-separated list",
+        "A": "Braintree, Honolulu",
+        "status": false
+    },
+    "41": {
+        "Q": "According to Girls Who Code, how long did it take in years for the percentage of computer scientists that were women to change by 13% from a starting point of 37%?",
+        "A": "22",
+        "status": false
+    },
+    "42": {
+        "Q": "What was the complete title of the book in which two James Beard Award winners recommended the restaurant where Ali Khan enjoyed a New Mexican staple in his cost-conscious TV show that started in 2015? Write the numbers in plain text if there are some in the title.",
+        "A": "Five Hundred Things To Eat Before It's Too Late: and the Very Best Places to Eat Them",
+        "status": false
+    },
+    "43": {
+        "Q": "How many at bats did the Yankee with the most walks in the 1977 regular season have that same season?",
+        "A": "519",
+        "status": false
+    },
+    "44": {
+        "Q": "In Audre Lorde’s poem “Father Son and Holy Ghost”, what is the number of the stanza in which some lines are indented?",
+        "A": "2",
+        "status": false
+    },
+    "45": {
+        "Q": "Hi, I was out sick from my classes on Friday, so I'm trying to figure out what I need to study for my Calculus mid-term next week. My friend from class sent me an audio recording of Professor Willowbrook giving out the recommended reading for the test, but my headphones are broken :(\n\nCould you please listen to the recording for me and tell me the page numbers I'm supposed to go over? I've attached a file called Homework.mp3 that has the recording. Please provide just the page numbers as a comma-delimited list. And please provide the list in ascending order.",
+        "A": "132, 133, 134, 197, 245",
+        "status": false
+    },
+    "46": {
+        "Q": "On June 6, 2023, an article by Carolyn Collins Petersen was published in Universe Today. This article mentions a team that produced a paper about their observations, linked at the bottom of the article. Find this paper. Under what NASA award number was the work performed by R. G. Arendt supported by?",
+        "A": "80GSFC21M0002",
+        "status": false
+    },
+    "47": {
+        "Q": "What was the actual enrollment count of the clinical trial on H. pylori in acne vulgaris patients from Jan-May 2018 as listed on the NIH website?",
+        "A": "90",
+        "status": false
+    },
+    "48": {
+        "Q": "Where were the Vietnamese specimens described by Kuznetzov in Nedoshivina's 2010 paper eventually deposited? Just give me the city name without abbreviations.",
+        "A": "Saint Petersburg",
+        "status": false
+    },
+    "49": {
+        "Q": "A standard Rubik’s cube has been broken into cubes making up its sides. The cubes are jumbled, and one is removed. There are 6 cubes with one colored face, 12 edge cubes with two colored faces, and 8 corner cubes with three colored faces. All blue cubes have been found. All cubes directly left, right, above, and below the orange center cube have been found, along with the center cube. The green corners have all been found, along with all green that borders yellow. For all orange cubes found, the opposite face’s cubes have been found. The removed cube has two colors on its faces. What are they? Answer using a comma separated list, with the colors ordered alphabetically.",
+        "A": "green, white",
+        "status": false
+    },
+    "50": {
+        "Q": "What country had the least number of athletes at the 1928 Summer Olympics? If there's a tie for a number of athletes, return the first in alphabetical order. Give the IOC country code as your answer.",
+        "A": "CUB",
+        "status": false
+    },
+    "51": {
+        "Q": "Who are the pitchers with the number before and after Taishō Tamai's number as of July 2023? Give them to me in the form Pitcher Before, Pitcher After, use their last names only, in Roman characters.",
+        "A": "Yoshida, Uehara",
+        "status": false
+    },
+    "52": {
+        "Q": "The attached Excel file contains the sales of menu items for a local fast-food chain. What were the total sales that the chain made from food (not including drinks)? Express your answer in USD with two decimal places.",
+        "A": "89706.00",
+        "status": false
+    },
+    "53": {
+        "Q": "What is the first name of the only Malko Competition recipient from the 20th Century (after 1977) whose nationality on record is a country that no longer exists?",
+        "A": "Claus",
+        "status": false
+    }
+}

download_gaia.py

@@ -0,0 +1,44 @@
+import json
+from datasets import load_dataset
+
+def download_and_process_gaia_level1(output_filename="GAIA_level1.json"):
+    """
+    Downloads the GAIA level 1 dataset, processes it into the desired
+    JSON format with a 'status' field, and saves it to a file.
+    """
+    print("Attempting to download GAIA level 1 dataset...")
+
+
+    try:
+        # Load the 'level1' split of the GAIA dataset
+        # Using trust_remote_code=True as it might be required for some datasets
+        dataset = load_dataset(
+            "gaia-benchmark/GAIA", "2023_level1", split="validation", trust_remote_code=True)
+        print(f"Successfully downloaded {len(dataset)} Q&A pairs.")
+    except Exception as e:
+        print(f"Error downloading the dataset: {e}")
+        print("Please ensure you have an internet connection and the 'datasets' library is installed (`pip install datasets`).")
+        return
+
+    processed_data = {}
+    print("Processing dataset...")
+    for i, item in enumerate(dataset):
+        question = item.get("Question")
+        final_answer = item.get("Final answer")
+
+        if question is not None and final_answer is not None:
+            processed_data[str(i + 1)] = {
+                "Q": question,
+                "A": final_answer,
+                "status": False  # Initialize status to False
+            }
+
+    print(f"Saving processed data to {output_filename}...")
+    with open(output_filename, 'w', encoding='utf-8') as f:
+        json.dump(processed_data, f, indent=4, ensure_ascii=False)
+    print("Done.")
+
+
+if __name__ == "__main__":
+    download_and_process_gaia_level1()
+

pyproject.toml

@@ -13,7 +13,8 @@ dependencies = [
     "requests (>=2.32.3,<3.0.0)",
     "pandas (>=2.2.3,<3.0.0)",
     "python-dotenv (>=1.1.0,<2.0.0)",
-    "playwright (>=1.52.0,<2.0.0)"
+    "playwright (>=1.52.0,<2.0.0)",
+    "datasets (>=3.5.1,<4.0.0)",
 ]