Skip to main content

Group Chat with Customized Speaker Selection Method

Open In Colab Open on GitHub

AutoGen offers conversable agents powered by LLM, tool or human, which can be used to perform tasks collectively via automated chat. This framework allows tool use and human participation through multi-agent conversation. Please find documentation about this feature here.

In this notebook, we demonstrate how to pass a cumstomized agent selection method to GroupChat. The customized function looks like this:

def custom_speaker_selection_func(last_speaker, groupchat):
"""Define a customized speaker selection function.
A recommended way is to define a transition for each speaker in the groupchat.

Parameters:
- last_speaker: Agent
The last speaker in the group chat.
- groupchat: GroupChat
The GroupChat object
Return:
Return one of the following:
1. an `Agent` class, it must be one of the agents in the group chat.
2. a string from ['auto', 'manual', 'random', 'round_robin'] to select a default method to use.
3. None, which indicates the chat should be terminated.
"""
pass

groupchat = autogen.GroupChat(
speaker_selection_method=custom_speaker_selection_func,
...,
)

The last speaker and the groupchat object are passed to the function. Commonly used variables from groupchat are groupchat.messages an groupchat.agents, which is the message history and the agents in the group chat respectively. You can access other attributes of the groupchat, such as groupchat.allowed_speaker_transitions_dict for pre-defined allowed_speaker_transitions_dict.

Requirements

Install autogen-agentchat:

pip install autogen-agentchat~=0.2

For more information, please refer to the installation guide.

Set your API Endpoint

The config_list_from_json function loads a list of configurations from an environment variable or a json file.

import autogen

config_list = autogen.config_list_from_json(
"OAI_CONFIG_LIST",
filter_dict={
"model": ["gpt-4", "gpt-4-1106-preview"],
},
)
tip

Learn more about configuring LLMs for agents here.

Construct Agents

  • Planner: Give a plan and revise.
  • Admin: Human in the loop to approve or terminate the process.
  • Engineer: Retrieve papers from the internet by writing code.
  • Executor: Execute the code.
  • Scientist: Read the papers and write a summary.

The pipeline is the following:

  1. The planner interact with Admin (user) to revise a plan. Only when the Admin types “Approve”, we can move to the next step.
  2. The engineer will write code to retrieve papers from the internet. The code will be executed by executor.
  3. When the code is executed successfully, the scientist will read the papers and write a summary.
  4. The summary will be reviewed by the Admin and give comments. When the Admin types “TERMINATE”, the process will be terminated.
gpt4_config = {
"cache_seed": 42, # change the cache_seed for different trials
"temperature": 0,
"config_list": config_list,
"timeout": 120,
}

planner = autogen.AssistantAgent(
name="Planner",
system_message="""Planner. Suggest a plan. Revise the plan based on feedback from admin and critic, until admin approval.
The plan may involve an engineer who can write code and a scientist who doesn't write code.
Explain the plan first. Be clear which step is performed by an engineer, and which step is performed by a scientist.
""",
llm_config=gpt4_config,
)

user_proxy = autogen.UserProxyAgent(
name="Admin",
system_message="A human admin. Interact with the planner to discuss the plan. Plan execution needs to be approved by this admin.",
code_execution_config=False,
)

engineer = autogen.AssistantAgent(
name="Engineer",
llm_config=gpt4_config,
system_message="""Engineer. You follow an approved plan. You write python/shell code to solve tasks. Wrap the code in a code block that specifies the script type. The user can't modify your code. So do not suggest incomplete code which requires others to modify. Don't use a code block if it's not intended to be executed by the executor.
Don't include multiple code blocks in one response. Do not ask others to copy and paste the result. Check the execution result returned by the executor.
If the result indicates there is an error, fix the error and output the code again. Suggest the full code instead of partial code or code changes. If the error can't be fixed or if the task is not solved even after the code is executed successfully, analyze the problem, revisit your assumption, collect additional info you need, and think of a different approach to try.
""",
)
scientist = autogen.AssistantAgent(
name="Scientist",
llm_config=gpt4_config,
system_message="""Scientist. You follow an approved plan. You are able to categorize papers after seeing their abstracts printed. You don't write code.""",
)

