feat: more infra, and a manual agent

notebooks/02-vi-handmade-agent.ipynb

@@ -0,0 +1,810 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "id": "initial_id",
+   "metadata": {
+    "collapsed": true,
+    "ExecuteTime": {
+     "end_time": "2025-06-16T14:22:31.049405Z",
+     "start_time": "2025-06-16T14:22:29.311032Z"
+    }
+   },
+   "source": [
+    "import time\n",
+    "import concurrent\n",
+    "from concurrent import futures\n",
+    "from typing import Any, List, Dict, Tuple\n",
+    "\n",
+    "from PIL import Image\n",
+    "from openai import AzureOpenAI\n",
+    "from tqdm.auto import tqdm\n",
+    "\n",
+    "import json\n",
+    "from pathlib import Path\n",
+    "\n",
+    "import pandas as pd\n",
+    "import requests\n",
+    "from dotenv import dotenv_values\n",
+    "from smolagents import tool, DuckDuckGoSearchTool\n",
+    "import wikipediaapi\n",
+    "\n",
+    "\n",
+    "\n",
+    "test_api_base = \"https://agents-course-unit4-scoring.hf.space\"\n",
+    "\n",
+    "def get_random_question():\n",
+    "    url = f\"{test_api_base}/random-question\"\n",
+    "\n",
+    "\n",
+    "    try:\n",
+    "        # Fetch the random question\n",
+    "        response = requests.get(url, timeout=10)\n",
+    "        response.raise_for_status()\n",
+    "        question_data = response.json()\n",
+    "\n",
+    "        # Check if there's an associated file to download\n",
+    "        if question_data.get(\"file_name\") and question_data.get(\"task_id\"):\n",
+    "            task_id = question_data[\"task_id\"]\n",
+    "            file_url = f\"{test_api_base}/files/{task_id}\"\n",
+    "\n",
+    "            # Create a directory for downloaded files if it doesn't exist\n",
+    "            download_dir = Path(\"downloaded_files\")\n",
+    "            download_dir.mkdir(exist_ok=True)\n",
+    "\n",
+    "            # Download the file\n",
+    "            file_response = requests.get(file_url, timeout=30)\n",
+    "            file_response.raise_for_status()\n",
+    "\n",
+    "            # Get filename from content-disposition header or use task_id\n",
+    "            content_disposition = file_response.headers.get('content-disposition', '')\n",
+    "            filename = content_disposition.split('filename=')[1].strip('\"')\n",
+    "            file_path = download_dir / filename\n",
+    "\n",
+    "            # Save the file\n",
+    "            with open(file_path, 'wb') as f:\n",
+    "                f.write(file_response.content)\n",
+    "\n",
+    "            # Add the file path to the question data\n",
+    "            question_data['downloaded_file_path'] = str(file_path)\n",
+    "            print(f\"Downloaded file to: {file_path}\")\n",
+    "\n",
+    "        return question_data\n",
+    "\n",
+    "    except requests.exceptions.RequestException as e:\n",
+    "        print(f\"Error fetching question: {e}\")\n",
+    "        return None\n",
+    "    except json.JSONDecodeError as e:\n",
+    "        print(f\"Error parsing JSON response: {e}\")\n",
+    "        return None\n",
+    "    except Exception as e:\n",
+    "        print(f\"Unexpected error: {e}\")\n",
+    "        return None"
+   ],
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "/Users/vladi/miniconda3/envs/gaia/lib/python3.11/site-packages/tqdm/auto.py:21: TqdmWarning: IProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.html\n",
+      "  from .autonotebook import tqdm as notebook_tqdm\n"
+     ]
+    }
+   ],
+   "execution_count": 1
+  },
+  {
+   "metadata": {
+    "ExecuteTime": {
+     "end_time": "2025-06-16T14:22:32.140555Z",
+     "start_time": "2025-06-16T14:22:31.056873Z"
+    }
+   },
+   "cell_type": "code",
+   "source": [
+    "question = get_random_question()\n",
+    "if question:\n",
+    "    print(f\"Task ID: {question.get('task_id')}\")\n",
+    "    print(f\"Question: {question.get('question')}\")\n",
+    "    print(f\"Level: {question.get('Level')}\")\n",
+    "    if 'downloaded_file_path' in question:\n",
+    "        print(f\"Downloaded file: {question['downloaded_file_path']}\")"
+   ],
+   "id": "55d7941445304e9b",
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Downloaded file to: downloaded_files/99c9cc74-fdc8-46c6-8f8d-3ce2d3bfeea3.mp3\n",
+      "Task ID: 99c9cc74-fdc8-46c6-8f8d-3ce2d3bfeea3\n",
+      "Question: 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",
+      "\n",
+      "In 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",
+      "\n",
+      "Please format your response as a comma separated list of ingredients. Also, please alphabetize the ingredients.\n",
+      "Level: 1\n",
+      "Downloaded file: downloaded_files/99c9cc74-fdc8-46c6-8f8d-3ce2d3bfeea3.mp3\n"
+     ]
+    }
+   ],
+   "execution_count": 2
+  },
+  {
+   "metadata": {
+    "ExecuteTime": {
+     "end_time": "2025-06-16T14:22:36.176421Z",
+     "start_time": "2025-06-16T14:22:36.148111Z"
+    }
+   },
+   "cell_type": "code",
+   "source": [
+    "config = dotenv_values()\n",
+    "\n",
+    "client = AzureOpenAI(\n",
+    "    api_key=config[\"AZURE_OPENAI_API_KEY\"],\n",
+    "    azure_endpoint=config[\"AZURE_OPENAI_API_BASE\"],\n",
+    "    api_version=config[\"AZURE_OPENAI_API_VERSION\"]\n",
+    ")\n",
+    "model_id=config[\"AZURE_OPENAI_CHAT_MODEL\"]"
+   ],
+   "id": "99393f634f21563f",
+   "outputs": [],
+   "execution_count": 3
+  },
+  {
+   "metadata": {},
+   "cell_type": "markdown",
+   "source": "# Question Processing",
+   "id": "1290570b730fda4"
+  },
+  {
+   "metadata": {
+    "ExecuteTime": {
+     "end_time": "2025-06-16T14:22:43.615612Z",
+     "start_time": "2025-06-16T14:22:43.204200Z"
+    }
+   },
+   "cell_type": "code",
+   "source": [
+    "response = requests.get(f\"{test_api_base}/questions\", timeout=15)\n",
+    "response.raise_for_status()\n",
+    "\n",
+    "questions_data = response.json()"
+   ],
+   "id": "9f6fe414bc8fb090",
+   "outputs": [],
+   "execution_count": 4
+  },
+  {
+   "metadata": {
+    "ExecuteTime": {
+     "end_time": "2025-06-16T14:22:43.692064Z",
+     "start_time": "2025-06-16T14:22:43.684720Z"
+    }
+   },
+   "cell_type": "code",
+   "source": "pd.DataFrame(questions_data)",
+   "id": "3d0ab116de02315e",
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "                                 task_id  \\\n",
+       "0   8e867cd7-cff9-4e6c-867a-ff5ddc2550be   \n",
+       "1   a1e91b78-d3d8-4675-bb8d-62741b4b68a6   \n",
+       "2   2d83110e-a098-4ebb-9987-066c06fa42d0   \n",
+       "3   cca530fc-4052-43b2-b130-b30968d8aa44   \n",
+       "4   4fc2f1ae-8625-45b5-ab34-ad4433bc21f8   \n",
+       "5   6f37996b-2ac7-44b0-8e68-6d28256631b4   \n",
+       "6   9d191bce-651d-4746-be2d-7ef8ecadb9c2   \n",
+       "7   cabe07ed-9eca-40ea-8ead-410ef5e83f91   \n",
+       "8   3cef3a44-215e-4aed-8e3b-b1e3f08063b7   \n",
+       "9   99c9cc74-fdc8-46c6-8f8d-3ce2d3bfeea3   \n",
+       "10  305ac316-eef6-4446-960a-92d80d542f82   \n",
+       "11  f918266a-b3e0-4914-865d-4faa564f1aef   \n",
+       "12  3f57289b-8c60-48be-bd80-01f8099ca449   \n",
+       "13  1f975693-876d-457b-a649-393859e79bf3   \n",
+       "14  840bfca7-4f7b-481a-8794-c560c340185d   \n",
+       "15  bda648d7-d618-4883-88f4-3466eabd860e   \n",
+       "16  cf106601-ab4f-4af9-b045-5295fe67b37d   \n",
+       "17  a0c07678-e491-4bbc-8f0b-07405144218f   \n",
+       "18  7bd855d8-463d-4ed5-93ca-5fe35145f733   \n",
+       "19  5a0c1adf-205e-4841-a666-7c3ef95def9d   \n",
+       "\n",
+       "                                             question Level  \\\n",
+       "0   How many studio albums were published by Merce...     1   \n",
+       "1   In the video https://www.youtube.com/watch?v=L...     1   \n",
+       "2   .rewsna eht sa \"tfel\" drow eht fo etisoppo eht...     1   \n",
+       "3   Review the chess position provided in the imag...     1   \n",
+       "4   Who nominated the only Featured Article on Eng...     1   \n",
+       "5   Given this table defining * on the set S = {a,...     1   \n",
+       "6   Examine the video at https://www.youtube.com/w...     1   \n",
+       "7   What is the surname of the equine veterinarian...     1   \n",
+       "8   I'm making a grocery list for my mom, but she'...     1   \n",
+       "9   Hi, I'm making a pie but I could use some help...     1   \n",
+       "10  Who did the actor who played Ray in the Polish...     1   \n",
+       "11  What is the final numeric output from the atta...     1   \n",
+       "12  How many at bats did the Yankee with the most ...     1   \n",
+       "13  Hi, I was out sick from my classes on Friday, ...     1   \n",
+       "14  On June 6, 2023, an article by Carolyn Collins...     1   \n",
+       "15  Where were the Vietnamese specimens described ...     1   \n",
+       "16  What country had the least number of athletes ...     1   \n",
+       "17  Who are the pitchers with the number before an...     1   \n",
+       "18  The attached Excel file contains the sales of ...     1   \n",
+       "19  What is the first name of the only Malko Compe...     1   \n",
+       "\n",
+       "                                    file_name  \n",
+       "0                                              \n",
+       "1                                              \n",
+       "2                                              \n",
+       "3    cca530fc-4052-43b2-b130-b30968d8aa44.png  \n",
+       "4                                              \n",
+       "5                                              \n",
+       "6                                              \n",
+       "7                                              \n",
+       "8                                              \n",
+       "9    99c9cc74-fdc8-46c6-8f8d-3ce2d3bfeea3.mp3  \n",
+       "10                                             \n",
+       "11    f918266a-b3e0-4914-865d-4faa564f1aef.py  \n",
+       "12                                             \n",
+       "13   1f975693-876d-457b-a649-393859e79bf3.mp3  \n",
+       "14                                             \n",
+       "15                                             \n",
+       "16                                             \n",
+       "17                                             \n",
+       "18  7bd855d8-463d-4ed5-93ca-5fe35145f733.xlsx  \n",
+       "19                                             "
+      ],
+      "text/html": [
+       "<div>\n",
+       "<style scoped>\n",
+       "    .dataframe tbody tr th:only-of-type {\n",
+       "        vertical-align: middle;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe tbody tr th {\n",
+       "        vertical-align: top;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe thead th {\n",
+       "        text-align: right;\n",
+       "    }\n",
+       "</style>\n",
+       "<table border=\"1\" class=\"dataframe\">\n",
+       "  <thead>\n",
+       "    <tr style=\"text-align: right;\">\n",
+       "      <th></th>\n",
+       "      <th>task_id</th>\n",
+       "      <th>question</th>\n",
+       "      <th>Level</th>\n",
+       "      <th>file_name</th>\n",
+       "    </tr>\n",
+       "  </thead>\n",
+       "  <tbody>\n",
+       "    <tr>\n",
+       "      <th>0</th>\n",
+       "      <td>8e867cd7-cff9-4e6c-867a-ff5ddc2550be</td>\n",
+       "      <td>How many studio albums were published by Merce...</td>\n",
+       "      <td>1</td>\n",
+       "      <td></td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>1</th>\n",
+       "      <td>a1e91b78-d3d8-4675-bb8d-62741b4b68a6</td>\n",
+       "      <td>In the video https://www.youtube.com/watch?v=L...</td>\n",
+       "      <td>1</td>\n",
+       "      <td></td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>2</th>\n",
+       "      <td>2d83110e-a098-4ebb-9987-066c06fa42d0</td>\n",
+       "      <td>.rewsna eht sa \"tfel\" drow eht fo etisoppo eht...</td>\n",
+       "      <td>1</td>\n",
+       "      <td></td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>3</th>\n",
+       "      <td>cca530fc-4052-43b2-b130-b30968d8aa44</td>\n",
+       "      <td>Review the chess position provided in the imag...</td>\n",
+       "      <td>1</td>\n",
+       "      <td>cca530fc-4052-43b2-b130-b30968d8aa44.png</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>4</th>\n",
+       "      <td>4fc2f1ae-8625-45b5-ab34-ad4433bc21f8</td>\n",
+       "      <td>Who nominated the only Featured Article on Eng...</td>\n",
+       "      <td>1</td>\n",
+       "      <td></td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>5</th>\n",
+       "      <td>6f37996b-2ac7-44b0-8e68-6d28256631b4</td>\n",
+       "      <td>Given this table defining * on the set S = {a,...</td>\n",
+       "      <td>1</td>\n",
+       "      <td></td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>6</th>\n",
+       "      <td>9d191bce-651d-4746-be2d-7ef8ecadb9c2</td>\n",
+       "      <td>Examine the video at https://www.youtube.com/w...</td>\n",
+       "      <td>1</td>\n",
+       "      <td></td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>7</th>\n",
+       "      <td>cabe07ed-9eca-40ea-8ead-410ef5e83f91</td>\n",
+       "      <td>What is the surname of the equine veterinarian...</td>\n",
+       "      <td>1</td>\n",
+       "      <td></td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>8</th>\n",
+       "      <td>3cef3a44-215e-4aed-8e3b-b1e3f08063b7</td>\n",
+       "      <td>I'm making a grocery list for my mom, but she'...</td>\n",
+       "      <td>1</td>\n",
+       "      <td></td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>9</th>\n",
+       "      <td>99c9cc74-fdc8-46c6-8f8d-3ce2d3bfeea3</td>\n",
+       "      <td>Hi, I'm making a pie but I could use some help...</td>\n",
+       "      <td>1</td>\n",
+       "      <td>99c9cc74-fdc8-46c6-8f8d-3ce2d3bfeea3.mp3</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>10</th>\n",
+       "      <td>305ac316-eef6-4446-960a-92d80d542f82</td>\n",
+       "      <td>Who did the actor who played Ray in the Polish...</td>\n",
+       "      <td>1</td>\n",
+       "      <td></td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>11</th>\n",
+       "      <td>f918266a-b3e0-4914-865d-4faa564f1aef</td>\n",
+       "      <td>What is the final numeric output from the atta...</td>\n",
+       "      <td>1</td>\n",
+       "      <td>f918266a-b3e0-4914-865d-4faa564f1aef.py</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>12</th>\n",
+       "      <td>3f57289b-8c60-48be-bd80-01f8099ca449</td>\n",
+       "      <td>How many at bats did the Yankee with the most ...</td>\n",
+       "      <td>1</td>\n",
+       "      <td></td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>13</th>\n",
+       "      <td>1f975693-876d-457b-a649-393859e79bf3</td>\n",
+       "      <td>Hi, I was out sick from my classes on Friday, ...</td>\n",
+       "      <td>1</td>\n",
+       "      <td>1f975693-876d-457b-a649-393859e79bf3.mp3</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>14</th>\n",
+       "      <td>840bfca7-4f7b-481a-8794-c560c340185d</td>\n",
+       "      <td>On June 6, 2023, an article by Carolyn Collins...</td>\n",
+       "      <td>1</td>\n",
+       "      <td></td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>15</th>\n",
+       "      <td>bda648d7-d618-4883-88f4-3466eabd860e</td>\n",
+       "      <td>Where were the Vietnamese specimens described ...</td>\n",
+       "      <td>1</td>\n",
+       "      <td></td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>16</th>\n",
+       "      <td>cf106601-ab4f-4af9-b045-5295fe67b37d</td>\n",
+       "      <td>What country had the least number of athletes ...</td>\n",
+       "      <td>1</td>\n",
+       "      <td></td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>17</th>\n",
+       "      <td>a0c07678-e491-4bbc-8f0b-07405144218f</td>\n",
+       "      <td>Who are the pitchers with the number before an...</td>\n",
+       "      <td>1</td>\n",
+       "      <td></td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>18</th>\n",
+       "      <td>7bd855d8-463d-4ed5-93ca-5fe35145f733</td>\n",
+       "      <td>The attached Excel file contains the sales of ...</td>\n",
+       "      <td>1</td>\n",
+       "      <td>7bd855d8-463d-4ed5-93ca-5fe35145f733.xlsx</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>19</th>\n",
+       "      <td>5a0c1adf-205e-4841-a666-7c3ef95def9d</td>\n",
+       "      <td>What is the first name of the only Malko Compe...</td>\n",
+       "      <td>1</td>\n",
+       "      <td></td>\n",
+       "    </tr>\n",
+       "  </tbody>\n",
+       "</table>\n",
+       "</div>"
+      ]
+     },
+     "execution_count": 5,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "execution_count": 5
+  },
+  {
+   "metadata": {
+    "ExecuteTime": {
+     "end_time": "2025-06-16T14:28:12.335526Z",
+     "start_time": "2025-06-16T14:28:12.332298Z"
+    }
+   },
+   "cell_type": "code",
+   "source": [
+    "df = pd.DataFrame(questions_data)\n",
+    "df[df[\"file_name\"] == \"\"][\"question\"].values"
+   ],
+   "id": "d624cc31dc5bb13d",
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "array(['How many studio albums were published by Mercedes Sosa between 2000 and 2009 (included)? You can use the latest 2022 version of english wikipedia.',\n",
+       "       'In the video https://www.youtube.com/watch?v=L1vXCYZAYYM, what is the highest number of bird species to be on camera simultaneously?',\n",
+       "       '.rewsna eht sa \"tfel\" drow eht fo etisoppo eht etirw ,ecnetnes siht dnatsrednu uoy fI',\n",
+       "       'Who nominated the only Featured Article on English Wikipedia about a dinosaur that was promoted in November 2016?',\n",
+       "       '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.',\n",
+       "       '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?\"',\n",
+       "       \"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?\",\n",
+       "       \"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.\",\n",
+       "       '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.',\n",
+       "       'How many at bats did the Yankee with the most walks in the 1977 regular season have that same season?',\n",
+       "       '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?',\n",
+       "       \"Where were the Vietnamese specimens described by Kuznetzov in Nedoshivina's 2010 paper eventually deposited? Just give me the city name without abbreviations.\",\n",
+       "       \"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.\",\n",
+       "       \"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.\",\n",
+       "       '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?'],\n",
+       "      dtype=object)"
+      ]
+     },
+     "execution_count": 11,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "execution_count": 11
+  },
+  {
+   "metadata": {
+    "ExecuteTime": {
+     "end_time": "2025-06-06T11:41:12.207070Z",
+     "start_time": "2025-06-06T11:41:12.193633Z"
+    }
+   },
+   "cell_type": "code",
+   "source": [
+    "@tool\n",
+    "def read_file(file_path_str: str) -> str:\n",
+    "    \"\"\"\n",
+    "    A tool that reads the contents of a file and returns them as text.\n",
+    "\n",
+    "    Args:\n",
+    "        file_path_str: The path to the file that should be read.\n",
+    "    \"\"\"\n",
+    "\n",
+    "    file_path = Path(file_path_str)\n",
+    "    file_path = file_path.resolve()\n",
+    "    if not file_path.exists() or not file_path.is_file():\n",
+    "        raise ValueError(f\"File {file_path} does not exist or is not a file.\")\n",
+    "\n",
+    "    switcher = {\n",
+    "        \".txt\": lambda: file_path.read_text(encoding=\"utf-8\"),\n",
+    "        \".csv\": lambda: file_path.read_text(encoding=\"utf-8\"),\n",
+    "        \".py\": lambda: file_path.read_text(encoding=\"utf-8\"),\n",
+    "        \".xlsx\": lambda: pd.read_excel(file_path).to_string(),\n",
+    "    }\n",
+    "\n",
+    "    return switcher.get(file_path.suffix, lambda: \"Unsupported file type\")()\n",
+    "\n",
+    "@tool\n",
+    "def wikipedia_search(page: str) -> str:\n",
+    "    \"\"\"\n",
+    "    A tool that returns the contents for a specific Wikipedia page if found, or \"This content has been truncated to stay below 0 characters\" if page not found.\n",
+    "\n",
+    "    Args:\n",
+    "        page: The Wikipedia page for which to retrieve the content.\n",
+    "    \"\"\"\n",
+    "    wiki_wiki = wikipediaapi.Wikipedia(user_agent='LLM Agents', language='en')\n",
+    "\n",
+    "    page_py = wiki_wiki.page(page)\n",
+    "    return page_py.text\n",
+    "\n",
+    "def process_question(question_data: dict[str, Any]) -> dict[str, str]:\n",
+    "    task_id = question_data.get(\"task_id\")\n",
+    "    question_text = question_data.get(\"question\")\n",
+    "    format_instructions = \"You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.\"\n",
+    "    adjusted_question_text = f\"{format_instructions}\\n\\n{question_text}\"\n",
+    "\n",
+    "    file_path = None\n",
+    "    if question_data.get(\"file_name\"):\n",
+    "        task_id = question_data[\"task_id\"]\n",
+    "        file_url = f\"{test_api_base}/files/{task_id}\"\n",
+    "\n",
+    "        download_dir = Path(\"downloaded_files\")\n",
+    "        download_dir.mkdir(exist_ok=True)\n",
+    "\n",
+    "        file_response = requests.get(file_url, timeout=30)\n",
+    "        file_response.raise_for_status()\n",
+    "\n",
+    "        file_path = download_dir / question_data.get(\"file_name\")\n",
+    "\n",
+    "        with open(file_path, 'wb') as f:\n",
+    "            f.write(file_response.content)\n",
+    "\n",
+    "    agent = CodeAgent(tools=[read_file,\n",
+    "                             # wikipedia_search\n",
+    "                                DuckDuckGoSearchTool(),\n",
+    "                                VisitWebpageTool()\n",
+    "                             ], model=model, max_steps=10,\n",
+    "                      # verbosity_level=0,\n",
+    "                      max_print_outputs_length=0)\n",
+    "\n",
+    "    if file_path and file_path.suffix in ['.png', '.jpg', '.jpeg']:   # I know, it's inconsistent\n",
+    "        answer = agent.run(task=adjusted_question_text, images=[Image.open(file_path)])\n",
+    "    else:\n",
+    "        answer = agent.run(task=f\"{adjusted_question_text}{f' File: |{file_path}|' if question_data.get('file_name') else ''}\", )\n",
+    "\n",
+    "    # print(f\"Task ID: {task_id}, Question: {question_text}, Answer: {answer}\")\n",
+    "\n",
+    "    return {\n",
+    "        \"task_id\": task_id,\n",
+    "        \"submitted_answer\": answer,\n",
+    "        \"question\": question_text\n",
+    "    }\n",
+    "\n",
+    "\n",
+    "def run_agents_parallel(questions_data: List[Dict[str, Any]], max_workers: int = 4) -> Tuple[Dict[str, Any], List[Dict[str, Any]]]:\n",
+    "    start = time.time()\n",
+    "\n",
+    "    answers = []\n",
+    "    results_log = []\n",
+    "\n",
+    "    with concurrent.futures.ThreadPoolExecutor(max_workers=max_workers) as executor:\n",
+    "        future_to_question = {executor.submit(process_question, q): q for q in questions_data}\n",
+    "\n",
+    "        for future in tqdm(concurrent.futures.as_completed(future_to_question)):\n",
+    "            try:\n",
+    "                answer = future.result()\n",
+    "                results_log.append(answer)\n",
+    "                answers.append(answer)\n",
+    "\n",
+    "            except Exception as e:\n",
+    "                print(f\"Question processing failed: {e}\")\n",
+    "\n",
+    "    submission_data = {\n",
+    "        \"username\": \"vladi\",\n",
+    "        \"agent_code\": \"https://huggingface.co/spaces/vladi/AgentsGAIAFun\",\n",
+    "        \"answers\": answers\n",
+    "    }\n",
+    "    end = time.time()\n",
+    "    print(f\"Processing time (parallel): {end - start:.2f} seconds\")\n",
+    "\n",
+    "    return submission_data, results_log\n",
+    "\n",
+    "def run_agents(questions_data: list[{}]):\n",
+    "    start = time.time()\n",
+    "\n",
+    "    answers = []\n",
+    "    results_log = []\n",
+    "    for question_data in tqdm(questions_data):\n",
+    "\n",
+    "        answer = process_question(question_data)\n",
+    "\n",
+    "        results_log.append(answer)\n",
+    "        answers.append(answer)\n",
+    "\n",
+    "    submission_data = {\n",
+    "        \"username\": \"vladi\",\n",
+    "        \"agent_code\": \"https://huggingface.co/spaces/vladi/AgentsGAIAFun\",\n",
+    "        \"answers\": answers\n",
+    "    }\n",
+    "\n",
+    "    end = time.time()\n",
+    "    print(f\"Processing time (sequential): {end - start:.2f} seconds\")\n",
+    "\n",
+    "    return submission_data, results_log\n",
+    "\n",
+    "def submit_answers(submission_data: dict):\n",
+    "    print(f\"Submitting {len(submission_data['answers'])} answers\")\n",
+    "\n",
+    "    response = requests.post(f\"{test_api_base}/submit\", json=submission_data, timeout=60)\n",
+    "    response.raise_for_status()\n",
+    "    result_data = response.json()\n",
+    "\n",
+    "    return result_data\n"
+   ],
+   "id": "74bce95503481798",
+   "outputs": [],
+   "execution_count": 28
+  },
+  {
+   "metadata": {},
+   "cell_type": "code",
+   "source": [
+    "submission_data, results_log = run_agents(questions_data[:1])\n",
+    "# submission_data, results_log = run_agents_parallel(questions_data)\n",
+    "results_df = pd.DataFrame(results_log)\n",
+    "\n",
+    "results_log"
+   ],
+   "id": "57e1c5515e9bf8a1",
+   "outputs": [],
+   "execution_count": null
+  },
+  {
+   "metadata": {
+    "ExecuteTime": {
+     "end_time": "2025-05-01T14:52:59.689261Z",
+     "start_time": "2025-05-01T14:52:59.686607Z"
+    }
+   },
+   "cell_type": "code",
+   "source": "submission_data",
+   "id": "d66d9affd58d3428",
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "{'username': 'vladi',\n",
+       " 'agent_code': 'https://huggingface.co/spaces/vladi/AgentsGAIAFun',\n",
+       " 'answers': [{'task_id': '2d83110e-a098-4ebb-9987-066c06fa42d0',\n",
+       "   'submitted_answer': 'right',\n",
+       "   'question': '.rewsna eht sa \"tfel\" drow eht fo etisoppo eht etirw ,ecnetnes siht dnatsrednu uoy fI'},\n",
+       "  {'task_id': 'a1e91b78-d3d8-4675-bb8d-62741b4b68a6',\n",
+       "   'submitted_answer': 'I lack tools for video analysis. This task requires external video analysis tools or manual review.',\n",
+       "   'question': 'In the video https://www.youtube.com/watch?v=L1vXCYZAYYM, what is the highest number of bird species to be on camera simultaneously?'},\n",
+       "  {'task_id': 'cca530fc-4052-43b2-b130-b30968d8aa44',\n",
+       "   'submitted_answer': 'Nd2',\n",
+       "   'question': \"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.\"},\n",
+       "  {'task_id': '6f37996b-2ac7-44b0-8e68-6d28256631b4',\n",
+       "   'submitted_answer': 'b,e',\n",
+       "   'question': '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.'},\n",
+       "  {'task_id': '9d191bce-651d-4746-be2d-7ef8ecadb9c2',\n",
+       "   'submitted_answer': 'I cannot directly analyze video content, including YouTube videos. Please provide a transcript or text if available.',\n",
+       "   'question': '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?\"'},\n",
+       "  {'task_id': '8e867cd7-cff9-4e6c-867a-ff5ddc2550be',\n",
+       "   'submitted_answer': \"I cannot directly access Wikipedia or other data sources to answer this task. Please refer to Mercedes Sosa's discography on Wikipedia or other trusted platforms. Look for studio albums released between 2000 and 2009 and count them.\",\n",
+       "   'question': 'How many studio albums were published by Mercedes Sosa between 2000 and 2009 (included)? You can use the latest 2022 version of english wikipedia.'},\n",
+       "  {'task_id': '3cef3a44-215e-4aed-8e3b-b1e3f08063b7',\n",
+       "   'submitted_answer': 'broccoli, celery, corn, green beans, lettuce, sweet potatoes, zucchini',\n",
+       "   'question': \"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.\"},\n",
+       "  {'task_id': '99c9cc74-fdc8-46c6-8f8d-3ce2d3bfeea3',\n",
+       "   'submitted_answer': \"I can't process the MP3 file directly. Could you provide me with a text transcription of the audio contents?\",\n",
+       "   'question': '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.'},\n",
+       "  {'task_id': '305ac316-eef6-4446-960a-92d80d542f82',\n",
+       "   'submitted_answer': 'Piotr',\n",
+       "   'question': '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.'},\n",
+       "  {'task_id': 'f918266a-b3e0-4914-865d-4faa564f1aef',\n",
+       "   'submitted_answer': 0,\n",
+       "   'question': 'What is the final numeric output from the attached Python code?'},\n",
+       "  {'task_id': '4fc2f1ae-8625-45b5-ab34-ad4433bc21f8',\n",
+       "   'submitted_answer': 'The only dinosaur-related article promoted to Featured Article status on English Wikipedia in November 2016 is **\"Dakota (dinosaur)\"**. The nomination for this article was made by **FunkMonk**, a prominent editor known for contributions to paleontology-related articles.',\n",
+       "   'question': 'Who nominated the only Featured Article on English Wikipedia about a dinosaur that was promoted in November 2016?'},\n",
+       "  {'task_id': '3f57289b-8c60-48be-bd80-01f8099ca449',\n",
+       "   'submitted_answer': 'The requested file does not exist. Please provide further instructions or clarify how to access the 1977 Yankees player statistics.',\n",
+       "   'question': 'How many at bats did the Yankee with the most walks in the 1977 regular season have that same season?'},\n",
+       "  {'task_id': '1f975693-876d-457b-a649-393859e79bf3',\n",
+       "   'submitted_answer': 'The provided file is an audio file (.mp3). Please use a transcription service like Otter.ai, Rev, or Whisper speech-to-text tools to extract the spoken information. Once transcribed, I can assist further in analyzing the text for the required page numbers.',\n",
+       "   'question': \"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.\"},\n",
+       "  {'task_id': 'bda648d7-d618-4883-88f4-3466eabd860e',\n",
+       "   'submitted_answer': 'Hanoi',\n",
+       "   'question': \"Where were the Vietnamese specimens described by Kuznetzov in Nedoshivina's 2010 paper eventually deposited? Just give me the city name without abbreviations.\"},\n",
+       "  {'task_id': 'cf106601-ab4f-4af9-b045-5295fe67b37d',\n",
+       "   'submitted_answer': 'AFG',\n",
+       "   'question': \"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.\"},\n",
+       "  {'task_id': '840bfca7-4f7b-481a-8794-c560c340185d',\n",
+       "   'submitted_answer': 'To answer your query, I conducted research based on the article mentioned and the linked paper. Here\\'s the result:\\n\\nCarolyn Collins Petersen\\'s article on *Universe Today*, published on June 6, 2023, references a team’s observations. The paper linked at the bottom of the article is titled:\\n\\n**\"Spitzer Observations of the Massive Infrared Dark Cloud G38.91-0.44: A Detailed Look at Star Formation.\"**\\n\\nIn this paper, the work performed by R. G. Arendt was supported under the **NASA award number 80GSFC21M0002**.\\n\\nLet me know if you need further clarification or assistance!',\n",
+       "   'question': '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?'},\n",
+       "  {'task_id': '7bd855d8-463d-4ed5-93ca-5fe35145f733',\n",
+       "   'submitted_answer': 'I cannot process the Excel file due to the limitations in using specific libraries or tools for Excel handling.',\n",
+       "   'question': '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.'},\n",
+       "  {'task_id': '5a0c1adf-205e-4841-a666-7c3ef95def9d',\n",
+       "   'submitted_answer': 'Anna',\n",
+       "   'question': '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?'},\n",
+       "  {'task_id': 'a0c07678-e491-4bbc-8f0b-07405144218f',\n",
+       "   'submitted_answer': 'To address the query effectively, I will answer based on standard baseball numbering and existing data from July 2023 regarding Taishō Tamai:\\n\\nAs of July 2023, Taishō Tamai wore jersey number **65** for the Yomiuri Giants, a Japanese professional baseball team. To determine the pitchers with numbers before and after his:\\n\\n1. **Pitcher Before** (jersey number **64**): **Ito**  \\n2. **Pitcher After** (jersey number **66**): **Takahashi**  \\n\\nTherefore, the result is:\\n\\n**Ito, Takahashi**',\n",
+       "   'question': \"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.\"}]}"
+      ]
+     },
+     "execution_count": 9,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "execution_count": 9
+  },
+  {
+   "metadata": {
+    "ExecuteTime": {
+     "end_time": "2025-05-01T14:53:15.147718Z",
+     "start_time": "2025-05-01T14:53:13.340660Z"
+    }
+   },
+   "cell_type": "code",
+   "source": "submit_answers(submission_data)",
+   "id": "ea6c404b932e9922",
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Submitting 19 answers\n"
+     ]
+    },
+    {
+     "data": {
+      "text/plain": [
+       "{'username': 'vladi',\n",
+       " 'score': 10.0,\n",
+       " 'correct_count': 2,\n",
+       " 'total_attempted': 19,\n",
+       " 'message': 'Score calculated successfully: 2/20 total questions answered correctly (19 valid tasks attempted). Score did not improve previous record, leaderboard not updated.',\n",
+       " 'timestamp': '2025-05-01T14:53:15.124360+00:00'}"
+      ]
+     },
+     "execution_count": 10,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "execution_count": 10
+  },
+  {
+   "metadata": {
+    "ExecuteTime": {
+     "end_time": "2025-06-06T11:39:35.785365Z",
+     "start_time": "2025-06-06T11:39:35.514993Z"
+    }
+   },
+   "cell_type": "code",
+   "source": "wikipedia_search(\"Mercedes Sosa\")",
+   "id": "57e58477ff361cff",
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "'Haydée Mercedes \"La Negra\" Sosa (Latin American Spanish: [meɾˈseðes ˈsosa]; 9 July 1935 – 4 October 2009) was an Argentine singer who was popular throughout Latin America and many countries outside the region. With her roots in Argentine folk music, Sosa became one of the preeminent exponents of El nuevo cancionero. She gave voice to songs written by many Latin American songwriters. Her music made people hail her as the \"voice of the voiceless ones\". She was often called \"the conscience of Latin America\".\\nSosa performed in venues such as the Lincoln Center in New York City, the Théâtre Mogador in Paris, the Sistine Chapel in Vatican City, as well as sold-out shows in New York\\'s Carnegie Hall and the Roman Colosseum during her final decade of life. Her career spanned four decades and she was the recipient of six Latin Grammy awards (2000, 2003, 2004, 2006, 2009, 2011), including a Latin Grammy Lifetime Achievement Award in 2004 and two posthumous Latin Grammy Award for Best Folk Album in 2009 and 2011. She won the Premio Gardel in 2000, the main musical award in Argentina. She served as an ambassador for UNICEF.\\n\\nLife\\nSosa was born on 9 July 1935, in San Miguel de Tucumán, in the northwestern Argentine province of Tucumán, of mestizo ancestry. She was of French, Spanish and Diaguita descent. Her nickname \"la negra\", which is a common nickname in Argentina for people with darker complexion, is a reference to her indigenous heritage. Her parents, a day laborer and a washerwoman, were Peronists, although they never registered in the party, and she started her career as a singer for the Peronist Party in Tucuman under the name Gladys Osorio. In 1950, at age fifteen, she won a singing competition organized by a local radio station and was given a contract to perform for two months. She recorded her first album, La Voz de la Zafra, in 1959. A performance at the 1965 Cosquín National Folklore Festival—where she was introduced and brought to the stage while sitting in the audience by fellow folk singer Jorge Cafrune— brought her to the attention of the Argentine public. Sosa and her first husband, Manuel Oscar Matus, with whom she had one son, were key players in the mid-60s nueva canción movement (which was called nuevo cancionero in Argentina). Her second record was Canciones con Fundamento, a collection of Argentine folk songs.\\n\\nSosa \"spent the late 1960s building her audience in Europe and among the cosmopolitan middle class in Buenos Aires, becoming in the process a much bigger star\" than her contemporaries. In 1967, Sosa toured the United States and Europe with great success. In later years, she performed and recorded extensively, broadening her repertoire to include material from throughout Latin America.\\nIn the early 1970s, Sosa released two concept albums in collaboration with composer Ariel Ramírez and lyricist Félix Luna: Cantata Sudamericana and Mujeres Argentinas (Argentine Women). She also recorded a tribute to Chilean musician Violeta Parra in 1971, including what was to become one of Sosa\\'s signature songs, Gracias a la vida. She further popularized of songs written by Milton Nascimento of Brazil and Pablo Milanés and Silvio Rodríguez both from Cuba. Throughout the decade, she released albums such as Hasta la Victoria in 1972 and Traigo un Pueblo en mi Voz in 1973. They featured songs like \"Cuando tenga la tierra\", written by Ariel Petrocelli and Daniel Toro, which tackles political and social issues like wealth and land inequality. During the 1970s she was a part of two films by the director Leopoldo Torre Nilsson: El Santo de la Espada in 1970 and Güemes, la tierra en armas in 1971, in which she portrayed Juana Azurduy de Padilla, the guerrilla military leader who fought for Argentine independence.\\n\\nAfter the military junta of Jorge Videla came to power in 1976, the atmosphere in Argentina grew increasingly oppressive. Sosa faced death threats against both her and her family, but refused for many years to leave the country. At a concert in La Plata in 1979, Sosa was searched and arrested on stage, along with all those attending the concert. Their release came about through international intervention. Despite attempts to hold more concerts, she was officially barred from performing by the military regime. Banned in her own country, she moved to Paris and then to Madrid. She has spoken publicly about her artistic and emotional struggles during this period of her life. While in exile, she released the album A Quien Doy in 1981. The album included a recording of the song \"Cuando Me Acuerdo de Mi Pais\" which was originally written by the prolific Chilean singer/songwriter, Patricio Manns. The song, which he wrote while also in political exile, expresses the sorrow he felt from being separated from his homeland. She related to this feeling and struggled to continue recording and performing. In an interview with the New York Times, she said, “It was a mental problem, a problem of morale...It wasn’t my throat, or anything physical\".\\nSosa returned to Argentina from her exile in Europe in February 1982, several months before the military regime collapsed as a result of the Falklands War, and gave a series of concerts at the Teatro Ópera in Buenos Aires, where she invited many of her younger colleagues to share the stage. A double album of recordings from these performances became an instant best seller. She then traveled to perform in her home province of Tucuman. However, these performances were largely ignored by mainstream media in the country. In subsequent years, Sosa continued to tour both in Argentina and abroad, performing in such venues as the Lincoln Center in New York City and the Théâtre Mogador in Paris. In poor health for much of the 1990s, she performed a comeback show in Argentina in 1998. In 1994, she played in the Sistine Chapel in Vatican City. In 2002, she sold out both Carnegie Hall in New York and the Colosseum in Rome in the same year.\\n\\nA supporter of Perón, she favored leftist causes throughout her life. She supported President Raul Alfonsin in the election of 1983 which marked the return of democracy in Argentina following the dictatorship. She referred to this election as \"Argentina\\'s Spring\" She opposed President Carlos Menem, who was in office from 1989 to 1999, and supported the election of Néstor Kirchner, who became president in 2003.\\nSosa was a UNESCO Goodwill Ambassador for Latin America and the Caribbean.\\nSosa disliked being identified as a protest singer. While she was outright in her political stances, Sosa said the following on the position of the artist:\\n\\n“An artist isn’t political in the party political sense – they have a constituency, which is their public – it is the poetry that matters most of all.”\\nIn a career spanning four decades, she worked with performers across several genres and generations, folk, opera, pop, rock, including Martha Argerich, Andrea Bocelli, David Broza, Franco Battiato, Jaime Roos, Joan Baez, Francis Cabrel, Gal Costa, Luz Casal, Lila Downs, Lucio Dalla, Maria Farantouri, Lucecita Benitez, Nilda Fernández, Charly Garcia, León Gieco, Gian Marco, Nana Mouskouri, Pablo Milanés, Holly Near, Milton Nascimento, Pata Negra, Fito Páez, Franco De Vita, Lourdes Pérez, Luciano Pavarotti, Silvio Rodríguez, Ismael Serrano, Shakira, Sting, Caetano Veloso, Julieta Venegas, Gustavo Cerati and Konstantin Wecker\\nSosa participated in a 1999 production of Ariel Ramírez\\'s Misa Criolla. Her song Balderrama is featured in the 2008 movie Che, starring Benicio del Toro as the Argentine Marxist revolutionary Che Guevara.\\nSosa was the co-chair of the Earth Charter International Commission.\\n\\nAwards\\nSosa won the Latin Grammy Award for Best Folk Album in 2000 (Misa Criolla), 2003 (Acústico), 2006 (Corazón Libre), 2009 (Cantora 1, which also won Best Recording Package and was nominated for Album of the Year), and 2011 (Deja La Vida Volar), as well as several international awards.\\nIn 1995, Konex Foundation from Argentina granted her the Diamond Konex Award, one of the most prestigious awards in Argentina, as the most important personality in the popular music of her country in the last decade.\\n\\nDeath\\nSuffering from recurrent endocrine and respiratory problems in later years, the 74-year-old Sosa was hospitalized in Buenos Aires on 18 September 2009. She died from multiple organ failure on 4 October 2009, at 5:15 am. She is survived by one son, Fabián Matus, born of her first marriage. He said: \"She lived her 74 years to the fullest. She had done practically everything she wanted, she didn\\'t have any type of barrier or any type of fear that limited her\". The hospital expressed its sympathies to her relatives. Her website featured the following: \"Her undisputed talent, her honesty and her profound convictions leave a great legacy to future generations\".\\nHer body was placed on display at the National Congress building in Buenos Aires for the public to pay their respects, and President Fernández de Kirchner ordered three days of national mourning. Thousands had queued by the end of the day.\\nSosa\\'s obituary in The Daily Telegraph said she was \"an unrivalled interpreter of works by her compatriot, the Argentine Atahualpa Yupanqui, and Chile\\'s Violeta Parra\". Helen Popper of Reuters reported her death by saying she \"fought South America\\'s dictators with her voice and became a giant of contemporary Latin American music\". Sosa received three Latin Grammy nominations for her album, in 2009 . She went on to win Best Folk Album about a month after her death.\\n\\nTributes\\nIn 2019, Sosa was celebrated by a Google Doodle. The doodle was showcased in Argentina, Chile, Uruguay, Paraguay, Bolivia, Peru, Ecuador, Cuba, Iceland, Sweden, Serbia, Greece, Israel and Vietnam.\\nIn 2023, Rolling Stone ranked Sosa at number 160 on its list of the 200 Greatest Singers of All Time.\\n\\nDiscography\\nSosa recorded forty albums.\\n\\nStudio albums\\nEPs\\nLive albums\\nCompilation albums\\nFilmography\\nGüemes, la tierra en armas (1971)\\nArgentinísima (1972)\\nEsta es mi Argentina (1974)\\nMercedes Sosa, como un pájaro libre (1983)\\nSerá possible el sur: Mercedes Sosa (1985)\\nHistorias de Argentina en vivo (2001)\\n\\nFurther reading\\nReferences\\nExternal links\\n\\nTribute to Mercedes Sosa (in Portuguese BR)\\nMercedes Sosa\\'s website (in Spanish)\\nMercedes Sosa\\'s News (in Spanish)\\nMercedes Sosa at IMDb\\nMercedes Sosa\\'s Discography on Discogs.com'"
+      ]
+     },
+     "execution_count": 27,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "execution_count": 27
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 2
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython2",
+   "version": "2.7.6"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

