chore: Release 1.6.3 (#3582)

# Changelog

## New Features:

- **User Control Flows on Playground**: The Agno Platform now support
user control flows on the playground.
- **Team & Agent Events on RunResponse:** Added `store_events` parameter
to optionally add all events that happened during the course of an
agent/team run on the `RunResponse`/`TeamRunResponse`.
- **Team Member Responses on Playground**: The Agno Platform now shows
member responses during team runs.
- **Behind-the-scenes on Playground**: The Agno Platform now shows what
is happening during runs, for both agents and teams.
- **Metadata filtering support for `csv` and `csv_url` knowledge
bases:** Add knowledge filters support for these knowledge base types.

## Bug Fixes:

- **AG-UI Fix**: Fixed issue related to missing messages when using the
Agno AG-UI app.
- **Chat History Fix**: Fixed issue related to history not available
when `agent_id` not set.
- **MongoDB ObjectId serialization issue when using with agent:** Fixed
issue while using mongodb vectordb with ObjectId present in the metadata
it throws Object of type `ObjectId` is not JSON serializable

---------

Co-authored-by: Kaustubh <shuklakaustubh84@gmail.com>

Author: dirkbrnd

cookbook/teams/team_events.py

@@ -8,16 +8,16 @@
 from agno.models.openai.chat import OpenAIChat
 from agno.team import Team, TeamRunEvent
 from agno.tools.duckduckgo import DuckDuckGoTools
-from agno.tools.wikipedia import WikipediaTools
+from agno.tools.hackernews import HackerNewsTools
 
 wikipedia_agent = Agent(
-    agent_id="wikipedia-agent",
-    name="Wikipedia Agent",
-    role="Search wikipedia for information",
+    agent_id="hacker-news-agent",
+    name="Hacker News Agent",
+    role="Search Hacker News for information",
     model=MistralChat(id="mistral-large-latest"),
-    tools=[WikipediaTools()],
+    tools=[HackerNewsTools()],
     instructions=[
-        "Find information about the company in the wikipedia",
+        "Find articles about the company in the Hacker News",
     ],
 )
 

libs/agno/agno/agent/agent.py

@@ -279,7 +279,7 @@ class Agent:
 
     # Persist the events on the run response
     store_events: bool = False
-    events_to_skip: Optional[List[str]] = None
+    events_to_skip: Optional[List[RunEvent]] = None
 
     # --- Agent Team ---
     # The team of agents that this agent can transfer tasks to.
@@ -399,7 +399,7 @@ def __init__(
         stream: Optional[bool] = None,
         stream_intermediate_steps: bool = False,
         store_events: bool = False,
-        events_to_skip: Optional[List[str]] = None,
+        events_to_skip: Optional[List[RunEvent]] = None,
         team: Optional[List[Agent]] = None,
         team_data: Optional[Dict[str, Any]] = None,
         role: Optional[str] = None,
@@ -504,7 +504,9 @@ def __init__(
 
         self.store_events = store_events
         # By default, we skip the run response content event
-        self.events_to_skip = events_to_skip or [RunEvent.run_response_content.value]
+        self.events_to_skip = events_to_skip
+        if self.events_to_skip is None:
+            self.events_to_skip = [RunEvent.run_response_content]
 
         self.team = team
 
@@ -633,7 +635,8 @@ def initialize_agent(self) -> None:
             from copy import deepcopy
 
             # We store a copy of memory to ensure different team instances reference unique memory copy
-            self.memory = deepcopy(self.memory)
+            if isinstance(self.memory, Memory):
+                self.memory = deepcopy(self.memory)
             self._memory_deepcopy_done = True
 
         # Default to the agent's model if no model is provided
@@ -2482,13 +2485,6 @@ def _handle_agent_run_paused_stream(
         if not run_response.content:
             run_response.content = self._get_paused_content(run_response)
 
-        # Save session to storage
-        self.write_to_storage(user_id=user_id, session_id=session_id)
-        # Log Agent Run
-        self._log_agent_run(user_id=user_id, session_id=session_id)
-
-        log_debug(f"Agent Run Paused: {run_response.run_id}", center=True, symbol="*")
-
         # Save output to file if save_response_to_file is set
         self.save_run_response_to_file(message=message, session_id=session_id)
 
@@ -2501,6 +2497,13 @@ def _handle_agent_run_paused_stream(
             run_response,
         )
 
+        # Save session to storage
+        self.write_to_storage(user_id=user_id, session_id=session_id)
+        # Log Agent Run
+        self._log_agent_run(user_id=user_id, session_id=session_id)
+
+        log_debug(f"Agent Run Paused: {run_response.run_id}", center=True, symbol="*")
+
     def _convert_response_to_structured_format(self, run_response: RunResponse):
         # Convert the response to the structured format if needed
         if self.response_model is not None and not isinstance(run_response.content, self.response_model):
@@ -6192,7 +6195,8 @@ async def areason(self, run_messages: RunMessages) -> Any:
 
     def _handle_event(self, event: RunResponseEvent, run_response: RunResponse):
         # We only store events that are not run_response_content events
-        if self.store_events and event.event not in (self.events_to_skip or []):
+        events_to_skip = [event.value for event in self.events_to_skip] if self.events_to_skip else []
+        if self.store_events and event.event not in events_to_skip:
             if run_response.events is None:
                 run_response.events = []
             run_response.events.append(event)

libs/agno/agno/team/team.py

@@ -267,7 +267,7 @@ class Team:
     # Store the events from the Team
     store_events: bool = False
     # List of events to skip from the Team
-    events_to_skip: Optional[List[str]] = None
+    events_to_skip: Optional[List[Union[RunEvent, TeamRunEvent]]] = None
 
     # Optional app ID. Indicates this team is part of an app.
     app_id: Optional[str] = None
@@ -347,7 +347,7 @@ def __init__(
         stream: Optional[bool] = None,
         stream_intermediate_steps: bool = False,
         store_events: bool = False,
-        events_to_skip: Optional[List[str]] = None,
+        events_to_skip: Optional[List[Union[RunEvent, TeamRunEvent]]] = None,
         stream_member_events: bool = True,
         debug_mode: bool = False,
         show_members_responses: bool = False,
@@ -433,10 +433,13 @@ def __init__(
         self.stream = stream
         self.stream_intermediate_steps = stream_intermediate_steps
         self.store_events = store_events
-        self.events_to_skip = events_to_skip or [
-            RunEvent.run_response_content.value,
-            TeamRunEvent.run_response_content.value,
-        ]
+
+        self.events_to_skip = events_to_skip
+        if self.events_to_skip is None:
+            self.events_to_skip = [
+                RunEvent.run_response_content,
+                TeamRunEvent.run_response_content,
+            ]
         self.stream_member_events = stream_member_events
 
         self.debug_mode = debug_mode
@@ -598,7 +601,8 @@ def initialize_team(self, session_id: Optional[str] = None) -> None:
             self.memory = Memory()
         elif not self._memory_deepcopy_done:
             # We store a copy of memory to ensure different team instances reference unique memory copy
-            self.memory = deepcopy(self.memory)
+            if isinstance(self.memory, Memory):
+                self.memory = deepcopy(self.memory)
             self._memory_deepcopy_done = True
 
         # Default to the team's model if no model is provided
@@ -2118,7 +2122,8 @@ def _get_response_format(self) -> Optional[Union[Dict, Type[BaseModel]]]:
 
     def _handle_event(self, event: Union[RunResponseEvent, TeamRunResponseEvent], run_response: TeamRunResponse):
         # We only store events that are not run_response_content events
-        if self.store_events and event.event not in (self.events_to_skip or []):
+        events_to_skip = [event.value for event in self.events_to_skip] if self.events_to_skip else []
+        if self.store_events and event.event not in events_to_skip:
             if run_response.events is None:
                 run_response.events = []
             run_response.events.append(event)

libs/agno/agno/vectordb/mongodb/mongodb.py

@@ -1,6 +1,7 @@
 import asyncio
 import time
 from typing import Any, Dict, List, Optional
+
 from bson import ObjectId
 
 from agno.document import Document
@@ -808,7 +809,7 @@ def hybrid_search(
 
         try:
             results = list(collection.aggregate(pipeline))
-            
+
             docs = []
             for doc in results:
                 # Convert ObjectIds to strings before creating Document
@@ -817,11 +818,10 @@ def hybrid_search(
                     id=str(clean_doc["_id"]),
                     name=clean_doc.get("name"),
                     content=clean_doc["content"],
-                    meta_data={
-                        **clean_doc.get("meta_data", {}), "score": clean_doc.get("score", 0.0)},
+                    meta_data={**clean_doc.get("meta_data", {}), "score": clean_doc.get("score", 0.0)},
                 )
                 docs.append(document)
-                
+
             log_info(f"Hybrid search completed. Found {len(docs)} documents.")
             return docs
         except errors.OperationFailure as e:

libs/agno/pyproject.toml

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

libs/agno/tests/integration/agent/test_event_streaming.py

@@ -227,7 +227,7 @@ def test_intermediate_steps_with_reasoning():
     assert events[RunEvent.reasoning_step][0].reasoning_content is not None
 
 
-def test_intermediate_steps_with_user_confirmation():
+def test_intermediate_steps_with_user_confirmation(agent_storage):
     """Test that the agent streams events."""
 
     @tool(requires_confirmation=True)
@@ -237,6 +237,8 @@ def get_the_weather(city: str):
     agent = Agent(
         model=OpenAIChat(id="gpt-4o-mini"),
         tools=[get_the_weather],
+        storage=agent_storage,
+        store_events=True,
         telemetry=False,
         monitoring=False,
     )
@@ -265,6 +267,13 @@ def get_the_weather(city: str):
     run_id = agent.run_response.run_id
     updated_tools[0].confirmed = True
 
+    # Check stored events
+    stored_session = agent_storage.get_all_sessions()[0]
+    assert stored_session.memory["runs"][0]["events"] is not None
+    assert len(stored_session.memory["runs"][0]["events"]) == 2
+    assert stored_session.memory["runs"][0]["events"][0]["event"] == RunEvent.run_started
+    assert stored_session.memory["runs"][0]["events"][1]["event"] == RunEvent.run_paused
+
     # Then we continue the run
     response_generator = agent.continue_run(
         run_id=run_id, updated_tools=updated_tools, stream=True, stream_intermediate_steps=True
@@ -297,6 +306,17 @@ def get_the_weather(city: str):
 
     assert agent.run_response.is_paused is False
 
+    # Check stored events
+    stored_session = agent_storage.get_all_sessions()[0]
+    assert stored_session.memory["runs"][0]["events"] is not None
+    assert len(stored_session.memory["runs"][0]["events"]) == 6
+    assert stored_session.memory["runs"][0]["events"][0]["event"] == RunEvent.run_started
+    assert stored_session.memory["runs"][0]["events"][1]["event"] == RunEvent.run_paused
+    assert stored_session.memory["runs"][0]["events"][2]["event"] == RunEvent.run_continued
+    assert stored_session.memory["runs"][0]["events"][3]["event"] == RunEvent.tool_call_started
+    assert stored_session.memory["runs"][0]["events"][4]["event"] == RunEvent.tool_call_completed
+    assert stored_session.memory["runs"][0]["events"][5]["event"] == RunEvent.run_completed
+
 
 def test_intermediate_steps_with_memory(agent_storage, memory):
     """Test that the agent streams events."""

libs/agno/tests/integration/knowledge/test_arxiv_knowledge_base.py

@@ -97,7 +97,7 @@ async def test_arxiv_knowledge_base_async_integration(setup_vector_db):
         search_knowledge=True,
         instructions=[
             "You are a helpful assistant that can answer questions.",
-            "You can use the asearch_knowledge_base tool to search the knowledge base of journal articles for information.",
+            "You can use the search_knowledge_base tool to search the knowledge base of journal articles for information.",
         ],
     )
     response = await agent.arun("What are the key capabilities of GPT-3?", markdown=True)
@@ -108,7 +108,7 @@ async def test_arxiv_knowledge_base_async_integration(setup_vector_db):
             tool_calls.extend(msg.tool_calls)
 
     function_calls = [call for call in tool_calls if call.get("type") == "function"]
-    assert any(call["function"]["name"] == "asearch_knowledge_base" for call in function_calls)
+    assert any(call["function"]["name"] == "search_knowledge_base" for call in function_calls)
 
 
 def test_arxiv_knowledge_base_empty_query_integration(setup_vector_db):

libs/agno/tests/integration/knowledge/test_csv_url_knowledge_base.py

@@ -81,7 +81,7 @@ async def test_csv_url_knowledge_base_async():
         search_knowledge=True,
         instructions=[
             "You are a helpful assistant that can answer questions.",
-            "You can use the asearch_knowledge_base tool to search the knowledge base of CSVs for information.",
+            "You can use the search_knowledge_base tool to search the knowledge base of CSVs for information.",
         ],
     )
     response = await agent.arun("Which employees have salaries above 50000?", markdown=True)

libs/agno/tests/integration/knowledge/test_docx_knowledge_base.py

@@ -112,8 +112,8 @@ async def test_docx_knowledge_base_async_directory(setup_vector_db):
             tool_calls.extend(msg.tool_calls)
 
     function_calls = [call for call in tool_calls if call.get("type") == "function"]
-    # For async operations, we use asearch_knowledge_base
-    assert any(call["function"]["name"] == "asearch_knowledge_base" for call in function_calls)
+    # For async operations, we use search_knowledge_base
+    assert any(call["function"]["name"] == "search_knowledge_base" for call in function_calls)
 
 
 # for the one with new knowledge filter DX- filters at initialization

libs/agno/tests/integration/knowledge/test_firecrawl_knowledge_base.py

@@ -87,7 +87,7 @@ async def test_firecrawl_knowledge_base_async_directory(setup_vector_db):
             tool_calls.extend(msg.tool_calls)
 
     function_calls = [call for call in tool_calls if call.get("type") == "function"]
-    assert any(call["function"]["name"] == "asearch_knowledge_base" for call in function_calls)
+    assert any(call["function"]["name"] == "search_knowledge_base" for call in function_calls)
 
 
 @pytest.mark.skip(reason="Skipping firecrawl knowledge base tests")
@@ -109,7 +109,7 @@ async def test_firecrawl_knowledge_base_async_single_url(setup_vector_db):
             tool_calls.extend(msg.tool_calls)
 
     function_calls = [call for call in tool_calls if call.get("type") == "function"]
-    assert any(call["function"]["name"] == "asearch_knowledge_base" for call in function_calls)
+    assert any(call["function"]["name"] == "search_knowledge_base" for call in function_calls)
 
 
 @pytest.mark.skip(reason="Skipping firecrawl knowledge base tests")