01 - Agents.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "6ed8f8da",
   "metadata": {},
   "source": [
    "# Agents\n",
    "`autogen` provides two classes of `ConversableAgent` that can communicate autonomously to  achieve a  goal.\n",
    "\n",
    "- `AssistantAgent` is  used to complete specific tasks – e.g. write code. They cannot execute code. \n",
    "\n",
    "- `UserProxyAgent` is a proxy for a human. By default human feedback is requested after each conversation turn, but it can operate autonomoudly with no human intervention needed. It can execute code and share the results with an `AssistantAgent`."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "2ff73d75",
   "metadata": {},
   "outputs": [],
   "source": [
    "# First step is to get environmental variables. In this notebook they are in .env file.\n",
    "from dotenv import load_dotenv\n",
    "import os\n",
    "\n",
    "load_dotenv()\n",
    "\n",
    "api_key = os.getenv(\"OPENAI_API_KEY\")"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "9840055c",
   "metadata": {},
   "source": [
    "The following will create an `AssistantAgent` and a `UserProxyAgent`."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "cf4cb7a2",
   "metadata": {},
   "outputs": [],
   "source": [
    "from autogen import AssistantAgent, UserProxyAgent\n",
    "\n",
    "assistant = AssistantAgent(name='assistant')\n",
    "\n",
    "user_agent = UserProxyAgent(name='user_agent')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "b17ac984",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\u001b[33muser_agent\u001b[0m (to assistant):\n",
      "\n",
      "What date is today? Which big tech stock has the largest year-to-date gain this year? How much is the gain?\n",
      "\n",
      "--------------------------------------------------------------------------------\n",
      "\u001b[33massistant\u001b[0m (to user_agent):\n",
      "\n",
      "To find out today's date, we can use Python's built-in `datetime` module. Here is the Python code to get today's date:\n",
      "\n",
      "```python\n",
      "import datetime\n",
      "\n",
      "today = datetime.date.today()\n",
      "print(\"Today's date:\", today)\n",
      "```\n",
      "\n",
      "For the second part of the query, we need to retrieve stock data for big tech companies. Unlike Python's built-in functionalities, there are no standard Python libraries for retrieving real-time or historical stock market data. However, there are Python-based platforms such as `yfinance` that allow you to connect with Yahoo Finance and retrieve such information.\n",
      "\n",
      "Unfortunately as an AI, I don't have real-time capabilities to directly fetch and analyze the data at any given moment. I suggest you to use Python's `yfinance` library to get the stock data of large tech companies (for example, FAANG: Facebook (FB), Amazon (AMZN), Apple (AAPL), Netflix (NFLX), and Google (GOOGL)), then determine which one has the largest year-to-date (YTD) gain. Here is how you can do it:\n",
      "\n",
      "First you have to install `yfinance` module, if you haven't already, using `pip`:\n",
      "\n",
      "```sh\n",
      "pip install yfinance\n",
      "```\n",
      "\n",
      "Then, you can use the following Python code to check the stock YTD gain for each of the FAANG stocks:\n",
      "\n",
      "```python\n",
      "import yfinance as yf\n",
      "from datetime import datetime\n",
      "\n",
      "# FAANG Stocks\n",
      "stocks = ['FB', 'AMZN', 'AAPL', 'NFLX', 'GOOGL']\n",
      "\n",
      "# Start date is the first trading day of this year\n",
      "start_date = datetime(datetime.now().year, 1, 1)\n",
      "\n",
      "max_gain_stock = ''\n",
      "max_gain_percent = 0\n",
      "\n",
      "# Check each stock\n",
      "for stock in stocks:\n",
      "    ticker = yf.Ticker(stock)\n",
      "    # Get historical market data\n",
      "    hist = ticker.history(start=start_date)\n",
      "    # Calculate YTD gain\n",
      "    if hist.index[0] != start_date:\n",
      "        continue\n",
      "    start_price = hist['Close'][0]\n",
      "    end_price = hist['Close'][-1]\n",
      "    gain_percent = 100 * (end_price - start_price) / start_price\n",
      "    if gain_percent > max_gain_percent:\n",
      "        max_gain_percent = gain_percent\n",
      "        max_gain_stock = stock\n",
      "\n",
      "print(f\"The stock with the largest YTD gain is: {max_gain_stock}\")\n",
      "print(f\"The YTD gain for {max_gain_stock} is: {max_gain_percent}%\")\n",
      "```\n",
      "\n",
      "Please note that the stock market data might be limited during the weekend or non-trading hours.\n",
      "\n",
      "--------------------------------------------------------------------------------\n",
      "Provide feedback to assistant. Press enter to skip and use auto-reply, or type 'exit' to end the conversation: \n",
      "\u001b[31m\n",
      ">>>>>>>> NO HUMAN INPUT RECEIVED.\u001b[0m\n",
      "\u001b[31m\n",
      ">>>>>>>> USING AUTO REPLY...\u001b[0m\n",
      "\u001b[31m\n",
      ">>>>>>>> EXECUTING CODE BLOCK 0 (inferred language is python)...\u001b[0m\n",
      "Pulling image python:3-alpine\n",
      "\u001b[31m\n",
      ">>>>>>>> EXECUTING CODE BLOCK 1 (inferred language is sh)...\u001b[0m\n",
      "\u001b[31m\n",
      ">>>>>>>> EXECUTING CODE BLOCK 2 (inferred language is python)...\u001b[0m\n",
      "\u001b[33muser_agent\u001b[0m (to assistant):\n",
      "\n",
      "exitcode: 1 (execution failed)\n",
      "Code output: \n",
      "Today's date: 2023-12-31\n",
      "\n",
      "Collecting yfinance\n",
      "  Obtaining dependency information for yfinance from https://files.pythonhosted.org/packages/09/82/5a5eee685095a6c1f8968db75bba8536baa8046c970be25e1ac91a843839/yfinance-0.2.33-py2.py3-none-any.whl.metadata\n",
      "  Downloading yfinance-0.2.33-py2.py3-none-any.whl.metadata (11 kB)\n",
      "Collecting pandas>=1.3.0 (from yfinance)\n",
      "  Obtaining dependency information for pandas>=1.3.0 from https://files.pythonhosted.org/packages/cc/a8/13dced3276ea4514909a80c8dd08b43ab23007b4949701e3d7ae2a8ccd2d/pandas-2.1.4-cp312-cp312-musllinux_1_1_x86_64.whl.metadata\n",
      "  Downloading pandas-2.1.4-cp312-cp312-musllinux_1_1_x86_64.whl.metadata (18 kB)\n",
      "Collecting numpy>=1.16.5 (from yfinance)\n",
      "  Obtaining dependency information for numpy>=1.16.5 from https://files.pythonhosted.org/packages/8c/9f/2f5c6b5f63cf006e6190bf750ade791d1fee353bab654bbde2f83a3ab92e/numpy-1.26.2-cp312-cp312-musllinux_1_1_x86_64.whl.metadata\n",
      "  Downloading numpy-1.26.2-cp312-cp312-musllinux_1_1_x86_64.whl.metadata (61 kB)\n",
      "     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 61.2/61.2 kB 1.1 MB/s eta 0:00:00\n",
      "Collecting requests>=2.31 (from yfinance)\n",
      "  Obtaining dependency information for requests>=2.31 from https://files.pythonhosted.org/packages/70/8e/0e2d847013cb52cd35b38c009bb167a1a26b2ce6cd6965bf26b47bc0bf44/requests-2.31.0-py3-none-any.whl.metadata\n",
      "  Downloading requests-2.31.0-py3-none-any.whl.metadata (4.6 kB)\n",
      "Collecting multitasking>=0.0.7 (from yfinance)\n",
      "  Downloading multitasking-0.0.11-py3-none-any.whl (8.5 kB)\n",
      "Collecting lxml>=4.9.1 (from yfinance)\n",
      "  Obtaining dependency information for lxml>=4.9.1 from https://files.pythonhosted.org/packages/2f/bf/b211c504820f73e812d40c638271725ffcb2081820e9d54b17a17d48282e/lxml-5.0.0-cp312-cp312-musllinux_1_1_x86_64.whl.metadata\n",
      "  Downloading lxml-5.0.0-cp312-cp312-musllinux_1_1_x86_64.whl.metadata (6.7 kB)\n",
      "Collecting appdirs>=1.4.4 (from yfinance)\n",
      "  Downloading appdirs-1.4.4-py2.py3-none-any.whl (9.6 kB)\n",
      "Collecting pytz>=2022.5 (from yfinance)\n",
      "  Obtaining dependency information for pytz>=2022.5 from https://files.pythonhosted.org/packages/32/4d/aaf7eff5deb402fd9a24a1449a8119f00d74ae9c2efa79f8ef9994261fc2/pytz-2023.3.post1-py2.py3-none-any.whl.metadata\n",
      "  Downloading pytz-2023.3.post1-py2.py3-none-any.whl.metadata (22 kB)\n",
      "Collecting frozendict>=2.3.4 (from yfinance)\n",
      "  Obtaining dependency information for frozendict>=2.3.4 from https://files.pythonhosted.org/packages/b6/2c/e391d6d57bbd837d383a6b69b54ecb4768d75b31d8bc7a380b4b4afe4cb2/frozendict-2.3.10-py3-none-any.whl.metadata\n",
      "  Downloading frozendict-2.3.10-py3-none-any.whl.metadata (20 kB)\n",
      "Collecting peewee>=3.16.2 (from yfinance)\n",
      "  Downloading peewee-3.17.0.tar.gz (2.9 MB)\n",
      "     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 2.9/2.9 MB 13.2 MB/s eta 0:00:00\n",
      "  Installing build dependencies: started\n",
      "  Installing build dependencies: finished with status 'done'\n",
      "  Getting requirements to build wheel: started\n",
      "  Getting requirements to build wheel: finished with status 'done'\n",
      "  Preparing metadata (pyproject.toml): started\n",
      "  Preparing metadata (pyproject.toml): finished with status 'done'\n",
      "Collecting beautifulsoup4>=4.11.1 (from yfinance)\n",
      "  Downloading beautifulsoup4-4.12.2-py3-none-any.whl (142 kB)\n",
      "     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 143.0/143.0 kB 13.4 MB/s eta 0:00:00\n",
      "Collecting html5lib>=1.1 (from yfinance)\n",
      "  Downloading html5lib-1.1-py2.py3-none-any.whl (112 kB)\n",
      "     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 112.2/112.2 kB 10.3 MB/s eta 0:00:00\n",
      "Collecting soupsieve>1.2 (from beautifulsoup4>=4.11.1->yfinance)\n",
      "  Obtaining dependency information for soupsieve>1.2 from https://files.pythonhosted.org/packages/4c/f3/038b302fdfbe3be7da016777069f26ceefe11a681055ea1f7817546508e3/soupsieve-2.5-py3-none-any.whl.metadata\n",
      "  Downloading soupsieve-2.5-py3-none-any.whl.metadata (4.7 kB)\n",
      "Collecting six>=1.9 (from html5lib>=1.1->yfinance)\n",
      "  Downloading six-1.16.0-py2.py3-none-any.whl (11 kB)\n",
      "Collecting webencodings (from html5lib>=1.1->yfinance)\n",
      "  Downloading webencodings-0.5.1-py2.py3-none-any.whl (11 kB)\n",
      "Collecting python-dateutil>=2.8.2 (from pandas>=1.3.0->yfinance)\n",
      "  Downloading python_dateutil-2.8.2-py2.py3-none-any.whl (247 kB)\n",
      "     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 247.7/247.7 kB 16.8 MB/s eta 0:00:00\n",
      "Collecting tzdata>=2022.1 (from pandas>=1.3.0->yfinance)\n",
      "  Obtaining dependency information for tzdata>=2022.1 from https://files.pythonhosted.org/packages/a3/fb/52b62131e21b24ee297e4e95ed41eba29647dad0e0051a92bb66b43c70ff/tzdata-2023.4-py2.py3-none-any.whl.metadata\n",
      "  Downloading tzdata-2023.4-py2.py3-none-any.whl.metadata (1.4 kB)\n",
      "Collecting charset-normalizer<4,>=2 (from requests>=2.31->yfinance)\n",
      "  Obtaining dependency information for charset-normalizer<4,>=2 from https://files.pythonhosted.org/packages/5b/ae/ce2c12fcac59cb3860b2e2d76dc405253a4475436b1861d95fe75bdea520/charset_normalizer-3.3.2-cp312-cp312-musllinux_1_1_x86_64.whl.metadata\n",
      "  Downloading charset_normalizer-3.3.2-cp312-cp312-musllinux_1_1_x86_64.whl.metadata (33 kB)\n",
      "Collecting idna<4,>=2.5 (from requests>=2.31->yfinance)\n",
      "  Obtaining dependency information for idna<4,>=2.5 from https://files.pythonhosted.org/packages/c2/e7/a82b05cf63a603df6e68d59ae6a68bf5064484a0718ea5033660af4b54a9/idna-3.6-py3-none-any.whl.metadata\n",
      "  Downloading idna-3.6-py3-none-any.whl.metadata (9.9 kB)\n",
      "Collecting urllib3<3,>=1.21.1 (from requests>=2.31->yfinance)\n",
      "  Obtaining dependency information for urllib3<3,>=1.21.1 from https://files.pythonhosted.org/packages/96/94/c31f58c7a7f470d5665935262ebd7455c7e4c7782eb525658d3dbf4b9403/urllib3-2.1.0-py3-none-any.whl.metadata\n",
      "  Downloading urllib3-2.1.0-py3-none-any.whl.metadata (6.4 kB)\n",
      "Collecting certifi>=2017.4.17 (from requests>=2.31->yfinance)\n",
      "  Obtaining dependency information for certifi>=2017.4.17 from https://files.pythonhosted.org/packages/64/62/428ef076be88fa93716b576e4a01f919d25968913e817077a386fcbe4f42/certifi-2023.11.17-py3-none-any.whl.metadata\n",
      "  Downloading certifi-2023.11.17-py3-none-any.whl.metadata (2.2 kB)\n",
      "Downloading yfinance-0.2.33-py2.py3-none-any.whl (69 kB)\n",
      "   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 69.0/69.0 kB 7.7 MB/s eta 0:00:00\n",
      "Downloading frozendict-2.3.10-py3-none-any.whl (14 kB)\n",
      "Downloading lxml-5.0.0-cp312-cp312-musllinux_1_1_x86_64.whl (8.0 MB)\n",
      "   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 8.0/8.0 MB 32.8 MB/s eta 0:00:00\n",
      "Downloading numpy-1.26.2-cp312-cp312-musllinux_1_1_x86_64.whl (17.8 MB)\n",
      "   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 17.8/17.8 MB 31.1 MB/s eta 0:00:00\n",
      "Downloading pandas-2.1.4-cp312-cp312-musllinux_1_1_x86_64.whl (12.4 MB)\n",
      "   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 12.4/12.4 MB 32.9 MB/s eta 0:00:00\n",
      "Downloading pytz-2023.3.post1-py2.py3-none-any.whl (502 kB)\n",
      "   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 502.5/502.5 kB 29.3 MB/s eta 0:00:00\n",
      "Downloading requests-2.31.0-py3-none-any.whl (62 kB)\n",
      "   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 62.6/62.6 kB 8.9 MB/s eta 0:00:00\n",
      "Downloading certifi-2023.11.17-py3-none-any.whl (162 kB)\n",
      "   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 162.5/162.5 kB 17.4 MB/s eta 0:00:00\n",
      "Downloading charset_normalizer-3.3.2-cp312-cp312-musllinux_1_1_x86_64.whl (142 kB)\n",
      "   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 142.2/142.2 kB 14.7 MB/s eta 0:00:00\n",
      "Downloading idna-3.6-py3-none-any.whl (61 kB)\n",
      "   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 61.6/61.6 kB 7.1 MB/s eta 0:00:00\n",
      "Downloading soupsieve-2.5-py3-none-any.whl (36 kB)\n",
      "Downloading tzdata-2023.4-py2.py3-none-any.whl (346 kB)\n",
      "   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 346.6/346.6 kB 27.1 MB/s eta 0:00:00\n",
      "Downloading urllib3-2.1.0-py3-none-any.whl (104 kB)\n",
      "   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 104.6/104.6 kB 10.4 MB/s eta 0:00:00\n",
      "Building wheels for collected packages: peewee\n",
      "  Building wheel for peewee (pyproject.toml): started\n",
      "  Building wheel for peewee (pyproject.toml): finished with status 'done'\n",
      "  Created wheel for peewee: filename=peewee-3.17.0-py3-none-any.whl size=135721 sha256=4049b9a41eb0f3e6dd2b2e1b4a91dd5a57ebe3648e869f3f89869bcc1291bf57\n",
      "  Stored in directory: /root/.cache/pip/wheels/35/c2/ce/d5490cdd71c6d4a993c87a5e172ec23c5d9cd287b59b7f8098\n",
      "Successfully built peewee\n",
      "Installing collected packages: webencodings, pytz, peewee, multitasking, appdirs, urllib3, tzdata, soupsieve, six, numpy, lxml, idna, frozendict, charset-normalizer, certifi, requests, python-dateutil, html5lib, beautifulsoup4, pandas, yfinance\n",
      "Successfully installed appdirs-1.4.4 beautifulsoup4-4.12.2 certifi-2023.11.17 charset-normalizer-3.3.2 frozendict-2.3.10 html5lib-1.1 idna-3.6 lxml-5.0.0 multitasking-0.0.11 numpy-1.26.2 pandas-2.1.4 peewee-3.17.0 python-dateutil-2.8.2 pytz-2023.3.post1 requests-2.31.0 six-1.16.0 soupsieve-2.5 tzdata-2023.4 urllib3-2.1.0 webencodings-0.5.1 yfinance-0.2.33\n",
      "WARNING: Running pip as the 'root' user can result in broken permissions and conflicting behaviour with the system package manager. It is recommended to use a virtual environment instead: https://pip.pypa.io/warnings/venv\n",
      "\n",
      "[notice] A new release of pip is available: 23.2.1 -> 23.3.2\n",
      "[notice] To update, run: pip install --upgrade pip\n",
      "\n",
      "FB: No timezone found, symbol may be delisted\n",
      "Traceback (most recent call last):\n",
      "  File \"\", line 19, in <module>\n",
      "    if hist.index[0] != start_date:\n",
      "       ~~~~~~~~~~^^^\n",
      "  File \"/usr/local/lib/python3.12/site-packages/pandas/core/indexes/base.py\", line 5366, in __getitem__\n",
      "    return getitem(key)\n",
      "           ^^^^^^^^^^^^\n",
      "IndexError: index 0 is out of bounds for axis 0 with size 0\n",
      "\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\n",
      "--------------------------------------------------------------------------------\n",
      "\u001b[33massistant\u001b[0m (to user_agent):\n",
      "\n",
      "It seems like there's an issue with retrieving data for some stocks. The error \"IndexError: index 0 is out of bounds for axis 0 with size 0\" typically indicates that the `hist` DataFrame (which should contain the historical price data for the stock) is empty, and the script is trying to access the first element of an empty list which leads to the error.\n",
      "\n",
      "Consequently, the stock symbol \"FB\" (Facebook) might be incorrect or the stock data can't be fetched due to certain restrictions (like being delisted, or restrictions in the data fetched by the Yahoo Finance API, or the Internet connection issue). To bypass this issue, I will add a try/except block to deal with such situations. So, the script continues to analyze remaining stocks even if it encounters a problem with a specific one.\n",
      "\n",
      "Let's modify the Python code accordingly:\n",
      "\n",
      "```python\n",
      "import yfinance as yf\n",
      "from datetime import datetime\n",
      "\n",
      "# FAANG Stocks\n",
      "stocks = ['FB', 'AMZN', 'AAPL', 'NFLX', 'GOOGL']\n",
      "\n",
      "# Start date is the first trading day of this year\n",
      "start_date = datetime(datetime.now().year, 1, 1)\n",
      "\n",
      "max_gain_stock = ''\n",
      "max_gain_percent = 0\n",
      "\n",
      "# Check each stock\n",
      "for stock in stocks:\n",
      "    try:\n",
      "        ticker = yf.Ticker(stock)\n",
      "        # Get historical market data\n",
      "        hist = ticker.history(start=start_date)\n",
      "        # Calculate YTD gain\n",
      "        if len(hist) == 0:\n",
      "            print(f\"Could not fetch data for {stock}\")\n",
      "            continue\n",
      "        start_price = hist['Close'][0]\n",
      "        end_price = hist['Close'][-1]\n",
      "        gain_percent = 100 * (end_price - start_price) / start_price\n",
      "        if gain_percent > max_gain_percent:\n",
      "            max_gain_percent = gain_percent\n",
      "            max_gain_stock = stock\n",
      "    except Exception as e:\n",
      "        print(f\"An error occurred when processing {stock}: {str(e)}\")\n",
      "\n",
      "print(f\"The stock with the largest YTD gain is: {max_gain_stock}\")\n",
      "print(f\"The YTD gain for {max_gain_stock} is: {max_gain_percent}%\")\n",
      "```\n",
      "\n",
      "This script will now output an error message and skip to the next stock if it encounters any problems when fetching or processing the data for a particular stock.\n",
      "\n",
      "--------------------------------------------------------------------------------\n",
      "Provide feedback to assistant. Press enter to skip and use auto-reply, or type 'exit' to end the conversation: \n",
      "\u001b[31m\n",
      ">>>>>>>> NO HUMAN INPUT RECEIVED.\u001b[0m\n",
      "\u001b[31m\n",
      ">>>>>>>> USING AUTO REPLY...\u001b[0m\n",
      "\u001b[31m\n",
      ">>>>>>>> EXECUTING CODE BLOCK 0 (inferred language is python)...\u001b[0m\n",
      "\u001b[33muser_agent\u001b[0m (to assistant):\n",
      "\n",
      "exitcode: 0 (execution succeeded)\n",
      "Code output: \n",
      "FB: No timezone found, symbol may be delisted\n",
      "/workspace/tmp_code_d20ef6d1be1d588d66994f0903d01c65.py:23: FutureWarning: Series.__getitem__ treating keys as positions is deprecated. In a future version, integer keys will always be treated as labels (consistent with DataFrame behavior). To access a value by position, use `ser.iloc[pos]`\n",
      "  start_price = hist['Close'][0]\n",
      "/workspace/tmp_code_d20ef6d1be1d588d66994f0903d01c65.py:24: FutureWarning: Series.__getitem__ treating keys as positions is deprecated. In a future version, integer keys will always be treated as labels (consistent with DataFrame behavior). To access a value by position, use `ser.iloc[pos]`\n",
      "  end_price = hist['Close'][-1]\n",
      "/workspace/tmp_code_d20ef6d1be1d588d66994f0903d01c65.py:23: FutureWarning: Series.__getitem__ treating keys as positions is deprecated. In a future version, integer keys will always be treated as labels (consistent with DataFrame behavior). To access a value by position, use `ser.iloc[pos]`\n",
      "  start_price = hist['Close'][0]\n",
      "/workspace/tmp_code_d20ef6d1be1d588d66994f0903d01c65.py:24: FutureWarning: Series.__getitem__ treating keys as positions is deprecated. In a future version, integer keys will always be treated as labels (consistent with DataFrame behavior). To access a value by position, use `ser.iloc[pos]`\n",
      "  end_price = hist['Close'][-1]\n",
      "/workspace/tmp_code_d20ef6d1be1d588d66994f0903d01c65.py:23: FutureWarning: Series.__getitem__ treating keys as positions is deprecated. In a future version, integer keys will always be treated as labels (consistent with DataFrame behavior). To access a value by position, use `ser.iloc[pos]`\n",
      "  start_price = hist['Close'][0]\n",
      "/workspace/tmp_code_d20ef6d1be1d588d66994f0903d01c65.py:24: FutureWarning: Series.__getitem__ treating keys as positions is deprecated. In a future version, integer keys will always be treated as labels (consistent with DataFrame behavior). To access a value by position, use `ser.iloc[pos]`\n",
      "  end_price = hist['Close'][-1]\n",
      "/workspace/tmp_code_d20ef6d1be1d588d66994f0903d01c65.py:23: FutureWarning: Series.__getitem__ treating keys as positions is deprecated. In a future version, integer keys will always be treated as labels (consistent with DataFrame behavior). To access a value by position, use `ser.iloc[pos]`\n",
      "  start_price = hist['Close'][0]\n",
      "/workspace/tmp_code_d20ef6d1be1d588d66994f0903d01c65.py:24: FutureWarning: Series.__getitem__ treating keys as positions is deprecated. In a future version, integer keys will always be treated as labels (consistent with DataFrame behavior). To access a value by position, use `ser.iloc[pos]`\n",
      "  end_price = hist['Close'][-1]\n",
      "Could not fetch data for FB\n",
      "The stock with the largest YTD gain is: AMZN\n",
      "The YTD gain for AMZN is: 77.04498133500891%\n",
      "\n",
      "\n",
      "--------------------------------------------------------------------------------\n",
      "\u001b[33massistant\u001b[0m (to user_agent):\n",
      "\n",
      "From the execution result:\n",
      "\n",
      "- The date today is: 31st December, 2023.\n",
      "- The tech stock with the largest YTD (Year To Date) gain is Amazon (AMZN) with a gain of approximately 77.04%.\n",
      "\n",
      "However, it seems like the data for Facebook (FB) could not be fetched, possibly due to the symbol being incorrect, or changes in the stock's status, or restrictions in the data fetched by the Yahoo Finance API. You may want to independently verify the performance of Facebook's stock or consult a financial advisor for the most accurate information.\n",
      "\n",
      "TERMINATE\n",
      "\n",
      "--------------------------------------------------------------------------------\n",
      "Provide feedback to assistant. Press enter to skip and use auto-reply, or type 'exit' to end the conversation: exit\n"
     ]
    }
   ],
   "source": [
    "# the assistant receives a message from the user, which contains the task description\n",
    "user_agent.initiate_chat(\n",
    "    assistant,\n",
    "    message=\"\"\"What date is today? Which big tech stock has the largest year-to-date gain this year? How much is the gain?\"\"\",\n",
    ")"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.11.7"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}