notebooks/03-vi-handmade-agent.ipynb

@@ -0,0 +1,326 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "id": "initial_id",
+   "metadata": {
+    "collapsed": true,
+    "ExecuteTime": {
+     "end_time": "2025-06-20T12:57:41.792135Z",
+     "start_time": "2025-06-20T12:57:40.889869Z"
+    }
+   },
+   "source": [
+    "import time\n",
+    "import concurrent\n",
+    "from concurrent import futures\n",
+    "from typing import Any, List, Dict, Tuple\n",
+    "\n",
+    "from PIL import Image\n",
+    "from openai import AzureOpenAI\n",
+    "from tqdm.auto import tqdm\n",
+    "\n",
+    "import json\n",
+    "from pathlib import Path\n",
+    "\n",
+    "import pandas as pd\n",
+    "import requests\n",
+    "from dotenv import dotenv_values\n",
+    "from smolagents import tool, DuckDuckGoSearchTool\n",
+    "import wikipediaapi\n"
+   ],
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "/Users/vladi/miniconda3/envs/gaia/lib/python3.11/site-packages/tqdm/auto.py:21: TqdmWarning: IProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.html\n",
+      "  from .autonotebook import tqdm as notebook_tqdm\n"
+     ]
+    }
+   ],
+   "execution_count": 1
+  },
+  {
+   "metadata": {
+    "ExecuteTime": {
+     "end_time": "2025-06-20T12:57:52.502521Z",
+     "start_time": "2025-06-20T12:57:52.498693Z"
+    }
+   },
+   "cell_type": "code",
+   "source": [
+    "questions = ['How many studio albums were published by Mercedes Sosa between 2000 and 2009 (included)? You can use the latest 2022 version of english wikipedia.',\n",
+    "       '.rewsna eht sa \"tfel\" drow eht fo etisoppo eht etirw ,ecnetnes siht dnatsrednu uoy fI',\n",
+    "       'Who nominated the only Featured Article on English Wikipedia about a dinosaur that was promoted in November 2016?',\n",
+    "       '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.',\n",
+    "       '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?\"',\n",
+    "       \"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?\",\n",
+    "       \"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.\",\n",
+    "       '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.',\n",
+    "       'How many at bats did the Yankee with the most walks in the 1977 regular season have that same season?',\n",
+    "       '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?',\n",
+    "       \"Where were the Vietnamese specimens described by Kuznetzov in Nedoshivina's 2010 paper eventually deposited? Just give me the city name without abbreviations.\",\n",
+    "       \"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.\",\n",
+    "       \"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.\",\n",
+    "       '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?']"
+   ],
+   "id": "55d7941445304e9b",
+   "outputs": [],
+   "execution_count": 2
+  },
+  {
+   "metadata": {
+    "ExecuteTime": {
+     "end_time": "2025-06-20T13:10:59.085162Z",
+     "start_time": "2025-06-20T13:10:59.079916Z"
+    }
+   },
+   "cell_type": "code",
+   "source": [
+    "from smolagents import GoogleSearchTool\n",
+    "import requests\n",
+    "import urllib.request\n",
+    "from markdownify import markdownify as md\n",
+    "from bs4 import BeautifulSoup\n",
+    "from dotenv import dotenv_values\n",
+    "from openai import AzureOpenAI\n",
+    "import json\n",
+    "\n",
+    "\n",
+    "\n",
+    "def get_search_results_for(query):\n",
+    "    encoded_query = urllib.parse.urlencode({'q': query})\n",
+    "    url = f'https://html.duckduckgo.com/html?q={encoded_query}'\n",
+    "\n",
+    "    request = urllib.request.Request(url)\n",
+    "    request.add_header('User-Agent', 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.88 Safari/537.36')\n",
+    "\n",
+    "    raw_response = urllib.request.urlopen(request).read()\n",
+    "    html = raw_response.decode(\"utf-8\")\n",
+    "\n",
+    "    soup = BeautifulSoup(html, 'html.parser')\n",
+    "    a_results = soup.select(\"a.result__a\")\n",
+    "\n",
+    "    links = []\n",
+    "    for a_result in a_results:\n",
+    "        # print(a_result)\n",
+    "        url = a_result.attrs['href']\n",
+    "        title = a_result.text\n",
+    "        links.append({\"title\": title,  \"url\": url} )\n",
+    "\n",
+    "    return links\n",
+    "\n",
+    "search_tool = GoogleSearchTool(\"serper\")\n",
+    "\n",
+    "def get_google_search_results_for(query: str):\n",
+    "    return search_tool.forward(query)\n",
+    "\n",
+    "\n",
+    "def load_page_content(url) -> str:\n",
+    "    response = requests.get(url, headers={'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.88 Safari/537.36'})\n",
+    "    page_content = response.content.decode('utf-8')\n",
+    "    page_content_md = md(page_content)\n",
+    "\n",
+    "    return page_content_md\n",
+    "\n"
+   ],
+   "id": "bf91141def5a906",
+   "outputs": [],
+   "execution_count": 24
+  },
+  {
+   "metadata": {
+    "ExecuteTime": {
+     "end_time": "2025-06-20T13:10:59.604614Z",
+     "start_time": "2025-06-20T13:10:59.600718Z"
+    }
+   },
+   "cell_type": "code",
+   "source": [
+    "tools = [{\n",
+    "        \"type\": \"function\",\n",
+    "        \"function\": {\n",
+    "            \"name\": \"get_search_results_for\",\n",
+    "            \"description\": \"Returns the top 10 results for a DuckDuckGo query.\",\n",
+    "            \"parameters\": {\n",
+    "                \"type\": \"object\",\n",
+    "                \"properties\": {\n",
+    "                    \"query\": {\n",
+    "                        \"type\": \"string\",\n",
+    "                        \"description\": \"query to search for on DuckDuckGo\"\n",
+    "                    }\n",
+    "                },\n",
+    "                \"required\": [\n",
+    "                    \"query\"\n",
+    "                ],\n",
+    "                \"additionalProperties\": False\n",
+    "            },\n",
+    "            \"strict\": True\n",
+    "        }\n",
+    "    },\n",
+    "    {\n",
+    "        \"type\": \"function\",\n",
+    "        \"function\": {\n",
+    "            \"name\": \"load_page_content\",\n",
+    "            \"description\": \"Returns the content of a particular webpage.\",\n",
+    "            \"parameters\": {\n",
+    "                \"type\": \"object\",\n",
+    "                \"properties\": {\n",
+    "                    \"url\": {\n",
+    "                        \"type\": \"string\",\n",
+    "                        \"description\": \"Url of the webpage for which to retrieve the content\"\n",
+    "                    }\n",
+    "                },\n",
+    "                \"required\": [\n",
+    "                    \"url\"\n",
+    "                ],\n",
+    "                \"additionalProperties\": False\n",
+    "            },\n",
+    "            \"strict\": True\n",
+    "        }\n",
+    "    }\n",
+    "]\n",
+    "\n",
+    "def call_function(name, args):\n",
+    "    if name == \"get_search_results_for\":\n",
+    "        return get_google_search_results_for(**args)\n",
+    "    if name == \"load_page_content\":\n",
+    "        return load_page_content(**args)\n",
+    "\n",
+    "    return None\n"
+   ],
+   "id": "45340dbef571198f",
+   "outputs": [],
+   "execution_count": 25
+  },
+  {
+   "metadata": {
+    "ExecuteTime": {
+     "end_time": "2025-06-20T13:22:00.080330Z",
+     "start_time": "2025-06-20T13:21:51.685133Z"
+    }
+   },
+   "cell_type": "code",
+   "source": [
+    "query = questions[2]\n",
+    "config = dotenv_values()\n",
+    "\n",
+    "client = AzureOpenAI(\n",
+    "    api_key=config[\"AZURE_OPENAI_API_KEY\"],\n",
+    "    azure_endpoint=config[\"AZURE_OPENAI_API_BASE\"],\n",
+    "    api_version=config[\"AZURE_OPENAI_API_VERSION\"]\n",
+    ")\n",
+    "openai_chatmodel = config[\"AZURE_OPENAI_CHAT_MODEL\"]\n",
+    "\n",
+    "GRAY = \"\\033[90m\"\n",
+    "BOLD = \"\\033[1m\"\n",
+    "RESET = \"\\033[0m\"\n",
+    "\n",
+    "\n",
+    "messages = [\n",
+    "    {\n",
+    "        \"role\": \"system\",\n",
+    "        \"content\": \"You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.\"\n",
+    "    },\n",
+    "    {\"role\": \"user\", \"content\": query}\n",
+    "]\n",
+    "\n",
+    "total_input_token_count = 0\n",
+    "total_output_token_count = 0\n",
+    "\n",
+    "while True:\n",
+    "    completion = client.chat.completions.create(\n",
+    "        model=openai_chatmodel,\n",
+    "        messages=messages,\n",
+    "        tools=tools\n",
+    "    )\n",
+    "\n",
+    "    total_input_token_count += completion.usage.prompt_tokens\n",
+    "    total_output_token_count += completion.usage.completion_tokens\n",
+    "\n",
+    "    if completion.choices[0].finish_reason == \"stop\":\n",
+    "        print(f\"{BOLD}Final answer: {completion.choices[0].message.content}{RESET}\")\n",
+    "        break\n",
+    "    elif completion.choices[0].finish_reason == \"tool_calls\":\n",
+    "        messages.append(completion.choices[0].message)\n",
+    "        for tool_call in completion.choices[0].message.tool_calls:\n",
+    "            name = tool_call.function.name\n",
+    "            args = json.loads(tool_call.function.arguments)\n",
+    "\n",
+    "            result = call_function(name, args)\n",
+    "            print(f\"Called {BOLD}{name}({args}){RESET} and it returned {GRAY}{str(result)[:300]}{RESET}\")\n",
+    "\n",
+    "            messages.append({\n",
+    "                \"role\": \"tool\",\n",
+    "                \"tool_call_id\": tool_call.id,\n",
+    "                \"content\": str(result)\n",
+    "            })\n",
+    "    else:\n",
+    "        raise Exception(\"We're not supposed to be here\")"
+   ],
+   "id": "adfd8a6e27fb6069",
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Called \u001B[1mget_search_results_for({'query': 'only Featured Article on English Wikipedia about a dinosaur promoted November 2016'})\u001B[0m and it returned \u001B[90m## Search Results\n",
+      "0. [Wikipedia:Featured article candidates/Giganotosaurus/archive1](https://en.wikipedia.org/wiki/Wikipedia:Featured_article_candidates/Giganotosaurus/archive1)\n",
+      "\n",
+      "The article was promoted by Ian Rose via FACBot (talk) 14:41, 19 November 2016 [1]. ... article is one of the most viewed\u001B[0m\n",
+      "Called \u001B[1mload_page_content({'url': 'https://en.wikipedia.org/wiki/Wikipedia:Featured_article_candidates/Giganotosaurus/archive1'})\u001B[0m and it returned \u001B[90mWikipedia:Featured article candidates/Giganotosaurus/archive1 - Wikipedia\n",
+      "\n",
+      "[Jump to content](#bodyContent)\n",
+      "\n",
+      "Main menu\n",
+      "\n",
+      "Main menu\n",
+      "\n",
+      "move to sidebar\n",
+      "hide\n",
+      "\n",
+      "Navigation\n",
+      "\n",
+      "* [Main page](/wiki/Main_Page \"Visit the main page [z]\")\n",
+      "* [Contents](/wiki/Wikipedia:Contents \"Guides to browsing Wikipedia\")\n",
+      "* [Curren\u001B[0m\n",
+      "\u001B[1mFinal answer: The only Featured Article on English Wikipedia about a dinosaur promoted in November 2016 is \"Giganotosaurus,\" and it was nominated by FunkMonk.\n",
+      "\n",
+      "FINAL ANSWER: FunkMonk\u001B[0m\n"
+     ]
+    }
+   ],
+   "execution_count": 30
+  },
+  {
+   "metadata": {},
+   "cell_type": "code",
+   "outputs": [],
+   "execution_count": null,
+   "source": "",
+   "id": "4518e95a30ad7b8a"
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 2
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython2",
+   "version": "2.7.6"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

requirements.txt

@@ -1,3 +1,5 @@
 gradio[oauth]==5.27.0
 requests==2.32.3
-smolagents[openai]==1.14.0
+smolagents[openai]==1.14.0
+wikipedia-api==0.8.1
+openai-function-calling==2.6.0