Tutorial 4 - Hierarchical Agents.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "2fa1e388-b8f4-4fa3-be5a-ba7c6caae038",
   "metadata": {
    "tags": []
   },
   "source": [
    "# Tutorial 4 - Hierarchical Agents\n",
    "\n",
    "- Video: https://www.youtube.com/watch?v=RkY0Et-JhKI\n",
    "\n",
    "## Functionalities:\n",
    "- An agent (referred as meta agent) can also assign other agents (referred as inner agents) as the function call, thereby creating a hierarchical structure\n",
    "- Whenever the inner agent is called, we will run the full inner agent's process until termination\n",
    "- Maximum number of steps for each agent (meta agent or inner agent) can be defined in `max_subtasks` (default: 5) variable when initialising agent\n",
    "\n",
    "## Additional Details:\n",
    "- The inner agent will take in meta agent's instruction as input\n",
    "- In addition, the inner agent will also have access to meta agent's task and subtasks completed for global context\n",
    "- Inner agent's output will directly go into meta agent's subtasks completed dictionary, enabling it to directly update global context\n",
    "\n",
    "## Design Philosophy:\n",
    "- Bottom layers can get context of the top layers for more context-specific generation\n",
    "- Top layers choose bottom layers in a hierarchical fashion so as to limit the number of choices at each level for better performance\n",
    "- `subtasks_completed` serves as the global context for the meta agents and the inner agents, and can be summarised as needed, e.g. when performing a new task to filter out irrelevant information "
   ]
  },
  {
   "cell_type": "markdown",
   "id": "0b305952-5ff3-4235-8ae0-c23f36b78aa2",
   "metadata": {},
   "source": [
    "# Setup Guide"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "666ab955-7060-4581-8539-95eff8b339dc",
   "metadata": {},
   "source": [
    "## Step 1: Install TaskGen"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "f542fc42-e84d-4302-9eda-d053815ea316",
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "# !pip install taskgen-ai"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "c6615712-c0a3-4e26-94f6-02878ae242cb",
   "metadata": {},
   "source": [
    "## Step 2: Import required functions and setup relevant API keys for your LLM"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "511f92ac-633a-46e4-ad1f-5db04c4b8674",
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "# Set up API key and do the necessary imports\n",
    "from taskgen import *\n",
    "import os\n",
    "\n",
    "# this is only if you use OpenAI as your LLM\n",
    "os.environ['OPENAI_API_KEY'] = '<YOUR API KEY HERE>'"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "4107c095-7879-49c9-8b80-894078202b08",
   "metadata": {},
   "source": [
    "## Step 3: Define your own LLM\n",
    "- Take in a `system_prompt`, `user_prompt`, and outputs llm response string"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "1a34fa2d-4339-47ad-a0b0-098d653b9f5c",
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "def llm(system_prompt: str, user_prompt: str) -> str:\n",
    "    ''' Here, we use OpenAI for illustration, you can change it to your own LLM '''\n",
    "    # ensure your LLM imports are all within this function\n",
    "    from openai import OpenAI\n",
    "    \n",
    "    # define your own LLM here\n",
    "    client = OpenAI()\n",
    "    response = client.chat.completions.create(\n",
    "        model='gpt-4o-mini',\n",
    "        temperature = 0,\n",
    "        messages=[\n",
    "            {\"role\": \"system\", \"content\": system_prompt},\n",
    "            {\"role\": \"user\", \"content\": user_prompt}\n",
    "        ]\n",
    "    )\n",
    "    return response.choices[0].message.content"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "4019a53b-da17-4971-ba8a-3d10294600ef",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "'Neutral'"
      ]
     },
     "execution_count": 5,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Verify that llm function is working\n",
    "llm(system_prompt = 'You are a classifier to classify the sentiment of a sentence', \n",
    "    user_prompt = 'It is a hot and sunny day')"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "d485cc60-ad94-49ad-8a2d-204368a7065d",
   "metadata": {},
   "source": [
    "# Define Agents and Meta Agents\n",
    "- You can also create a Meta Agent that uses other Agents (referred to as Inner Agents) as functions\n",
    "- Create your Meta agent using `Agent()` (Note: No different from usual process of creating Agents - your Meta Agent is also an Agent)\n",
    "- Set up an Inner Agent list and assign it to your Meta agent using `assign_agents(agent_list)`\n",
    "\n",
    "## Example Meta Agent Setup\n",
    "```python\n",
    "# Define your meta-agent\n",
    "my_agent = Agent('Menu Creator', \n",
    "                 'Creates a menu for a restaurant. Menu item includes Name, Description, Ingredients, Pricing.',\n",
    "                 llm = llm)\n",
    "\n",
    "# Define your agent list. Note you can just assign functions to the agent in place using .assign_functions(function_list)\n",
    "agent_list = [\n",
    "    Agent('Chef', 'Takes in dish names and generates ingredients for each of them. Does not generate prices.', llm = llm),\n",
    "    Agent('Boss', 'Makes decision on which items to include in menu.', llm = llm),\n",
    "    Agent('Creative Writer', 'Takes in a cuisine type and generates interesting dish names and descriptions.', max_subtasks = 1, llm = llm),\n",
    "    Agent('Economist', 'Takes in dish names and comes up with pricing for each of them', llm = llm).assign_functions([dish_price])\n",
    "    ]\n",
    "\n",
    "my_agent.assign_agents(agent_list)\n",
    "```\n",
    "\n",
    "## Run the Meta Agent\n",
    "- Let us run the agent and see the interactions between the Meta Agent and Inner Agents to solve the task!\n",
    "```python\n",
    "output = my_agent.run('Generate 5 menu items with name, description, ingredients and price based on Italian food choices. Output in menu format.')\n",
    "```"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "1f2304be-b9fa-4292-b20f-2ac6c6909015",
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "def random_number_from_string(input_string) -> int:\n",
    "    ''' Returns a random number between 1 and 10 when given an input string '''\n",
    "    import hashlib\n",
    "    import random\n",
    "    # Hash the input string\n",
    "    hashed_string = hashlib.sha256(input_string.encode()).hexdigest()\n",
    "    \n",
    "    # Convert the hashed string to an integer\n",
    "    hashed_integer = int(hashed_string, 16)\n",
    "    \n",
    "    # Seed the random number generator with the hashed integer\n",
    "    random.seed(hashed_integer)\n",
    "    \n",
    "    # Generate a random number between 1 and 10\n",
    "    return random.randint(1, 10)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "id": "7222e82b-e030-4494-894d-41a61370c747",
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "# assign a dish pricing function for economist\n",
    "# Note: Try to make the output informative that is understandable by the Agent - use free text or json to explain the output so that someone can understand without the input\n",
    "def dish_price(list_of_dish_names: list) -> dict:\n",
    "    ''' Takes in list_of_dish_names and outputs price of each dish '''\n",
    "    if not isinstance(list_of_dish_names, list):\n",
    "        list_of_food_items = [list_of_dish_names]\n",
    "    output = {}\n",
    "    for each in list_of_dish_names:\n",
    "        output[each] = '$'+str(random_number_from_string(each))\n",
    "    return output"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "id": "e56b3ab0-9cfc-414b-8386-4ac76a5ce5c2",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'spaghetti': '$8', 'tomato': '$1'}"
      ]
     },
     "execution_count": 8,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "dish_price(['spaghetti', 'tomato'])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "id": "fc770857-5ca0-43a1-a72f-5df7a97f7c02",
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "# Define your meta-agent\n",
    "my_agent = Agent('Menu Creator', \n",
    "                 'Creates a menu for a restaurant. Menu item includes Name, Description, Ingredients, Pricing.',\n",
    "                 llm = llm)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "id": "7302c9ec-6e27-40c0-8a3c-3cd350964027",
   "metadata": {
    "jp-MarkdownHeadingCollapsed": true,
    "tags": []
   },
   "outputs": [],
   "source": [
    "# Define your agent list. Note you can just assign functions to the agent in place using .assign_functions(function_list)\n",
    "agent_list = [\n",
    "    Agent('Chef', 'Takes in dish names and generates ingredients for each of them. Does not generate prices.', llm = llm),\n",
    "    Agent('Boss', 'Makes decisions on which items to include in Menu', llm = llm),\n",
    "    Agent('Creative Writer', 'Takes in a cuisine type and generates interesting dish names and descriptions. Does not generate prices or ingredients.', \n",
    "          max_subtasks = 1, llm = llm),\n",
    "    Agent('Economist', 'Takes in dish names and comes up with pricing for each of them', llm = llm).assign_functions([dish_price])\n",
    "    ]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "id": "d4e9c84e-0ffa-4450-9522-6e21d86bad29",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "<taskgen.agent.Agent at 0x1249e4e50>"
      ]
     },
     "execution_count": 11,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "my_agent.assign_agents(agent_list)"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "b3a63cf7-0c50-4c6d-993a-8cee1de738d9",
   "metadata": {},
   "source": [
    "# Approach 1: Run a generic task with the Meta-Agent\n",
    "- Let us run a general task to create 5 menu items for an Italian restaurant\n",
    "- You can also specify how Agents interact with one another in the task, otherwise the Meta-Agent will infer what should be done"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "id": "4aabd0c2-6470-43e1-b476-d77e2c85c0bb",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\u001b[1m\u001b[30mObservation: No subtasks have been completed yet for the assigned task of generating 5 menu items based on Italian food choices.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: To complete the assigned task, I need to first generate interesting dish names and descriptions for Italian cuisine. After that, I will generate the ingredients for each dish and finally determine the pricing.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: Generate interesting dish names and descriptions for 5 Italian menu items.\u001b[0m\n",
      "Calling function Creative Writer with parameters {'instruction': 'Generate interesting dish names and descriptions for 5 Italian menu items.'}\n",
      "\n",
      "### Start of Inner Agent: Creative Writer ###\n",
      "\u001b[1m\u001b[30mObservation: No subtasks have been completed yet for generating interesting dish names and descriptions for Italian menu items.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: To complete the assigned task, I need to generate five unique dish names along with their descriptions that reflect Italian cuisine. This can be done by using the general language model to create creative and appealing names and descriptions.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: Generate five interesting dish names and descriptions for Italian menu items using the language model.\u001b[0m\n",
      "Getting LLM to perform the following task: Generate five interesting dish names and descriptions for Italian menu items using the language model.\n",
      "> 1. **Trattoria di Sogni**: A delightful fusion of flavors, this dish features handmade fettuccine tossed in a velvety truffle cream sauce, garnished with wild mushrooms and a sprinkle of fresh parsley, transporting you to a rustic Italian dream. \n",
      "\n",
      "2. **Sole di Napoli**: Experience the essence of Naples with this vibrant dish of pan-seared sole fillet, served atop a bed of zesty lemon and caper-infused risotto, finished with a drizzle of extra virgin olive oil and a hint of fresh basil. \n",
      "\n",
      "3. **Fior di Latte Fantasia**: Indulge in this creamy burrata cheese served with heirloom tomatoes, drizzled with a balsamic reduction and a touch of pesto, creating a colorful and refreshing appetizer that sings of summer in Italy. \n",
      "\n",
      "4. **Pasta al Vento**: A whimsical take on traditional pasta, this dish features spiraled zucchini noodles tossed with a light garlic and white wine sauce, topped with grilled shrimp and a sprinkle of chili flakes for a delightful kick. \n",
      "\n",
      "5. **Dolce di Amore**: End your meal on a sweet note with this decadent dessert, a rich chocolate lava cake served with a scoop of homemade vanilla gelato and a raspberry coulis, embodying the passion and romance of Italian cuisine.\n",
      "\n",
      "\u001b[1m\u001b[35m###\n",
      "Reply from Creative Writer to Menu Creator:\n",
      "1. **Trattoria di Sogni**: A delightful fusion of flavors, this dish features handmade fettuccine tossed in a velvety truffle cream sauce, garnished with wild mushrooms and a sprinkle of fresh parsley, transporting you to a rustic Italian dream. \n",
      "\n",
      "2. **Sole di Napoli**: Experience the essence of Naples with this vibrant dish of pan-seared sole fillet, served atop a bed of zesty lemon and caper-infused risotto, finished with a drizzle of extra virgin olive oil and a hint of fresh basil. \n",
      "\n",
      "3. **Fior di Latte Fantasia**: Indulge in this creamy burrata cheese served with heirloom tomatoes, drizzled with a balsamic reduction and a touch of pesto, creating a colorful and refreshing appetizer that sings of summer in Italy. \n",
      "\n",
      "4. **Pasta al Vento**: A whimsical take on traditional pasta, this dish features spiraled zucchini noodles tossed with a light garlic and white wine sauce, topped with grilled shrimp and a sprinkle of chili flakes for a delightful kick. \n",
      "\n",
      "5. **Dolce di Amore**: End your meal on a sweet note with this decadent dessert, a rich chocolate lava cake served with a scoop of homemade vanilla gelato and a raspberry coulis, embodying the passion and romance of Italian cuisine.\n",
      "###\n",
      "\u001b[0m\n",
      "### End of Inner Agent: Creative Writer ###\n",
      "\n",
      "\u001b[1m\u001b[30mObservation: The Creative Writer has successfully generated interesting dish names and descriptions for 5 Italian menu items.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: Next, I need to generate the ingredients for each of the 5 menu items. The Chef function can be utilized to create a list of ingredients based on the dish names provided.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: Use the Chef function to generate ingredients for the 5 Italian menu items: Trattoria di Sogni, Sole di Napoli, Fior di Latte Fantasia, Pasta al Vento, and Dolce di Amore.\u001b[0m\n",
      "Calling function Chef with parameters {'instruction': 'Use the Chef function to generate ingredients for the 5 Italian menu items: Trattoria di Sogni, Sole di Napoli, Fior di Latte Fantasia, Pasta al Vento, and Dolce di Amore.'}\n",
      "\n",
      "### Start of Inner Agent: Chef ###\n",
      "\u001b[1m\u001b[30mObservation: The creative writer has successfully generated interesting dish names and descriptions for 5 Italian menu items, but no ingredients have been provided yet.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: To complete the assigned task, I need to generate the ingredients for each of the 5 Italian dishes mentioned. Since there are no specific functions available for generating ingredients, I will use the general function to create a list of ingredients for each dish.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: Generate ingredients for the following dishes: Trattoria di Sogni, Sole di Napoli, Fior di Latte Fantasia, Pasta al Vento, and Dolce di Amore.\u001b[0m\n",
      "Getting LLM to perform the following task: Generate ingredients for the following dishes: Trattoria di Sogni, Sole di Napoli, Fior di Latte Fantasia, Pasta al Vento, and Dolce di Amore.\n",
      "> 1. **Trattoria di Sogni**: Ingredients include handmade fettuccine, truffle cream, wild mushrooms, fresh parsley, garlic, and olive oil. \n",
      "\n",
      "2. **Sole di Napoli**: Ingredients consist of sole fillet, lemon, capers, risotto rice, extra virgin olive oil, and fresh basil. \n",
      "\n",
      "3. **Fior di Latte Fantasia**: Ingredients are burrata cheese, heirloom tomatoes, balsamic reduction, pesto, and fresh basil. \n",
      "\n",
      "4. **Pasta al Vento**: Ingredients include zucchini, garlic, white wine, shrimp, chili flakes, and olive oil. \n",
      "\n",
      "5. **Dolce di Amore**: Ingredients consist of chocolate, eggs, sugar, flour, vanilla gelato, and raspberry coulis.\n",
      "\n",
      "\u001b[1m\u001b[30mObservation: The ingredients for the 5 Italian menu items have been successfully generated, including Trattoria di Sogni, Sole di Napoli, Fior di Latte Fantasia, Pasta al Vento, and Dolce di Amore.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: Since the ingredients have been generated, the next step is to finalize the output and present it to the user.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: End Task\u001b[0m\n",
      "Task completed successfully!\n",
      "\n",
      "\u001b[1m\u001b[35m###\n",
      "Reply from Chef to Menu Creator:\n",
      "1. **Trattoria di Sogni**: Ingredients include handmade fettuccine, truffle cream, wild mushrooms, fresh parsley, garlic, and olive oil. \n",
      "\n",
      "2. **Sole di Napoli**: Ingredients consist of sole fillet, lemon, capers, risotto rice, extra virgin olive oil, and fresh basil. \n",
      "\n",
      "3. **Fior di Latte Fantasia**: Ingredients are burrata cheese, heirloom tomatoes, balsamic reduction, pesto, and fresh basil. \n",
      "\n",
      "4. **Pasta al Vento**: Ingredients include zucchini, garlic, white wine, shrimp, chili flakes, and olive oil. \n",
      "\n",
      "5. **Dolce di Amore**: Ingredients consist of chocolate, eggs, sugar, flour, vanilla gelato, and raspberry coulis.\n",
      "###\n",
      "\u001b[0m\n",
      "### End of Inner Agent: Chef ###\n",
      "\n",
      "\u001b[1m\u001b[30mObservation: The Creative Writer has successfully generated interesting dish names and descriptions for 5 Italian menu items. The Chef has also provided the ingredients for each of these dishes.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: The next step is to determine the pricing for each of the menu items. Since the dish names and ingredients are already available, I can use the Economist function to generate appropriate prices for each item.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: Use the Economist function to generate pricing for the 5 Italian menu items: Trattoria di Sogni, Sole di Napoli, Fior di Latte Fantasia, Pasta al Vento, and Dolce di Amore.\u001b[0m\n",
      "Calling function Economist with parameters {'instruction': 'Use the Economist function to generate pricing for the 5 Italian menu items: Trattoria di Sogni, Sole di Napoli, Fior di Latte Fantasia, Pasta al Vento, and Dolce di Amore.'}\n",
      "\n",
      "### Start of Inner Agent: Economist ###\n",
      "\u001b[1m\u001b[30mObservation: No subtasks have been completed yet for the assigned task of generating pricing for the 5 Italian menu items.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: To complete the assigned task, I need to use the dish_price function to generate pricing for the dishes that have already been created: Trattoria di Sogni, Sole di Napoli, Fior di Latte Fantasia, Pasta al Vento, and Dolce di Amore.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: Use the dish_price function to input the list of dish names and obtain the pricing for each dish.\u001b[0m\n",
      "Calling function dish_price with parameters {'list_of_dish_names': ['Trattoria di Sogni', 'Sole di Napoli', 'Fior di Latte Fantasia', 'Pasta al Vento', 'Dolce di Amore']}\n",
      "> {'output_1': {'Trattoria di Sogni': '$6', 'Sole di Napoli': '$8', 'Fior di Latte Fantasia': '$6', 'Pasta al Vento': '$3', 'Dolce di Amore': '$2'}}\n",
      "\n",
      "\u001b[1m\u001b[30mObservation: The pricing for the 5 Italian menu items has been successfully generated, with each dish assigned a price: Trattoria di Sogni: $6, Sole di Napoli: $8, Fior di Latte Fantasia: $6, Pasta al Vento: $3, and Dolce di Amore: $2.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: Since the pricing has already been completed for all the assigned dishes, the next step is to finalize the output and present it to the user.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: End Task\u001b[0m\n",
      "Task completed successfully!\n",
      "\n",
      "\u001b[1m\u001b[35m###\n",
      "Reply from Economist to Menu Creator:\n",
      "The pricing for the 5 Italian menu items is as follows: \n",
      "\n",
      "1. **Trattoria di Sogni**: $6 \n",
      "2. **Sole di Napoli**: $8 \n",
      "3. **Fior di Latte Fantasia**: $6 \n",
      "4. **Pasta al Vento**: $3 \n",
      "5. **Dolce di Amore**: $2 \n",
      "\n",
      "These prices reflect the value of the ingredients and the culinary experience each dish offers, aligning with the descriptions and ingredients provided in the previous subtasks.\n",
      "###\n",
      "\u001b[0m\n",
      "### End of Inner Agent: Economist ###\n",
      "\n",
      "\u001b[1m\u001b[30mObservation: The Creative Writer has successfully generated interesting dish names and descriptions for 5 Italian menu items. The Chef has provided the necessary ingredients for each dish, and the Economist has determined the pricing for all items.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: With all the required information gathered, the next step is to compile the menu items into a structured format that includes the name, description, ingredients, and pricing for each dish.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: End Task\u001b[0m\n",
      "Task completed successfully!\n",
      "\n"
     ]
    }
   ],
   "source": [
    "output = my_agent.run('Generate 5 menu items with name, description, ingredients and price based on Italian food choices. Output in Menu format.')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "id": "5a4a96d3-2580-4989-8e3b-f240d1f12190",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Agent Name: Menu Creator\n",
      "Agent Description: Creates a menu for a restaurant. Menu item includes Name, Description, Ingredients, Pricing.\n",
      "Available Functions: ['use_llm', 'end_task', 'Chef', 'Boss', 'Creative Writer', 'Economist']\n",
      "Shared Variables: ['agent']\n",
      "\u001b[1m\u001b[32mTask: Generate 5 menu items with name, description, ingredients and price based on Italian food choices. Output in Menu format.\u001b[0m\n",
      "\u001b[1m\u001b[30mSubtasks Completed:\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask: Creative Writer(instruction=\"Generate interesting dish names and descriptions for 5 Italian menu items.\")\u001b[0m\n",
      "{'Agent Output': '1. **Trattoria di Sogni**: A delightful fusion of flavors, this dish features handmade fettuccine tossed in a velvety truffle cream sauce, garnished with wild mushrooms and a sprinkle of fresh parsley, transporting you to a rustic Italian dream. \\n\\n2. **Sole di Napoli**: Experience the essence of Naples with this vibrant dish of pan-seared sole fillet, served atop a bed of zesty lemon and caper-infused risotto, finished with a drizzle of extra virgin olive oil and a hint of fresh basil. \\n\\n3. **Fior di Latte Fantasia**: Indulge in this creamy burrata cheese served with heirloom tomatoes, drizzled with a balsamic reduction and a touch of pesto, creating a colorful and refreshing appetizer that sings of summer in Italy. \\n\\n4. **Pasta al Vento**: A whimsical take on traditional pasta, this dish features spiraled zucchini noodles tossed with a light garlic and white wine sauce, topped with grilled shrimp and a sprinkle of chili flakes for a delightful kick. \\n\\n5. **Dolce di Amore**: End your meal on a sweet note with this decadent dessert, a rich chocolate lava cake served with a scoop of homemade vanilla gelato and a raspberry coulis, embodying the passion and romance of Italian cuisine.'}\n",
      "\n",
      "\u001b[1m\u001b[34mSubtask: Chef(instruction=\"Use the Chef function to generate ingredients for the 5 Italian menu items: Trattoria di Sogni, Sole di Napoli, Fior di Latte Fantasia, Pasta al Vento, and Dolce di Amore.\")\u001b[0m\n",
      "{'Agent Output': '1. **Trattoria di Sogni**: Ingredients include handmade fettuccine, truffle cream, wild mushrooms, fresh parsley, garlic, and olive oil. \\n\\n2. **Sole di Napoli**: Ingredients consist of sole fillet, lemon, capers, risotto rice, extra virgin olive oil, and fresh basil. \\n\\n3. **Fior di Latte Fantasia**: Ingredients are burrata cheese, heirloom tomatoes, balsamic reduction, pesto, and fresh basil. \\n\\n4. **Pasta al Vento**: Ingredients include zucchini, garlic, white wine, shrimp, chili flakes, and olive oil. \\n\\n5. **Dolce di Amore**: Ingredients consist of chocolate, eggs, sugar, flour, vanilla gelato, and raspberry coulis.'}\n",
      "\n",
      "\u001b[1m\u001b[34mSubtask: Economist(instruction=\"Use the Economist function to generate pricing for the 5 Italian menu items: Trattoria di Sogni, Sole di Napoli, Fior di Latte Fantasia, Pasta al Vento, and Dolce di Amore.\")\u001b[0m\n",
      "{'Agent Output': 'The pricing for the 5 Italian menu items is as follows: \\n\\n1. **Trattoria di Sogni**: $6 \\n2. **Sole di Napoli**: $8 \\n3. **Fior di Latte Fantasia**: $6 \\n4. **Pasta al Vento**: $3 \\n5. **Dolce di Amore**: $2 \\n\\nThese prices reflect the value of the ingredients and the culinary experience each dish offers, aligning with the descriptions and ingredients provided in the previous subtasks.'}\n",
      "\n",
      "Is Task Completed: True\n"
     ]
    }
   ],
   "source": [
    "my_agent.status()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "id": "1857e28c-e6ae-43cf-89ed-baabd72289de",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[{'Observation': 'No subtasks have been completed yet for the assigned task of generating 5 menu items based on Italian food choices.',\n",
       "  'Thoughts': 'To complete the assigned task, I need to first generate interesting dish names and descriptions for Italian cuisine. After that, I will generate the ingredients for each dish and finally determine the pricing.',\n",
       "  'Current Subtask': 'Generate interesting dish names and descriptions for 5 Italian menu items.',\n",
       "  'Equipped Function Name': 'Creative Writer',\n",
       "  'Equipped Function Inputs': {'instruction': 'Generate interesting dish names and descriptions for 5 Italian menu items.'}},\n",
       " {'Observation': 'The Creative Writer has successfully generated interesting dish names and descriptions for 5 Italian menu items.',\n",
       "  'Thoughts': 'Next, I need to generate the ingredients for each of the 5 menu items. The Chef function can be utilized to create a list of ingredients based on the dish names provided.',\n",
       "  'Current Subtask': 'Use the Chef function to generate ingredients for the 5 Italian menu items: Trattoria di Sogni, Sole di Napoli, Fior di Latte Fantasia, Pasta al Vento, and Dolce di Amore.',\n",
       "  'Equipped Function Name': 'Chef',\n",
       "  'Equipped Function Inputs': {'instruction': 'Use the Chef function to generate ingredients for the 5 Italian menu items: Trattoria di Sogni, Sole di Napoli, Fior di Latte Fantasia, Pasta al Vento, and Dolce di Amore.'}},\n",
       " {'Observation': 'The Creative Writer has successfully generated interesting dish names and descriptions for 5 Italian menu items. The Chef has also provided the ingredients for each of these dishes.',\n",
       "  'Thoughts': 'The next step is to determine the pricing for each of the menu items. Since the dish names and ingredients are already available, I can use the Economist function to generate appropriate prices for each item.',\n",
       "  'Current Subtask': 'Use the Economist function to generate pricing for the 5 Italian menu items: Trattoria di Sogni, Sole di Napoli, Fior di Latte Fantasia, Pasta al Vento, and Dolce di Amore.',\n",
       "  'Equipped Function Name': 'Economist',\n",
       "  'Equipped Function Inputs': {'instruction': 'Use the Economist function to generate pricing for the 5 Italian menu items: Trattoria di Sogni, Sole di Napoli, Fior di Latte Fantasia, Pasta al Vento, and Dolce di Amore.'}},\n",
       " {'Observation': 'The Creative Writer has successfully generated interesting dish names and descriptions for 5 Italian menu items. The Chef has provided the necessary ingredients for each dish, and the Economist has determined the pricing for all items.',\n",
       "  'Thoughts': 'With all the required information gathered, the next step is to compile the menu items into a structured format that includes the name, description, ingredients, and pricing for each dish.',\n",
       "  'Current Subtask': 'Compile the generated information into a final menu format that presents each of the 5 Italian dishes with their respective names, descriptions, ingredients, and prices.',\n",
       "  'Equipped Function Name': 'end_task',\n",
       "  'Equipped Function Inputs': {}}]"
      ]
     },
     "execution_count": 14,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "my_agent.thoughts"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "id": "b6ec6b0f-9a70-4b47-b6ea-1c29cd88116f",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "1. **Trattoria di Sogni**  \n",
      "   - **Description**: A delightful fusion of flavors, this dish features handmade fettuccine tossed in a velvety truffle cream sauce, garnished with wild mushrooms and a sprinkle of fresh parsley, transporting you to a rustic Italian dream.  \n",
      "   - **Ingredients**: Handmade fettuccine, truffle cream, wild mushrooms, fresh parsley, garlic, and olive oil.  \n",
      "   - **Pricing**: $6  \n",
      "\n",
      "2. **Sole di Napoli**  \n",
      "   - **Description**: Experience the essence of Naples with this vibrant dish of pan-seared sole fillet, served atop a bed of zesty lemon and caper-infused risotto, finished with a drizzle of extra virgin olive oil and a hint of fresh basil.  \n",
      "   - **Ingredients**: Sole fillet, lemon, capers, risotto rice, extra virgin olive oil, and fresh basil.  \n",
      "   - **Pricing**: $8  \n",
      "\n",
      "3. **Fior di Latte Fantasia**  \n",
      "   - **Description**: Indulge in this creamy burrata cheese served with heirloom tomatoes, drizzled with a balsamic reduction and a touch of pesto, creating a colorful and refreshing appetizer that sings of summer in Italy.  \n",
      "   - **Ingredients**: Burrata cheese, heirloom tomatoes, balsamic reduction, pesto, and fresh basil.  \n",
      "   - **Pricing**: $6  \n",
      "\n",
      "4. **Pasta al Vento**  \n",
      "   - **Description**: A whimsical take on traditional pasta, this dish features spiraled zucchini noodles tossed with a light garlic and white wine sauce, topped with grilled shrimp and a sprinkle of chili flakes for a delightful kick.  \n",
      "   - **Ingredients**: Zucchini, garlic, white wine, shrimp, chili flakes, and olive oil.  \n",
      "   - **Pricing**: $3  \n",
      "\n",
      "5. **Dolce di Amore**  \n",
      "   - **Description**: End your meal on a sweet note with this decadent dessert, a rich chocolate lava cake served with a scoop of homemade vanilla gelato and a raspberry coulis, embodying the passion and romance of Italian cuisine.  \n",
      "   - **Ingredients**: Chocolate, eggs, sugar, flour, vanilla gelato, and raspberry coulis.  \n",
      "   - **Pricing**: $2\n"
     ]
    }
   ],
   "source": [
    "output = my_agent.reply_user()"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "18be4496-fc6b-4d61-957a-e42809989f89",
   "metadata": {},
   "source": [
    "# Approach 2: Step through the process yourself as the Meta Agent\n",
    "- You can use agents manually using `use_agent(agent_name: str, agent_task: str)` The outputs of the agents will be stored into `subtasks_completed` of the meta agent automatically\n",
    "- You can also interact with the meta agent using `reply_user(task: str, stateful: bool = False)`. This generates a reply for the agent to the `task` using `subtasks_completed` as context. If `stateful` is true, this will also store the output into `subtasks_completed`\n",
    "- If `subtasks_completed` is too lengthy, you can run `summarise_subtasks_completed(task: str)` This summarises `subtasks_completed` according to what is relevant to the `task`\n",
    "- If you want to redo a subtask, you can remove the most recent subtask from `subtasks_completed` using `remove_last_subtask()`"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "id": "9b1828fd-975e-41c1-ad6c-5f1552286a82",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Agent Name: Menu Creator\n",
      "Agent Description: Creates a menu for a restaurant. Menu item includes Name, Description, Ingredients, Pricing.\n",
      "Available Functions: ['use_llm', 'end_task', 'Chef', 'Boss', 'Creative Writer', 'Economist']\n",
      "Shared Variables: ['agent']\n",
      "\u001b[1m\u001b[32mTask: No task assigned\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtasks Completed: None\u001b[0m\n",
      "Is Task Completed: False\n"
     ]
    }
   ],
   "source": [
    "# reset the state of the agent\n",
    "my_agent.reset()\n",
    "# currently should have no task and nothing in subtasks_completed\n",
    "my_agent.status()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "id": "83e65804-6519-4160-81ac-bb05ad10237b",
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "# Assign a task for the meta agent - this helps to contextualise what you give each of the inner agents later on\n",
    "my_agent.assign_task('Generate 5 menu items with name, description, ingredients and price based on Italian food choices. Output in Menu format.')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "id": "1cc2a7e3-407f-448b-a49b-92260f433527",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Calling function Creative Writer with parameters {'instruction': 'Generate only 5 Italian dish names and descriptions'}\n",
      "\n",
      "### Start of Inner Agent: Creative Writer ###\n",
      "\u001b[1m\u001b[30mObservation: No subtasks have been completed yet for generating Italian dish names and descriptions.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: To complete the assigned task, I need to generate 5 unique Italian dish names along with their descriptions. Since there are no specific guidelines provided for the names or descriptions, I can use my creative abilities to come up with interesting and appealing options.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: Generate 5 Italian dish names and their descriptions using the equipped function.\u001b[0m\n",
      "Getting LLM to perform the following task: Generate 5 Italian dish names and their descriptions using the equipped function.\n",
      "> 1. **Sogni di Pomodoro**: A vibrant tomato-based risotto that dances on the palate, infused with fresh basil and a hint of garlic, evoking the warmth of a sun-kissed Italian summer. \n",
      "\n",
      "2. **Frittata di Mare**: A delightful seafood frittata, featuring tender shrimp and succulent calamari, all enveloped in a fluffy egg mixture, garnished with a sprinkle of parsley and a drizzle of lemon. \n",
      "\n",
      "3. **Cappelli di Angelo al Pesto**: Angel hair pasta elegantly tossed in a fragrant basil pesto, adorned with pine nuts and a touch of Parmesan, creating a dish that is both light and bursting with flavor. \n",
      "\n",
      "4. **Pollo alla Romana**: Juicy chicken thighs braised with bell peppers, olives, and capers, simmered in a rich tomato sauce that captures the essence of Roman cuisine, served with crusty bread for dipping. \n",
      "\n",
      "5. **Torta di Ricotta e Limone**: A luscious ricotta cheesecake infused with zesty lemon, baked to perfection and topped with a delicate glaze, offering a sweet finish that is both refreshing and indulgent.\n",
      "\n",
      "\u001b[1m\u001b[35m###\n",
      "Reply from Creative Writer to Menu Creator:\n",
      "1. **Sogni di Pomodoro**: A vibrant tomato-based risotto that dances on the palate, infused with fresh basil and a hint of garlic, evoking the warmth of a sun-kissed Italian summer. \n",
      "\n",
      "2. **Frittata di Mare**: A delightful seafood frittata, featuring tender shrimp and succulent calamari, all enveloped in a fluffy egg mixture, garnished with a sprinkle of parsley and a drizzle of lemon. \n",
      "\n",
      "3. **Cappelli di Angelo al Pesto**: Angel hair pasta elegantly tossed in a fragrant basil pesto, adorned with pine nuts and a touch of Parmesan, creating a dish that is both light and bursting with flavor. \n",
      "\n",
      "4. **Pollo alla Romana**: Juicy chicken thighs braised with bell peppers, olives, and capers, simmered in a rich tomato sauce that captures the essence of Roman cuisine, served with crusty bread for dipping. \n",
      "\n",
      "5. **Torta di Ricotta e Limone**: A luscious ricotta cheesecake infused with zesty lemon, baked to perfection and topped with a delicate glaze, offering a sweet finish that is both refreshing and indulgent.\n",
      "###\n",
      "\u001b[0m\n",
      "### End of Inner Agent: Creative Writer ###\n",
      "\n"
     ]
    }
   ],
   "source": [
    "# This will run the Agent Creative Writer (with shared subtasks_completed as meta agent)\n",
    "# Output of task will go directly into subtasks_completed - we view the inner agent as simply an extension of the meta agent\n",
    "my_agent.use_agent('Creative Writer', 'Generate only 5 Italian dish names and descriptions')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "id": "d7baae92-097a-47db-a083-58e3328451db",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Agent Name: Menu Creator\n",
      "Agent Description: Creates a menu for a restaurant. Menu item includes Name, Description, Ingredients, Pricing.\n",
      "Available Functions: ['use_llm', 'end_task', 'Chef', 'Boss', 'Creative Writer', 'Economist']\n",
      "Shared Variables: ['agent']\n",
      "\u001b[1m\u001b[32mTask: Generate 5 menu items with name, description, ingredients and price based on Italian food choices. Output in Menu format.\u001b[0m\n",
      "\u001b[1m\u001b[30mSubtasks Completed:\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask: Creative Writer(instruction=\"Generate only 5 Italian dish names and descriptions\")\u001b[0m\n",
      "{'Agent Output': '1. **Sogni di Pomodoro**: A vibrant tomato-based risotto that dances on the palate, infused with fresh basil and a hint of garlic, evoking the warmth of a sun-kissed Italian summer. \\n\\n2. **Frittata di Mare**: A delightful seafood frittata, featuring tender shrimp and succulent calamari, all enveloped in a fluffy egg mixture, garnished with a sprinkle of parsley and a drizzle of lemon. \\n\\n3. **Cappelli di Angelo al Pesto**: Angel hair pasta elegantly tossed in a fragrant basil pesto, adorned with pine nuts and a touch of Parmesan, creating a dish that is both light and bursting with flavor. \\n\\n4. **Pollo alla Romana**: Juicy chicken thighs braised with bell peppers, olives, and capers, simmered in a rich tomato sauce that captures the essence of Roman cuisine, served with crusty bread for dipping. \\n\\n5. **Torta di Ricotta e Limone**: A luscious ricotta cheesecake infused with zesty lemon, baked to perfection and topped with a delicate glaze, offering a sweet finish that is both refreshing and indulgent.'}\n",
      "\n",
      "Is Task Completed: False\n"
     ]
    }
   ],
   "source": [
    "# subtasks_completed should be updated with the Creative Agent's outputs\n",
    "my_agent.status()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 20,
   "id": "db731e85-444e-4a9d-8f31-e990594a6bf1",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Calling function Chef with parameters {'instruction': 'Generate ingredients for the 5 Italian dishes'}\n",
      "\n",
      "### Start of Inner Agent: Chef ###\n",
      "\u001b[1m\u001b[30mObservation: The task of generating 5 Italian dish names and descriptions has been completed successfully, providing a rich variety of dishes to work with.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: Now, I need to generate the ingredients for each of the 5 Italian dishes that have been described. This will involve creating a list of ingredients that correspond to each dish, ensuring they are appropriate and authentic to the Italian cuisine.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: Generate a list of ingredients for each of the following dishes: Sogni di Pomodoro, Frittata di Mare, Cappelli di Angelo al Pesto, Pollo alla Romana, and Torta di Ricotta e Limone.\u001b[0m\n",
      "Getting LLM to perform the following task: Generate a list of ingredients for each of the following dishes: Sogni di Pomodoro, Frittata di Mare, Cappelli di Angelo al Pesto, Pollo alla Romana, and Torta di Ricotta e Limone.\n",
      "> 1. **Sogni di Pomodoro**: Ingredients include Arborio rice, fresh tomatoes, vegetable broth, garlic, fresh basil, olive oil, onion, salt, and pepper.\n",
      "\n",
      "2. **Frittata di Mare**: Ingredients consist of eggs, shrimp, calamari, parsley, lemon, olive oil, salt, and pepper.\n",
      "\n",
      "3. **Cappelli di Angelo al Pesto**: Ingredients feature angel hair pasta, fresh basil, pine nuts, Parmesan cheese, garlic, olive oil, salt, and pepper.\n",
      "\n",
      "4. **Pollo alla Romana**: Ingredients include chicken thighs, bell peppers, olives, capers, garlic, onion, tomato sauce, olive oil, salt, and pepper.\n",
      "\n",
      "5. **Torta di Ricotta e Limone**: Ingredients consist of ricotta cheese, lemon zest, sugar, eggs, vanilla extract, graham cracker crust, and a glaze made from powdered sugar and lemon juice.\n",
      "\n",
      "\u001b[1m\u001b[30mObservation: The ingredients for the 5 Italian dishes have been successfully generated, including Sogni di Pomodoro, Frittata di Mare, Cappelli di Angelo al Pesto, Pollo alla Romana, and Torta di Ricotta e Limone.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: Since the ingredients for all dishes have been generated, the next step is to finalize the output and present it to the user.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: End Task\u001b[0m\n",
      "Task completed successfully!\n",
      "\n",
      "\u001b[1m\u001b[35m###\n",
      "Reply from Chef to Menu Creator:\n",
      "1. **Sogni di Pomodoro**: Ingredients include Arborio rice, fresh tomatoes, vegetable broth, garlic, fresh basil, olive oil, onion, salt, and pepper.\n",
      "\n",
      "2. **Frittata di Mare**: Ingredients consist of eggs, shrimp, calamari, parsley, lemon, olive oil, salt, and pepper.\n",
      "\n",
      "3. **Cappelli di Angelo al Pesto**: Ingredients feature angel hair pasta, fresh basil, pine nuts, Parmesan cheese, garlic, olive oil, salt, and pepper.\n",
      "\n",
      "4. **Pollo alla Romana**: Ingredients include chicken thighs, bell peppers, olives, capers, garlic, onion, tomato sauce, olive oil, salt, and pepper.\n",
      "\n",
      "5. **Torta di Ricotta e Limone**: Ingredients consist of ricotta cheese, lemon zest, sugar, eggs, vanilla extract, graham cracker crust, and a glaze made from powdered sugar and lemon juice.\n",
      "###\n",
      "\u001b[0m\n",
      "### End of Inner Agent: Chef ###\n",
      "\n"
     ]
    }
   ],
   "source": [
    "my_agent.use_agent('Chef', 'Generate ingredients for the 5 Italian dishes')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "id": "fd1b1fda-7e72-4669-8e40-f01207ad198f",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Calling function Economist with parameters {'instruction': 'Generate prices for the 5 Italian dishes'}\n",
      "\n",
      "### Start of Inner Agent: Economist ###\n",
      "\u001b[1m\u001b[30mObservation: No subtasks have been completed yet for generating prices for the 5 Italian dishes.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: To complete the assigned task, I need to take the list of the 5 Italian dish names and use the dish_price function to generate their prices.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: Use the dish_price function to input the list of the 5 Italian dish names and obtain their corresponding prices.\u001b[0m\n",
      "Calling function dish_price with parameters {'list_of_dish_names': ['Sogni di Pomodoro', 'Frittata di Mare', 'Cappelli di Angelo al Pesto', 'Pollo alla Romana', 'Torta di Ricotta e Limone']}\n",
      "> {'output_1': {'Sogni di Pomodoro': '$9', 'Frittata di Mare': '$2', 'Cappelli di Angelo al Pesto': '$4', 'Pollo alla Romana': '$8', 'Torta di Ricotta e Limone': '$7'}}\n",
      "\n",
      "\u001b[1m\u001b[30mObservation: The prices for the 5 Italian dishes have been successfully generated, with each dish assigned a specific price.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: Since the task of generating prices for the dishes is complete, I can now proceed to finalize the output and present it to the user.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: End Task\u001b[0m\n",
      "Task completed successfully!\n",
      "\n",
      "\u001b[1m\u001b[35m###\n",
      "Reply from Economist to Menu Creator:\n",
      "The assigned task was to generate prices for the following 5 Italian dishes:\n",
      "\n",
      "1. **Sogni di Pomodoro**: Priced at $9. This dish is a vibrant tomato-based risotto infused with fresh basil and garlic.\n",
      "\n",
      "2. **Frittata di Mare**: Priced at $2. This delightful seafood frittata features shrimp and calamari in a fluffy egg mixture.\n",
      "\n",
      "3. **Cappelli di Angelo al Pesto**: Priced at $4. This dish consists of angel hair pasta tossed in a fragrant basil pesto with pine nuts and Parmesan.\n",
      "\n",
      "4. **Pollo alla Romana**: Priced at $8. This dish includes juicy chicken thighs braised with bell peppers, olives, and capers in a rich tomato sauce.\n",
      "\n",
      "5. **Torta di Ricotta e Limone**: Priced at $7. This is a luscious ricotta cheesecake infused with zesty lemon and topped with a delicate glaze.\n",
      "###\n",
      "\u001b[0m\n",
      "### End of Inner Agent: Economist ###\n",
      "\n"
     ]
    }
   ],
   "source": [
    "my_agent.use_agent('Economist', 'Generate prices for the 5 Italian dishes')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 22,
   "id": "3d0ff21d-3684-49ad-9a59-8d63e8c181aa",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Calling function Boss with parameters {'instruction': 'Choose only 3 Italian dishes to serve that will please Gordon Ramsay'}\n",
      "\n",
      "### Start of Inner Agent: Boss ###\n",
      "\u001b[1m\u001b[30mObservation: I have generated names, descriptions, ingredients, and prices for 5 Italian dishes, but no dishes have been selected yet for serving.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: To choose 3 Italian dishes that will please Gordon Ramsay, I should consider the complexity, presentation, and flavor profiles of the dishes. I can select dishes that are well-balanced and showcase a variety of ingredients and cooking techniques.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: Evaluate the 5 Italian dishes based on their descriptions and select 3 that would likely impress Gordon Ramsay, considering factors such as taste, presentation, and uniqueness.\u001b[0m\n",
      "Getting LLM to perform the following task: Evaluate the 5 Italian dishes based on their descriptions and select 3 that would likely impress Gordon Ramsay, considering factors such as taste, presentation, and uniqueness.\n",
      "> After evaluating the five Italian dishes based on their descriptions, the following three dishes stand out as likely to impress Gordon Ramsay due to their taste, presentation, and uniqueness:\n",
      "\n",
      "1. **Sogni di Pomodoro**: This dish is a vibrant tomato-based risotto that not only showcases the freshness of its ingredients but also evokes the warmth of Italian summers. The infusion of fresh basil and garlic adds depth to the flavor profile, making it a delightful and comforting dish that Ramsay would appreciate for its balance and authenticity.\n",
      "\n",
      "2. **Pollo alla Romana**: The combination of juicy chicken thighs braised with bell peppers, olives, and capers in a rich tomato sauce presents a hearty and visually appealing dish. The use of bold flavors and the rustic presentation would likely resonate with Ramsay's appreciation for traditional yet elevated cuisine.\n",
      "\n",
      "3. **Torta di Ricotta e Limone**: This luscious ricotta cheesecake infused with zesty lemon offers a refreshing and indulgent finish to a meal. The delicate glaze adds an elegant touch, making it not only a treat for the palate but also a feast for the eyes. Its uniqueness as a dessert would likely impress Ramsay, who values creativity in culinary presentations.\n",
      "\n",
      "\u001b[1m\u001b[30mObservation: The evaluation of the five Italian dishes has been completed, and three dishes have been selected that are likely to impress Gordon Ramsay: Sogni di Pomodoro, Pollo alla Romana, and Torta di Ricotta e Limone. These selections were made based on their taste, presentation, and uniqueness.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: The next step is to finalize the selection of the three dishes and present them in a concise format that highlights their appeal to Gordon Ramsay. This will involve summarizing the chosen dishes and their key attributes to ensure clarity and impact.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: Prepare a final presentation of the three selected Italian dishes, including their names and a brief description of why they were chosen to please Gordon Ramsay.\u001b[0m\n",
      "Getting LLM to perform the following task: Prepare a final presentation of the three selected Italian dishes, including their names and a brief description of why they were chosen to please Gordon Ramsay.\n",
      "> The final presentation of the three selected Italian dishes that are likely to impress Gordon Ramsay is as follows:\n",
      "\n",
      "1. **Sogni di Pomodoro**: This vibrant tomato-based risotto is a celebration of fresh ingredients, showcasing the essence of Italian cuisine. The infusion of fresh basil and garlic not only enhances the flavor but also evokes the warmth of a sun-kissed Italian summer. Its comforting nature and balance of flavors make it a dish that Ramsay would appreciate for its authenticity and depth.\n",
      "\n",
      "2. **Pollo alla Romana**: A hearty dish featuring juicy chicken thighs braised with bell peppers, olives, and capers in a rich tomato sauce. The bold flavors and rustic presentation create a visually appealing plate that resonates with Ramsay's appreciation for traditional yet elevated cuisine. This dish stands out for its comforting qualities and the skillful use of classic Italian ingredients.\n",
      "\n",
      "3. **Torta di Ricotta e Limone**: This luscious ricotta cheesecake, infused with zesty lemon, offers a refreshing and indulgent finish to any meal. The delicate glaze adds an elegant touch, making it not only a treat for the palate but also a feast for the eyes. Its uniqueness as a dessert, combined with its creative presentation, would likely impress Ramsay, who values innovation and artistry in culinary creations.\n",
      "\n",
      "\u001b[1m\u001b[30mObservation: The three Italian dishes selected to please Gordon Ramsay are Sogni di Pomodoro, Pollo alla Romana, and Torta di Ricotta e Limone. These dishes were chosen based on their taste, presentation, and uniqueness, which align with Ramsay's culinary standards.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: To complete the assigned task, I need to finalize the presentation of the selected dishes, ensuring that the descriptions highlight their appeal and why they would impress Gordon Ramsay.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: Prepare the final presentation document that includes the names of the three selected Italian dishes along with their descriptions and reasons for selection, formatted for clarity and impact.\u001b[0m\n",
      "Getting LLM to perform the following task: Prepare the final presentation document that includes the names of the three selected Italian dishes along with their descriptions and reasons for selection, formatted for clarity and impact.\n",
      "> Final Presentation Document of Selected Italian Dishes:\n",
      "\n",
      "**1. Sogni di Pomodoro**  \n",
      "*Description*: A vibrant tomato-based risotto that dances on the palate, infused with fresh basil and a hint of garlic, evoking the warmth of a sun-kissed Italian summer.  \n",
      "*Reason for Selection*: This dish showcases the freshness of its ingredients and offers a delightful and comforting experience. Its balance and authenticity make it a standout choice that Gordon Ramsay would appreciate.\n",
      "\n",
      "**2. Pollo alla Romana**  \n",
      "*Description*: Juicy chicken thighs braised with bell peppers, olives, and capers, simmered in a rich tomato sauce that captures the essence of Roman cuisine, served with crusty bread for dipping.  \n",
      "*Reason for Selection*: The hearty nature and rustic presentation of this dish, combined with bold flavors, resonate with Ramsay's appreciation for traditional yet elevated cuisine. It stands out for its comforting qualities and skillful use of classic Italian ingredients.\n",
      "\n",
      "**3. Torta di Ricotta e Limone**  \n",
      "*Description*: A luscious ricotta cheesecake infused with zesty lemon, baked to perfection and topped with a delicate glaze, offering a sweet finish that is both refreshing and indulgent.  \n",
      "*Reason for Selection*: This dessert not only provides a refreshing end to a meal but also features an elegant presentation. Its uniqueness and creative flair would likely impress Ramsay, who values innovation and artistry in culinary creations.\n",
      "\n",
      "\u001b[1m\u001b[30mObservation: The three Italian dishes selected for presentation are Sogni di Pomodoro, Pollo alla Romana, and Torta di Ricotta e Limone. Each dish was chosen based on its taste, presentation, and uniqueness, which are qualities that would likely impress Gordon Ramsay.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: To complete the assigned task, I need to prepare a final presentation document that clearly outlines the names, descriptions, and reasons for selecting these three dishes. This document should be formatted for clarity and impact to ensure it meets the expectations of the task.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: Prepare the final presentation document that includes the names of the three selected Italian dishes along with their descriptions and reasons for selection, formatted for clarity and impact.\u001b[0m\n",
      "Getting LLM to perform the following task: Prepare the final presentation document that includes the names of the three selected Italian dishes along with their descriptions and reasons for selection, formatted for clarity and impact.\n",
      "> Final Presentation Document of Selected Italian Dishes:\n",
      "\n",
      "**1. Sogni di Pomodoro**  \n",
      "*Description*: A vibrant tomato-based risotto that dances on the palate, infused with fresh basil and a hint of garlic, evoking the warmth of a sun-kissed Italian summer.  \n",
      "*Reason for Selection*: This dish showcases the freshness of its ingredients and offers a delightful and comforting experience. Its balance and authenticity make it a standout choice that Gordon Ramsay would appreciate.\n",
      "\n",
      "**2. Pollo alla Romana**  \n",
      "*Description*: Juicy chicken thighs braised with bell peppers, olives, and capers, simmered in a rich tomato sauce that captures the essence of Roman cuisine, served with crusty bread for dipping.  \n",
      "*Reason for Selection*: The hearty nature and rustic presentation of this dish, combined with bold flavors, resonate with Ramsay's appreciation for traditional yet elevated cuisine. It stands out for its comforting qualities and skillful use of classic Italian ingredients.\n",
      "\n",
      "**3. Torta di Ricotta e Limone**  \n",
      "*Description*: A luscious ricotta cheesecake infused with zesty lemon, baked to perfection and topped with a delicate glaze, offering a sweet finish that is both refreshing and indulgent.  \n",
      "*Reason for Selection*: This dessert not only provides a refreshing end to a meal but also features an elegant presentation. Its uniqueness and creative flair would likely impress Ramsay, who values innovation and artistry in culinary creations.\n",
      "\n",
      "\u001b[1m\u001b[30mObservation: The three selected Italian dishes, Sogni di Pomodoro, Pollo alla Romana, and Torta di Ricotta e Limone, have been carefully chosen based on their descriptions and the likelihood of impressing Gordon Ramsay. Each dish showcases unique flavors, appealing presentations, and a balance of traditional and innovative elements.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: To complete the assigned task, I need to prepare a final presentation document that clearly outlines the names, descriptions, and reasons for selecting these three dishes. This document should be formatted for clarity and impact to ensure it effectively communicates the choices made.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: Prepare the final presentation document that includes the names of the three selected Italian dishes along with their descriptions and reasons for selection, formatted for clarity and impact.\u001b[0m\n",
      "Getting LLM to perform the following task: Prepare the final presentation document that includes the names of the three selected Italian dishes along with their descriptions and reasons for selection, formatted for clarity and impact.\n",
      "> Final Presentation Document of Selected Italian Dishes:\n",
      "\n",
      "**1. Sogni di Pomodoro**  \n",
      "*Description*: A vibrant tomato-based risotto that dances on the palate, infused with fresh basil and a hint of garlic, evoking the warmth of a sun-kissed Italian summer.  \n",
      "*Reason for Selection*: This dish showcases the freshness of its ingredients and offers a delightful and comforting experience. Its balance and authenticity make it a standout choice that Gordon Ramsay would appreciate.\n",
      "\n",
      "**2. Pollo alla Romana**  \n",
      "*Description*: Juicy chicken thighs braised with bell peppers, olives, and capers, simmered in a rich tomato sauce that captures the essence of Roman cuisine, served with crusty bread for dipping.  \n",
      "*Reason for Selection*: The hearty nature and rustic presentation of this dish, combined with bold flavors, resonate with Ramsay's appreciation for traditional yet elevated cuisine. It stands out for its comforting qualities and skillful use of classic Italian ingredients.\n",
      "\n",
      "**3. Torta di Ricotta e Limone**  \n",
      "*Description*: A luscious ricotta cheesecake infused with zesty lemon, baked to perfection and topped with a delicate glaze, offering a sweet finish that is both refreshing and indulgent.  \n",
      "*Reason for Selection*: This dessert not only provides a refreshing end to a meal but also features an elegant presentation. Its uniqueness and creative flair would likely impress Ramsay, who values innovation and artistry in culinary creations.\n",
      "\n",
      "\u001b[1m\u001b[35m###\n",
      "Reply from Boss to Menu Creator:\n",
      "After evaluating the five Italian dishes based on their descriptions, the following three dishes stand out as likely to impress Gordon Ramsay due to their taste, presentation, and uniqueness:\n",
      "\n",
      "1. **Sogni di Pomodoro**: This vibrant tomato-based risotto is a celebration of fresh ingredients, showcasing the essence of Italian cuisine. The infusion of fresh basil and garlic not only enhances the flavor but also evokes the warmth of a sun-kissed Italian summer. Its comforting nature and balance of flavors make it a dish that Ramsay would appreciate for its authenticity and depth.\n",
      "\n",
      "2. **Pollo alla Romana**: A hearty dish featuring juicy chicken thighs braised with bell peppers, olives, and capers in a rich tomato sauce. The bold flavors and rustic presentation create a visually appealing plate that resonates with Ramsay's appreciation for traditional yet elevated cuisine. This dish stands out for its comforting qualities and the skillful use of classic Italian ingredients.\n",
      "\n",
      "3. **Torta di Ricotta e Limone**: This luscious ricotta cheesecake, infused with zesty lemon, offers a refreshing and indulgent finish to any meal. The delicate glaze adds an elegant touch, making it not only a treat for the palate but also a feast for the eyes. Its uniqueness as a dessert, combined with its creative presentation, would likely impress Ramsay, who values innovation and artistry in culinary creations.\n",
      "###\n",
      "\u001b[0m\n",
      "### End of Inner Agent: Boss ###\n",
      "\n"
     ]
    }
   ],
   "source": [
    "my_agent.use_agent('Boss', 'Choose only 3 Italian dishes to serve that will please Gordon Ramsay')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 23,
   "id": "9290ba86-c1a7-4d68-ba2e-5ffbe292f403",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "1. **Sogni di Pomodoro**: A vibrant tomato-based risotto that dances on the palate, infused with fresh basil and a hint of garlic, evoking the warmth of a sun-kissed Italian summer. \n",
      "   Ingredients: Arborio rice, fresh tomatoes, vegetable broth, garlic, fresh basil, olive oil, onion, salt, and pepper. \n",
      "   Price: $9.\n",
      "\n",
      "2. **Pollo alla Romana**: Juicy chicken thighs braised with bell peppers, olives, and capers, simmered in a rich tomato sauce that captures the essence of Roman cuisine, served with crusty bread for dipping. \n",
      "   Ingredients: Chicken thighs, bell peppers, olives, capers, garlic, onion, tomato sauce, olive oil, salt, and pepper. \n",
      "   Price: $8.\n",
      "\n",
      "3. **Torta di Ricotta e Limone**: A luscious ricotta cheesecake infused with zesty lemon, baked to perfection and topped with a delicate glaze, offering a sweet finish that is both refreshing and indulgent. \n",
      "   Ingredients: Ricotta cheese, lemon zest, sugar, eggs, vanilla extract, graham cracker crust, and a glaze made from powdered sugar and lemon juice. \n",
      "   Price: $7.\n"
     ]
    }
   ],
   "source": [
    "# Get a targeted response from your agent, and add this to subtask_completed\n",
    "output = my_agent.reply_user('Choose the 3 menu items for Gordon Ramsay line by line in this format - Dish Name, Dish Description, Dish Ingredients, Dish Price')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 24,
   "id": "8c134d97-36fc-4f33-880a-fbb79e2388a5",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Agent Name: Menu Creator\n",
      "Agent Description: Creates a menu for a restaurant. Menu item includes Name, Description, Ingredients, Pricing.\n",
      "Available Functions: ['use_llm', 'end_task', 'Chef', 'Boss', 'Creative Writer', 'Economist']\n",
      "Shared Variables: ['agent']\n",
      "\u001b[1m\u001b[32mTask: Generate 5 menu items with name, description, ingredients and price based on Italian food choices. Output in Menu format.\u001b[0m\n",
      "\u001b[1m\u001b[30mSubtasks Completed:\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask: Creative Writer(instruction=\"Generate only 5 Italian dish names and descriptions\")\u001b[0m\n",
      "{'Agent Output': '1. **Sogni di Pomodoro**: A vibrant tomato-based risotto that dances on the palate, infused with fresh basil and a hint of garlic, evoking the warmth of a sun-kissed Italian summer. \\n\\n2. **Frittata di Mare**: A delightful seafood frittata, featuring tender shrimp and succulent calamari, all enveloped in a fluffy egg mixture, garnished with a sprinkle of parsley and a drizzle of lemon. \\n\\n3. **Cappelli di Angelo al Pesto**: Angel hair pasta elegantly tossed in a fragrant basil pesto, adorned with pine nuts and a touch of Parmesan, creating a dish that is both light and bursting with flavor. \\n\\n4. **Pollo alla Romana**: Juicy chicken thighs braised with bell peppers, olives, and capers, simmered in a rich tomato sauce that captures the essence of Roman cuisine, served with crusty bread for dipping. \\n\\n5. **Torta di Ricotta e Limone**: A luscious ricotta cheesecake infused with zesty lemon, baked to perfection and topped with a delicate glaze, offering a sweet finish that is both refreshing and indulgent.'}\n",
      "\n",
      "\u001b[1m\u001b[34mSubtask: Chef(instruction=\"Generate ingredients for the 5 Italian dishes\")\u001b[0m\n",
      "{'Agent Output': '1. **Sogni di Pomodoro**: Ingredients include Arborio rice, fresh tomatoes, vegetable broth, garlic, fresh basil, olive oil, onion, salt, and pepper.\\n\\n2. **Frittata di Mare**: Ingredients consist of eggs, shrimp, calamari, parsley, lemon, olive oil, salt, and pepper.\\n\\n3. **Cappelli di Angelo al Pesto**: Ingredients feature angel hair pasta, fresh basil, pine nuts, Parmesan cheese, garlic, olive oil, salt, and pepper.\\n\\n4. **Pollo alla Romana**: Ingredients include chicken thighs, bell peppers, olives, capers, garlic, onion, tomato sauce, olive oil, salt, and pepper.\\n\\n5. **Torta di Ricotta e Limone**: Ingredients consist of ricotta cheese, lemon zest, sugar, eggs, vanilla extract, graham cracker crust, and a glaze made from powdered sugar and lemon juice.'}\n",
      "\n",
      "\u001b[1m\u001b[34mSubtask: Economist(instruction=\"Generate prices for the 5 Italian dishes\")\u001b[0m\n",
      "{'Agent Output': 'The assigned task was to generate prices for the following 5 Italian dishes:\\n\\n1. **Sogni di Pomodoro**: Priced at $9. This dish is a vibrant tomato-based risotto infused with fresh basil and garlic.\\n\\n2. **Frittata di Mare**: Priced at $2. This delightful seafood frittata features shrimp and calamari in a fluffy egg mixture.\\n\\n3. **Cappelli di Angelo al Pesto**: Priced at $4. This dish consists of angel hair pasta tossed in a fragrant basil pesto with pine nuts and Parmesan.\\n\\n4. **Pollo alla Romana**: Priced at $8. This dish includes juicy chicken thighs braised with bell peppers, olives, and capers in a rich tomato sauce.\\n\\n5. **Torta di Ricotta e Limone**: Priced at $7. This is a luscious ricotta cheesecake infused with zesty lemon and topped with a delicate glaze.'}\n",
      "\n",
      "\u001b[1m\u001b[34mSubtask: Boss(instruction=\"Choose only 3 Italian dishes to serve that will please Gordon Ramsay\")\u001b[0m\n",
      "{'Agent Output': \"After evaluating the five Italian dishes based on their descriptions, the following three dishes stand out as likely to impress Gordon Ramsay due to their taste, presentation, and uniqueness:\\n\\n1. **Sogni di Pomodoro**: This vibrant tomato-based risotto is a celebration of fresh ingredients, showcasing the essence of Italian cuisine. The infusion of fresh basil and garlic not only enhances the flavor but also evokes the warmth of a sun-kissed Italian summer. Its comforting nature and balance of flavors make it a dish that Ramsay would appreciate for its authenticity and depth.\\n\\n2. **Pollo alla Romana**: A hearty dish featuring juicy chicken thighs braised with bell peppers, olives, and capers in a rich tomato sauce. The bold flavors and rustic presentation create a visually appealing plate that resonates with Ramsay's appreciation for traditional yet elevated cuisine. This dish stands out for its comforting qualities and the skillful use of classic Italian ingredients.\\n\\n3. **Torta di Ricotta e Limone**: This luscious ricotta cheesecake, infused with zesty lemon, offers a refreshing and indulgent finish to any meal. The delicate glaze adds an elegant touch, making it not only a treat for the palate but also a feast for the eyes. Its uniqueness as a dessert, combined with its creative presentation, would likely impress Ramsay, who values innovation and artistry in culinary creations.\"}\n",
      "\n",
      "\u001b[1m\u001b[34mSubtask: Choose the 3 menu items for Gordon Ramsay line by line in this format - Dish Name, Dish Description, Dish Ingredients, Dish Price\u001b[0m\n",
      "1. **Sogni di Pomodoro**: A vibrant tomato-based risotto that dances on the palate, infused with fresh basil and a hint of garlic, evoking the warmth of a sun-kissed Italian summer. \n",
      "   Ingredients: Arborio rice, fresh tomatoes, vegetable broth, garlic, fresh basil, olive oil, onion, salt, and pepper. \n",
      "   Price: $9.\n",
      "\n",
      "2. **Pollo alla Romana**: Juicy chicken thighs braised with bell peppers, olives, and capers, simmered in a rich tomato sauce that captures the essence of Roman cuisine, served with crusty bread for dipping. \n",
      "   Ingredients: Chicken thighs, bell peppers, olives, capers, garlic, onion, tomato sauce, olive oil, salt, and pepper. \n",
      "   Price: $8.\n",
      "\n",
      "3. **Torta di Ricotta e Limone**: A luscious ricotta cheesecake infused with zesty lemon, baked to perfection and topped with a delicate glaze, offering a sweet finish that is both refreshing and indulgent. \n",
      "   Ingredients: Ricotta cheese, lemon zest, sugar, eggs, vanilla extract, graham cracker crust, and a glaze made from powdered sugar and lemon juice. \n",
      "   Price: $7.\n",
      "\n",
      "Is Task Completed: False\n"
     ]
    }
   ],
   "source": [
    "my_agent.status()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 25,
   "id": "8bdfad22-3b67-42d5-a4c6-cfd607dd22b1",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "1. **Sogni di Pomodoro**: A vibrant tomato-based risotto that dances on the palate, infused with fresh basil and a hint of garlic, evoking the warmth of a sun-kissed Italian summer. \n",
      "   Ingredients: Arborio rice, fresh tomatoes, vegetable broth, garlic, fresh basil, olive oil, onion, salt, and pepper. \n",
      "   Price: $9.\n",
      "\n",
      "2. **Pollo alla Romana**: Juicy chicken thighs braised with bell peppers, olives, and capers, simmered in a rich tomato sauce that captures the essence of Roman cuisine, served with crusty bread for dipping. \n",
      "   Ingredients: Chicken thighs, bell peppers, olives, capers, garlic, onion, tomato sauce, olive oil, salt, and pepper. \n",
      "   Price: $8.\n",
      "\n",
      "3. **Torta di Ricotta e Limone**: A luscious ricotta cheesecake infused with zesty lemon, baked to perfection and topped with a delicate glaze, offering a sweet finish that is both refreshing and indulgent. \n",
      "   Ingredients: Ricotta cheese, lemon zest, sugar, eggs, vanilla extract, graham cracker crust, and a glaze made from powdered sugar and lemon juice. \n",
      "   Price: $7.\n"
     ]
    }
   ],
   "source": [
    "# you can summarise the subtasks_history for the next task\n",
    "my_agent.summarise_subtasks_completed('Generate the 3 Italian dishes for Gordon Ramsay in menu format')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 26,
   "id": "ea8ebd26-25e2-41b3-a2a3-78f03d324f3b",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Agent Name: Menu Creator\n",
      "Agent Description: Creates a menu for a restaurant. Menu item includes Name, Description, Ingredients, Pricing.\n",
      "Available Functions: ['use_llm', 'end_task', 'Chef', 'Boss', 'Creative Writer', 'Economist']\n",
      "Shared Variables: ['agent']\n",
      "\u001b[1m\u001b[32mTask: Generate 5 menu items with name, description, ingredients and price based on Italian food choices. Output in Menu format.\u001b[0m\n",
      "\u001b[1m\u001b[30mSubtasks Completed:\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask: Current Results for 'Generate the 3 Italian dishes for Gordon Ramsay in menu format'\u001b[0m\n",
      "1. **Sogni di Pomodoro**: A vibrant tomato-based risotto that dances on the palate, infused with fresh basil and a hint of garlic, evoking the warmth of a sun-kissed Italian summer. \n",
      "   Ingredients: Arborio rice, fresh tomatoes, vegetable broth, garlic, fresh basil, olive oil, onion, salt, and pepper. \n",
      "   Price: $9.\n",
      "\n",
      "2. **Pollo alla Romana**: Juicy chicken thighs braised with bell peppers, olives, and capers, simmered in a rich tomato sauce that captures the essence of Roman cuisine, served with crusty bread for dipping. \n",
      "   Ingredients: Chicken thighs, bell peppers, olives, capers, garlic, onion, tomato sauce, olive oil, salt, and pepper. \n",
      "   Price: $8.\n",
      "\n",
      "3. **Torta di Ricotta e Limone**: A luscious ricotta cheesecake infused with zesty lemon, baked to perfection and topped with a delicate glaze, offering a sweet finish that is both refreshing and indulgent. \n",
      "   Ingredients: Ricotta cheese, lemon zest, sugar, eggs, vanilla extract, graham cracker crust, and a glaze made from powdered sugar and lemon juice. \n",
      "   Price: $7.\n",
      "\n",
      "Is Task Completed: False\n"
     ]
    }
   ],
   "source": [
    "# subtask has been summarised, helping to reduce the context significantly for later task generation\n",
    "my_agent.status()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 27,
   "id": "8283f502-512f-47fc-9270-5ac9457d8a01",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{\"Current Results for 'Generate the 3 Italian dishes for Gordon Ramsay in menu format'\": '1. **Sogni di Pomodoro**: A vibrant tomato-based risotto that dances on the palate, infused with fresh basil and a hint of garlic, evoking the warmth of a sun-kissed Italian summer. \\n   Ingredients: Arborio rice, fresh tomatoes, vegetable broth, garlic, fresh basil, olive oil, onion, salt, and pepper. \\n   Price: $9.\\n\\n2. **Pollo alla Romana**: Juicy chicken thighs braised with bell peppers, olives, and capers, simmered in a rich tomato sauce that captures the essence of Roman cuisine, served with crusty bread for dipping. \\n   Ingredients: Chicken thighs, bell peppers, olives, capers, garlic, onion, tomato sauce, olive oil, salt, and pepper. \\n   Price: $8.\\n\\n3. **Torta di Ricotta e Limone**: A luscious ricotta cheesecake infused with zesty lemon, baked to perfection and topped with a delicate glaze, offering a sweet finish that is both refreshing and indulgent. \\n   Ingredients: Ricotta cheese, lemon zest, sugar, eggs, vanilla extract, graham cracker crust, and a glaze made from powdered sugar and lemon juice. \\n   Price: $7.'}"
      ]
     },
     "execution_count": 27,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "my_agent.subtasks_completed"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 28,
   "id": "c7887297-c007-42fd-a21e-5196afef605a",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Removed last subtask from subtasks_completed: (\"Current Results for 'Generate the 3 Italian dishes for Gordon Ramsay in menu format'\", '1. **Sogni di Pomodoro**: A vibrant tomato-based risotto that dances on the palate, infused with fresh basil and a hint of garlic, evoking the warmth of a sun-kissed Italian summer. \\n   Ingredients: Arborio rice, fresh tomatoes, vegetable broth, garlic, fresh basil, olive oil, onion, salt, and pepper. \\n   Price: $9.\\n\\n2. **Pollo alla Romana**: Juicy chicken thighs braised with bell peppers, olives, and capers, simmered in a rich tomato sauce that captures the essence of Roman cuisine, served with crusty bread for dipping. \\n   Ingredients: Chicken thighs, bell peppers, olives, capers, garlic, onion, tomato sauce, olive oil, salt, and pepper. \\n   Price: $8.\\n\\n3. **Torta di Ricotta e Limone**: A luscious ricotta cheesecake infused with zesty lemon, baked to perfection and topped with a delicate glaze, offering a sweet finish that is both refreshing and indulgent. \\n   Ingredients: Ricotta cheese, lemon zest, sugar, eggs, vanilla extract, graham cracker crust, and a glaze made from powdered sugar and lemon juice. \\n   Price: $7.')\n"
     ]
    }
   ],
   "source": [
    "# we also allow you to manually remove the last item added to subtasks_completed, so you can revert your agent state if needed for backtracking\n",
    "my_agent.remove_last_subtask()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 29,
   "id": "35340b71-d7ea-4f28-ad0a-dddf6099737f",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Agent Name: Menu Creator\n",
      "Agent Description: Creates a menu for a restaurant. Menu item includes Name, Description, Ingredients, Pricing.\n",
      "Available Functions: ['use_llm', 'end_task', 'Chef', 'Boss', 'Creative Writer', 'Economist']\n",
      "Shared Variables: ['agent']\n",
      "\u001b[1m\u001b[32mTask: Generate 5 menu items with name, description, ingredients and price based on Italian food choices. Output in Menu format.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtasks Completed: None\u001b[0m\n",
      "Is Task Completed: False\n"
     ]
    }
   ],
   "source": [
    "# visualise your agent's status - should have empty subtasks_completed\n",
    "my_agent.status()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 30,
   "id": "ca73a9cb-73c2-4768-8e75-eb42063afb2e",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{}"
      ]
     },
     "execution_count": 30,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "my_agent.subtasks_completed"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "8a03f820-c405-4004-982a-b46dc7c9a191",
   "metadata": {},
   "source": [
    "# More Advanced Agents with Shared Variables\n",
    "- Each Inner Agent has access to a specific part of Shared Variables via Global Context\n",
    "- Each Inner Agent will update the Shared Variables as it is changed by their actions\n",
    "- Meta Agent will know of the change in inventory via Shared Variables, as well as the reply of the Inner Agent"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 31,
   "id": "e3b4c40c-a742-4909-82c9-3c351eda4aac",
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "def add_item_to_inventory(shared_variables, item: str) -> str:\n",
    "    ''' Adds item to inventory, and returns outcome of action '''\n",
    "    shared_variables['Inventory A'].append(item)\n",
    "    return f'{item} successfully added to Inventory A'\n",
    "\n",
    "def remove_item_from_inventory(shared_variables, item: str) -> str:\n",
    "    ''' Removes item from inventory and returns outcome of action '''\n",
    "    if item in shared_variables['Inventory A']:\n",
    "        shared_variables['Inventory A'].remove(item)\n",
    "        return f'{item} successfully removed from Inventory A'\n",
    "    else:\n",
    "        return f'{item} not found in Inventory A, unable to remove'"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 32,
   "id": "fd56209d-21c8-4630-ab4d-5f819789d980",
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "def add_item_to_inventory_2(shared_variables, item: str) -> str:\n",
    "    ''' Adds item to inventory, and returns outcome of action '''\n",
    "    shared_variables['Inventory B'].append(item)\n",
    "    return f'{item} successfully added to Inventory B'\n",
    "    \n",
    "def remove_item_from_inventory_2(shared_variables, item: str) -> str:\n",
    "    ''' Removes item from inventory and returns outcome of action '''\n",
    "    if item in shared_variables['Inventory B']:\n",
    "        shared_variables['Inventory B'].remove(item)\n",
    "        return f'{item} successfully removed from Inventory B'\n",
    "    else:\n",
    "        return f'{item} not found in Inventory B, unable to remove'"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 33,
   "id": "ddd6a358-be78-4f49-871b-674456b08c1c",
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "agent_a = Agent('Factory A Inventory Manager', \n",
    "      'Adds and removes items in Inventory A. Only able to remove items if present in Inventory A',\n",
    "      global_context = 'Inventory of Factory A: <Inventory A>', # Add in Global Context here with shared_variables Inventory\n",
    "      llm = llm).assign_functions([add_item_to_inventory, remove_item_from_inventory])\n",
    "\n",
    "agent_b = Agent('Factory B Inventory Manager', \n",
    "      'Adds and removes items in Inventory B. Only able to remove items if present in Inventory B',\n",
    "      global_context = 'Inventory of Factory B: <Inventory B>', # Add in Global Context here with shared_variables Inventory\n",
    "      llm = llm).assign_functions([add_item_to_inventory_2, remove_item_from_inventory_2])\n",
    "\n",
    "meta_agent = Agent('Inventory Manager', 'Manages inventory of all factories', \n",
    "       shared_variables = {'Inventory A': [],\n",
    "                           'Inventory B': []},\n",
    "        llm = llm,\n",
    "       global_context = 'Inventory of Factory A: <Inventory A>\\nInventory of Factory B: <Inventory B>').assign_functions([agent_a, agent_b])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 34,
   "id": "12f2bd5f-4b8c-44d0-9c96-90a667a38a93",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Name: use_llm\n",
      "Description: For general tasks. Used only when no other function can do the task\n",
      "Input: []\n",
      "Output: {'Output': 'Output of LLM'}\n",
      "\n",
      "Name: end_task\n",
      "Description: Passes the final output to the user\n",
      "Input: []\n",
      "Output: {}\n",
      "\n",
      "Name: Factory A Inventory Manager\n",
      "Description: Agent Description: ```Adds and removes items in Inventory A. Only able to remove items if present in Inventory A```\n",
      "Executes the given <instruction>\n",
      "Input: ['instruction']\n",
      "Output: {'Agent Output': 'Output of instruction'}\n",
      "\n",
      "Name: Factory B Inventory Manager\n",
      "Description: Agent Description: ```Adds and removes items in Inventory B. Only able to remove items if present in Inventory B```\n",
      "Executes the given <instruction>\n",
      "Input: ['instruction']\n",
      "Output: {'Agent Output': 'Output of instruction'}\n",
      "\n"
     ]
    }
   ],
   "source": [
    "meta_agent.print_functions()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 35,
   "id": "e7d4ffdf-031d-4f54-802e-c0447b67172a",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\u001b[1m\u001b[30mObservation: No items have been added to either Factory A or Factory B yet.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: To complete the Assigned Task, I need to add apples to Factory A and pears to Factory B. Since both inventories are currently empty, I can proceed to add the items directly.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: Add apples to Factory A.\u001b[0m\n",
      "Calling function Factory A Inventory Manager with parameters {'instruction': 'Add apples to Factory A.'}\n",
      "\n",
      "### Start of Inner Agent: Factory A Inventory Manager ###\n",
      "\u001b[1m\u001b[30mObservation: No items have been added to Inventory A yet, and the inventory is currently empty.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: To complete the assigned task of adding apples to Factory A, I need to use the function that allows me to add items to the inventory.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: Add apples to Inventory A using the add_item_to_inventory function.\u001b[0m\n",
      "Calling function add_item_to_inventory with parameters {'item': 'apples'}\n",
      "> {'output_1': 'apples successfully added to Inventory A'}\n",
      "\n",
      "\u001b[1m\u001b[30mObservation: The task to add apples to Inventory A has been completed successfully, as indicated by the output stating that apples were successfully added.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: Since the task of adding apples has been completed, there is no further action required to fulfill the Assigned Task.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: End Task\u001b[0m\n",
      "Task completed successfully!\n",
      "\n",
      "\u001b[1m\u001b[35m###\n",
      "Reply from Factory A Inventory Manager to Inventory Manager:\n",
      "The assigned task was to add apples to Factory A. However, according to the global context, the current inventory of Factory A already contains apples. The subtask completed indicates that apples were successfully added to Inventory A, but since they were already present, this action does not change the inventory. Therefore, the inventory remains as follows: Inventory A: ['apples'].\n",
      "###\n",
      "\u001b[0m\n",
      "### End of Inner Agent: Factory A Inventory Manager ###\n",
      "\n",
      "\u001b[1m\u001b[30mObservation: The task to add apples to Factory A was completed, but the inventory already contained apples, so no change occurred. The inventory for Factory A remains as is: Inventory A: ['apples'].\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: The only remaining part of the assigned task is to add pears to Factory B, which is currently empty. I need to execute the appropriate function to add pears to Factory B.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: Add pears to Factory B.\u001b[0m\n",
      "Calling function Factory B Inventory Manager with parameters {'instruction': 'Add pears to Factory B.'}\n",
      "\n",
      "### Start of Inner Agent: Factory B Inventory Manager ###\n",
      "\u001b[1m\u001b[30mObservation: No subtasks have been completed yet, and the current inventory of Factory B is empty.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: Since the inventory of Factory B is empty, I can proceed to add pears to it without any issues.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: Add pears to Inventory B.\u001b[0m\n",
      "Calling function add_item_to_inventory_2 with parameters {'item': 'pears'}\n",
      "> {'output_1': 'pears successfully added to Inventory B'}\n",
      "\n",
      "\u001b[1m\u001b[30mObservation: The assigned task was to add pears to Factory B. The subtask completed indicates that pears were successfully added to Inventory B.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: Since the task of adding pears to Inventory B has been completed, there is no further action required. The task can now be concluded.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: End Task\u001b[0m\n",
      "Task completed successfully!\n",
      "\n",
      "\u001b[1m\u001b[35m###\n",
      "Reply from Factory B Inventory Manager to Inventory Manager:\n",
      "The assigned task was to add pears to Factory B. According to the global context, the current inventory of Factory B already contains pears. The subtask completed indicates that pears were successfully added to Inventory B. However, since pears were already present in the inventory, this action does not change the inventory. Therefore, the inventory remains as follows: Inventory B: ['pears'].\n",
      "###\n",
      "\u001b[0m\n",
      "### End of Inner Agent: Factory B Inventory Manager ###\n",
      "\n",
      "\u001b[1m\u001b[30mObservation: The assigned tasks to add apples to Factory A and pears to Factory B have been completed. However, both items were already present in their respective inventories, so no changes were made to the inventories.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: Since both inventories already contain the items that were supposed to be added, there are no further actions needed to complete the assigned task. The task can be considered complete.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: End Task\u001b[0m\n",
      "Task completed successfully!\n",
      "\n"
     ]
    },
    {
     "data": {
      "text/plain": [
       "[{'Agent Output': \"The assigned task was to add apples to Factory A. However, according to the global context, the current inventory of Factory A already contains apples. The subtask completed indicates that apples were successfully added to Inventory A, but since they were already present, this action does not change the inventory. Therefore, the inventory remains as follows: Inventory A: ['apples'].\"},\n",
       " {'Agent Output': \"The assigned task was to add pears to Factory B. According to the global context, the current inventory of Factory B already contains pears. The subtask completed indicates that pears were successfully added to Inventory B. However, since pears were already present in the inventory, this action does not change the inventory. Therefore, the inventory remains as follows: Inventory B: ['pears'].\"}]"
      ]
     },
     "execution_count": 35,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "meta_agent.run('Add apples to Factory A, add pears to Factory B')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 36,
   "id": "7b3f56dc-ecb8-4b85-bc34-46c7040d8c82",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'Inventory A': ['apples'],\n",
       " 'Inventory B': ['pears'],\n",
       " 'agent': <taskgen.agent.Agent at 0x124ab7510>}"
      ]
     },
     "execution_count": 36,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "meta_agent.shared_variables"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 37,
   "id": "8aef8c60-dc47-40cd-a2b4-31299390adf3",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\u001b[1m\u001b[30mObservation: The subtasks completed involved adding items to both Factory A and Factory B, but since the items were already present in the inventories, no changes were made. Factory A still contains apples, and Factory B still contains pears.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: To complete the assigned task, I need to remove apples from Factory A, and then add bananas and kiwi to the inventory. Since the removal of apples is the first step, I should focus on that before adding the new items.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: Remove apples from Factory A.\u001b[0m\n",
      "Calling function Factory A Inventory Manager with parameters {'instruction': 'Remove apples from Factory A.'}\n",
      "\n",
      "### Start of Inner Agent: Factory A Inventory Manager ###\n",
      "\u001b[1m\u001b[30mObservation: The assigned task is to remove apples from Factory A. The current inventory of Factory A contains apples, which means the removal can be executed.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: Since the inventory contains the item to be removed, I can proceed to remove apples from Inventory A. This will complete the assigned task.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: Remove apples from Inventory A by using the remove_item_from_inventory function.\u001b[0m\n",
      "Calling function remove_item_from_inventory with parameters {'item': 'apples'}\n",
      "> {'output_1': 'apples successfully removed from Inventory A'}\n",
      "\n",
      "\u001b[1m\u001b[30mObservation: The assigned task was to remove apples from Factory A. The subtask completed indicates that apples were successfully removed from Inventory A, which means the action has been completed as intended.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: Since the task of removing apples from Inventory A has been successfully completed, there is no further action required. The task can be concluded.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: End Task\u001b[0m\n",
      "Task completed successfully!\n",
      "\n",
      "\u001b[1m\u001b[35m###\n",
      "Reply from Factory A Inventory Manager to Inventory Manager:\n",
      "The assigned task was to remove apples from Factory A. According to the global context, the current inventory of Factory A contains apples. The subtask completed indicates that apples were successfully removed from Inventory A. Therefore, after this action, the inventory of Factory A is now empty: Inventory A: [].\n",
      "###\n",
      "\u001b[0m\n",
      "### End of Inner Agent: Factory A Inventory Manager ###\n",
      "\n",
      "\u001b[1m\u001b[30mObservation: The apples were successfully removed from Factory A, leaving the inventory empty. The previous attempts to add apples and pears did not change the inventories as they were already present.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: Now that Factory A is empty, I need to add bananas and kiwi to the inventory. Since there are no restrictions on adding items, I can proceed with this action.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: Add bananas and kiwi to Factory A.\u001b[0m\n",
      "Calling function Factory A Inventory Manager with parameters {'instruction': 'Add bananas and kiwi to Factory A.'}\n",
      "\n",
      "### Start of Inner Agent: Factory A Inventory Manager ###\n",
      "\u001b[1m\u001b[30mObservation: The assigned task is to add bananas and kiwi to Factory A. Currently, the inventory of Factory A is empty, which means there are no items present to conflict with the addition of new items.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: Since the inventory is empty, I can proceed to add both bananas and kiwi to Factory A without any issues. I will first add bananas and then kiwi to ensure both items are included in the inventory.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: Add bananas to Factory A inventory.\u001b[0m\n",
      "Calling function add_item_to_inventory with parameters {'item': 'bananas'}\n",
      "> {'output_1': 'bananas successfully added to Inventory A'}\n",
      "\n",
      "\u001b[1m\u001b[30mObservation: The assigned task was to add bananas and kiwi to Factory A. The subtask completed indicates that bananas were successfully added to Inventory A, which now contains bananas. However, the task of adding kiwi has not yet been addressed.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: To complete the remainder of the assigned task, I need to add kiwi to Inventory A. Since kiwi is not currently in the inventory, I can proceed with adding it.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: Add kiwi to Inventory A.\u001b[0m\n",
      "Calling function add_item_to_inventory with parameters {'item': 'kiwi'}\n",
      "> {'output_1': 'kiwi successfully added to Inventory A'}\n",
      "\n",
      "\u001b[1m\u001b[30mObservation: The assigned task was to add bananas and kiwi to Factory A. Both items were successfully added to Inventory A, which already contained these items. Therefore, the inventory remains unchanged: Inventory A: ['bananas', 'kiwi'].\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: Since the items to be added are already present in the inventory, there is no further action needed to complete the assigned task. The task can be considered complete.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: End Task\u001b[0m\n",
      "Task completed successfully!\n",
      "\n",
      "\u001b[1m\u001b[35m###\n",
      "Reply from Factory A Inventory Manager to Inventory Manager:\n",
      "The assigned task was to add bananas and kiwi to Factory A. However, according to the global context, the current inventory of Factory A already contains bananas and kiwi. The subtasks completed indicate that bananas were successfully added to Inventory A and kiwi was also successfully added to Inventory A. Since both items were already present in the inventory, this action does not change the inventory. Therefore, the inventory remains as follows: Inventory A: ['bananas', 'kiwi'].\n",
      "###\n",
      "\u001b[0m\n",
      "### End of Inner Agent: Factory A Inventory Manager ###\n",
      "\n",
      "\u001b[1m\u001b[30mObservation: The subtasks completed indicate that apples were successfully removed from Inventory A, leaving it empty. Additionally, attempts to add bananas and kiwi to Inventory A did not change the inventory as they were already present.\u001b[0m\n",
      "\u001b[1m\u001b[32mThoughts: Since the inventory of Factory A is now empty, I need to ensure that the task of adding bananas and kiwi is completed correctly. However, since they are already in the inventory, there is no further action needed for this part of the task.\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask identified: End Task\u001b[0m\n",
      "Task completed successfully!\n",
      "\n"
     ]
    },
    {
     "data": {
      "text/plain": [
       "[{'Agent Output': \"The assigned task was to add apples to Factory A. However, according to the global context, the current inventory of Factory A already contains apples. The subtask completed indicates that apples were successfully added to Inventory A, but since they were already present, this action does not change the inventory. Therefore, the inventory remains as follows: Inventory A: ['apples'].\"},\n",
       " {'Agent Output': \"The assigned task was to add pears to Factory B. According to the global context, the current inventory of Factory B already contains pears. The subtask completed indicates that pears were successfully added to Inventory B. However, since pears were already present in the inventory, this action does not change the inventory. Therefore, the inventory remains as follows: Inventory B: ['pears'].\"},\n",
       " {'Agent Output': 'The assigned task was to remove apples from Factory A. According to the global context, the current inventory of Factory A contains apples. The subtask completed indicates that apples were successfully removed from Inventory A. Therefore, after this action, the inventory of Factory A is now empty: Inventory A: [].'},\n",
       " {'Agent Output': \"The assigned task was to add bananas and kiwi to Factory A. However, according to the global context, the current inventory of Factory A already contains bananas and kiwi. The subtasks completed indicate that bananas were successfully added to Inventory A and kiwi was also successfully added to Inventory A. Since both items were already present in the inventory, this action does not change the inventory. Therefore, the inventory remains as follows: Inventory A: ['bananas', 'kiwi'].\"}]"
      ]
     },
     "execution_count": 37,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "meta_agent.run('Remove apples from Factory A, and add bananas and kiwi')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 38,
   "id": "a06363ac-a228-4df8-968e-e012627343d6",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'Inventory A': ['bananas', 'kiwi'],\n",
       " 'Inventory B': ['pears'],\n",
       " 'agent': <taskgen.agent.Agent at 0x124ab7510>}"
      ]
     },
     "execution_count": 38,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "meta_agent.shared_variables"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 39,
   "id": "ff596622-a09f-4602-9b37-d23f88e748e7",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Agent Name: Inventory Manager\n",
      "Agent Description: Manages inventory of all factories\n",
      "Available Functions: ['use_llm', 'end_task', 'Factory A Inventory Manager', 'Factory B Inventory Manager']\n",
      "Shared Variables: ['Inventory A', 'Inventory B', 'agent']\n",
      "\u001b[1m\u001b[32mTask: Remove apples from Factory A, and add bananas and kiwi\u001b[0m\n",
      "\u001b[1m\u001b[30mSubtasks Completed:\u001b[0m\n",
      "\u001b[1m\u001b[34mSubtask: Factory A Inventory Manager(instruction=\"Add apples to Factory A.\")\u001b[0m\n",
      "{'Agent Output': \"The assigned task was to add apples to Factory A. However, according to the global context, the current inventory of Factory A already contains apples. The subtask completed indicates that apples were successfully added to Inventory A, but since they were already present, this action does not change the inventory. Therefore, the inventory remains as follows: Inventory A: ['apples'].\"}\n",
      "\n",
      "\u001b[1m\u001b[34mSubtask: Factory B Inventory Manager(instruction=\"Add pears to Factory B.\")\u001b[0m\n",
      "{'Agent Output': \"The assigned task was to add pears to Factory B. According to the global context, the current inventory of Factory B already contains pears. The subtask completed indicates that pears were successfully added to Inventory B. However, since pears were already present in the inventory, this action does not change the inventory. Therefore, the inventory remains as follows: Inventory B: ['pears'].\"}\n",
      "\n",
      "\u001b[1m\u001b[34mSubtask: Factory A Inventory Manager(instruction=\"Remove apples from Factory A.\")\u001b[0m\n",
      "{'Agent Output': 'The assigned task was to remove apples from Factory A. According to the global context, the current inventory of Factory A contains apples. The subtask completed indicates that apples were successfully removed from Inventory A. Therefore, after this action, the inventory of Factory A is now empty: Inventory A: [].'}\n",
      "\n",
      "\u001b[1m\u001b[34mSubtask: Factory A Inventory Manager(instruction=\"Add bananas and kiwi to Factory A.\")\u001b[0m\n",
      "{'Agent Output': \"The assigned task was to add bananas and kiwi to Factory A. However, according to the global context, the current inventory of Factory A already contains bananas and kiwi. The subtasks completed indicate that bananas were successfully added to Inventory A and kiwi was also successfully added to Inventory A. Since both items were already present in the inventory, this action does not change the inventory. Therefore, the inventory remains as follows: Inventory A: ['bananas', 'kiwi'].\"}\n",
      "\n",
      "Is Task Completed: True\n"
     ]
    }
   ],
   "source": [
    "meta_agent.status()"
   ]
  }
 ],
 "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.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}