Release 1.7.1

Author: dirkbrnd

libs/agno/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agno"
-version = "1.7.0"
+version = "1.7.1"
 description = "Agno: a lightweight library for building Multi-Agent Systems"
 requires-python = ">=3.7,<4"
 readme = "README.md"

libs/agno/tests/integration/teams/test_team_with_storage_and_memory.py

@@ -8,6 +8,7 @@
 from agno.memory.v2.db.sqlite import SqliteMemoryDb
 from agno.memory.v2.memory import Memory
 from agno.models.anthropic.claude import Claude
+from agno.models.google.gemini import Gemini
 from agno.models.openai.chat import OpenAIChat
 from agno.storage.sqlite import SqliteStorage
 from agno.team.team import Team
@@ -77,6 +78,36 @@ def route_team(team_storage, memory):
     )
 
 
+@pytest.fixture
+def route_team_with_members(team_storage, agent_storage, memory):
+    """Create a route team with storage and memory for testing."""
+    
+    def get_weather(city: str) -> str:
+        return f"The weather in {city} is sunny."
+    
+    def get_open_restaurants(city: str) -> str:
+        return f"The open restaurants in {city} are: {', '.join(['Restaurant 1', 'Restaurant 2', 'Restaurant 3'])}"
+
+    travel_agent = Agent(
+        name="Travel Agent",
+        model=Gemini(id="gemini-2.0-flash-001"),
+        storage=agent_storage,
+        memory=memory,
+        add_history_to_messages=True,
+        role="Search the web for travel information. Don't call multiple tools at once. First get weather, then restaurants.",
+        tools=[get_weather, get_open_restaurants],
+    )
+    return Team(
+        name="Route Team",
+        mode="route",
+        model=Gemini(id="gemini-2.0-flash-001"),
+        members=[travel_agent],
+        storage=team_storage,
+        memory=memory,
+        instructions="Route a single question to the travel agent. Don't make multiple requests.",
+        enable_user_memories=True,
+    )
+
 @pytest.mark.asyncio
 async def test_run_history_persistence(route_team, team_storage, memory):
     """Test that all runs within a session are persisted in storage."""
@@ -142,6 +173,57 @@ async def test_run_session_summary(route_team, team_storage, memory):
     assert len(team_session.memory["summaries"][user_id][session_id]) > 0
 
 
+@pytest.mark.asyncio
+async def test_member_run_history_persistence(route_team_with_members, agent_storage, memory):
+    """Test that all runs within a member's session are persisted in storage."""
+    user_id = "[email protected]"
+    session_id = "session_123"
+
+    # Clear memory for this specific test case
+    memory.clear()
+
+    # First request 
+    await route_team_with_members.arun("I'm traveling to Tokyo, what is the weather and open restaurants?", user_id=user_id, session_id=session_id)
+
+    agent_session = agent_storage.read(session_id=session_id)
+
+    stored_memory_data = agent_session.memory
+    assert stored_memory_data is not None, "Memory data not found in stored session."
+
+    agent_runs = stored_memory_data["runs"]
+    
+    assert len(agent_runs[-1]["messages"]) == 7, "Only system message, user message, two tool calls (and results), and response"
+
+    first_user_message_content = agent_runs[0]["messages"][1]["content"]
+    assert "I'm traveling to Tokyo, what is the weather and open restaurants?" in first_user_message_content
+
+    # Second request 
+    await route_team_with_members.arun("I'm traveling to Munich, what is the weather and open restaurants?", user_id=user_id, session_id=session_id)
+
+    agent_session = agent_storage.read(session_id=session_id)
+
+    stored_memory_data = agent_session.memory
+    assert stored_memory_data is not None, "Memory data not found in stored session."
+
+    agent_runs = stored_memory_data["runs"]
+    
+    assert len(agent_runs[-1]["messages"]) == 13, "Full history of messages"
+
+    # Third request (to the member directly)
+    await route_team_with_members.members[0].arun("Write me a report about all the places I have requested information about", user_id=user_id, session_id=session_id)
+
+    agent_session = agent_storage.read(session_id=session_id)
+
+    stored_memory_data = agent_session.memory
+    assert stored_memory_data is not None, "Memory data not found in stored session."
+
+    agent_runs = stored_memory_data["runs"]
+    
+    assert len(agent_runs[-1]["messages"]) == 15, "Full history of messages"
+
+
+
+
 @pytest.mark.asyncio
 async def test_multi_user_multi_session_route_team(route_team, team_storage, memory):
     """Test multi-user multi-session route team with storage and memory."""