async_toolkit_context.py
"""
Async Toolkit Context
=====================
Demonstrates that async-only toolkit functions are correctly included
in the team system message when add_member_tools_to_context=True.
"""
from agno.agent import Agent
from agno.team import Team
from agno.team.mode import TeamMode
from agno.tools import Toolkit
# ---------------------------------------------------------------------------
# Define an async-only toolkit
# ---------------------------------------------------------------------------
class AsyncResearchTools(Toolkit):
def __init__(self):
super().__init__(name="async_research_tools")
self.register(self.async_search)
self.register(self.async_summarize)
async def async_search(self, query: str) -> str:
"""Search for information on a topic."""
return f"Search results for: {query}"
async def async_summarize(self, text: str) -> str:
"""Summarize a block of text."""
return f"Summary of: {text}"
# ---------------------------------------------------------------------------
# Create Members
# ---------------------------------------------------------------------------
research_agent = Agent(
name="Research Agent",
role="Research topics using async tools",
tools=[AsyncResearchTools()],
)
# ---------------------------------------------------------------------------
# Create Team
# ---------------------------------------------------------------------------
team = Team(
name="Research Team",
members=[research_agent],
mode=TeamMode.coordinate,
add_member_tools_to_context=True,
)
# ---------------------------------------------------------------------------
# Verify
# ---------------------------------------------------------------------------
if __name__ == "__main__":
# async_mode=True shows async tools (used by aget_system_message / team.arun)
content = team.get_members_system_message_content(async_mode=True)
print("Team system message content (async mode):")
print(content)
# Verify async tool names are present in async mode
assert "async_search" in content, "async_search should appear in async team context"
assert "async_summarize" in content, (
"async_summarize should appear in async team context"
)
print("PASS: Async toolkit functions are visible in the team system message.")
Run the Example
Set up your virtual environment
uv venv --python 3.12
source .venv/bin/activate
uv venv --python 3.12
.venv\Scripts\activate