Upload folder using huggingface_hub

benchmarking.ipynb

@@ -9,13 +9,19 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 34,
+   "execution_count": 1,
    "metadata": {},
    "outputs": [],
    "source": [
+    "import os\n",
     "import numpy as np\n",
-    "import json\n",
     "import pandas as pd\n",
+    "from transformers.agents import agent_types\n",
+    "from tqdm.notebook import tqdm\n",
+    "import logging\n",
+    "from semscore import EmbeddingModelWrapper\n",
+    "from statistics import mean\n",
+    "\n",
     "\n",
     "def display_text_df(df):\n",
     "    display(df.style.set_properties(**{'white-space': 'pre-wrap'}).set_table_styles(\n",
@@ -27,9 +33,23 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 2,
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Initializing Data...\n",
+      "Download: False\n",
+      "Loading data...\n",
+      "Raw Data loaded\n",
+      "Chroma DB already exists\n",
+      "Loading index...\n",
+      "Index loaded\n"
+     ]
+    }
+   ],
    "source": [
     "from data import get_data\n",
     "data = get_data(download=False)\n"
@@ -37,97 +57,160 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": 3,
    "metadata": {},
    "outputs": [
     {
      "data": {
+      "text/html": [
+       "<style type=\"text/css\">\n",
+       "#T_3b140 th {\n",
+       "  text-align: left;\n",
+       "}\n",
+       "#T_3b140 td {\n",
+       "  text-align: left;\n",
+       "}\n",
+       "#T_3b140_row0_col0, #T_3b140_row0_col1, #T_3b140_row0_col2, #T_3b140_row0_col3, #T_3b140_row1_col0, #T_3b140_row1_col1, #T_3b140_row1_col2, #T_3b140_row1_col3, #T_3b140_row2_col0, #T_3b140_row2_col1, #T_3b140_row2_col2, #T_3b140_row2_col3 {\n",
+       "  white-space: pre-wrap;\n",
+       "}\n",
+       "</style>\n",
+       "<table id=\"T_3b140\">\n",
+       "  <thead>\n",
+       "    <tr>\n",
+       "      <th id=\"T_3b140_level0_col0\" class=\"col_heading level0 col0\" >Title</th>\n",
+       "      <th id=\"T_3b140_level0_col1\" class=\"col_heading level0 col1\" >Context</th>\n",
+       "      <th id=\"T_3b140_level0_col2\" class=\"col_heading level0 col2\" >Question</th>\n",
+       "      <th id=\"T_3b140_level0_col3\" class=\"col_heading level0 col3\" >Answer</th>\n",
+       "    </tr>\n",
+       "  </thead>\n",
+       "  <tbody>\n",
+       "    <tr>\n",
+       "      <td id=\"T_3b140_row0_col0\" class=\"data row0 col0\" >University_of_Notre_Dame</td>\n",
+       "      <td id=\"T_3b140_row0_col1\" class=\"data row0 col1\" >Architecturally, the school has a Catholic character. Atop the Main Building's gold dome is a golden statue of the Virgin Mary. Immediately in front of the Main Building and facing it, is a copper statue of Christ with arms upraised with the legend \"Venite Ad Me Omnes\". Next to the Main Building is the Basilica of the Sacred Heart. Immediately behind the basilica is the Grotto, a Marian place of prayer and reflection. It is a replica of the grotto at Lourdes, France where the Virgin Mary reputedly appeared to Saint Bernadette Soubirous in 1858. At the end of the main drive (and in a direct line that connects through 3 statues and the Gold Dome), is a simple, modern stone statue of Mary.</td>\n",
+       "      <td id=\"T_3b140_row0_col2\" class=\"data row0 col2\" >To whom did the Virgin Mary allegedly appear in 1858 in Lourdes France?</td>\n",
+       "      <td id=\"T_3b140_row0_col3\" class=\"data row0 col3\" >Saint Bernadette Soubirous</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <td id=\"T_3b140_row1_col0\" class=\"data row1 col0\" >University_of_Notre_Dame</td>\n",
+       "      <td id=\"T_3b140_row1_col1\" class=\"data row1 col1\" >Architecturally, the school has a Catholic character. Atop the Main Building's gold dome is a golden statue of the Virgin Mary. Immediately in front of the Main Building and facing it, is a copper statue of Christ with arms upraised with the legend \"Venite Ad Me Omnes\". Next to the Main Building is the Basilica of the Sacred Heart. Immediately behind the basilica is the Grotto, a Marian place of prayer and reflection. It is a replica of the grotto at Lourdes, France where the Virgin Mary reputedly appeared to Saint Bernadette Soubirous in 1858. At the end of the main drive (and in a direct line that connects through 3 statues and the Gold Dome), is a simple, modern stone statue of Mary.</td>\n",
+       "      <td id=\"T_3b140_row1_col2\" class=\"data row1 col2\" >What is in front of the Notre Dame Main Building?</td>\n",
+       "      <td id=\"T_3b140_row1_col3\" class=\"data row1 col3\" >a copper statue of Christ</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <td id=\"T_3b140_row2_col0\" class=\"data row2 col0\" >University_of_Notre_Dame</td>\n",
+       "      <td id=\"T_3b140_row2_col1\" class=\"data row2 col1\" >Architecturally, the school has a Catholic character. Atop the Main Building's gold dome is a golden statue of the Virgin Mary. Immediately in front of the Main Building and facing it, is a copper statue of Christ with arms upraised with the legend \"Venite Ad Me Omnes\". Next to the Main Building is the Basilica of the Sacred Heart. Immediately behind the basilica is the Grotto, a Marian place of prayer and reflection. It is a replica of the grotto at Lourdes, France where the Virgin Mary reputedly appeared to Saint Bernadette Soubirous in 1858. At the end of the main drive (and in a direct line that connects through 3 statues and the Gold Dome), is a simple, modern stone statue of Mary.</td>\n",
+       "      <td id=\"T_3b140_row2_col2\" class=\"data row2 col2\" >The Basilica of the Sacred heart at Notre Dame is beside to which structure?</td>\n",
+       "      <td id=\"T_3b140_row2_col3\" class=\"data row2 col3\" >the Main Building</td>\n",
+       "    </tr>\n",
+       "  </tbody>\n",
+       "</table>\n"
+      ],
       "text/plain": [
-       "('To whom did the Virgin Mary allegedly appear in 1858 in Lourdes France?',\n",
-       " 'Saint Bernadette Soubirous')"
+       "<pandas.io.formats.style.Styler at 0x16b7441d0>"
       ]
      },
-     "execution_count": 6,
      "metadata": {},
-     "output_type": "execute_result"
+     "output_type": "display_data"
     }
    ],
    "source": [
-    "data.question_answer_pairs[0]"
+    "display_text_df(data.df.head(3))\n"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 35,
+   "execution_count": 4,
    "metadata": {},
    "outputs": [
     {
      "data": {
       "text/html": [
        "<style type=\"text/css\">\n",
-       "#T_fc111 th {\n",
+       "#T_8f656 th {\n",
        "  text-align: left;\n",
        "}\n",
-       "#T_fc111 td {\n",
+       "#T_8f656 td {\n",
        "  text-align: left;\n",
        "}\n",
-       "#T_fc111_row0_col0, #T_fc111_row0_col1, #T_fc111_row1_col0, #T_fc111_row1_col1, #T_fc111_row2_col0, #T_fc111_row2_col1, #T_fc111_row3_col0, #T_fc111_row3_col1, #T_fc111_row4_col0, #T_fc111_row4_col1, #T_fc111_row5_col0, #T_fc111_row5_col1, #T_fc111_row6_col0, #T_fc111_row6_col1, #T_fc111_row7_col0, #T_fc111_row7_col1, #T_fc111_row8_col0, #T_fc111_row8_col1, #T_fc111_row9_col0, #T_fc111_row9_col1 {\n",
+       "#T_8f656_row0_col0, #T_8f656_row0_col1, #T_8f656_row0_col2, #T_8f656_row0_col3, #T_8f656_row1_col0, #T_8f656_row1_col1, #T_8f656_row1_col2, #T_8f656_row1_col3, #T_8f656_row2_col0, #T_8f656_row2_col1, #T_8f656_row2_col2, #T_8f656_row2_col3, #T_8f656_row3_col0, #T_8f656_row3_col1, #T_8f656_row3_col2, #T_8f656_row3_col3, #T_8f656_row4_col0, #T_8f656_row4_col1, #T_8f656_row4_col2, #T_8f656_row4_col3, #T_8f656_row5_col0, #T_8f656_row5_col1, #T_8f656_row5_col2, #T_8f656_row5_col3, #T_8f656_row6_col0, #T_8f656_row6_col1, #T_8f656_row6_col2, #T_8f656_row6_col3, #T_8f656_row7_col0, #T_8f656_row7_col1, #T_8f656_row7_col2, #T_8f656_row7_col3, #T_8f656_row8_col0, #T_8f656_row8_col1, #T_8f656_row8_col2, #T_8f656_row8_col3, #T_8f656_row9_col0, #T_8f656_row9_col1, #T_8f656_row9_col2, #T_8f656_row9_col3 {\n",
        "  white-space: pre-wrap;\n",
        "}\n",
        "</style>\n",
-       "<table id=\"T_fc111\">\n",
+       "<table id=\"T_8f656\">\n",
        "  <thead>\n",
        "    <tr>\n",
-       "      <th id=\"T_fc111_level0_col0\" class=\"col_heading level0 col0\" >Question</th>\n",
-       "      <th id=\"T_fc111_level0_col1\" class=\"col_heading level0 col1\" >Answer</th>\n",
+       "      <th id=\"T_8f656_level0_col0\" class=\"col_heading level0 col0\" >Title</th>\n",
+       "      <th id=\"T_8f656_level0_col1\" class=\"col_heading level0 col1\" >Context</th>\n",
+       "      <th id=\"T_8f656_level0_col2\" class=\"col_heading level0 col2\" >Question</th>\n",
+       "      <th id=\"T_8f656_level0_col3\" class=\"col_heading level0 col3\" >Answer</th>\n",
        "    </tr>\n",
        "  </thead>\n",
        "  <tbody>\n",
        "    <tr>\n",
-       "      <td id=\"T_fc111_row0_col0\" class=\"data row0 col0\" >What year was the Banská Akadémia founded?</td>\n",
-       "      <td id=\"T_fc111_row0_col1\" class=\"data row0 col1\" >1735</td>\n",
+       "      <td id=\"T_8f656_row0_col0\" class=\"data row0 col0\" >Institute_of_technology</td>\n",
+       "      <td id=\"T_8f656_row0_col1\" class=\"data row0 col1\" >The world's first institution of technology or technical university with tertiary technical education is the Banská Akadémia in Banská Štiavnica, Slovakia, founded in 1735, Academy since December 13, 1762 established by queen Maria Theresa in order to train specialists of silver and gold mining and metallurgy in neighbourhood. Teaching started in 1764. Later the department of Mathematics, Mechanics and Hydraulics and department of Forestry were settled. University buildings are still at their place today and are used for teaching. University has launched the first book of electrotechnics in the world.</td>\n",
+       "      <td id=\"T_8f656_row0_col2\" class=\"data row0 col2\" >What year was the Banská Akadémia founded?</td>\n",
+       "      <td id=\"T_8f656_row0_col3\" class=\"data row0 col3\" >1735</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <td id=\"T_fc111_row1_col0\" class=\"data row1 col0\" >What is another speed that can also be reported by the camera?</td>\n",
-       "      <td id=\"T_fc111_row1_col1\" class=\"data row1 col1\" >SOS-based speed</td>\n",
+       "      <td id=\"T_8f656_row1_col0\" class=\"data row1 col0\" >Film_speed</td>\n",
+       "      <td id=\"T_8f656_row1_col1\" class=\"data row1 col1\" >The standard specifies how speed ratings should be reported by the camera. If the noise-based speed (40:1) is higher than the saturation-based speed, the noise-based speed should be reported, rounded downwards to a standard value (e.g. 200, 250, 320, or 400). The rationale is that exposure according to the lower saturation-based speed would not result in a visibly better image. In addition, an exposure latitude can be specified, ranging from the saturation-based speed to the 10:1 noise-based speed. If the noise-based speed (40:1) is lower than the saturation-based speed, or undefined because of high noise, the saturation-based speed is specified, rounded upwards to a standard value, because using the noise-based speed would lead to overexposed images. The camera may also report the SOS-based speed (explicitly as being an SOS speed), rounded to the nearest standard speed rating.</td>\n",
+       "      <td id=\"T_8f656_row1_col2\" class=\"data row1 col2\" >What is another speed that can also be reported by the camera?</td>\n",
+       "      <td id=\"T_8f656_row1_col3\" class=\"data row1 col3\" >SOS-based speed</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <td id=\"T_fc111_row2_col0\" class=\"data row2 col0\" >Where were the use of advanced materials and techniques on display in Sumer?</td>\n",
-       "      <td id=\"T_fc111_row2_col1\" class=\"data row2 col1\" >Sumerian temples and palaces</td>\n",
+       "      <td id=\"T_8f656_row2_col0\" class=\"data row2 col0\" >Sumer</td>\n",
+       "      <td id=\"T_8f656_row2_col1\" class=\"data row2 col1\" >The most impressive and famous of Sumerian buildings are the ziggurats, large layered platforms which supported temples. Sumerian cylinder seals also depict houses built from reeds not unlike those built by the Marsh Arabs of Southern Iraq until as recently as 400 CE. The Sumerians also developed the arch, which enabled them to develop a strong type of dome. They built this by constructing and linking several arches. Sumerian temples and palaces made use of more advanced materials and techniques,[citation needed] such as buttresses, recesses, half columns, and clay nails.</td>\n",
+       "      <td id=\"T_8f656_row2_col2\" class=\"data row2 col2\" >Where were the use of advanced materials and techniques on display in Sumer?</td>\n",
+       "      <td id=\"T_8f656_row2_col3\" class=\"data row2 col3\" >Sumerian temples and palaces</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <td id=\"T_fc111_row3_col0\" class=\"data row3 col0\" >Who is elected every even numbered year?</td>\n",
-       "      <td id=\"T_fc111_row3_col1\" class=\"data row3 col1\" >mayor</td>\n",
+       "      <td id=\"T_8f656_row3_col0\" class=\"data row3 col0\" >Ann_Arbor,_Michigan</td>\n",
+       "      <td id=\"T_8f656_row3_col1\" class=\"data row3 col1\" >Ann Arbor has a council-manager form of government. The City Council has 11 voting members: the mayor and 10 city council members. The mayor and city council members serve two-year terms: the mayor is elected every even-numbered year, while half of the city council members are up for election annually (five in even-numbered and five in odd-numbered years). Two council members are elected from each of the city's five wards. The mayor is elected citywide. The mayor is the presiding officer of the City Council and has the power to appoint all Council committee members as well as board and commission members, with the approval of the City Council. The current mayor of Ann Arbor is Christopher Taylor, a Democrat who was elected as mayor in 2014. Day-to-day city operations are managed by a city administrator chosen by the city council.</td>\n",
+       "      <td id=\"T_8f656_row3_col2\" class=\"data row3 col2\" >Who is elected every even numbered year?</td>\n",
+       "      <td id=\"T_8f656_row3_col3\" class=\"data row3 col3\" >mayor</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <td id=\"T_fc111_row4_col0\" class=\"data row4 col0\" >What was the purpose of top secret ICBM committee?</td>\n",
-       "      <td id=\"T_fc111_row4_col1\" class=\"data row4 col1\" >decide on the feasibility of building an ICBM large enough to carry a thermonuclear weapon</td>\n",
+       "      <td id=\"T_8f656_row4_col0\" class=\"data row4 col0\" >John_von_Neumann</td>\n",
+       "      <td id=\"T_8f656_row4_col1\" class=\"data row4 col1\" >Shortly before his death, when he was already quite ill, von Neumann headed the United States government's top secret ICBM committee, and it would sometimes meet in his home. Its purpose was to decide on the feasibility of building an ICBM large enough to carry a thermonuclear weapon. Von Neumann had long argued that while the technical obstacles were sizable, they could be overcome in time. The SM-65 Atlas passed its first fully functional test in 1959, two years after his death. The feasibility of an ICBM owed as much to improved, smaller warheads as it did to developments in rocketry, and his understanding of the former made his advice invaluable.</td>\n",
+       "      <td id=\"T_8f656_row4_col2\" class=\"data row4 col2\" >What was the purpose of top secret ICBM committee?</td>\n",
+       "      <td id=\"T_8f656_row4_col3\" class=\"data row4 col3\" >decide on the feasibility of building an ICBM large enough to carry a thermonuclear weapon</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <td id=\"T_fc111_row5_col0\" class=\"data row5 col0\" >What conferences became a requirement after Vatican II?</td>\n",
-       "      <td id=\"T_fc111_row5_col1\" class=\"data row5 col1\" >National Bishop Conferences</td>\n",
+       "      <td id=\"T_8f656_row5_col0\" class=\"data row5 col0\" >Pope_Paul_VI</td>\n",
+       "      <td id=\"T_8f656_row5_col1\" class=\"data row5 col1\" >Some critiqued Paul VI's decision; the newly created Synod of Bishops had an advisory role only and could not make decisions on their own, although the Council decided exactly that. During the pontificate of Paul VI, five such synods took place, and he is on record of implementing all their decisions. Related questions were raised about the new National Bishop Conferences, which became mandatory after Vatican II. Others questioned his Ostpolitik and contacts with Communism and the deals he engaged in for the faithful.</td>\n",
+       "      <td id=\"T_8f656_row5_col2\" class=\"data row5 col2\" >What conferences became a requirement after Vatican II?</td>\n",
+       "      <td id=\"T_8f656_row5_col3\" class=\"data row5 col3\" >National Bishop Conferences</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <td id=\"T_fc111_row6_col0\" class=\"data row6 col0\" >Who does M fight with?</td>\n",
-       "      <td id=\"T_fc111_row6_col1\" class=\"data row6 col1\" >C</td>\n",
+       "      <td id=\"T_8f656_row6_col0\" class=\"data row6 col0\" >Spectre_(2015_film)</td>\n",
+       "      <td id=\"T_8f656_row6_col1\" class=\"data row6 col1\" >Bond and Swann return to London where they meet M, Bill Tanner, Q, and Moneypenny; they intend to arrest C and stop Nine Eyes from going online. Swann leaves Bond, telling him she cannot be part of a life involving espionage, and is subsequently kidnapped. On the way, the group is ambushed and Bond is kidnapped, but the rest still proceed with the plan. After Q succeeds in preventing the Nine Eyes from going online, a brief struggle between M and C ends with the latter falling to his death. Meanwhile, Bond is taken to the old MI6 building, which is scheduled for demolition, and frees himself. Moving throughout the ruined labyrinth, he encounters a disfigured Blofeld, who tells him that he has three minutes to escape the building before explosives are detonated or die trying to save Swann. Bond finds Swann and the two escape by boat as the building collapses. Bond shoots down Blofeld's helicopter, which crashes onto Westminster Bridge. As Blofeld crawls away from the wreckage, Bond confronts him but ultimately leaves him to be arrested by M. Bond leaves the bridge with Swann.</td>\n",
+       "      <td id=\"T_8f656_row6_col2\" class=\"data row6 col2\" >Who does M fight with?</td>\n",
+       "      <td id=\"T_8f656_row6_col3\" class=\"data row6 col3\" >C</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <td id=\"T_fc111_row7_col0\" class=\"data row7 col0\" >How many species of fungi have been found on Antarctica?</td>\n",
-       "      <td id=\"T_fc111_row7_col1\" class=\"data row7 col1\" >1150</td>\n",
+       "      <td id=\"T_8f656_row7_col0\" class=\"data row7 col0\" >Antarctica</td>\n",
+       "      <td id=\"T_8f656_row7_col1\" class=\"data row7 col1\" >About 1150 species of fungi have been recorded from Antarctica, of which about 750 are non-lichen-forming and 400 are lichen-forming. Some of these species are cryptoendoliths as a result of evolution under extreme conditions, and have significantly contributed to shaping the impressive rock formations of the McMurdo Dry Valleys and surrounding mountain ridges. The apparently simple morphology, scarcely differentiated structures, metabolic systems and enzymes still active at very low temperatures, and reduced life cycles shown by such fungi make them particularly suited to harsh environments such as the McMurdo Dry Valleys. In particular, their thick-walled and strongly melanized cells make them resistant to UV light. Those features can also be observed in algae and cyanobacteria, suggesting that these are adaptations to the conditions prevailing in Antarctica. This has led to speculation that, if life ever occurred on Mars, it might have looked similar to Antarctic fungi such as Cryomyces minteri. Some of these fungi are also apparently endemic to Antarctica. Endemic Antarctic fungi also include certain dung-inhabiting species which have had to evolve in response to the double challenge of extreme cold while growing on dung, and the need to survive passage through the gut of warm-blooded animals.</td>\n",
+       "      <td id=\"T_8f656_row7_col2\" class=\"data row7 col2\" >How many species of fungi have been found on Antarctica?</td>\n",
+       "      <td id=\"T_8f656_row7_col3\" class=\"data row7 col3\" >1150</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <td id=\"T_fc111_row8_col0\" class=\"data row8 col0\" >After losing the battle of Guilford Courthouse, Cornawallis moved his troops where?</td>\n",
-       "      <td id=\"T_fc111_row8_col1\" class=\"data row8 col1\" >Virginia coastline</td>\n",
+       "      <td id=\"T_8f656_row8_col0\" class=\"data row8 col0\" >North_Carolina</td>\n",
+       "      <td id=\"T_8f656_row8_col1\" class=\"data row8 col1\" >In the Battle of Cowan's Ford, Cornwallis met resistance along the banks of the Catawba River at Cowan's Ford on February 1, 1781, in an attempt to engage General Morgan's forces during a tactical withdrawal. Morgan had moved to the northern part of the state to combine with General Greene's newly recruited forces. Generals Greene and Cornwallis finally met at the Battle of Guilford Courthouse in present-day Greensboro on March 15, 1781. Although the British troops held the field at the end of the battle, their casualties at the hands of the numerically superior Continental Army were crippling. Following this \"Pyrrhic victory\", Cornwallis chose to move to the Virginia coastline to get reinforcements, and to allow the Royal Navy to protect his battered army. This decision would result in Cornwallis' eventual defeat at Yorktown, Virginia, later in 1781. The Patriots' victory there guaranteed American independence.</td>\n",
+       "      <td id=\"T_8f656_row8_col2\" class=\"data row8 col2\" >After losing the battle of Guilford Courthouse, Cornawallis moved his troops where?</td>\n",
+       "      <td id=\"T_8f656_row8_col3\" class=\"data row8 col3\" >Virginia coastline</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <td id=\"T_fc111_row9_col0\" class=\"data row9 col0\" >What is the Olympic Torch made from?</td>\n",
-       "      <td id=\"T_fc111_row9_col1\" class=\"data row9 col1\" >aluminum.</td>\n",
+       "      <td id=\"T_8f656_row9_col0\" class=\"data row9 col0\" >2008_Summer_Olympics_torch_relay</td>\n",
+       "      <td id=\"T_8f656_row9_col1\" class=\"data row9 col1\" >The Olympic Torch is based on traditional scrolls and uses a traditional Chinese design known as \"Lucky Cloud\". It is made from aluminum. It is 72 centimetres high and weighs 985 grams. The torch is designed to remain lit in 65 kilometre per hour (37 mile per hour) winds, and in rain of up to 50 millimetres (2 inches) per hour. An ignition key is used to ignite and extinguish the flame. The torch is fueled by cans of propane. Each can will light the torch for 15 minutes. It is designed by a team from Lenovo Group. The Torch is designed in reference to the traditional Chinese concept of the 5 elements that make up the entire universe.</td>\n",
+       "      <td id=\"T_8f656_row9_col2\" class=\"data row9 col2\" >What is the Olympic Torch made from?</td>\n",
+       "      <td id=\"T_8f656_row9_col3\" class=\"data row9 col3\" >aluminum.</td>\n",
        "    </tr>\n",
        "  </tbody>\n",
        "</table>\n"
       ],
       "text/plain": [
-       "<pandas.io.formats.style.Styler at 0x3afc43c80>"
+       "<pandas.io.formats.style.Styler at 0x32b08f050>"
       ]
      },
      "metadata": {},
@@ -136,12 +219,8 @@
    ],
    "source": [
     "np.random.seed(42)\n",
-    "arr =np.array(data.question_answer_pairs)\n",
-    "n_samples = 10\n",
-    "indices = np.random.choice(len(arr), n_samples, replace=False)\n",
-    "random_sample = arr[indices]\n",
-    "# Display the questions and answers in the random sample as a dataframe\n",
-    "dfSample = pd.DataFrame(random_sample, columns=[\"Question\", \"Answer\"])\n",
+    "# Select 10 random rows from data.df\n",
+    "dfSample = data.df.sample(n=10)\n",
     "display_text_df(dfSample)"
    ]
   },
@@ -154,30 +233,35 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": 5,
    "metadata": {},
    "outputs": [],
-   "source": [
-    "from agent import get_agent\n",
-    "agent = get_agent()"
-   ]
+   "source": []
   },
   {
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "### Run the agent on the random sample of questions"
+    "### Run the agent on the random sample of questions\n",
+    "\n",
+    "* Unlike the default Retrieval QA or Open Generative QA of SQuAD, in our use case, the agent would normally be given context in the course of a natural conversation, as the user elaborates on what they want to know. \n",
+    "* Therefore, for benchmarking, we will provide the context to answer the question in the prompt.\n",
+    "\n",
+    "### Use semantic similarity to evaluate the agent's answers against the reference answers\n",
+    "\n",
+    "* One flaw of this approach is that it does not take into account the existence of multiple acceptable answers.\n",
+    "* Another flaw is that the agent me be unfairly penalized for elaborating on the answer, while this benchmark focuses on only and exactly the one canonical answer given.\n"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 36,
+   "execution_count": 6,
    "metadata": {},
    "outputs": [
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "4bce5a5c2449435dbd058ed938db2a91",
+       "model_id": "c8e4bf7687534df284b075480f4dba88",
        "version_major": 2,
        "version_minor": 0
       },
@@ -190,188 +274,216 @@
     }
    ],
    "source": [
-    "from gradio import ChatMessage\n",
-    "from transformers.agents import agent_types\n",
-    "from tqdm.notebook import tqdm\n",
-    "import logging\n",
+    "BENCHMARKS_DIR = \"benchmarks\"\n",
+    "\n",
+    "def benchmark_agent(agent, dfSample, name):\n",
+    "    answers_ref, answers_pred = [], []        \n",
+    "\n",
+    "    # Suppress logging from the agent, which can be quite verbose\n",
+    "    agent.logger.setLevel(logging.CRITICAL)\n",
+    "\n",
+    "    for title, context, question, answer in tqdm(dfSample.values):\n",
+    "        class Output:\n",
+    "            output: agent_types.AgentType | str = None\n",
+    "\n",
+    "        prompt = f\"\"\"\n",
+    "            Read the following document and answer the question.\n",
+    "\n",
+    "            Document Title: {title}\n",
+    "            Document Content: {context}\n",
+    "\n",
+    "            Question: {question}\n",
+    "        \"\"\"\n",
+    "        answers_ref.append(answer)\n",
+    "        final_answer = agent.run(prompt, stream=False, reset=True)\n",
+    "        answers_pred.append(final_answer)\n",
     "\n",
-    "answers_ref, answers_pred = [], []        \n",
+    "        answers_ref = [str(answer) for answer in answers_ref]\n",
+    "        answers_pred = [str(answer) for answer in answers_pred]\n",
     "\n",
-    "# Suppress logging from the agent, which can be quite verbose\n",
-    "agent.logger.setLevel(logging.CRITICAL)\n",
+    "        em = EmbeddingModelWrapper()\n",
+    "        similarities = em.get_similarities(\n",
+    "            em.get_embeddings( answers_pred ),\n",
+    "            em.get_embeddings( answers_ref ),\n",
+    "        )\n",
     "\n",
-    "for question, answer in tqdm(random_sample):\n",
-    "    class Output:\n",
-    "        output: agent_types.AgentType | str = None\n",
+    "        dfAnswers = dfSample.copy()\n",
+    "        dfAnswers[\"Predicted Answer\"] = answers_pred\n",
+    "        dfAnswers[\"Similarity\"] = similarities\n",
     "\n",
-    "    prompt = question\n",
-    "    answers_ref.append(answer)\n",
-    "    final_answer = agent.run(prompt, stream=False, reset=True)\n",
-    "    answers_pred.append(final_answer)"
+    "        os.makedirs(BENCHMARKS_DIR, exist_ok=True)\n",
+    "        dfAnswers.to_pickle(os.path.join(BENCHMARKS_DIR, f\"{name}.pkl\"))\n"
    ]
   },
   {
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "### Use semantic similarity to evaluate the agent's answers against the reference answers\n",
-    "\n",
-    "* One flaw of this approach is that it does not take into account the existence of multiple acceptable answers.\n",
-    "* It also does not benefit from having the context of the question. "
+    "### Set up and run the benchmarks"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 37,
+   "execution_count": 7,
    "metadata": {},
    "outputs": [],
    "source": [
-    "from semscore import EmbeddingModelWrapper\n",
-    "from statistics import mean\n",
+    "from agent import get_agent\n",
     "\n",
-    "answers_ref = [str(answer) for answer in answers_ref]\n",
-    "answers_pred = [str(answer) for answer in answers_pred]\n",
+    "benchmarks = [\n",
+    "    (get_agent(), \"baseline\"),\n",
+    "]\n",
     "\n",
-    "em = EmbeddingModelWrapper()\n",
-    "similarities = em.get_similarities(\n",
-    "    em.get_embeddings( answers_pred ),\n",
-    "    em.get_embeddings( answers_ref ),\n",
-    ")"
+    "for agent, name in tqdm(benchmarks):\n",
+    "    benchmark_agent(agent, dfSample, name)\n"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 39,
+   "execution_count": 11,
    "metadata": {},
    "outputs": [
     {
      "data": {
       "text/html": [
        "<style type=\"text/css\">\n",
-       "#T_67704 th {\n",
+       "#T_7cf04 th {\n",
        "  text-align: left;\n",
        "}\n",
-       "#T_67704 td {\n",
+       "#T_7cf04 td {\n",
        "  text-align: left;\n",
        "}\n",
-       "#T_67704_row0_col0, #T_67704_row0_col1, #T_67704_row0_col2, #T_67704_row0_col3, #T_67704_row1_col0, #T_67704_row1_col1, #T_67704_row1_col2, #T_67704_row1_col3, #T_67704_row2_col0, #T_67704_row2_col1, #T_67704_row2_col2, #T_67704_row2_col3, #T_67704_row3_col0, #T_67704_row3_col1, #T_67704_row3_col2, #T_67704_row3_col3, #T_67704_row4_col0, #T_67704_row4_col1, #T_67704_row4_col2, #T_67704_row4_col3, #T_67704_row5_col0, #T_67704_row5_col1, #T_67704_row5_col2, #T_67704_row5_col3, #T_67704_row6_col0, #T_67704_row6_col1, #T_67704_row6_col2, #T_67704_row6_col3, #T_67704_row7_col0, #T_67704_row7_col1, #T_67704_row7_col2, #T_67704_row7_col3, #T_67704_row8_col0, #T_67704_row8_col1, #T_67704_row8_col2, #T_67704_row8_col3, #T_67704_row9_col0, #T_67704_row9_col1, #T_67704_row9_col2, #T_67704_row9_col3 {\n",
+       "#T_7cf04_row0_col0, #T_7cf04_row0_col1, #T_7cf04_row0_col2, #T_7cf04_row0_col3, #T_7cf04_row0_col4, #T_7cf04_row0_col5, #T_7cf04_row1_col0, #T_7cf04_row1_col1, #T_7cf04_row1_col2, #T_7cf04_row1_col3, #T_7cf04_row1_col4, #T_7cf04_row1_col5, #T_7cf04_row2_col0, #T_7cf04_row2_col1, #T_7cf04_row2_col2, #T_7cf04_row2_col3, #T_7cf04_row2_col4, #T_7cf04_row2_col5, #T_7cf04_row3_col0, #T_7cf04_row3_col1, #T_7cf04_row3_col2, #T_7cf04_row3_col3, #T_7cf04_row3_col4, #T_7cf04_row3_col5, #T_7cf04_row4_col0, #T_7cf04_row4_col1, #T_7cf04_row4_col2, #T_7cf04_row4_col3, #T_7cf04_row4_col4, #T_7cf04_row4_col5, #T_7cf04_row5_col0, #T_7cf04_row5_col1, #T_7cf04_row5_col2, #T_7cf04_row5_col3, #T_7cf04_row5_col4, #T_7cf04_row5_col5, #T_7cf04_row6_col0, #T_7cf04_row6_col1, #T_7cf04_row6_col2, #T_7cf04_row6_col3, #T_7cf04_row6_col4, #T_7cf04_row6_col5, #T_7cf04_row7_col0, #T_7cf04_row7_col1, #T_7cf04_row7_col2, #T_7cf04_row7_col3, #T_7cf04_row7_col4, #T_7cf04_row7_col5, #T_7cf04_row8_col0, #T_7cf04_row8_col1, #T_7cf04_row8_col2, #T_7cf04_row8_col3, #T_7cf04_row8_col4, #T_7cf04_row8_col5, #T_7cf04_row9_col0, #T_7cf04_row9_col1, #T_7cf04_row9_col2, #T_7cf04_row9_col3, #T_7cf04_row9_col4, #T_7cf04_row9_col5 {\n",
        "  white-space: pre-wrap;\n",
        "}\n",
        "</style>\n",
-       "<table id=\"T_67704\">\n",
+       "<table id=\"T_7cf04\">\n",
        "  <thead>\n",
        "    <tr>\n",
-       "      <th id=\"T_67704_level0_col0\" class=\"col_heading level0 col0\" >Question</th>\n",
-       "      <th id=\"T_67704_level0_col1\" class=\"col_heading level0 col1\" >Reference Answer</th>\n",
-       "      <th id=\"T_67704_level0_col2\" class=\"col_heading level0 col2\" >Predicted Answer</th>\n",
-       "      <th id=\"T_67704_level0_col3\" class=\"col_heading level0 col3\" >Similarity</th>\n",
+       "      <th id=\"T_7cf04_level0_col0\" class=\"col_heading level0 col0\" >Title</th>\n",
+       "      <th id=\"T_7cf04_level0_col1\" class=\"col_heading level0 col1\" >Context</th>\n",
+       "      <th id=\"T_7cf04_level0_col2\" class=\"col_heading level0 col2\" >Question</th>\n",
+       "      <th id=\"T_7cf04_level0_col3\" class=\"col_heading level0 col3\" >Answer</th>\n",
+       "      <th id=\"T_7cf04_level0_col4\" class=\"col_heading level0 col4\" >Predicted Answer</th>\n",
+       "      <th id=\"T_7cf04_level0_col5\" class=\"col_heading level0 col5\" >Similarity</th>\n",
        "    </tr>\n",
        "  </thead>\n",
        "  <tbody>\n",
        "    <tr>\n",
-       "      <td id=\"T_67704_row0_col0\" class=\"data row0 col0\" >What year was the Banská Akadémia founded?</td>\n",
-       "      <td id=\"T_67704_row0_col1\" class=\"data row0 col1\" >1735</td>\n",
-       "      <td id=\"T_67704_row0_col2\" class=\"data row0 col2\" >1735</td>\n",
-       "      <td id=\"T_67704_row0_col3\" class=\"data row0 col3\" >1.000000</td>\n",
+       "      <td id=\"T_7cf04_row0_col0\" class=\"data row0 col0\" >Institute_of_technology</td>\n",
+       "      <td id=\"T_7cf04_row0_col1\" class=\"data row0 col1\" >The world's first institution of technology or technical university with tertiary technical education is the Banská Akadémia in Banská Štiavnica, Slovakia, founded in 1735, Academy since December 13, 1762 established by queen Maria Theresa in order to train specialists of silver and gold mining and metallurgy in neighbourhood. Teaching started in 1764. Later the department of Mathematics, Mechanics and Hydraulics and department of Forestry were settled. University buildings are still at their place today and are used for teaching. University has launched the first book of electrotechnics in the world.</td>\n",
+       "      <td id=\"T_7cf04_row0_col2\" class=\"data row0 col2\" >What year was the Banská Akadémia founded?</td>\n",
+       "      <td id=\"T_7cf04_row0_col3\" class=\"data row0 col3\" >1735</td>\n",
+       "      <td id=\"T_7cf04_row0_col4\" class=\"data row0 col4\" >1735</td>\n",
+       "      <td id=\"T_7cf04_row0_col5\" class=\"data row0 col5\" >1.000000</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <td id=\"T_67704_row1_col0\" class=\"data row1 col0\" >What is another speed that can also be reported by the camera?</td>\n",
-       "      <td id=\"T_67704_row1_col1\" class=\"data row1 col1\" >SOS-based speed</td>\n",
-       "      <td id=\"T_67704_row1_col2\" class=\"data row1 col2\" >Average speed</td>\n",
-       "      <td id=\"T_67704_row1_col3\" class=\"data row1 col3\" >0.433297</td>\n",
+       "      <td id=\"T_7cf04_row1_col0\" class=\"data row1 col0\" >Film_speed</td>\n",
+       "      <td id=\"T_7cf04_row1_col1\" class=\"data row1 col1\" >The standard specifies how speed ratings should be reported by the camera. If the noise-based speed (40:1) is higher than the saturation-based speed, the noise-based speed should be reported, rounded downwards to a standard value (e.g. 200, 250, 320, or 400). The rationale is that exposure according to the lower saturation-based speed would not result in a visibly better image. In addition, an exposure latitude can be specified, ranging from the saturation-based speed to the 10:1 noise-based speed. If the noise-based speed (40:1) is lower than the saturation-based speed, or undefined because of high noise, the saturation-based speed is specified, rounded upwards to a standard value, because using the noise-based speed would lead to overexposed images. The camera may also report the SOS-based speed (explicitly as being an SOS speed), rounded to the nearest standard speed rating.</td>\n",
+       "      <td id=\"T_7cf04_row1_col2\" class=\"data row1 col2\" >What is another speed that can also be reported by the camera?</td>\n",
+       "      <td id=\"T_7cf04_row1_col3\" class=\"data row1 col3\" >SOS-based speed</td>\n",
+       "      <td id=\"T_7cf04_row1_col4\" class=\"data row1 col4\" >saturation-based speed</td>\n",
+       "      <td id=\"T_7cf04_row1_col5\" class=\"data row1 col5\" >0.555529</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <td id=\"T_67704_row2_col0\" class=\"data row2 col0\" >Where were the use of advanced materials and techniques on display in Sumer?</td>\n",
-       "      <td id=\"T_67704_row2_col1\" class=\"data row2 col1\" >Sumerian temples and palaces</td>\n",
-       "      <td id=\"T_67704_row2_col2\" class=\"data row2 col2\" >Based on the information provided, it appears that the Sumerians developed and displayed advanced materials and techniques such as metrology, writing, and astronomy throughout their city-states. The specific locations where these advanced materials and techniques were on display are not explicitly mentioned.\n",
-       "\n",
-       "However, considering the context of the question, I would argue that the city-states of Sumer itself is the most relevant answer. The city-states of Sumer were the hub of Sumerian civilization, culture, and innovation, and it was likely there that these advanced materials and techniques were developed, displayed, and showcased.\n",
-       "\n",
-       "Therefore, my final answer to the user request is:\n",
-       "\n",
-       "The city-states of Sumer</td>\n",
-       "      <td id=\"T_67704_row2_col3\" class=\"data row2 col3\" >0.545807</td>\n",
+       "      <td id=\"T_7cf04_row2_col0\" class=\"data row2 col0\" >Sumer</td>\n",
+       "      <td id=\"T_7cf04_row2_col1\" class=\"data row2 col1\" >The most impressive and famous of Sumerian buildings are the ziggurats, large layered platforms which supported temples. Sumerian cylinder seals also depict houses built from reeds not unlike those built by the Marsh Arabs of Southern Iraq until as recently as 400 CE. The Sumerians also developed the arch, which enabled them to develop a strong type of dome. They built this by constructing and linking several arches. Sumerian temples and palaces made use of more advanced materials and techniques,[citation needed] such as buttresses, recesses, half columns, and clay nails.</td>\n",
+       "      <td id=\"T_7cf04_row2_col2\" class=\"data row2 col2\" >Where were the use of advanced materials and techniques on display in Sumer?</td>\n",
+       "      <td id=\"T_7cf04_row2_col3\" class=\"data row2 col3\" >Sumerian temples and palaces</td>\n",
+       "      <td id=\"T_7cf04_row2_col4\" class=\"data row2 col4\" >temples and palaces</td>\n",
+       "      <td id=\"T_7cf04_row2_col5\" class=\"data row2 col5\" >0.726322</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <td id=\"T_67704_row3_col0\" class=\"data row3 col0\" >Who is elected every even numbered year?</td>\n",
-       "      <td id=\"T_67704_row3_col1\" class=\"data row3 col1\" >mayor</td>\n",
-       "      <td id=\"T_67704_row3_col2\" class=\"data row3 col2\" >mayor</td>\n",
-       "      <td id=\"T_67704_row3_col3\" class=\"data row3 col3\" >1.000000</td>\n",
+       "      <td id=\"T_7cf04_row3_col0\" class=\"data row3 col0\" >Ann_Arbor,_Michigan</td>\n",
+       "      <td id=\"T_7cf04_row3_col1\" class=\"data row3 col1\" >Ann Arbor has a council-manager form of government. The City Council has 11 voting members: the mayor and 10 city council members. The mayor and city council members serve two-year terms: the mayor is elected every even-numbered year, while half of the city council members are up for election annually (five in even-numbered and five in odd-numbered years). Two council members are elected from each of the city's five wards. The mayor is elected citywide. The mayor is the presiding officer of the City Council and has the power to appoint all Council committee members as well as board and commission members, with the approval of the City Council. The current mayor of Ann Arbor is Christopher Taylor, a Democrat who was elected as mayor in 2014. Day-to-day city operations are managed by a city administrator chosen by the city council.</td>\n",
+       "      <td id=\"T_7cf04_row3_col2\" class=\"data row3 col2\" >Who is elected every even numbered year?</td>\n",
+       "      <td id=\"T_7cf04_row3_col3\" class=\"data row3 col3\" >mayor</td>\n",
+       "      <td id=\"T_7cf04_row3_col4\" class=\"data row3 col4\" >The mayor is elected every even-numbered year.</td>\n",
+       "      <td id=\"T_7cf04_row3_col5\" class=\"data row3 col5\" >0.493396</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <td id=\"T_67704_row4_col0\" class=\"data row4 col0\" >What was the purpose of top secret ICBM committee?</td>\n",
-       "      <td id=\"T_67704_row4_col1\" class=\"data row4 col1\" >decide on the feasibility of building an ICBM large enough to carry a thermonuclear weapon</td>\n",
-       "      <td id=\"T_67704_row4_col2\" class=\"data row4 col2\" >decide on the feasibility of building an ICBM large enough to carry a thermonuclear weapon</td>\n",
-       "      <td id=\"T_67704_row4_col3\" class=\"data row4 col3\" >1.000000</td>\n",
+       "      <td id=\"T_7cf04_row4_col0\" class=\"data row4 col0\" >John_von_Neumann</td>\n",
+       "      <td id=\"T_7cf04_row4_col1\" class=\"data row4 col1\" >Shortly before his death, when he was already quite ill, von Neumann headed the United States government's top secret ICBM committee, and it would sometimes meet in his home. Its purpose was to decide on the feasibility of building an ICBM large enough to carry a thermonuclear weapon. Von Neumann had long argued that while the technical obstacles were sizable, they could be overcome in time. The SM-65 Atlas passed its first fully functional test in 1959, two years after his death. The feasibility of an ICBM owed as much to improved, smaller warheads as it did to developments in rocketry, and his understanding of the former made his advice invaluable.</td>\n",
+       "      <td id=\"T_7cf04_row4_col2\" class=\"data row4 col2\" >What was the purpose of top secret ICBM committee?</td>\n",
+       "      <td id=\"T_7cf04_row4_col3\" class=\"data row4 col3\" >decide on the feasibility of building an ICBM large enough to carry a thermonuclear weapon</td>\n",
+       "      <td id=\"T_7cf04_row4_col4\" class=\"data row4 col4\" >decide on the feasibility of building an ICBM large enough to carry a thermonuclear weapon</td>\n",
+       "      <td id=\"T_7cf04_row4_col5\" class=\"data row4 col5\" >1.000000</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <td id=\"T_67704_row5_col0\" class=\"data row5 col0\" >What conferences became a requirement after Vatican II?</td>\n",
-       "      <td id=\"T_67704_row5_col1\" class=\"data row5 col1\" >National Bishop Conferences</td>\n",
-       "      <td id=\"T_67704_row5_col2\" class=\"data row5 col2\" >['National Bishop Conferences']</td>\n",
-       "      <td id=\"T_67704_row5_col3\" class=\"data row5 col3\" >0.937632</td>\n",
+       "      <td id=\"T_7cf04_row5_col0\" class=\"data row5 col0\" >Pope_Paul_VI</td>\n",
+       "      <td id=\"T_7cf04_row5_col1\" class=\"data row5 col1\" >Some critiqued Paul VI's decision; the newly created Synod of Bishops had an advisory role only and could not make decisions on their own, although the Council decided exactly that. During the pontificate of Paul VI, five such synods took place, and he is on record of implementing all their decisions. Related questions were raised about the new National Bishop Conferences, which became mandatory after Vatican II. Others questioned his Ostpolitik and contacts with Communism and the deals he engaged in for the faithful.</td>\n",
+       "      <td id=\"T_7cf04_row5_col2\" class=\"data row5 col2\" >What conferences became a requirement after Vatican II?</td>\n",
+       "      <td id=\"T_7cf04_row5_col3\" class=\"data row5 col3\" >National Bishop Conferences</td>\n",
+       "      <td id=\"T_7cf04_row5_col4\" class=\"data row5 col4\" >The National Bishop Conferences became mandatory after Vatican II.</td>\n",
+       "      <td id=\"T_7cf04_row5_col5\" class=\"data row5 col5\" >0.442729</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <td id=\"T_67704_row6_col0\" class=\"data row6 col0\" >Who does M fight with?</td>\n",
-       "      <td id=\"T_67704_row6_col1\" class=\"data row6 col1\" >C</td>\n",
-       "      <td id=\"T_67704_row6_col2\" class=\"data row6 col2\" >C</td>\n",
-       "      <td id=\"T_67704_row6_col3\" class=\"data row6 col3\" >1.000000</td>\n",
+       "      <td id=\"T_7cf04_row6_col0\" class=\"data row6 col0\" >Spectre_(2015_film)</td>\n",
+       "      <td id=\"T_7cf04_row6_col1\" class=\"data row6 col1\" >Bond and Swann return to London where they meet M, Bill Tanner, Q, and Moneypenny; they intend to arrest C and stop Nine Eyes from going online. Swann leaves Bond, telling him she cannot be part of a life involving espionage, and is subsequently kidnapped. On the way, the group is ambushed and Bond is kidnapped, but the rest still proceed with the plan. After Q succeeds in preventing the Nine Eyes from going online, a brief struggle between M and C ends with the latter falling to his death. Meanwhile, Bond is taken to the old MI6 building, which is scheduled for demolition, and frees himself. Moving throughout the ruined labyrinth, he encounters a disfigured Blofeld, who tells him that he has three minutes to escape the building before explosives are detonated or die trying to save Swann. Bond finds Swann and the two escape by boat as the building collapses. Bond shoots down Blofeld's helicopter, which crashes onto Westminster Bridge. As Blofeld crawls away from the wreckage, Bond confronts him but ultimately leaves him to be arrested by M. Bond leaves the bridge with Swann.</td>\n",
+       "      <td id=\"T_7cf04_row6_col2\" class=\"data row6 col2\" >Who does M fight with?</td>\n",
+       "      <td id=\"T_7cf04_row6_col3\" class=\"data row6 col3\" >C</td>\n",
+       "      <td id=\"T_7cf04_row6_col4\" class=\"data row6 col4\" >C</td>\n",
+       "      <td id=\"T_7cf04_row6_col5\" class=\"data row6 col5\" >1.000000</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <td id=\"T_67704_row7_col0\" class=\"data row7 col0\" >How many species of fungi have been found on Antarctica?</td>\n",
-       "      <td id=\"T_67704_row7_col1\" class=\"data row7 col1\" >1150</td>\n",
-       "      <td id=\"T_67704_row7_col2\" class=\"data row7 col2\" >Based on the output from the `squad_retriever` tool, I can see that there are two documents in the SQuAD dataset that answer the question \"How many species of fungi have been found on Antarctica?\".\n",
-       "\n",
-       "The first document states that about 1150 species of fungi have been recorded from Antarctica. The second document does not provide a different answer to this question.\n",
-       "\n",
-       "Therefore, my final answer is:\n",
-       "\n",
-       "There are approximately 1150 species of fungi that have been found on Antarctica.</td>\n",
-       "      <td id=\"T_67704_row7_col3\" class=\"data row7 col3\" >-0.020657</td>\n",
+       "      <td id=\"T_7cf04_row7_col0\" class=\"data row7 col0\" >Antarctica</td>\n",
+       "      <td id=\"T_7cf04_row7_col1\" class=\"data row7 col1\" >About 1150 species of fungi have been recorded from Antarctica, of which about 750 are non-lichen-forming and 400 are lichen-forming. Some of these species are cryptoendoliths as a result of evolution under extreme conditions, and have significantly contributed to shaping the impressive rock formations of the McMurdo Dry Valleys and surrounding mountain ridges. The apparently simple morphology, scarcely differentiated structures, metabolic systems and enzymes still active at very low temperatures, and reduced life cycles shown by such fungi make them particularly suited to harsh environments such as the McMurdo Dry Valleys. In particular, their thick-walled and strongly melanized cells make them resistant to UV light. Those features can also be observed in algae and cyanobacteria, suggesting that these are adaptations to the conditions prevailing in Antarctica. This has led to speculation that, if life ever occurred on Mars, it might have looked similar to Antarctic fungi such as Cryomyces minteri. Some of these fungi are also apparently endemic to Antarctica. Endemic Antarctic fungi also include certain dung-inhabiting species which have had to evolve in response to the double challenge of extreme cold while growing on dung, and the need to survive passage through the gut of warm-blooded animals.</td>\n",
+       "      <td id=\"T_7cf04_row7_col2\" class=\"data row7 col2\" >How many species of fungi have been found on Antarctica?</td>\n",
+       "      <td id=\"T_7cf04_row7_col3\" class=\"data row7 col3\" >1150</td>\n",
+       "      <td id=\"T_7cf04_row7_col4\" class=\"data row7 col4\" >1150</td>\n",
+       "      <td id=\"T_7cf04_row7_col5\" class=\"data row7 col5\" >1.000000</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <td id=\"T_67704_row8_col0\" class=\"data row8 col0\" >After losing the battle of Guilford Courthouse, Cornawallis moved his troops where?</td>\n",
-       "      <td id=\"T_67704_row8_col1\" class=\"data row8 col1\" >Virginia coastline</td>\n",
-       "      <td id=\"T_67704_row8_col2\" class=\"data row8 col2\" >The Virginia coastline</td>\n",
-       "      <td id=\"T_67704_row8_col3\" class=\"data row8 col3\" >0.948570</td>\n",
+       "      <td id=\"T_7cf04_row8_col0\" class=\"data row8 col0\" >North_Carolina</td>\n",
+       "      <td id=\"T_7cf04_row8_col1\" class=\"data row8 col1\" >In the Battle of Cowan's Ford, Cornwallis met resistance along the banks of the Catawba River at Cowan's Ford on February 1, 1781, in an attempt to engage General Morgan's forces during a tactical withdrawal. Morgan had moved to the northern part of the state to combine with General Greene's newly recruited forces. Generals Greene and Cornwallis finally met at the Battle of Guilford Courthouse in present-day Greensboro on March 15, 1781. Although the British troops held the field at the end of the battle, their casualties at the hands of the numerically superior Continental Army were crippling. Following this \"Pyrrhic victory\", Cornwallis chose to move to the Virginia coastline to get reinforcements, and to allow the Royal Navy to protect his battered army. This decision would result in Cornwallis' eventual defeat at Yorktown, Virginia, later in 1781. The Patriots' victory there guaranteed American independence.</td>\n",
+       "      <td id=\"T_7cf04_row8_col2\" class=\"data row8 col2\" >After losing the battle of Guilford Courthouse, Cornawallis moved his troops where?</td>\n",
+       "      <td id=\"T_7cf04_row8_col3\" class=\"data row8 col3\" >Virginia coastline</td>\n",
+       "      <td id=\"T_7cf04_row8_col4\" class=\"data row8 col4\" >the Virginia coastline</td>\n",
+       "      <td id=\"T_7cf04_row8_col5\" class=\"data row8 col5\" >0.948570</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <td id=\"T_67704_row9_col0\" class=\"data row9 col0\" >What is the Olympic Torch made from?</td>\n",
-       "      <td id=\"T_67704_row9_col1\" class=\"data row9 col1\" >aluminum.</td>\n",
-       "      <td id=\"T_67704_row9_col2\" class=\"data row9 col2\" >aluminum</td>\n",
-       "      <td id=\"T_67704_row9_col3\" class=\"data row9 col3\" >0.973508</td>\n",
+       "      <td id=\"T_7cf04_row9_col0\" class=\"data row9 col0\" >2008_Summer_Olympics_torch_relay</td>\n",
+       "      <td id=\"T_7cf04_row9_col1\" class=\"data row9 col1\" >The Olympic Torch is based on traditional scrolls and uses a traditional Chinese design known as \"Lucky Cloud\". It is made from aluminum. It is 72 centimetres high and weighs 985 grams. The torch is designed to remain lit in 65 kilometre per hour (37 mile per hour) winds, and in rain of up to 50 millimetres (2 inches) per hour. An ignition key is used to ignite and extinguish the flame. The torch is fueled by cans of propane. Each can will light the torch for 15 minutes. It is designed by a team from Lenovo Group. The Torch is designed in reference to the traditional Chinese concept of the 5 elements that make up the entire universe.</td>\n",
+       "      <td id=\"T_7cf04_row9_col2\" class=\"data row9 col2\" >What is the Olympic Torch made from?</td>\n",
+       "      <td id=\"T_7cf04_row9_col3\" class=\"data row9 col3\" >aluminum.</td>\n",
+       "      <td id=\"T_7cf04_row9_col4\" class=\"data row9 col4\" >aluminum</td>\n",
+       "      <td id=\"T_7cf04_row9_col5\" class=\"data row9 col5\" >0.973508</td>\n",
        "    </tr>\n",
        "  </tbody>\n",
        "</table>\n"
       ],
       "text/plain": [
-       "<pandas.io.formats.style.Styler at 0x3b0db7320>"
+       "<pandas.io.formats.style.Styler at 0x33e917d10>"
       ]
      },
      "metadata": {},
      "output_type": "display_data"
-    },
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "Mean similarity: 0.78\n"
-     ]
     }
    ],
    "source": [
-    "import pandas as pd\n",
-    "questions = [question for question, _ in random_sample]\n",
-    "dfAnswers = pd.DataFrame(list(zip(questions, answers_ref, answers_pred)), columns=[\"Question\", \"Reference Answer\", \"Predicted Answer\"])\n",
-    "dfAnswers[\"Similarity\"] = similarities\n",
-    "display(dfAnswers.style.set_properties(**{'white-space': 'pre-wrap'}).set_table_styles(\n",
-    "    [{'selector': 'th', 'props': [('text-align', 'left')]},\n",
-    "     {'selector': 'td', 'props': [('text-align', 'left')]}\n",
-    "    ]\n",
-    ").hide())\n",
-    "print(f\"Mean similarity: {round(mean(similarities), 2)}\")\n",
-    "\n"
+    "# Load and display all benchmarks\n",
+    "def load_benchmarks():\n",
+    "    benchmarks_dir = \"benchmarks\"\n",
+    "    benchmarks = []\n",
+    "    for file in os.listdir(benchmarks_dir):\n",
+    "        if file.endswith(\".pkl\"):\n",
+    "            df = pd.read_pickle(os.path.join(benchmarks_dir, file))\n",
+    "            benchmarks.append(df)\n",
+    "    return benchmarks\n",
+    "\n",
+    "benchmarks = load_benchmarks()\n",
+    "\n",
+    "for benchmark in benchmarks:\n",
+    "    display_text_df(benchmark)\n"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": 10,
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  },
   {
    "cell_type": "code",
    "execution_count": null,

benchmarks/baseline.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:eac426004c5fb5452866d7d767c3ee286d01e3ade51497a9003a255594c70ae7
+size 10430

data.py

@@ -1,6 +1,7 @@
 import os
 import json
 import chromadb
+import pandas as pd
 from llama_index.core import VectorStoreIndex
 from llama_index.vector_stores.chroma import ChromaVectorStore
 from llama_index.core import StorageContext
@@ -52,8 +53,7 @@ class Data:
         with open('data/train-v1.1.json', 'r') as f:
             raw_data = json.load(f)  
             
-        extracted_question = []
-        extracted_answer = []
+        raw_documents = []
         documents = []
 
         for data in raw_data['data']:
@@ -67,8 +67,7 @@ class Data:
                         if ans['text'] not in answers:
                             answers.append(ans['text'])
                     for answer in answers:
-                        extracted_question.append(question)
-                        extracted_answer.append(answer)
+                        raw_documents.append([title, context, question, answer])
                     
                     doc = f"""
                         Title: {title}
@@ -81,8 +80,8 @@ class Data:
                     doc = "\n".join([line.strip() for line in doc.split("\n")])
                     documents.append(doc)
 
+        self.df = pd.DataFrame(raw_documents, columns=["Title", "Context", "Question", "Answer"])
         self.documents = [Document(text=t) for t in documents]
-        self.question_answer_pairs = list(zip(extracted_question, extracted_answer))
 
         print("Raw Data loaded")