yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-10-26 23:39:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Allow SocietyOfMind agent to work with function calling and tools. (#1569)

Browse Source 
* Allow SocietyOfMind agent to work with function calling and tools.

* Correctly skip OpenAI tests.
This commit is contained in:
afourney 2024-02-06 17:07:35 -08:00 committed by GitHub
parent 3e33a2c410
commit 53e6d28b9d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 169 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
autogen/agentchat/contrib/society_of_mind_agent.py
View File

@ -95,6 +95,26 @@ class SocietyOfMindAgent(ConversableAgent):
for message in messages:
message = copy.deepcopy(message)
message["role"] = "user"
# Convert tool and function calls to basic messages to avoid an error on the LLM call
if "content" not in message:
message["content"] = ""
if "tool_calls" in message:
del message["tool_calls"]
if "tool_responses" in message:
del message["tool_responses"]
if "function_call" in message:
if message["content"] == "":
try:
message["content"] = (
message["function_call"]["name"] + "(" + message["function_call"]["arguments"] + ")"
)
except KeyError:
pass
del message["function_call"]
# Add the modified message to the transcript
_messages.append(message)
_messages.append(

151
test/agentchat/contrib/test_society_of_mind_agent.py
View File

@ -2,8 +2,18 @@ import pytest
import sys
import autogen
import os
from typing_extensions import Annotated
from autogen.agentchat.contrib.society_of_mind_agent import SocietyOfMindAgent
sys.path.append(os.path.join(os.path.dirname(__file__), ".."))
from test_assistant_agent import KEY_LOC, OAI_CONFIG_LIST # noqa: E402
sys.path.append(os.path.join(os.path.dirname(__file__), "../.."))
from conftest import skip_openai # noqa: E402
if not skip_openai:
config_list = autogen.config_list_from_json(env_or_file=OAI_CONFIG_LIST, file_location=KEY_LOC)
def test_society_of_mind_agent():
external_agent = autogen.ConversableAgent(
@ -195,6 +205,143 @@ def test_custom_preparer():
assert external_agent.chat_messages[soc][-1]["content"] == "All tests passed."
@pytest.mark.skipif(
skip_openai,
reason="do not run openai tests",
)
def test_function_calling():
llm_config = {"config_list": config_list}
inner_llm_config = {
"config_list": config_list,
"functions": [
{
"name": "reverse_print",
"description": "Prints a string in reverse",
"parameters": {
"type": "object",
"properties": {
"str": {
"type": "string",
"description": "The string to reverse-print",
}
},
},
"required": ["str"],
}
],
}
external_agent = autogen.ConversableAgent(
"external_agent",
max_consecutive_auto_reply=10,
human_input_mode="NEVER",
llm_config=False,
default_auto_reply="This is an external agent speaking.",
)
agent1 = autogen.ConversableAgent(
"alice",
max_consecutive_auto_reply=10,
human_input_mode="NEVER",
llm_config=inner_llm_config,
default_auto_reply="This is alice speaking.",
)
agent2 = autogen.ConversableAgent(
"bob",
max_consecutive_auto_reply=10,
human_input_mode="NEVER",
llm_config=False,
default_auto_reply="This is bob speaking.",
)
agent3 = autogen.ConversableAgent(
"sam",
max_consecutive_auto_reply=10,
human_input_mode="NEVER",
llm_config=False,
default_auto_reply="TERMINATE",
)
agent2.register_function(
function_map={
"reverse_print": lambda str: str[::-1],
}
)
groupchat = autogen.GroupChat(
agents=[agent1, agent2, agent3], messages=[], speaker_selection_method="round_robin", max_round=10
)
group_chat_manager = autogen.GroupChatManager(
groupchat=groupchat,
llm_config=False,
is_termination_msg=lambda x: x.get("content", "").find("TERMINATE") >= 0,
)
soc = SocietyOfMindAgent("soc_agent", chat_manager=group_chat_manager, llm_config=llm_config)
external_agent.send(
"Call the reverse_print function with the str 'Hello world.'", soc, request_reply=True, silent=False
)
@pytest.mark.skipif(
skip_openai,
reason="do not run openai tests",
)
def test_tool_use():
llm_config = {"config_list": config_list}
inner_llm_config = {"config_list": config_list}
external_agent = autogen.ConversableAgent(
"external_agent",
max_consecutive_auto_reply=10,
human_input_mode="NEVER",
llm_config=False,
default_auto_reply="This is an external agent speaking.",
)
agent1 = autogen.ConversableAgent(
"alice",
max_consecutive_auto_reply=10,
human_input_mode="NEVER",
llm_config=inner_llm_config,
default_auto_reply="This is alice speaking.",
)
agent2 = autogen.ConversableAgent(
"bob",
max_consecutive_auto_reply=10,
human_input_mode="NEVER",
llm_config=False,
default_auto_reply="This is bob speaking.",
)
agent3 = autogen.ConversableAgent(
"sam",
max_consecutive_auto_reply=10,
human_input_mode="NEVER",
llm_config=False,
default_auto_reply="TERMINATE",
)
@agent2.register_for_execution()
@agent1.register_for_llm(name="reverse_print", description="Prints a string in reverse")
def _reverse_print(s: Annotated[str, "The string to reverse-print."]) -> str:
return s[::-1]
groupchat = autogen.GroupChat(
agents=[agent1, agent2, agent3], messages=[], speaker_selection_method="round_robin", max_round=10
)
group_chat_manager = autogen.GroupChatManager(
groupchat=groupchat,
llm_config=False,
is_termination_msg=lambda x: x.get("content", "").find("TERMINATE") >= 0,
)
soc = SocietyOfMindAgent("soc_agent", chat_manager=group_chat_manager, llm_config=llm_config)
external_agent.send("Call reverse_print with the str 'Hello world.'", soc, request_reply=True, silent=False)
if __name__ == "__main__":
test_society_of_mind_agent()
test_custom_preparer()
# test_society_of_mind_agent()
# test_custom_preparer()
test_function_calling()
test_tool_use()