executor = autogen.UserProxyAgent(
name="Executor",
system_message="Executor. Execute the code written by the engineer and report the result.",
human_input_mode="NEVER",
code_execution_config={
"last_n_messages": 3,
"work_dir": "paper",
"use_docker": False,
}, # Please set use_docker=True if docker is available to run the generated code. Using docker is safer than running the generated code directly.
)

from typing import Dict, List

from autogen import Agent


def custom_speaker_selection_func(last_speaker: Agent, groupchat: autogen.GroupChat):
"""Define a customized speaker selection function.
A recommended way is to define a transition for each speaker in the groupchat.

Returns:
Return an `Agent` class or a string from ['auto', 'manual', 'random', 'round_robin'] to select a default method to use.
"""
messages = groupchat.messages

if len(messages) <= 1:
return planner

if last_speaker is user_proxy:
if "Approve" in messages[-1]["content"]:
# If the last message is approved, let the engineer to speak
return engineer
elif messages[-2]["name"] == "Planner":
# If it is the planning stage, let the planner to continue
return planner
elif messages[-2]["name"] == "Scientist":
# If the last message is from the scientist, let the scientist to continue
return scientist

elif last_speaker is planner:
# Always let the user to speak after the planner
return user_proxy

elif last_speaker is engineer:
if "```python" in messages[-1]["content"]:
# If the last message is a python code block, let the executor to speak
return executor
else:
# Otherwise, let the engineer to continue
return engineer

elif last_speaker is executor:
if "exitcode: 1" in messages[-1]["content"]:
# If the last message indicates an error, let the engineer to improve the code
return engineer
else:
# Otherwise, let the scientist to speak
return scientist

elif last_speaker is scientist:
# Always let the user to speak after the scientist
return user_proxy

else:
return "random"


groupchat = autogen.GroupChat(
agents=[user_proxy, engineer, scientist, planner, executor],
messages=[],
max_round=20,
speaker_selection_method=custom_speaker_selection_func,
)
manager = autogen.GroupChatManager(groupchat=groupchat, llm_config=gpt4_config)

Start Chat

user_proxy.initiate_chat(
manager, message="Find a latest paper about gpt-4 on arxiv and find its potential applications in software."
)
# type exit to terminate the chat
Admin (to chat_manager):

Find a latest paper about gpt-4 on arxiv and find its potential applications in software.

--------------------------------------------------------------------------------
Planner (to chat_manager):

**Initial Plan:**

1. **Scientist's Task: Literature Review**
- The scientist will conduct a comprehensive literature review to find the latest paper about GPT-4 on arXiv. This involves using search queries related to GPT-4 and filtering results by the most recent publications.

2. **Scientist's Task: Analysis of the Paper**
- Once the latest paper is identified, the scientist will read through the paper to understand its contents, focusing on the methodology, results, and discussions about potential applications in software.

3. **Scientist's Task: Identifying Potential Applications**
- The scientist will then brainstorm and list potential applications of GPT-4 in software, based on the findings from the paper. This may include applications in natural language processing, code generation, chatbots, and more.

4. **Engineer's Task: Technical Feasibility Assessment**
- The engineer will review the list of potential applications provided by the scientist and assess the technical feasibility of each application. This involves considering the current state of software technology, the capabilities of GPT-4, and the practicality of integrating GPT-4 into existing systems.

5. **Engineer's Task: Prototype Development Plan**
- For applications deemed technically feasible, the engineer will draft a plan for developing a prototype that demonstrates the use of GPT-4 in a software application. This plan will outline the required resources, estimated timeline, and the steps for implementation.

