update

Author: dirkbrnd

cookbook/agent_os/handle_custom_events.py

@@ -9,10 +9,10 @@
 from typing import Optional
 
 from agno.agent import Agent
-from agno.team import Team
 from agno.db.postgres import PostgresDb
 from agno.os import AgentOS
 from agno.run.agent import CustomEvent
+from agno.team import Team
 
 # Setup the database
 db = PostgresDb(id="basic-db", db_url="postgresql+psycopg://ai:ai@localhost:5532/ai")
@@ -93,4 +93,4 @@ async def get_customer_profile(session_state: dict):
         --data-urlencode 'user_id=user_123.' \
         --data-urlencode 'session_state={"customer_name": "John Doe", "customer_email": "john.doe@example.com", "customer_phone": "1234567890"}'
     """
-    agent_os.serve(app="handle_custom_events:app", reload=True)
+    agent_os.serve(app="handle_custom_events:app", reload=True)

libs/agno/agno/run/agent.py

@@ -384,7 +384,7 @@ class OutputModelResponseCompletedEvent(BaseAgentRunEvent):
 @dataclass
 class CustomEvent(BaseAgentRunEvent):
     event: str = RunEvent.custom_event.value
-    
+
     def __init__(self, **kwargs):
         # Store arbitrary attributes directly on the instance
         for key, value in kwargs.items():

libs/agno/agno/run/team.py

@@ -367,14 +367,13 @@ class OutputModelResponseCompletedEvent(BaseTeamRunEvent):
 @dataclass
 class CustomEvent(BaseTeamRunEvent):
     event: str = TeamRunEvent.custom_event.value
-    
+
     def __init__(self, **kwargs):
         # Store arbitrary attributes directly on the instance
         for key, value in kwargs.items():
             setattr(self, key, value)
 
 
-
 TeamRunOutputEvent = Union[
     RunStartedEvent,
     RunContentEvent,

libs/agno/agno/run/workflow.py

@@ -6,9 +6,9 @@
 from pydantic import BaseModel
 
 from agno.media import Audio, Image, Video
-from agno.run.agent import RunOutput, RunEvent, run_output_event_from_dict
+from agno.run.agent import RunEvent, RunOutput, run_output_event_from_dict
 from agno.run.base import BaseRunOutputEvent, RunStatus
-from agno.run.team import TeamRunOutput, TeamRunEvent, team_run_output_event_from_dict
+from agno.run.team import TeamRunEvent, TeamRunOutput, team_run_output_event_from_dict
 
 if TYPE_CHECKING:
     from agno.workflow.types import StepOutput, WorkflowMetrics
@@ -394,7 +394,6 @@ def __init__(self, **kwargs):
             setattr(self, key, value)
 
 
-
 # Union type for all workflow run response events
 WorkflowRunOutputEvent = Union[
     WorkflowStartedEvent,

libs/agno/agno/utils/message.py

@@ -26,8 +26,8 @@ def filter_tool_calls(messages: List[Message], max_tool_calls: int) -> None:
     tool_call_ids_list: List[str] = []
     for msg in reversed(messages):
         if msg.role == "tool" and len(tool_call_ids_list) < max_tool_calls:
-                if msg.tool_call_id:
-                    tool_call_ids_list.append(msg.tool_call_id)
+            if msg.tool_call_id:
+                tool_call_ids_list.append(msg.tool_call_id)
 
     tool_call_ids_to_keep: set[str] = set(tool_call_ids_list)
 
@@ -45,9 +45,7 @@ def filter_tool_calls(messages: List[Message], max_tool_calls: int) -> None:
             # Filter tool_calls
             if filtered_msg.tool_calls is not None:
                 filtered_msg.tool_calls = [
-                    tc
-                    for tc in filtered_msg.tool_calls
-                    if tc.get("id") in tool_call_ids_to_keep
+                    tc for tc in filtered_msg.tool_calls if tc.get("id") in tool_call_ids_to_keep
                 ]
 
             if filtered_msg.tool_calls: