fix: handle async tools when running async agents on playground (#3356)

## Summary

Fixes a regression where using Agents with async tools (e.g. MCP tools)
was breaking in the Playground.

## Type of change

- [ x ] Bug fix
- [ ] New feature
- [ ] Breaking change
- [ ] Improvement
- [ ] Model update
- [ ] Other:

---

## Checklist

- [ ] Code complies with style guidelines
- [ ] Ran format/validation scripts (`./scripts/format.sh` and
`./scripts/validate.sh`)
- [ ] Self-review completed
- [ ] Documentation updated (comments, docstrings)
- [ ] Examples and guides: Relevant cookbook examples have been included
or updated (if applicable)
- [ ] Tested in clean environment
- [ ] Tests added/updated (if applicable)

Author: manuhortet

cookbook/apps/playground/mcp_demo.py

@@ -1,3 +1,12 @@
+"""This example shows how to run an Agent using our MCP integration in the Agno Playground.
+
+For this example to run you need:
+- Create a GitHub personal access token following these steps:
+    - https://github.com/modelcontextprotocol/servers/tree/main/src/github#setup
+- Set the GITHUB_TOKEN environment variable: `export GITHUB_TOKEN=<Your GitHub access token>`
+- Run: `pip install agno mcp openai` to install the dependencies
+"""
+
 import asyncio
 from os import getenv
 from textwrap import dedent

libs/agno/agno/app/playground/async_router.py

@@ -616,15 +616,15 @@ async def get_teams():
         if teams is None:
             return []
 
-        return [TeamGetResponse.from_team(team) for team in teams]
+        return [TeamGetResponse.from_team(team, async_mode=True) for team in teams]
 
     @playground_router.get("/teams/{team_id}")
     async def get_team(team_id: str):
         team = get_team_by_id(team_id, teams)
         if team is None:
             raise HTTPException(status_code=404, detail="Team not found")
 
-        return TeamGetResponse.from_team(team)
+        return TeamGetResponse.from_team(team, async_mode=True)
 
     @playground_router.post("/teams/{team_id}/runs")
     async def create_team_run(

libs/agno/agno/app/playground/schemas.py

@@ -32,7 +32,7 @@ class AgentGetResponse(BaseModel):
     instructions: Optional[Union[List[str], str, Callable]] = None
 
     @classmethod
-    def from_agent(self, agent: Agent) -> "AgentGetResponse":
+    def from_agent(self, agent: Agent, async_mode: bool = False) -> "AgentGetResponse":
         if agent.memory:
             memory_dict: Optional[Dict[str, Any]] = {}
             if isinstance(agent.memory, AgentMemory) and agent.memory.db:
@@ -52,7 +52,7 @@ def from_agent(self, agent: Agent) -> "AgentGetResponse":
                 memory_dict = None
         else:
             memory_dict = None
-        tools = agent.get_tools(session_id=str(uuid4()))
+        tools = agent.get_tools(session_id=str(uuid4()), async_mode=async_mode)
         return AgentGetResponse(
             agent_id=agent.agent_id,
             name=agent.name,
@@ -151,9 +151,10 @@ class TeamGetResponse(BaseModel):
     response_model: Optional[str] = None
     storage: Optional[Dict[str, Any]] = None
     memory: Optional[Dict[str, Any]] = None
+    async_mode: bool = False
 
     @classmethod
-    def from_team(self, team: Team) -> "TeamGetResponse":
+    def from_team(self, team: Team, async_mode: bool = False) -> "TeamGetResponse":
         import json
 
         memory_dict: Optional[Dict[str, Any]] = {}
@@ -191,9 +192,9 @@ def from_team(self, team: Team) -> "TeamGetResponse":
             storage={"name": team.storage.__class__.__name__} if team.storage else None,
             memory=memory_dict,
             members=[
-                AgentGetResponse.from_agent(member)
+                AgentGetResponse.from_agent(member, async_mode=async_mode)
                 if isinstance(member, Agent)
-                else TeamGetResponse.from_team(member)
+                else TeamGetResponse.from_team(member, async_mode=async_mode)
                 if isinstance(member, Team)
                 else None
                 for member in team.members