6. **Joint Task: Finalizing the Plan**
- The scientist and engineer will collaborate to finalize the plan, ensuring that it is scientifically sound and technically viable. They will prepare a document detailing the plan for potential applications and the prototype development.

7. **Presentation to Admin**
- The finalized plan will be presented to the admin for approval. The admin will review the plan and provide feedback.

8. **Revisions Based on Feedback**
- Based on the admin's feedback, the scientist and engineer will make necessary revisions to the plan. This iterative process will continue until the admin approves the plan.

**Awaiting Admin's Feedback:** Please review the initial plan and provide feedback on any adjustments or additional details you would like to see.

--------------------------------------------------------------------------------
Admin (to chat_manager):

Approve

--------------------------------------------------------------------------------
Engineer (to chat_manager):

Since the plan has been approved, I will now proceed with the first step, which is to find the latest paper about GPT-4 on arXiv. To do this, I will write a Python script that uses the arXiv API to search for papers related to GPT-4 and filter them by the most recent publications.

Here is the Python script that accomplishes this task:

```python
import requests
from datetime import datetime

# Define the URL for the arXiv API
ARXIV_API_URL = "http://export.arxiv.org/api/query"

# Define the search parameters
search_query = "all:gpt-4"
start = 0
max_results = 1
sort_by = "submittedDate"
sort_order = "descending"

# Construct the query
query_params = {
"search_query": search_query,
"start": start,
"max_results": max_results,
"sortBy": sort_by,
"sortOrder": sort_order
}

# Send the request to the arXiv API
response = requests.get(ARXIV_API_URL, params=query_params)

# Check if the request was successful
if response.status_code == 200:
# Parse the response
feed = response.text
# Find the entry element, which contains the paper information
start_entry = feed.find('<entry>')
end_entry = feed.find('</entry>')
entry = feed[start_entry:end_entry]

# Extract the title
start_title = entry.find('<title>') + 7
end_title = entry.find('</title>')
title = entry[start_title:end_title].strip()

# Extract the published date
start_published = entry.find('<published>') + 12
end_published = entry.find('</published>')
published = entry[start_published:end_published].strip()

# Extract the summary
start_summary = entry.find('<summary>') + 9
end_summary = entry.find('</summary>')
summary = entry[start_summary:end_summary].strip()

# Extract the authors
authors = []
start_author = entry.find('<author>')
end_author = entry.find('</author>')
while start_author != -1 and end_author != -1:
start_name = entry.find('<name>', start_author) + 6
end_name = entry.find('</name>', start_author)
author_name = entry[start_name:end_name].strip()
authors.append(author_name)
start_author = entry.find('<author>', end_author)
end_author = entry.find('</author>', start_author)

# Print the results
print(f"Title: {title}")
print(f"Published Date: {published}")
print(f"Authors: {', '.join(authors)}")
print(f"Summary: {summary}")
else:
print("Failed to retrieve data from arXiv API.")
```

This script will output the title, published date, authors, and summary of the most recent paper related to GPT-4 on arXiv. Please note that the actual content of the paper and its potential applications in software will need to be analyzed manually after retrieving the paper information.

--------------------------------------------------------------------------------

>>>>>>>> EXECUTING CODE BLOCK 0 (inferred language is python)...
Executor (to chat_manager):

