autogen/python/docs/src/user-guide/agentchat-user-guide/selector-group-chat.ipynb

{
    "cells": [
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "# Selector Group Chat"
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "{py:class}`~autogen_agentchat.teams.SelectorGroupChat` implements a team where participants take turns broadcasting messages to all other members. A generative model (e.g., an LLM) selects the next speaker based on the shared context, enabling dynamic, context-aware collaboration.\n",
                "\n",
                "Key features include:\n",
                "\n",
                "- Model-based speaker selection\n",
                "- Configurable participant roles and descriptions\n",
                "- Prevention of consecutive turns by the same speaker (optional)\n",
                "- Customizable selection prompting\n",
                "- Customizable selection function to override the default model-based selection\n",
                "- Customizable candidate function to narrow-down the set of agents for selection using model\n",
                "\n",
                "```{note}\n",
                "{py:class}`~autogen_agentchat.teams.SelectorGroupChat` is a high-level API. For more control and customization, refer to the [Group Chat Pattern](../core-user-guide/design-patterns/group-chat.ipynb) in the Core API documentation to implement your own group chat logic.\n",
                "```\n",
                "\n",
                "## How Does it Work?\n",
                "\n",
                "{py:class}`~autogen_agentchat.teams.SelectorGroupChat` is a group chat similar to {py:class}`~autogen_agentchat.teams.RoundRobinGroupChat`,\n",
                "but with a model-based next speaker selection mechanism.\n",
                "When the team receives a task through {py:meth}`~autogen_agentchat.teams.BaseGroupChat.run` or {py:meth}`~autogen_agentchat.teams.BaseGroupChat.run_stream`,\n",
                "the following steps are executed:\n",
                "\n",
                "1. The team analyzes the current conversation context, including the conversation history and participants' {py:attr}`~autogen_agentchat.base.ChatAgent.name` and {py:attr}`~autogen_agentchat.base.ChatAgent.description` attributes, to determine the next speaker using a model. By default, the team will not select the same speak consecutively unless it is the only agent available. This can be changed by setting `allow_repeated_speaker=True`. You can also override the model by providing a custom selection function.\n",
                "2. The team prompts the selected speaker agent to provide a response, which is then **broadcasted** to all other participants.\n",
                "3. The termination condition is checked to determine if the conversation should end, if not, the process repeats from step 1.\n",
                "4. When the conversation ends, the team returns the {py:class}`~autogen_agentchat.base.TaskResult` containing the conversation history from this task.\n",
                "\n",
                "Once the team finishes the task, the conversation context is kept within the team and all participants, so the next task can continue from the previous conversation context.\n",
                "You can reset the conversation context by calling {py:meth}`~autogen_agentchat.teams.BaseGroupChat.reset`.\n",
                "\n",
                "In this section, we will demonstrate how to use {py:class}`~autogen_agentchat.teams.SelectorGroupChat` with a simple example for a web search and data analysis task."
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "## Example: Web Search/Analysis"
            ]
        },
        {
            "cell_type": "code",
            "execution_count": 1,
            "metadata": {},
            "outputs": [],
            "source": [
                "from typing import List, Sequence\n",
                "\n",
                "from autogen_agentchat.agents import AssistantAgent, UserProxyAgent\n",
                "from autogen_agentchat.conditions import MaxMessageTermination, TextMentionTermination\n",
                "from autogen_agentchat.messages import BaseAgentEvent, BaseChatMessage\n",
                "from autogen_agentchat.teams import SelectorGroupChat\n",
                "from autogen_agentchat.ui import Console\n",
                "from autogen_ext.models.openai import OpenAIChatCompletionClient"
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "### Agents\n",
                "\n",
                "![Selector Group Chat](selector-group-chat.svg)\n",
                "\n",
                "This system uses three specialized agents:\n",
                "\n",
                "- **Planning Agent**: The strategic coordinator that breaks down complex tasks into manageable subtasks. \n",
                "- **Web Search Agent**: An information retrieval specialist that interfaces with the `search_web_tool`.\n",
                "- **Data Analyst Agent**: An agent specialist in performing calculations equipped with `percentage_change_tool`. "
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "The tools `search_web_tool` and `percentage_change_tool` are external tools that the agents can use to perform their tasks."
            ]
        },
        {
            "cell_type": "code",
            "execution_count": 2,
            "metadata": {},
            "outputs": [],
            "source": [
                "# Note: This example uses mock tools instead of real APIs for demonstration purposes\n",
                "def search_web_tool(query: str) -> str:\n",
                "    if \"2006-2007\" in query:\n",
                "        return \"\"\"Here are the total points scored by Miami Heat players in the 2006-2007 season:\n",
                "        Udonis Haslem: 844 points\n",
                "        Dwayne Wade: 1397 points\n",
                "        James Posey: 550 points\n",
                "        ...\n",
                "        \"\"\"\n",
                "    elif \"2007-2008\" in query:\n",
                "        return \"The number of total rebounds for Dwayne Wade in the Miami Heat season 2007-2008 is 214.\"\n",
                "    elif \"2008-2009\" in query:\n",
                "        return \"The number of total rebounds for Dwayne Wade in the Miami Heat season 2008-2009 is 398.\"\n",
                "    return \"No data found.\"\n",
                "\n",
                "\n",
                "def percentage_change_tool(start: float, end: float) -> float:\n",
                "    return ((end - start) / start) * 100"
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "Let's create the specialized agents using the {py:class}`~autogen_agentchat.agents.AssistantAgent` class.\n",
                "It is important to note that the agents' {py:attr}`~autogen_agentchat.base.ChatAgent.name` and {py:attr}`~autogen_agentchat.base.ChatAgent.description` attributes are used by the model to determine the next speaker,\n",
                "so it is recommended to provide meaningful names and descriptions."
            ]
        },
        {
            "cell_type": "code",
            "execution_count": 3,
            "metadata": {},
            "outputs": [],
            "source": [
                "model_client = OpenAIChatCompletionClient(model=\"gpt-4o\")\n",
                "\n",
                "planning_agent = AssistantAgent(\n",
                "    \"PlanningAgent\",\n",
                "    description=\"An agent for planning tasks, this agent should be the first to engage when given a new task.\",\n",
                "    model_client=model_client,\n",
                "    system_message=\"\"\"\n",
                "    You are a planning agent.\n",
                "    Your job is to break down complex tasks into smaller, manageable subtasks.\n",
                "    Your team members are:\n",
                "        WebSearchAgent: Searches for information\n",
                "        DataAnalystAgent: Performs calculations\n",
                "\n",
                "    You only plan and delegate tasks - you do not execute them yourself.\n",
                "\n",
                "    When assigning tasks, use this format:\n",
                "    1. <agent> : <task>\n",
                "\n",
                "    After all tasks are complete, summarize the findings and end with \"TERMINATE\".\n",
                "    \"\"\",\n",
                ")\n",
                "\n",
                "web_search_agent = AssistantAgent(\n",
                "    \"WebSearchAgent\",\n",
                "    description=\"An agent for searching information on the web.\",\n",
                "    tools=[search_web_tool],\n",
                "    model_client=model_client,\n",
                "    system_message=\"\"\"\n",
                "    You are a web search agent.\n",
                "    Your only tool is search_tool - use it to find information.\n",
                "    You make only one search call at a time.\n",
                "    Once you have the results, you never do calculations based on them.\n",
                "    \"\"\",\n",
                ")\n",
                "\n",
                "data_analyst_agent = AssistantAgent(\n",
                "    \"DataAnalystAgent\",\n",
                "    description=\"An agent for performing calculations.\",\n",
                "    model_client=model_client,\n",
                "    tools=[percentage_change_tool],\n",
                "    system_message=\"\"\"\n",
                "    You are a data analyst.\n",
                "    Given the tasks you have been assigned, you should analyze the data and provide results using the tools provided.\n",
                "    If you have not seen the data, ask for it.\n",
                "    \"\"\",\n",
                ")"
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "```{note}\n",
                "By default, {py:class}`~autogen_agentchat.agents.AssistantAgent` returns the\n",
                "tool output as the response. If your tool does not return a well-formed\n",
                "string in natural language format, you may want to add a reflection step\n",
                "within the agent by setting `reflect_on_tool_use=True` when creating the agent.\n",
                "This will allow the agent to reflect on the tool output and provide a natural\n",
                "language response.\n",
                "```"
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "### Workflow\n",
                "\n",
                "1. The task is received by the {py:class}`~autogen_agentchat.teams.SelectorGroupChat` which, based on agent descriptions, selects the most appropriate agent to handle the initial task (typically the Planning Agent).\n",
                "\n",
                "2. The **Planning Agent** analyzes the task and breaks it down into subtasks, assigning each to the most appropriate agent using the format:\n",
                "   `<agent> : <task>`\n",
                "\n",
                "3. Based on the conversation context and agent descriptions, the {py:class}`~autogen_agent.teams.SelectorGroupChat` manager dynamically selects the next agent to handle their assigned subtask.\n",
                "\n",
                "4. The **Web Search Agent** performs searches one at a time, storing results in the shared conversation history.\n",
                "\n",
                "5. The **Data Analyst** processes the gathered information using available calculation tools when selected.\n",
                "\n",
                "6. The workflow continues with agents being dynamically selected until either:\n",
                "   - The Planning Agent determines all subtasks are complete and sends \"TERMINATE\"\n",
                "   - An alternative termination condition is met (e.g., a maximum number of messages)\n",
                "\n",
                "When defining your agents, make sure to include a helpful {py:attr}`~autogen_agentchat.base.ChatAgent.description` since this is used to decide which agent to select next."
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "### Termination Conditions\n",
                "\n",
                "Let's use two termination conditions:\n",
                "{py:class}`~autogen_agentchat.conditions.TextMentionTermination` to end the conversation when the Planning Agent sends \"TERMINATE\",\n",
                "and {py:class}`~autogen_agentchat.conditions.MaxMessageTermination` to limit the conversation to 25 messages to avoid infinite loop."
            ]
        },
        {
            "cell_type": "code",
            "execution_count": 4,
            "metadata": {},
            "outputs": [],
            "source": [
                "text_mention_termination = TextMentionTermination(\"TERMINATE\")\n",
                "max_messages_termination = MaxMessageTermination(max_messages=25)\n",
                "termination = text_mention_termination | max_messages_termination"
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "### Selector Prompt\n",
                "\n",
                "{py:class}`~autogen_agentchat.teams.SelectorGroupChat` uses a model to select\n",
                "the next speaker based on the conversation context.\n",
                "We will use a custom selector prompt to properly align with the workflow."
            ]
        },
        {
            "cell_type": "code",
            "execution_count": 5,
            "metadata": {},
            "outputs": [],
            "source": [
                "selector_prompt = \"\"\"Select an agent to perform task.\n",
                "\n",
                "{roles}\n",
                "\n",
                "Current conversation context:\n",
                "{history}\n",
                "\n",
                "Read the above conversation, then select an agent from {participants} to perform the next task.\n",
                "Make sure the planner agent has assigned tasks before other agents start working.\n",
                "Only select one agent.\n",
                "\"\"\""
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "The string variables available in the selector prompt are:\n",
                "- `{participants}`: The names of candidates for selection. The format is `[\"<name1>\", \"<name2>\", ...]`.\n",
                "- `{roles}`: A newline-separated list of names and descriptions of the candidate agents. The format for each line is: `\"<name> : <description>\"`.\n",
                "- `{history}`: The conversation history formatted as a double newline separated of names and message content. The format for each message is: `\"<name> : <message content>\"`.\n",
                "\n",
                "```{tip}\n",
                "Try not to overload the model with too much instruction in the selector prompt.\n",
                "\n",
                "What is too much? It depends on the capabilities of the model you are using.\n",
                "For GPT-4o and equivalents, you can use a selector prompt with a condition for when each speaker should be selected.\n",
                "For smaller models such as Phi-4, you should keep the selector prompt as simple as possible\n",
                "such as the one used in this example.\n",
                "\n",
                "Generally, if you find yourself writing multiple conditions for each agent,\n",
                "it is a sign that you should consider using a custom selection function,\n",
                "or breaking down the task into smaller, sequential tasks to be handled by\n",
                "separate agents or teams.\n",
                "```"
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "### Running the Team\n",
                "\n",
                "Let's create the team with the agents, termination conditions, and custom selector prompt."
            ]
        },
        {
            "cell_type": "code",
            "execution_count": 6,
            "metadata": {},
            "outputs": [],
            "source": [
                "team = SelectorGroupChat(\n",
                "    [planning_agent, web_search_agent, data_analyst_agent],\n",
                "    model_client=model_client,\n",
                "    termination_condition=termination,\n",
                "    selector_prompt=selector_prompt,\n",
                "    allow_repeated_speaker=True,  # Allow an agent to speak multiple turns in a row.\n",
                ")"
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "Now we run the team with a task to find information about an NBA player."
            ]
        },
        {
            "cell_type": "code",
            "execution_count": 7,
            "metadata": {},
            "outputs": [],
            "source": [
                "task = \"Who was the Miami Heat player with the highest points in the 2006-2007 season, and what was the percentage change in his total rebounds between the 2007-2008 and 2008-2009 seasons?\""
            ]
        },
        {
            "cell_type": "code",
            "execution_count": 8,
            "metadata": {},
            "outputs": [
                {
                    "name": "stdout",
                    "output_type": "stream",
                    "text": [
                        "---------- user ----------\n",
                        "Who was the Miami Heat player with the highest points in the 2006-2007 season, and what was the percentage change in his total rebounds between the 2007-2008 and 2008-2009 seasons?\n",
                        "---------- PlanningAgent ----------\n",
                        "To complete this task, we need to perform the following subtasks:\n",
                        "\n",
                        "1. Find out which Miami Heat player had the highest points in the 2006-2007 season.\n",
                        "2. Gather data on this player's total rebounds for the 2007-2008 season.\n",
                        "3. Gather data on this player's total rebounds for the 2008-2009 season.\n",
                        "4. Calculate the percentage change in the player's total rebounds between the 2007-2008 and 2008-2009 seasons.\n",
                        "\n",
                        "I'll assign these tasks accordingly:\n",
                        "\n",
                        "1. WebSearchAgent: Search for the Miami Heat player with the highest points in the 2006-2007 NBA season.\n",
                        "2. WebSearchAgent: Find the total rebounds for this player in the 2007-2008 NBA season.\n",
                        "3. WebSearchAgent: Find the total rebounds for this player in the 2008-2009 NBA season.\n",
                        "4. DataAnalystAgent: Calculate the percentage change in total rebounds from the 2007-2008 season to the 2008-2009 season for this player.\n",
                        "---------- WebSearchAgent ----------\n",
                        "[FunctionCall(id='call_89tUNHaAM0kKQYPJLleGUKK7', arguments='{\"query\":\"Miami Heat player highest points 2006-2007 season\"}', name='search_web_tool')]\n",
                        "---------- WebSearchAgent ----------\n",
                        "[FunctionExecutionResult(content='Here are the total points scored by Miami Heat players in the 2006-2007 season:\\n        Udonis Haslem: 844 points\\n        Dwayne Wade: 1397 points\\n        James Posey: 550 points\\n        ...\\n        ', name='search_web_tool', call_id='call_89tUNHaAM0kKQYPJLleGUKK7', is_error=False)]\n",
                        "---------- WebSearchAgent ----------\n",
                        "Here are the total points scored by Miami Heat players in the 2006-2007 season:\n",
                        "        Udonis Haslem: 844 points\n",
                        "        Dwayne Wade: 1397 points\n",
                        "        James Posey: 550 points\n",
                        "        ...\n",
                        "        \n",
                        "---------- WebSearchAgent ----------\n",
                        "The Miami Heat player with the highest points in the 2006-2007 season was Dwyane Wade, with 1,397 points.\n",
                        "\n",
                        "Next, I will search for Dwyane Wade's total rebounds for the 2007-2008 season.\n",
                        "---------- WebSearchAgent ----------\n",
                        "[FunctionCall(id='call_RC55TkSjG3JXRuVOTPrcE1RL', arguments='{\"query\":\"Dwyane Wade total rebounds 2007-2008 season\"}', name='search_web_tool')]\n",
                        "---------- WebSearchAgent ----------\n",
                        "[FunctionExecutionResult(content='The number of total rebounds for Dwayne Wade in the Miami Heat season 2007-2008 is 214.', name='search_web_tool', call_id='call_RC55TkSjG3JXRuVOTPrcE1RL', is_error=False)]\n",
                        "---------- WebSearchAgent ----------\n",
                        "The number of total rebounds for Dwayne Wade in the Miami Heat season 2007-2008 is 214.\n",
                        "---------- WebSearchAgent ----------\n",
                        "[FunctionCall(id='call_pBXoABrErDow0rZjw3tjOZol', arguments='{\"query\":\"Dwyane Wade total rebounds 2008-2009 season\"}', name='search_web_tool')]\n",
                        "---------- WebSearchAgent ----------\n",
                        "[FunctionExecutionResult(content='The number of total rebounds for Dwayne Wade in the Miami Heat season 2008-2009 is 398.', name='search_web_tool', call_id='call_pBXoABrErDow0rZjw3tjOZol', is_error=False)]\n",
                        "---------- WebSearchAgent ----------\n",
                        "The number of total rebounds for Dwayne Wade in the Miami Heat season 2008-2009 is 398.\n",
                        "---------- DataAnalystAgent ----------\n",
                        "[FunctionCall(id='call_qMxxXtcJsiK8KFSSCx3zm0is', arguments='{\"start\":214,\"end\":398}', name='percentage_change_tool')]\n",
                        "---------- DataAnalystAgent ----------\n",
                        "[FunctionExecutionResult(content='85.98130841121495', name='percentage_change_tool', call_id='call_qMxxXtcJsiK8KFSSCx3zm0is', is_error=False)]\n",
                        "---------- DataAnalystAgent ----------\n",
                        "85.98130841121495\n",
                        "---------- PlanningAgent ----------\n",
                        "The player with the highest points for the Miami Heat in the 2006-2007 NBA season was Dwyane Wade, who scored 1,397 points. The percentage change in Dwyane Wade's total rebounds from 214 in the 2007-2008 season to 398 in the 2008-2009 season is approximately 85.98%.\n",
                        "\n",
                        "TERMINATE\n"
                    ]
                },
                {
                    "data": {
                        "text/plain": [
                            "TaskResult(messages=[TextMessage(source='user', models_usage=None, metadata={}, content='Who was the Miami Heat player with the highest points in the 2006-2007 season, and what was the percentage change in his total rebounds between the 2007-2008 and 2008-2009 seasons?', type='TextMessage'), TextMessage(source='PlanningAgent', models_usage=RequestUsage(prompt_tokens=161, completion_tokens=220), metadata={}, content=\"To complete this task, we need to perform the following subtasks:\\n\\n1. Find out which Miami Heat player had the highest points in the 2006-2007 season.\\n2. Gather data on this player's total rebounds for the 2007-2008 season.\\n3. Gather data on this player's total rebounds for the 2008-2009 season.\\n4. Calculate the percentage change in the player's total rebounds between the 2007-2008 and 2008-2009 seasons.\\n\\nI'll assign these tasks accordingly:\\n\\n1. WebSearchAgent: Search for the Miami Heat player with the highest points in the 2006-2007 NBA season.\\n2. WebSearchAgent: Find the total rebounds for this player in the 2007-2008 NBA season.\\n3. WebSearchAgent: Find the total rebounds for this player in the 2008-2009 NBA season.\\n4. DataAnalystAgent: Calculate the percentage change in total rebounds from the 2007-2008 season to the 2008-2009 season for this player.\", type='TextMessage'), ToolCallRequestEvent(source='WebSearchAgent', models_usage=RequestUsage(prompt_tokens=368, completion_tokens=27), metadata={}, content=[FunctionCall(id='call_89tUNHaAM0kKQYPJLleGUKK7', arguments='{\"query\":\"Miami Heat player highest points 2006-2007 season\"}', name='search_web_tool')], type='ToolCallRequestEvent'), ToolCallExecutionEvent(source='WebSearchAgent', models_usage=None, metadata={}, content=[FunctionExecutionResult(content='Here are the total points scored by Miami Heat players in the 2006-2007 season:\\n        Udonis Haslem: 844 points\\n        Dwayne Wade: 1397 points\\n        James Posey: 550 points\\n        ...\\n        ', name='search_web_tool', call_id='call_89tUNHaAM0kKQYPJLleGUKK7', is_error=False)], type='ToolCallExecutionEvent'), ToolCallSummaryMessage(source='WebSearchAgent', models_usage=None, metadata={}, content='Here are the total points scored by Miami Heat players in the 2006-2007 season:\\n        Udonis Haslem: 844 points\\n        Dwayne Wade: 1397 points\\n        James Posey: 550 points\\n        ...\\n        ', type='ToolCallSummaryMessage'), ThoughtEvent(source='WebSearchAgent', models_usage=None, metadata={}, content=\"The Miami Heat player with the highest points in the 2006-2007 season was Dwyane Wade, with 1,397 points.\\n\\nNext, I will search for Dwyane Wade's total rebounds for the 2007-2008 season.\", type='ThoughtEvent'), ToolCallRequestEvent(source='WebSearchAgent', models_usage=RequestUsage(prompt_tokens=460, completion_tokens=83), metadata={}, content=[FunctionCall(id='call_RC55TkSjG3JXRuVOTPrcE1RL', arguments='{\"query\":\"Dwyane Wade total rebounds 2007-2008 season\"}', name='search_web_tool')], type='ToolCallRequestEvent'), ToolCallExecutionEvent(source='WebSearchAgent', models_usage=None, metadata={}, content=[FunctionExecutionResult(content='The number of total rebounds for Dwayne Wade in the Miami Heat season 2007-2008 is 214.', name='search_web_tool', call_id='call_RC55TkSjG3JXRuVOTPrcE1RL', is_error=False)], type='ToolCallExecutionEvent'), ToolCallSummaryMessage(source='WebSearchAgent', models_usage=None, metadata={}, content='The number of total rebounds for Dwayne Wade in the Miami Heat season 2007-2008 is 214.', type='ToolCallSummaryMessage'), ToolCallRequestEvent(source='WebSearchAgent', models_usage=RequestUsage(prompt_tokens=585, completion_tokens=28), metadata={}, content=[FunctionCall(id='call_pBXoABrErDow0rZjw3tjOZol', arguments='{\"query\":\"Dwyane Wade total rebounds 2008-2009 season\"}', name='search_web_tool')], type='ToolCallRequestEvent'), ToolCallExecutionEvent(source='WebSearchAgent', models_usage=None, metadata={}, content=[FunctionExecutionResult(content='The number of total rebounds for Dwayne Wade i
                        ]
                    },
                    "execution_count": 8,
                    "metadata": {},
                    "output_type": "execute_result"
                }
            ],
            "source": [
                "# Use asyncio.run(...) if you are running this in a script.\n",
                "await Console(team.run_stream(task=task))"
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "As we can see, after the Web Search Agent conducts the necessary searches and the Data Analyst Agent completes the necessary calculations, we find that Dwayne Wade was the Miami Heat player with the highest points in the 2006-2007 season, and the percentage change in his total rebounds between the 2007-2008 and 2008-2009 seasons is 85.98%!"
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "## Custom Selector Function"
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "Often times we want better control over the selection process.\n",
                "To this end, we can set the `selector_func` argument with a custom selector function to override the default model-based selection.\n",
                "This allows us to implement more complex selection logic and state-based transitions.\n",
                "\n",
                "For instance, we want the Planning Agent to speak immediately after any specialized agent to check the progress.\n",
                "\n",
                "```{note}\n",
                "Returning `None` from the custom selector function will use the default model-based selection.\n",
                "``` \n",
                "\n",
                "```{note}\n",
                "Custom selector functions are not [serialized](https://microsoft.github.io/autogen/stable/user-guide/agentchat-user-guide/serialize-components.html) when `.dump_component()` is called on the SelectorGroupChat team . If you need to serialize team configurations with custom selector functions, consider implementing custom workflows and serialization logic.\n",
                "```"
            ]
        },
        {
            "cell_type": "code",
            "execution_count": 10,
            "metadata": {},
            "outputs": [
                {
                    "name": "stdout",
                    "output_type": "stream",
                    "text": [
                        "---------- user ----------\n",
                        "Who was the Miami Heat player with the highest points in the 2006-2007 season, and what was the percentage change in his total rebounds between the 2007-2008 and 2008-2009 seasons?\n",
                        "---------- PlanningAgent ----------\n",
                        "To answer this question, we need to follow these steps: \n",
                        "\n",
                        "1. Identify the Miami Heat player with the highest points in the 2006-2007 season.\n",
                        "2. Retrieve the total rebounds of that player for the 2007-2008 and 2008-2009 seasons.\n",
                        "3. Calculate the percentage change in his total rebounds between the two seasons.\n",
                        "\n",
                        "Let's delegate these tasks:\n",
                        "\n",
                        "1. WebSearchAgent: Find the Miami Heat player with the highest points in the 2006-2007 NBA season.\n",
                        "2. WebSearchAgent: Retrieve the total rebounds for the identified player during the 2007-2008 NBA season.\n",
                        "3. WebSearchAgent: Retrieve the total rebounds for the identified player during the 2008-2009 NBA season.\n",
                        "4. DataAnalystAgent: Calculate the percentage change in total rebounds between the 2007-2008 and 2008-2009 seasons for the player found.\n",
                        "---------- WebSearchAgent ----------\n",
                        "[FunctionCall(id='call_Pz82ndNLSV4cH0Sg6g7ArP4L', arguments='{\"query\":\"Miami Heat player highest points 2006-2007 season\"}', name='search_web_tool')]\n",
                        "---------- WebSearchAgent ----------\n",
                        "[FunctionExecutionResult(content='Here are the total points scored by Miami Heat players in the 2006-2007 season:\\n        Udonis Haslem: 844 points\\n        Dwayne Wade: 1397 points\\n        James Posey: 550 points\\n        ...\\n        ', call_id='call_Pz82ndNLSV4cH0Sg6g7ArP4L')]\n",
                        "---------- WebSearchAgent ----------\n",
                        "Here are the total points scored by Miami Heat players in the 2006-2007 season:\n",
                        "        Udonis Haslem: 844 points\n",
                        "        Dwayne Wade: 1397 points\n",
                        "        James Posey: 550 points\n",
                        "        ...\n",
                        "        \n",
                        "---------- PlanningAgent ----------\n",
                        "Great! Dwyane Wade was the Miami Heat player with the highest points in the 2006-2007 season. Now, let's continue with the next tasks:\n",
                        "\n",
                        "2. WebSearchAgent: Retrieve the total rebounds for Dwyane Wade during the 2007-2008 NBA season.\n",
                        "3. WebSearchAgent: Retrieve the total rebounds for Dwyane Wade during the 2008-2009 NBA season.\n",
                        "---------- WebSearchAgent ----------\n",
                        "[FunctionCall(id='call_3qv9so2DXFZIHtzqDIfXoFID', arguments='{\"query\": \"Dwyane Wade total rebounds 2007-2008 season\"}', name='search_web_tool'), FunctionCall(id='call_Vh7zzzWUeiUAvaYjP0If0k1k', arguments='{\"query\": \"Dwyane Wade total rebounds 2008-2009 season\"}', name='search_web_tool')]\n",
                        "---------- WebSearchAgent ----------\n",
                        "[FunctionExecutionResult(content='The number of total rebounds for Dwayne Wade in the Miami Heat season 2007-2008 is 214.', call_id='call_3qv9so2DXFZIHtzqDIfXoFID'), FunctionExecutionResult(content='The number of total rebounds for Dwayne Wade in the Miami Heat season 2008-2009 is 398.', call_id='call_Vh7zzzWUeiUAvaYjP0If0k1k')]\n",
                        "---------- WebSearchAgent ----------\n",
                        "The number of total rebounds for Dwayne Wade in the Miami Heat season 2007-2008 is 214.\n",
                        "The number of total rebounds for Dwayne Wade in the Miami Heat season 2008-2009 is 398.\n",
                        "---------- PlanningAgent ----------\n",
                        "Now let's calculate the percentage change in total rebounds between the 2007-2008 and 2008-2009 seasons for Dwyane Wade.\n",
                        "\n",
                        "4. DataAnalystAgent: Calculate the percentage change in total rebounds for Dwyane Wade between the 2007-2008 and 2008-2009 seasons.\n",
                        "---------- DataAnalystAgent ----------\n",
                        "[FunctionCall(id='call_FXnPSr6JVGfAWs3StIizbt2V', arguments='{\"start\":214,\"end\":398}', name='percentage_change_tool')]\n",
                        "---------- DataAnalystAgent ----------\n",
                        "[FunctionExecutionResult(content='85.98130841121495', call_id='call_FXnPSr6JVGfAWs3StIizbt2V')]\n",
                        "---------- DataAnalystAgent ----------\n",
                        "85.98130841121495\n",
                        "---------- PlanningAgent ----------\n",
                        "Dwyane Wade was the Miami Heat player with the highest points in the 2006-2007 season, scoring a total of 1397 points. The percentage change in his total rebounds from the 2007-2008 season (214 rebounds) to the 2008-2009 season (398 rebounds) is approximately 86.0%.\n",
                        "\n",
                        "TERMINATE\n"
                    ]
                },
                {
                    "data": {
                        "text/plain": [
                            "TaskResult(messages=[TextMessage(source='user', models_usage=None, content='Who was the Miami Heat player with the highest points in the 2006-2007 season, and what was the percentage change in his total rebounds between the 2007-2008 and 2008-2009 seasons?', type='TextMessage'), TextMessage(source='PlanningAgent', models_usage=RequestUsage(prompt_tokens=161, completion_tokens=192), content=\"To answer this question, we need to follow these steps: \\n\\n1. Identify the Miami Heat player with the highest points in the 2006-2007 season.\\n2. Retrieve the total rebounds of that player for the 2007-2008 and 2008-2009 seasons.\\n3. Calculate the percentage change in his total rebounds between the two seasons.\\n\\nLet's delegate these tasks:\\n\\n1. WebSearchAgent: Find the Miami Heat player with the highest points in the 2006-2007 NBA season.\\n2. WebSearchAgent: Retrieve the total rebounds for the identified player during the 2007-2008 NBA season.\\n3. WebSearchAgent: Retrieve the total rebounds for the identified player during the 2008-2009 NBA season.\\n4. DataAnalystAgent: Calculate the percentage change in total rebounds between the 2007-2008 and 2008-2009 seasons for the player found.\", type='TextMessage'), ToolCallRequestEvent(source='WebSearchAgent', models_usage=RequestUsage(prompt_tokens=340, completion_tokens=27), content=[FunctionCall(id='call_Pz82ndNLSV4cH0Sg6g7ArP4L', arguments='{\"query\":\"Miami Heat player highest points 2006-2007 season\"}', name='search_web_tool')], type='ToolCallRequestEvent'), ToolCallExecutionEvent(source='WebSearchAgent', models_usage=None, content=[FunctionExecutionResult(content='Here are the total points scored by Miami Heat players in the 2006-2007 season:\\n        Udonis Haslem: 844 points\\n        Dwayne Wade: 1397 points\\n        James Posey: 550 points\\n        ...\\n        ', call_id='call_Pz82ndNLSV4cH0Sg6g7ArP4L')], type='ToolCallExecutionEvent'), ToolCallSummaryMessage(source='WebSearchAgent', models_usage=None, content='Here are the total points scored by Miami Heat players in the 2006-2007 season:\\n        Udonis Haslem: 844 points\\n        Dwayne Wade: 1397 points\\n        James Posey: 550 points\\n        ...\\n        ', type='ToolCallSummaryMessage'), TextMessage(source='PlanningAgent', models_usage=RequestUsage(prompt_tokens=420, completion_tokens=87), content=\"Great! Dwyane Wade was the Miami Heat player with the highest points in the 2006-2007 season. Now, let's continue with the next tasks:\\n\\n2. WebSearchAgent: Retrieve the total rebounds for Dwyane Wade during the 2007-2008 NBA season.\\n3. WebSearchAgent: Retrieve the total rebounds for Dwyane Wade during the 2008-2009 NBA season.\", type='TextMessage'), ToolCallRequestEvent(source='WebSearchAgent', models_usage=RequestUsage(prompt_tokens=525, completion_tokens=71), content=[FunctionCall(id='call_3qv9so2DXFZIHtzqDIfXoFID', arguments='{\"query\": \"Dwyane Wade total rebounds 2007-2008 season\"}', name='search_web_tool'), FunctionCall(id='call_Vh7zzzWUeiUAvaYjP0If0k1k', arguments='{\"query\": \"Dwyane Wade total rebounds 2008-2009 season\"}', name='search_web_tool')], type='ToolCallRequestEvent'), ToolCallExecutionEvent(source='WebSearchAgent', models_usage=None, content=[FunctionExecutionResult(content='The number of total rebounds for Dwayne Wade in the Miami Heat season 2007-2008 is 214.', call_id='call_3qv9so2DXFZIHtzqDIfXoFID'), FunctionExecutionResult(content='The number of total rebounds for Dwayne Wade in the Miami Heat season 2008-2009 is 398.', call_id='call_Vh7zzzWUeiUAvaYjP0If0k1k')], type='ToolCallExecutionEvent'), ToolCallSummaryMessage(source='WebSearchAgent', models_usage=None, content='The number of total rebounds for Dwayne Wade in the Miami Heat season 2007-2008 is 214.\\nThe number of total rebounds for Dwayne Wade in the Miami Heat season 2008-2009 is 398.', type='ToolCallSummaryMessage'), TextMessage(source='PlanningAgent', models_usage=RequestUsage(prompt_tokens=569, completion_tokens=68), content=\"Now let's calculate the percentage change in total 
                        ]
                    },
                    "execution_count": 10,
                    "metadata": {},
                    "output_type": "execute_result"
                }
            ],
            "source": [
                "def selector_func(messages: Sequence[BaseAgentEvent | BaseChatMessage]) -> str | None:\n",
                "    if messages[-1].source != planning_agent.name:\n",
                "        return planning_agent.name\n",
                "    return None\n",
                "\n",
                "\n",
                "# Reset the previous team and run the chat again with the selector function.\n",
                "await team.reset()\n",
                "team = SelectorGroupChat(\n",
                "    [planning_agent, web_search_agent, data_analyst_agent],\n",
                "    model_client=model_client,\n",
                "    termination_condition=termination,\n",
                "    selector_prompt=selector_prompt,\n",
                "    allow_repeated_speaker=True,\n",
                "    selector_func=selector_func,\n",
                ")\n",
                "\n",
                "await Console(team.run_stream(task=task))"
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "You can see from the conversation log that the Planning Agent always speaks immediately after the specialized agents.\n",
                "\n",
                "```{tip}\n",
                "Each participant agent only makes one step (executing tools, generating a response, etc.)\n",
                "on each turn. \n",
                "If you want an {py:class}`~autogen_agentchat.agents.AssistantAgent` to repeat\n",
                "until it stop returning a {py:class}`~autogen_agentchat.messages.ToolCallSummaryMessage`\n",
                "when it has finished running all the tools it needs to run, you can do so by\n",
                "checking the last message and returning the agent if it is a\n",
                "{py:class}`~autogen_agentchat.messages.ToolCallSummaryMessage`.\n",
                "```"
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "## Custom Candidate Function"
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "One more possible requirement might be to automatically select the next speaker from a filtered list of agents.\n",
                "For this, we can set `candidate_func` parameter with a custom candidate function to filter down the list of potential agents for speaker selection for each turn of groupchat.\n",
                "\n",
                "This allow us to restrict speaker selection to a specific set of agents after a given agent.\n",
                "\n",
                "\n",
                "```{note}\n",
                "The `candidate_func` is only valid if `selector_func` is not set.\n",
                "Returning `None` or an empty list `[]` from the custom candidate function will raise a `ValueError`.\n",
                "```"
            ]
        },
        {
            "cell_type": "code",
            "execution_count": null,
            "metadata": {},
            "outputs": [
                {
                    "name": "stdout",
                    "output_type": "stream",
                    "text": [
                        "---------- user ----------\n",
                        "Who was the Miami Heat player with the highest points in the 2006-2007 season, and what was the percentage change in his total rebounds between the 2007-2008 and 2008-2009 seasons?\n",
                        "---------- PlanningAgent ----------\n",
                        "To answer this question, we'll break it down into two main subtasks:\n",
                        "\n",
                        "1. Identify the Miami Heat player with the highest points in the 2006-2007 season.\n",
                        "2. Calculate the percentage change in that player's total rebounds between the 2007-2008 and 2008-2009 seasons.\n",
                        "\n",
                        "Let's assign these tasks:\n",
                        "\n",
                        "1. WebSearchAgent: Search for the Miami Heat player with the highest points in the 2006-2007 NBA season.\n",
                        "2. WebSearchAgent: Find the total rebound statistics for that identified player for both the 2007-2008 and 2008-2009 NBA seasons.\n",
                        "3. DataAnalystAgent: Calculate the percentage change in the player's total rebounds between the 2007-2008 and 2008-2009 seasons once the data is retrieved.\n",
                        "---------- WebSearchAgent ----------\n",
                        "[FunctionCall(id='call_WtR5KTfEIxs3jIO25gjAw7dF', arguments='{\"query\":\"Miami Heat highest points scorer 2006-2007 NBA season\"}', name='search_web_tool')]\n",
                        "---------- WebSearchAgent ----------\n",
                        "[FunctionExecutionResult(content='Here are the total points scored by Miami Heat players in the 2006-2007 season:\\n        Udonis Haslem: 844 points\\n        Dwayne Wade: 1397 points\\n        James Posey: 550 points\\n        ...\\n        ', name='search_web_tool', call_id='call_WtR5KTfEIxs3jIO25gjAw7dF', is_error=False)]\n",
                        "---------- WebSearchAgent ----------\n",
                        "Here are the total points scored by Miami Heat players in the 2006-2007 season:\n",
                        "        Udonis Haslem: 844 points\n",
                        "        Dwayne Wade: 1397 points\n",
                        "        James Posey: 550 points\n",
                        "        ...\n",
                        "        \n",
                        "---------- DataAnalystAgent ----------\n",
                        "[FunctionCall(id='call_9HA3DEacUl4WuG2G2PtRkXAO', arguments='{\"start\": 432, \"end\": 527}', name='percentage_change_tool')]\n",
                        "---------- DataAnalystAgent ----------\n",
                        "[FunctionExecutionResult(content='21.99074074074074', name='percentage_change_tool', call_id='call_9HA3DEacUl4WuG2G2PtRkXAO', is_error=False)]\n",
                        "---------- DataAnalystAgent ----------\n",
                        "21.99074074074074\n",
                        "---------- PlanningAgent ----------\n",
                        "It seems we've missed some context there, so let's assign the subtasks again for clarity:\n",
                        "\n",
                        "Based on the search results, Dwyane Wade had the highest points for the Miami Heat in the 2006-2007 season with 1397 points.\n",
                        "\n",
                        "Now, let's find the necessary rebound statistics:\n",
                        "\n",
                        "2. WebSearchAgent: Find Dwyane Wade's total rebound statistics for both the 2007-2008 and 2008-2009 NBA seasons.\n",
                        "3. DataAnalystAgent: Once the data is retrieved, calculate the percentage change in Dwyane Wade's total rebounds between the 2007-2008 and 2008-2009 seasons.\n",
                        "---------- WebSearchAgent ----------\n",
                        "[FunctionCall(id='call_3i1wTDSjkGg6Ev8YKYWkZK55', arguments='{\"query\": \"Dwyane Wade total rebounds 2007-2008 NBA season\"}', name='search_web_tool'), FunctionCall(id='call_NRAs6jHxXRi8zsvpW5WlHAaU', arguments='{\"query\": \"Dwyane Wade total rebounds 2008-2009 NBA season\"}', name='search_web_tool')]\n",
                        "---------- WebSearchAgent ----------\n",
                        "[FunctionExecutionResult(content='The number of total rebounds for Dwayne Wade in the Miami Heat season 2007-2008 is 214.', name='search_web_tool', call_id='call_3i1wTDSjkGg6Ev8YKYWkZK55', is_error=False), FunctionExecutionResult(content='The number of total rebounds for Dwayne Wade in the Miami Heat season 2008-2009 is 398.', name='search_web_tool', call_id='call_NRAs6jHxXRi8zsvpW5WlHAaU', is_error=False)]\n",
                        "---------- WebSearchAgent ----------\n",
                        "The number of total rebounds for Dwayne Wade in the Miami Heat season 2007-2008 is 214.\n",
                        "The number of total rebounds for Dwayne Wade in the Miami Heat season 2008-2009 is 398.\n",
                        "---------- PlanningAgent ----------\n",
                        "The total rebounds for Dwyane Wade in the 2007-2008 season were 214, and in the 2008-2009 season, they were 398.\n",
                        "\n",
                        "Now, let's calculate the percentage change.\n",
                        "\n",
                        "3. DataAnalystAgent: Calculate the percentage change in Dwyane Wade's total rebounds from the 2007-2008 season to the 2008-2009 season.\n",
                        "---------- DataAnalystAgent ----------\n",
                        "[FunctionCall(id='call_XECA7ezz7VIKbf8IbZYSCSpI', arguments='{\"start\":214,\"end\":398}', name='percentage_change_tool')]\n",
                        "---------- DataAnalystAgent ----------\n",
                        "[FunctionExecutionResult(content='85.98130841121495', name='percentage_change_tool', call_id='call_XECA7ezz7VIKbf8IbZYSCSpI', is_error=False)]\n",
                        "---------- DataAnalystAgent ----------\n",
                        "85.98130841121495\n",
                        "---------- PlanningAgent ----------\n",
                        "The Miami Heat player with the highest points in the 2006-2007 season was Dwyane Wade, with 1397 points. The percentage change in his total rebounds between the 2007-2008 and 2008-2009 seasons was approximately 85.98%.\n",
                        "\n",
                        "TERMINATE\n"
                    ]
                },
                {
                    "data": {
                        "text/plain": [
                            "TaskResult(messages=[TextMessage(source='user', models_usage=None, metadata={}, content='Who was the Miami Heat player with the highest points in the 2006-2007 season, and what was the percentage change in his total rebounds between the 2007-2008 and 2008-2009 seasons?', type='TextMessage'), TextMessage(source='PlanningAgent', models_usage=RequestUsage(prompt_tokens=161, completion_tokens=169), metadata={}, content=\"To answer this question, we'll break it down into two main subtasks:\\n\\n1. Identify the Miami Heat player with the highest points in the 2006-2007 season.\\n2. Calculate the percentage change in that player's total rebounds between the 2007-2008 and 2008-2009 seasons.\\n\\nLet's assign these tasks:\\n\\n1. WebSearchAgent: Search for the Miami Heat player with the highest points in the 2006-2007 NBA season.\\n2. WebSearchAgent: Find the total rebound statistics for that identified player for both the 2007-2008 and 2008-2009 NBA seasons.\\n3. DataAnalystAgent: Calculate the percentage change in the player's total rebounds between the 2007-2008 and 2008-2009 seasons once the data is retrieved.\", type='TextMessage'), ToolCallRequestEvent(source='WebSearchAgent', models_usage=RequestUsage(prompt_tokens=324, completion_tokens=28), metadata={}, content=[FunctionCall(id='call_WtR5KTfEIxs3jIO25gjAw7dF', arguments='{\"query\":\"Miami Heat highest points scorer 2006-2007 NBA season\"}', name='search_web_tool')], type='ToolCallRequestEvent'), ToolCallExecutionEvent(source='WebSearchAgent', models_usage=None, metadata={}, content=[FunctionExecutionResult(content='Here are the total points scored by Miami Heat players in the 2006-2007 season:\\n        Udonis Haslem: 844 points\\n        Dwayne Wade: 1397 points\\n        James Posey: 550 points\\n        ...\\n        ', name='search_web_tool', call_id='call_WtR5KTfEIxs3jIO25gjAw7dF', is_error=False)], type='ToolCallExecutionEvent'), ToolCallSummaryMessage(source='WebSearchAgent', models_usage=None, metadata={}, content='Here are the total points scored by Miami Heat players in the 2006-2007 season:\\n        Udonis Haslem: 844 points\\n        Dwayne Wade: 1397 points\\n        James Posey: 550 points\\n        ...\\n        ', type='ToolCallSummaryMessage'), ToolCallRequestEvent(source='DataAnalystAgent', models_usage=RequestUsage(prompt_tokens=390, completion_tokens=37), metadata={}, content=[FunctionCall(id='call_9HA3DEacUl4WuG2G2PtRkXAO', arguments='{\"start\": 432, \"end\": 527}', name='percentage_change_tool')], type='ToolCallRequestEvent'), ToolCallExecutionEvent(source='DataAnalystAgent', models_usage=None, metadata={}, content=[FunctionExecutionResult(content='21.99074074074074', name='percentage_change_tool', call_id='call_9HA3DEacUl4WuG2G2PtRkXAO', is_error=False)], type='ToolCallExecutionEvent'), ToolCallSummaryMessage(source='DataAnalystAgent', models_usage=None, metadata={}, content='21.99074074074074', type='ToolCallSummaryMessage'), TextMessage(source='PlanningAgent', models_usage=RequestUsage(prompt_tokens=413, completion_tokens=137), metadata={}, content=\"It seems we've missed some context there, so let's assign the subtasks again for clarity:\\n\\nBased on the search results, Dwyane Wade had the highest points for the Miami Heat in the 2006-2007 season with 1397 points.\\n\\nNow, let's find the necessary rebound statistics:\\n\\n2. WebSearchAgent: Find Dwyane Wade's total rebound statistics for both the 2007-2008 and 2008-2009 NBA seasons.\\n3. DataAnalystAgent: Once the data is retrieved, calculate the percentage change in Dwyane Wade's total rebounds between the 2007-2008 and 2008-2009 seasons.\", type='TextMessage'), ToolCallRequestEvent(source='WebSearchAgent', models_usage=RequestUsage(prompt_tokens=576, completion_tokens=73), metadata={}, content=[FunctionCall(id='call_3i1wTDSjkGg6Ev8YKYWkZK55', arguments='{\"query\": \"Dwyane Wade total rebounds 2007-2008 NBA season\"}', name='search_web_tool'), FunctionCall(id='call_NRAs6jHxXRi8zsvpW5WlHAaU', arguments='{\"query\": \"Dwyane Wade total rebounds 2008-2009 NBA se
                        ]
                    },
                    "metadata": {},
                    "output_type": "display_data"
                }
            ],
            "source": [
                "def candidate_func(messages: Sequence[BaseAgentEvent | BaseChatMessage]) -> List[str]:\n",
                "    # keep planning_agent first one to plan out the tasks\n",
                "    if messages[-1].source == \"user\":\n",
                "        return [planning_agent.name]\n",
                "\n",
                "    # if previous agent is planning_agent and if it explicitely asks for web_search_agent\n",
                "    # or data_analyst_agent or both (in-case of re-planning or re-assignment of tasks)\n",
                "    # then return those specific agents\n",
                "    last_message = messages[-1]\n",
                "    if last_message.source == planning_agent.name:\n",
                "        participants = []\n",
                "        if web_search_agent.name in last_message.to_text():\n",
                "            participants.append(web_search_agent.name)\n",
                "        if data_analyst_agent.name in last_message.to_text():\n",
                "            participants.append(data_analyst_agent.name)\n",
                "        if participants:\n",
                "            return participants  # SelectorGroupChat will select from the remaining two agents.\n",
                "\n",
                "    # we can assume that the task is finished once the web_search_agent\n",
                "    # and data_analyst_agent have took their turns, thus we send\n",
                "    # in planning_agent to terminate the chat\n",
                "    previous_set_of_agents = set(message.source for message in messages)\n",
                "    if web_search_agent.name in previous_set_of_agents and data_analyst_agent.name in previous_set_of_agents:\n",
                "        return [planning_agent.name]\n",
                "\n",
                "    # if no-conditions are met then return all the agents\n",
                "    return [planning_agent.name, web_search_agent.name, data_analyst_agent.name]\n",
                "\n",
                "\n",
                "# Reset the previous team and run the chat again with the selector function.\n",
                "await team.reset()\n",
                "team = SelectorGroupChat(\n",
                "    [planning_agent, web_search_agent, data_analyst_agent],\n",
                "    model_client=model_client,\n",
                "    termination_condition=termination,\n",
                "    candidate_func=candidate_func,\n",
                ")\n",
                "\n",
                "await Console(team.run_stream(task=task))"
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "You can see from the conversation log that the Planning Agent returns to conversation once the Web Search Agent and Data Analyst Agent took their turns and it finds that the task was not finished as expected so it called the WebSearchAgent again to get rebound values and then called DataAnalysetAgent to get the percentage change."
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "## User Feedback\n",
                "\n",
                "We can add {py:class}`~autogen_agentchat.agents.UserProxyAgent` to the team to\n",
                "provide user feedback during a run.\n",
                "See [Human-in-the-Loop](./tutorial/human-in-the-loop.ipynb) for more details\n",
                "about {py:class}`~autogen_agentchat.agents.UserProxyAgent`.\n",
                "\n",
                "To use the {py:class}`~autogen_agentchat.agents.UserProxyAgent` in the \n",
                "web search example, we simply add it to the team and update the selector function\n",
                "to always check for user feedback after the planning agent speaks.\n",
                "If the user responds with `\"APPROVE\"`, the conversation continues, otherwise,\n",
                "the planning agent tries again, until the user approves."
            ]
        },
        {
            "cell_type": "code",
            "execution_count": null,
            "metadata": {},
            "outputs": [
                {
                    "name": "stdout",
                    "output_type": "stream",
                    "text": [
                        "---------- user ----------\n",
                        "Who was the Miami Heat player with the highest points in the 2006-2007 season, and what was the percentage change in his total rebounds between the 2007-2008 and 2008-2009 seasons?\n"
                    ]
                },
                {
                    "name": "stdout",
                    "output_type": "stream",
                    "text": [
                        "---------- PlanningAgent ----------\n",
                        "To address the user's query, we will need to perform the following tasks:\n",
                        "\n",
                        "1. Identify the Miami Heat player with the highest points in the 2006-2007 season.\n",
                        "2. Find the total rebounds for that player in the 2007-2008 season.\n",
                        "3. Find the total rebounds for that player in the 2008-2009 season.\n",
                        "4. Calculate the percentage change in the total rebounds between the 2007-2008 and 2008-2009 seasons.\n",
                        "\n",
                        "Let's assign these tasks:\n",
                        "\n",
                        "1. **WebSearchAgent**: Identify the Miami Heat player with the highest points in the 2006-2007 season.\n",
                        "   \n",
                        "(Task 2 and 3 depend on the result of Task 1. We'll proceed with Tasks 2 and 3 once Task 1 is complete.)\n",
                        "---------- UserProxyAgent ----------\n",
                        "approve\n",
                        "---------- WebSearchAgent ----------\n",
                        "[FunctionCall(id='call_0prr3fUnG5CtisUG7QeygW0w', arguments='{\"query\":\"Miami Heat highest points scorer 2006-2007 NBA season\"}', name='search_web_tool')]\n",
                        "---------- WebSearchAgent ----------\n",
                        "[FunctionExecutionResult(content='Here are the total points scored by Miami Heat players in the 2006-2007 season:\\n        Udonis Haslem: 844 points\\n        Dwayne Wade: 1397 points\\n        James Posey: 550 points\\n        ...\\n        ', call_id='call_0prr3fUnG5CtisUG7QeygW0w')]\n",
                        "---------- WebSearchAgent ----------\n",
                        "Here are the total points scored by Miami Heat players in the 2006-2007 season:\n",
                        "        Udonis Haslem: 844 points\n",
                        "        Dwayne Wade: 1397 points\n",
                        "        James Posey: 550 points\n",
                        "        ...\n",
                        "        \n",
                        "---------- PlanningAgent ----------\n",
                        "Dwyane Wade was the Miami Heat player with the highest points in the 2006-2007 season, scoring 1397 points.\n",
                        "\n",
                        "Next, we need to find Dwyane Wade's total rebounds for the 2007-2008 and 2008-2009 seasons:\n",
                        "\n",
                        "2. **WebSearchAgent**: Find Dwyane Wade's total rebounds for the 2007-2008 season.\n",
                        "3. **WebSearchAgent**: Find Dwyane Wade's total rebounds for the 2008-2009 season.\n",
                        "---------- UserProxyAgent ----------\n",
                        "approve\n",
                        "---------- WebSearchAgent ----------\n",
                        "[FunctionCall(id='call_fBZe80NaBfruOVGwRWbhXyRm', arguments='{\"query\": \"Dwyane Wade total rebounds 2007-2008 NBA season\"}', name='search_web_tool'), FunctionCall(id='call_cURYibna4fGxySiL7IYt0c3s', arguments='{\"query\": \"Dwyane Wade total rebounds 2008-2009 NBA season\"}', name='search_web_tool')]\n",
                        "---------- WebSearchAgent ----------\n",
                        "[FunctionExecutionResult(content='The number of total rebounds for Dwayne Wade in the Miami Heat season 2007-2008 is 214.', call_id='call_fBZe80NaBfruOVGwRWbhXyRm'), FunctionExecutionResult(content='The number of total rebounds for Dwayne Wade in the Miami Heat season 2008-2009 is 398.', call_id='call_cURYibna4fGxySiL7IYt0c3s')]\n",
                        "---------- WebSearchAgent ----------\n",
                        "The number of total rebounds for Dwayne Wade in the Miami Heat season 2007-2008 is 214.\n",
                        "The number of total rebounds for Dwayne Wade in the Miami Heat season 2008-2009 is 398.\n",
                        "---------- PlanningAgent ----------\n",
                        "Now that we have Dwyane Wade's total rebounds for both seasons, we can calculate the percentage change:\n",
                        "\n",
                        "4. **DataAnalystAgent**: Calculate the percentage change in Dwyane Wade's total rebounds from the 2007-2008 season (214 rebounds) to the 2008-2009 season (398 rebounds).\n",
                        "---------- UserProxyAgent ----------\n",
                        "approve\n",
                        "---------- DataAnalystAgent ----------\n",
                        "[FunctionCall(id='call_z3uog7t2x0z1Suzl5hACF9hY', arguments='{\"start\":214,\"end\":398}', name='percentage_change_tool')]\n",
                        "---------- DataAnalystAgent ----------\n",
                        "[FunctionExecutionResult(content='85.98130841121495', call_id='call_z3uog7t2x0z1Suzl5hACF9hY')]\n",
                        "---------- DataAnalystAgent ----------\n",
                        "85.98130841121495\n",
                        "---------- PlanningAgent ----------\n",
                        "Dwyane Wade was the Miami Heat player with the highest points in the 2006-2007 season, scoring 1397 points. His total rebounds increased from 214 in the 2007-2008 season to 398 in the 2008-2009 season, which is a percentage change of approximately 85.98%.\n",
                        "\n",
                        "TERMINATE\n"
                    ]
                },
                {
                    "data": {
                        "text/plain": [
                            "TaskResult(messages=[TextMessage(source='user', models_usage=None, content='Who was the Miami Heat player with the highest points in the 2006-2007 season, and what was the percentage change in his total rebounds between the 2007-2008 and 2008-2009 seasons?', type='TextMessage'), TextMessage(source='PlanningAgent', models_usage=RequestUsage(prompt_tokens=161, completion_tokens=166), content=\"To address the user's query, we will need to perform the following tasks:\\n\\n1. Identify the Miami Heat player with the highest points in the 2006-2007 season.\\n2. Find the total rebounds for that player in the 2007-2008 season.\\n3. Find the total rebounds for that player in the 2008-2009 season.\\n4. Calculate the percentage change in the total rebounds between the 2007-2008 and 2008-2009 seasons.\\n\\nLet's assign these tasks:\\n\\n1. **WebSearchAgent**: Identify the Miami Heat player with the highest points in the 2006-2007 season.\\n   \\n(Task 2 and 3 depend on the result of Task 1. We'll proceed with Tasks 2 and 3 once Task 1 is complete.)\", type='TextMessage'), UserInputRequestedEvent(source='UserProxyAgent', models_usage=None, request_id='2a433f88-f886-4b39-a078-ea1acdcb2f9d', content='', type='UserInputRequestedEvent'), TextMessage(source='UserProxyAgent', models_usage=None, content='approve', type='TextMessage'), ToolCallRequestEvent(source='WebSearchAgent', models_usage=RequestUsage(prompt_tokens=323, completion_tokens=28), content=[FunctionCall(id='call_0prr3fUnG5CtisUG7QeygW0w', arguments='{\"query\":\"Miami Heat highest points scorer 2006-2007 NBA season\"}', name='search_web_tool')], type='ToolCallRequestEvent'), ToolCallExecutionEvent(source='WebSearchAgent', models_usage=None, content=[FunctionExecutionResult(content='Here are the total points scored by Miami Heat players in the 2006-2007 season:\\n        Udonis Haslem: 844 points\\n        Dwayne Wade: 1397 points\\n        James Posey: 550 points\\n        ...\\n        ', call_id='call_0prr3fUnG5CtisUG7QeygW0w')], type='ToolCallExecutionEvent'), ToolCallSummaryMessage(source='WebSearchAgent', models_usage=None, content='Here are the total points scored by Miami Heat players in the 2006-2007 season:\\n        Udonis Haslem: 844 points\\n        Dwayne Wade: 1397 points\\n        James Posey: 550 points\\n        ...\\n        ', type='ToolCallSummaryMessage'), TextMessage(source='PlanningAgent', models_usage=RequestUsage(prompt_tokens=403, completion_tokens=112), content=\"Dwyane Wade was the Miami Heat player with the highest points in the 2006-2007 season, scoring 1397 points.\\n\\nNext, we need to find Dwyane Wade's total rebounds for the 2007-2008 and 2008-2009 seasons:\\n\\n2. **WebSearchAgent**: Find Dwyane Wade's total rebounds for the 2007-2008 season.\\n3. **WebSearchAgent**: Find Dwyane Wade's total rebounds for the 2008-2009 season.\", type='TextMessage'), UserInputRequestedEvent(source='UserProxyAgent', models_usage=None, request_id='23dd4570-2391-41e9-aeea-86598499792c', content='', type='UserInputRequestedEvent'), TextMessage(source='UserProxyAgent', models_usage=None, content='approve', type='TextMessage'), ToolCallRequestEvent(source='WebSearchAgent', models_usage=RequestUsage(prompt_tokens=543, completion_tokens=73), content=[FunctionCall(id='call_fBZe80NaBfruOVGwRWbhXyRm', arguments='{\"query\": \"Dwyane Wade total rebounds 2007-2008 NBA season\"}', name='search_web_tool'), FunctionCall(id='call_cURYibna4fGxySiL7IYt0c3s', arguments='{\"query\": \"Dwyane Wade total rebounds 2008-2009 NBA season\"}', name='search_web_tool')], type='ToolCallRequestEvent'), ToolCallExecutionEvent(source='WebSearchAgent', models_usage=None, content=[FunctionExecutionResult(content='The number of total rebounds for Dwayne Wade in the Miami Heat season 2007-2008 is 214.', call_id='call_fBZe80NaBfruOVGwRWbhXyRm'), FunctionExecutionResult(content='The number of total rebounds for Dwayne Wade in the Miami Heat season 2008-2009 is 398.', call_id='call_cURYibna4fGxySiL7IYt0c3s')], type='ToolCallExecutionEvent'), ToolCallSummaryMessage(sourc
                        ]
                    },
                    "execution_count": 9,
                    "metadata": {},
                    "output_type": "execute_result"
                }
            ],
            "source": [
                "user_proxy_agent = UserProxyAgent(\"UserProxyAgent\", description=\"A proxy for the user to approve or disapprove tasks.\")\n",
                "\n",
                "\n",
                "def selector_func_with_user_proxy(messages: Sequence[BaseAgentEvent | BaseChatMessage]) -> str | None:\n",
                "    if messages[-1].source != planning_agent.name and messages[-1].source != user_proxy_agent.name:\n",
                "        # Planning agent should be the first to engage when given a new task, or check progress.\n",
                "        return planning_agent.name\n",
                "    if messages[-1].source == planning_agent.name:\n",
                "        if messages[-2].source == user_proxy_agent.name and \"APPROVE\" in messages[-1].content.upper():  # type: ignore\n",
                "            # User has approved the plan, proceed to the next agent.\n",
                "            return None\n",
                "        # Use the user proxy agent to get the user's approval to proceed.\n",
                "        return user_proxy_agent.name\n",
                "    if messages[-1].source == user_proxy_agent.name:\n",
                "        # If the user does not approve, return to the planning agent.\n",
                "        if \"APPROVE\" not in messages[-1].content.upper():  # type: ignore\n",
                "            return planning_agent.name\n",
                "    return None\n",
                "\n",
                "\n",
                "# Reset the previous agents and run the chat again with the user proxy agent and selector function.\n",
                "await team.reset()\n",
                "team = SelectorGroupChat(\n",
                "    [planning_agent, web_search_agent, data_analyst_agent, user_proxy_agent],\n",
                "    model_client=model_client,\n",
                "    termination_condition=termination,\n",
                "    selector_prompt=selector_prompt,\n",
                "    selector_func=selector_func_with_user_proxy,\n",
                "    allow_repeated_speaker=True,\n",
                ")\n",
                "\n",
                "await Console(team.run_stream(task=task))"
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "Now, the user's feedback is incorporated into the conversation flow,\n",
                "and the user can approve or reject the planning agent's decisions."
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "## Using Reasoning Models\n",
                "\n",
                "So far in the examples, we have used a `gpt-4o` model. Models like `gpt-4o`\n",
                "and `gemini-1.5-flash` are great at following instructions, so you can\n",
                "have relatively detailed instructions in the selector prompt for the team and the \n",
                "system messages for each agent to guide their behavior.\n",
                "\n",
                "However, if you are using a reasoning model like `o3-mini`, you will need to\n",
                "keep the selector prompt and system messages as simple and to the point as possible.\n",
                "This is because the reasoning models are already good at coming up with their own \n",
                "instructions given the context provided to them.\n",
                "\n",
                "This also means that we don't need a planning agent to break down the task\n",
                "anymore, since the {py:class}`~autogen_agentchat.teams.SelectorGroupChat` that\n",
                "uses a reasoning model can do that on its own.\n",
                "\n",
                "In the following example, we will use `o3-mini` as the model for the\n",
                "agents and the team, and we will not use a planning agent.\n",
                "Also, we are keeping the selector prompt and system messages as simple as possible."
            ]
        },
        {
            "cell_type": "code",
            "execution_count": 8,
            "metadata": {},
            "outputs": [],
            "source": [
                "model_client = OpenAIChatCompletionClient(model=\"o3-mini\")\n",
                "\n",
                "web_search_agent = AssistantAgent(\n",
                "    \"WebSearchAgent\",\n",
                "    description=\"An agent for searching information on the web.\",\n",
                "    tools=[search_web_tool],\n",
                "    model_client=model_client,\n",
                "    system_message=\"\"\"Use web search tool to find information.\"\"\",\n",
                ")\n",
                "\n",
                "data_analyst_agent = AssistantAgent(\n",
                "    \"DataAnalystAgent\",\n",
                "    description=\"An agent for performing calculations.\",\n",
                "    model_client=model_client,\n",
                "    tools=[percentage_change_tool],\n",
                "    system_message=\"\"\"Use tool to perform calculation. If you have not seen the data, ask for it.\"\"\",\n",
                ")\n",
                "\n",
                "user_proxy_agent = UserProxyAgent(\n",
                "    \"UserProxyAgent\",\n",
                "    description=\"A user to approve or disapprove tasks.\",\n",
                ")\n",
                "\n",
                "selector_prompt = \"\"\"Select an agent to perform task.\n",
                "\n",
                "{roles}\n",
                "\n",
                "Current conversation context:\n",
                "{history}\n",
                "\n",
                "Read the above conversation, then select an agent from {participants} to perform the next task.\n",
                "When the task is complete, let the user approve or disapprove the task.\n",
                "\"\"\"\n",
                "\n",
                "team = SelectorGroupChat(\n",
                "    [web_search_agent, data_analyst_agent, user_proxy_agent],\n",
                "    model_client=model_client,\n",
                "    termination_condition=termination,  # Use the same termination condition as before.\n",
                "    selector_prompt=selector_prompt,\n",
                "    allow_repeated_speaker=True,\n",
                ")"
            ]
        },
        {
            "cell_type": "code",
            "execution_count": 9,
            "metadata": {},
            "outputs": [
                {
                    "name": "stdout",
                    "output_type": "stream",
                    "text": [
                        "---------- user ----------\n",
                        "Who was the Miami Heat player with the highest points in the 2006-2007 season, and what was the percentage change in his total rebounds between the 2007-2008 and 2008-2009 seasons?\n",
                        "---------- WebSearchAgent ----------\n",
                        "[FunctionCall(id='call_hl7EP6Lp5jj5wEdxeNHTwUVG', arguments='{\"query\": \"Who was the Miami Heat player with the highest points in the 2006-2007 season Miami Heat statistics Dwyane Wade rebounds percentage change 2007-2008 2008-2009 seasons\"}', name='search_web_tool')]\n",
                        "---------- WebSearchAgent ----------\n",
                        "[FunctionExecutionResult(content='Here are the total points scored by Miami Heat players in the 2006-2007 season:\\n        Udonis Haslem: 844 points\\n        Dwayne Wade: 1397 points\\n        James Posey: 550 points\\n        ...\\n        ', call_id='call_hl7EP6Lp5jj5wEdxeNHTwUVG', is_error=False)]\n",
                        "---------- WebSearchAgent ----------\n",
                        "Here are the total points scored by Miami Heat players in the 2006-2007 season:\n",
                        "        Udonis Haslem: 844 points\n",
                        "        Dwayne Wade: 1397 points\n",
                        "        James Posey: 550 points\n",
                        "        ...\n",
                        "        \n",
                        "---------- DataAnalystAgent ----------\n",
                        "I found that in the 2006–2007 season the player with the highest points was Dwyane Wade (with 1,397 points). Could you please provide Dwyane Wade’s total rebounds for the 2007–2008 and the 2008–2009 seasons so I can calculate the percentage change?\n",
                        "---------- WebSearchAgent ----------\n",
                        "[FunctionCall(id='call_lppGTILXDvO9waPwKO66ehK6', arguments='{\"query\": \"Dwyane Wade total rebounds 2007-2008 and 2008-2009 seasons for Miami Heat\"}', name='search_web_tool')]\n",
                        "---------- WebSearchAgent ----------\n",
                        "[FunctionExecutionResult(content='The number of total rebounds for Dwayne Wade in the Miami Heat season 2007-2008 is 214.', call_id='call_lppGTILXDvO9waPwKO66ehK6', is_error=False)]\n",
                        "---------- WebSearchAgent ----------\n",
                        "The number of total rebounds for Dwayne Wade in the Miami Heat season 2007-2008 is 214.\n",
                        "---------- DataAnalystAgent ----------\n",
                        "Could you please provide Dwyane Wade’s total rebounds in the 2008-2009 season?\n",
                        "---------- WebSearchAgent ----------\n",
                        "[FunctionCall(id='call_r8DBcbJtQfdtugLtyTrqOvoK', arguments='{\"query\": \"Dwyane Wade total rebounds 2008-2009 season Miami Heat\"}', name='search_web_tool')]\n",
                        "---------- WebSearchAgent ----------\n",
                        "[FunctionExecutionResult(content='The number of total rebounds for Dwayne Wade in the Miami Heat season 2008-2009 is 398.', call_id='call_r8DBcbJtQfdtugLtyTrqOvoK', is_error=False)]\n",
                        "---------- WebSearchAgent ----------\n",
                        "The number of total rebounds for Dwayne Wade in the Miami Heat season 2008-2009 is 398.\n",
                        "---------- DataAnalystAgent ----------\n",
                        "[FunctionCall(id='call_4jejv1wM7V1osbBCxJze8aQM', arguments='{\"start\": 214, \"end\": 398}', name='percentage_change_tool')]\n",
                        "---------- DataAnalystAgent ----------\n",
                        "[FunctionExecutionResult(content='85.98130841121495', call_id='call_4jejv1wM7V1osbBCxJze8aQM', is_error=False)]\n",
                        "---------- DataAnalystAgent ----------\n",
                        "85.98130841121495\n",
                        "---------- DataAnalystAgent ----------\n",
                        "Dwyane Wade was the Miami Heat player with the highest total points (1,397) during the 2006-2007 season. His total rebounds increased by approximately 86% from 214 in the 2007-2008 season to 398 in the 2008-2009 season.\n",
                        "---------- UserProxyAgent ----------\n",
                        "Approve. TERMINATE\n"
                    ]
                },
                {
                    "data": {
                        "text/plain": [
                            "TaskResult(messages=[TextMessage(source='user', models_usage=None, content='Who was the Miami Heat player with the highest points in the 2006-2007 season, and what was the percentage change in his total rebounds between the 2007-2008 and 2008-2009 seasons?', type='TextMessage'), ToolCallRequestEvent(source='WebSearchAgent', models_usage=RequestUsage(prompt_tokens=103, completion_tokens=384), content=[FunctionCall(id='call_hl7EP6Lp5jj5wEdxeNHTwUVG', arguments='{\"query\": \"Who was the Miami Heat player with the highest points in the 2006-2007 season Miami Heat statistics Dwyane Wade rebounds percentage change 2007-2008 2008-2009 seasons\"}', name='search_web_tool')], type='ToolCallRequestEvent'), ToolCallExecutionEvent(source='WebSearchAgent', models_usage=None, content=[FunctionExecutionResult(content='Here are the total points scored by Miami Heat players in the 2006-2007 season:\\n        Udonis Haslem: 844 points\\n        Dwayne Wade: 1397 points\\n        James Posey: 550 points\\n        ...\\n        ', call_id='call_hl7EP6Lp5jj5wEdxeNHTwUVG', is_error=False)], type='ToolCallExecutionEvent'), ToolCallSummaryMessage(source='WebSearchAgent', models_usage=None, content='Here are the total points scored by Miami Heat players in the 2006-2007 season:\\n        Udonis Haslem: 844 points\\n        Dwayne Wade: 1397 points\\n        James Posey: 550 points\\n        ...\\n        ', type='ToolCallSummaryMessage'), TextMessage(source='DataAnalystAgent', models_usage=RequestUsage(prompt_tokens=183, completion_tokens=1038), content='I found that in the 2006–2007 season the player with the highest points was Dwyane Wade (with 1,397 points). Could you please provide Dwyane Wade’s total rebounds for the 2007–2008 and the 2008–2009 seasons so I can calculate the percentage change?', type='TextMessage'), ToolCallRequestEvent(source='WebSearchAgent', models_usage=RequestUsage(prompt_tokens=299, completion_tokens=109), content=[FunctionCall(id='call_lppGTILXDvO9waPwKO66ehK6', arguments='{\"query\": \"Dwyane Wade total rebounds 2007-2008 and 2008-2009 seasons for Miami Heat\"}', name='search_web_tool')], type='ToolCallRequestEvent'), ToolCallExecutionEvent(source='WebSearchAgent', models_usage=None, content=[FunctionExecutionResult(content='The number of total rebounds for Dwayne Wade in the Miami Heat season 2007-2008 is 214.', call_id='call_lppGTILXDvO9waPwKO66ehK6', is_error=False)], type='ToolCallExecutionEvent'), ToolCallSummaryMessage(source='WebSearchAgent', models_usage=None, content='The number of total rebounds for Dwayne Wade in the Miami Heat season 2007-2008 is 214.', type='ToolCallSummaryMessage'), TextMessage(source='DataAnalystAgent', models_usage=RequestUsage(prompt_tokens=291, completion_tokens=224), content='Could you please provide Dwyane Wade’s total rebounds in the 2008-2009 season?', type='TextMessage'), ToolCallRequestEvent(source='WebSearchAgent', models_usage=RequestUsage(prompt_tokens=401, completion_tokens=37), content=[FunctionCall(id='call_r8DBcbJtQfdtugLtyTrqOvoK', arguments='{\"query\": \"Dwyane Wade total rebounds 2008-2009 season Miami Heat\"}', name='search_web_tool')], type='ToolCallRequestEvent'), ToolCallExecutionEvent(source='WebSearchAgent', models_usage=None, content=[FunctionExecutionResult(content='The number of total rebounds for Dwayne Wade in the Miami Heat season 2008-2009 is 398.', call_id='call_r8DBcbJtQfdtugLtyTrqOvoK', is_error=False)], type='ToolCallExecutionEvent'), ToolCallSummaryMessage(source='WebSearchAgent', models_usage=None, content='The number of total rebounds for Dwayne Wade in the Miami Heat season 2008-2009 is 398.', type='ToolCallSummaryMessage'), ToolCallRequestEvent(source='DataAnalystAgent', models_usage=RequestUsage(prompt_tokens=353, completion_tokens=158), content=[FunctionCall(id='call_4jejv1wM7V1osbBCxJze8aQM', arguments='{\"start\": 214, \"end\": 398}', name='percentage_change_tool')], type='ToolCallRequestEvent'), ToolCallExecutionEvent(source='DataAnalystAgent', models_usage=None, content=[FunctionExecutionResult(con
                        ]
                    },
                    "execution_count": 9,
                    "metadata": {},
                    "output_type": "execute_result"
                }
            ],
            "source": [
                "await Console(team.run_stream(task=task))"
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "```{tip}\n",
                "For more guidance on how to prompt reasoning models, see the\n",
                "Azure AI Services Blog on [Prompt Engineering for OpenAI's O1 and O3-mini Reasoning Models](https://techcommunity.microsoft.com/blog/azure-ai-services-blog/prompt-engineering-for-openai%E2%80%99s-o1-and-o3-mini-reasoning-models/4374010)\n",
                "```"
            ]
        }
    ],
    "metadata": {
        "kernelspec": {
            "display_name": ".venv",
            "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.12.3"
        }
    },
    "nbformat": 4,
    "nbformat_minor": 2
}