exitcode: 0 (execution succeeded)
Code output:
Title: A Data-Centric Approach To Generate Faithful and High Quality Patient
Summaries with Large Language Models
Published Date: 024-02-23T16:32:28Z
Authors: Stefan Hegselmann, Shannon Zejiang Shen, Florian Gierse, Monica Agrawal, David Sontag, Xiaoyi Jiang
Summary: Patients often face difficulties in understanding their hospitalizations,
while healthcare workers have limited resources to provide explanations. In
this work, we investigate the potential of large language models to generate
patient summaries based on doctors' notes and study the effect of training data
on the faithfulness and quality of the generated summaries. To this end, we
develop a rigorous labeling protocol for hallucinations, and have two medical
experts annotate 100 real-world summaries and 100 generated summaries. We show
that fine-tuning on hallucination-free data effectively reduces hallucinations
from 2.60 to 1.55 per summary for Llama 2, while preserving relevant
information. Although the effect is still present, it is much smaller for GPT-4
when prompted with five examples (0.70 to 0.40). We also conduct a qualitative
evaluation using hallucination-free and improved training data. GPT-4 shows
very good results even in the zero-shot setting. We find that common
quantitative metrics do not correlate well with faithfulness and quality.
Finally, we test GPT-4 for automatic hallucination detection, which yields
promising results.


--------------------------------------------------------------------------------
Scientist (to chat_manager):

Based on the abstract provided, the paper titled "A Data-Centric Approach To Generate Faithful and High Quality Patient Summaries with Large Language Models" explores the use of large language models, including GPT-4, to generate patient summaries from doctors' notes. The study focuses on the impact of training data on the faithfulness and quality of the generated summaries and also investigates the potential of GPT-4 for automatic hallucination detection.

**Potential Applications in Software:**

1. **Healthcare Documentation Automation:**
- GPT-4 could be used to develop software that assists healthcare professionals in creating accurate and comprehensive patient summaries by automatically processing doctors' notes and other medical records.

2. **Clinical Decision Support Systems:**
- Integrating GPT-4 into clinical decision support systems could provide healthcare workers with insights and suggestions based on a patient's medical history, potentially improving diagnosis and treatment planning.

3. **Patient Education and Communication:**
- Software applications could leverage GPT-4 to translate complex medical information into patient-friendly summaries, enhancing patient understanding of their health conditions and treatments.

4. **Medical Training and Simulation:**
- GPT-4 could be used to create realistic medical scenarios for training medical students and professionals, simulating patient interactions and generating case studies.

5. **Data Quality Assurance:**
- The paper suggests that GPT-4 can be used for automatic hallucination detection, which refers to the identification of inaccuracies or fabrications in generated text. This could be applied to software that ensures the quality and reliability of medical documentation.

6. **Research and Development:**
- GPT-4 could assist researchers in summarizing and synthesizing large volumes of medical literature, aiding in the discovery of new insights and the development of novel treatments.

7. **Personalized Health Monitoring:**
- Software applications could use GPT-4 to provide personalized health monitoring and advice by analyzing user input, such as symptoms or lifestyle factors, and generating tailored health recommendations.

These potential applications highlight the versatility of GPT-4 in the realm of healthcare software, offering opportunities to enhance patient care, improve healthcare workflows, and support medical education and research.

--------------------------------------------------------------------------------
Admin (to chat_manager):

TERMINATE

--------------------------------------------------------------------------------
ChatResult(chat_id=None, chat_history=[{'content': 'Find a latest paper about gpt-4 on arxiv and find its potential applications in software.', 'role': 'assistant'}, {'content': "**Initial Plan:**\n\n1. **Scientist's Task: Literature Review**\n   - The scientist will conduct a comprehensive literature review to find the latest paper about GPT-4 on arXiv. This involves using search queries related to GPT-4 and filtering results by the most recent publications.\n\n2. **Scientist's Task: Analysis of the Paper**\n   - Once the latest paper is identified, the scientist will read through the paper to understand its contents, focusing on the methodology, results, and discussions about potential applications in software.\n\n3. **Scientist's Task: Identifying Potential Applications**\n   - The scientist will then brainstorm and list potential applications of GPT-4 in software, based on the findings from the paper. This may include applications in natural language processing, code generation, chatbots, and more.\n\n4. **Engineer's Task: Technical Feasibility Assessment**\n   - The engineer will review the list of potential applications provided by the scientist and assess the technical feasibility of each application. This involves considering the current state of software technology, the capabilities of GPT-4, and the practicality of integrating GPT-4 into existing systems.\n\n5. **Engineer's Task: Prototype Development Plan**\n   - For applications deemed technically feasible, the engineer will draft a plan for developing a prototype that demonstrates the use of GPT-4 in a software application. This plan will outline the required resources, estimated timeline, and the steps for implementation.\n\n6. **Joint Task: Finalizing the Plan**\n   - The scientist and engineer will collaborate to finalize the plan, ensuring that it is scientifically sound and technically viable. They will prepare a document detailing the plan for potential applications and the prototype development.\n\n7. **Presentation to Admin**\n   - The finalized plan will be presented to the admin for approval. The admin will review the plan and provide feedback.\n\n8. **Revisions Based on Feedback**\n   - Based on the admin's feedback, the scientist and engineer will make necessary revisions to the plan. This iterative process will continue until the admin approves the plan.\n\n**Awaiting Admin's Feedback:** Please review the initial plan and provide feedback on any adjustments or additional details you would like to see.", 'name': 'Planner', 'role': 'user'}, {'content': 'Approve', 'role': 'assistant'}, {'content': 'Since the plan has been approved, I will now proceed with the first step, which is to find the latest paper about GPT-4 on arXiv. To do this, I will write a Python script that uses the arXiv API to search for papers related to GPT-4 and filter them by the most recent publications.\n\nHere is the Python script that accomplishes this task:\n\n```python\nimport requests\nfrom datetime import datetime\n\n# Define the URL for the arXiv API\nARXIV_API_URL = "http://export.arxiv.org/api/query"\n\n# Define the search parameters\nsearch_query = "all:gpt-4"\nstart = 0\nmax_results = 1\nsort_by = "submittedDate"\nsort_order = "descending"\n\n# Construct the query\nquery_params = {\n    "search_query": search_query,\n    "start": start,\n    "max_results": max_results,\n    "sortBy": sort_by,\n    "sortOrder": sort_order\n}\n\n# Send the request to the arXiv API\nresponse = requests.get(ARXIV_API_URL, params=query_params)\n\n# Check if the request was successful\nif response.status_code == 200:\n    # Parse the response\n    feed = response.text\n    # Find the entry element, which contains the paper information\n    start_entry = feed.find(\'<entry>\')\n    end_entry = feed.find(\'</entry>\')\n    entry = feed[start_entry:end_entry]\n    \n    # Extract the title\n    start_title = entry.find(\'<title>\') + 7\n    end_title = entry.find(\'</title>\')\n    title = entry[start_title:end_title].strip()\n    \n    # Extract the published date\n    start_published = entry.find(\'<published>\') + 12\n    end_published = entry.find(\'</published>\')\n    published = entry[start_published:end_published].strip()\n    \n    # Extract the summary\n    start_summary = entry.find(\'<summary>\') + 9\n    end_summary = entry.find(\'</summary>\')\n    summary = entry[start_summary:end_summary].strip()\n    \n    # Extract the authors\n    authors = []\n    start_author = entry.find(\'<author>\')\n    end_author = entry.find(\'</author>\')\n    while start_author != -1 and end_author != -1:\n        start_name = entry.find(\'<name>\', start_author) + 6\n        end_name = entry.find(\'</name>\', start_author)\n        author_name = entry[start_name:end_name].strip()\n        authors.append(author_name)\n        start_author = entry.find(\'<author>\', end_author)\n        end_author = entry.find(\'</author>\', start_author)\n    \n    # Print the results\n    print(f"Title: {title}")\n    print(f"Published Date: {published}")\n    print(f"Authors: {\', \'.join(authors)}")\n    print(f"Summary: {summary}")\nelse:\n    print("Failed to retrieve data from arXiv API.")\n```\n\nThis script will output the title, published date, authors, and summary of the most recent paper related to GPT-4 on arXiv. Please note that the actual content of the paper and its potential applications in software will need to be analyzed manually after retrieving the paper information.', 'name': 'Engineer', 'role': 'user'}, {'content': "exitcode: 0 (execution succeeded)\nCode output: \nTitle: A Data-Centric Approach To Generate Faithful and High Quality Patient\n  Summaries with Large Language Models\nPublished Date: 024-02-23T16:32:28Z\nAuthors: Stefan Hegselmann, Shannon Zejiang Shen, Florian Gierse, Monica Agrawal, David Sontag, Xiaoyi Jiang\nSummary: Patients often face difficulties in understanding their hospitalizations,\nwhile healthcare workers have limited resources to provide explanations. In\nthis work, we investigate the potential of large language models to generate\npatient summaries based on doctors' notes and study the effect of training data\non the faithfulness and quality of the generated summaries. To this end, we\ndevelop a rigorous labeling protocol for hallucinations, and have two medical\nexperts annotate 100 real-world summaries and 100 generated summaries. We show\nthat fine-tuning on hallucination-free data effectively reduces hallucinations\nfrom 2.60 to 1.55 per summary for Llama 2, while preserving relevant\ninformation. Although the effect is still present, it is much smaller for GPT-4\nwhen prompted with five examples (0.70 to 0.40). We also conduct a qualitative\nevaluation using hallucination-free and improved training data. GPT-4 shows\nvery good results even in the zero-shot setting. We find that common\nquantitative metrics do not correlate well with faithfulness and quality.\nFinally, we test GPT-4 for automatic hallucination detection, which yields\npromising results.\n", 'name': 'Executor', 'role': 'user'}, {'content': 'Based on the abstract provided, the paper titled "A Data-Centric Approach To Generate Faithful and High Quality Patient Summaries with Large Language Models" explores the use of large language models, including GPT-4, to generate patient summaries from doctors\' notes. The study focuses on the impact of training data on the faithfulness and quality of the generated summaries and also investigates the potential of GPT-4 for automatic hallucination detection.\n\n**Potential Applications in Software:**\n\n1. **Healthcare Documentation Automation:**\n   - GPT-4 could be used to develop software that assists healthcare professionals in creating accurate and comprehensive patient summaries by automatically processing doctors\' notes and other medical records.\n\n2. **Clinical Decision Support Systems:**\n   - Integrating GPT-4 into clinical decision support systems could provide healthcare workers with insights and suggestions based on a patient\'s medical history, potentially improving diagnosis and treatment planning.\n\n3. **Patient Education and Communication:**\n   - Software applications could leverage GPT-4 to translate complex medical information into patient-friendly summaries, enhancing patient understanding of their health conditions and treatments.\n\n4. **Medical Training and Simulation:**\n   - GPT-4 could be used to create realistic medical scenarios for training medical students and professionals, simulating patient interactions and generating case studies.\n\n5. **Data Quality Assurance:**\n   - The paper suggests that GPT-4 can be used for automatic hallucination detection, which refers to the identification of inaccuracies or fabrications in generated text. This could be applied to software that ensures the quality and reliability of medical documentation.\n\n6. **Research and Development:**\n   - GPT-4 could assist researchers in summarizing and synthesizing large volumes of medical literature, aiding in the discovery of new insights and the development of novel treatments.\n\n7. **Personalized Health Monitoring:**\n   - Software applications could use GPT-4 to provide personalized health monitoring and advice by analyzing user input, such as symptoms or lifestyle factors, and generating tailored health recommendations.\n\nThese potential applications highlight the versatility of GPT-4 in the realm of healthcare software, offering opportunities to enhance patient care, improve healthcare workflows, and support medical education and research.', 'name': 'Scientist', 'role': 'user'}, {'content': 'TERMINATE', 'role': 'assistant'}], summary='', cost=({'total_cost': 0}, {'total_cost': 0}), human_input=['Approve', 'TERMINATE'])