agno 2.0.1__py3-none-any.whl → 2.3.0__py3-none-any.whl

agno/utils/print_response/team.py

@@ -2,6 +2,7 @@ from typing import TYPE_CHECKING, Any, Dict, List, Optional, Sequence, Set, Unio
 
 from pydantic import BaseModel
 
+from agno.filters import FilterExpr
 from agno.media import Audio, File, Image, Video
 from agno.models.message import Message
 from agno.models.response import ToolExecution
@@ -24,6 +25,7 @@ def print_response(
     show_message: bool = True,
     show_reasoning: bool = True,
     show_full_reasoning: bool = False,
+    show_member_responses: Optional[bool] = None,
     tags_to_include_in_markdown: Optional[Set[str]] = None,
     session_id: Optional[str] = None,
     session_state: Optional[Dict[str, Any]] = None,
@@ -33,9 +35,11 @@ def print_response(
     videos: Optional[Sequence[Video]] = None,
     files: Optional[Sequence[File]] = None,
     markdown: bool = False,
-    knowledge_filters: Optional[Dict[str, Any]] = None,
+    knowledge_filters: Optional[Union[Dict[str, Any], List[FilterExpr]]] = None,
     add_history_to_context: Optional[bool] = None,
     dependencies: Optional[Dict[str, Any]] = None,
+    add_dependencies_to_context: Optional[bool] = None,
+    add_session_state_to_context: Optional[bool] = None,
     metadata: Optional[Dict[str, Any]] = None,
     debug_mode: Optional[bool] = None,
     **kwargs: Any,
@@ -82,18 +86,35 @@ def print_response(
             videos=videos,
             files=files,
             stream=False,
+            stream_events=True,
             session_id=session_id,
             session_state=session_state,
             user_id=user_id,
             knowledge_filters=knowledge_filters,
             add_history_to_context=add_history_to_context,
             dependencies=dependencies,
+            add_dependencies_to_context=add_dependencies_to_context,
+            add_session_state_to_context=add_session_state_to_context,
             metadata=metadata,
             debug_mode=debug_mode,
             **kwargs,
         )
         response_timer.stop()
 
+        if run_response.input is not None and run_response.input.input_content != input:
+            # Input was modified during the run
+            panels = [status]
+            if show_message:
+                # Convert message to a panel
+                message_content = get_text_from_message(run_response.input.input_content)
+                message_panel = create_panel(
+                    content=Text(message_content, style="green"),
+                    title="Message",
+                    border_style="cyan",
+                )
+                panels.append(message_panel)  # type: ignore
+                live_console.update(Group(*panels))
+
         team_markdown = False
         member_markdown = {}
         if markdown:
@@ -121,7 +142,7 @@ def print_response(
                 panels.append(reasoning_panel)
             live_console.update(Group(*panels))
 
-        if isinstance(run_response, TeamRunOutput) and run_response.reasoning_content is not None:
+        if isinstance(run_response, TeamRunOutput) and run_response.reasoning_content is not None and show_reasoning:
             # Create panel for thinking
             thinking_panel = create_panel(
                 content=Text(run_response.reasoning_content),
@@ -133,7 +154,7 @@ def print_response(
 
         if isinstance(run_response, TeamRunOutput):
             # Handle member responses
-            if team.show_members_responses:
+            if show_member_responses:
                 for member_response in run_response.member_responses:
                     # Handle member reasoning
                     reasoning_steps = []
@@ -304,6 +325,7 @@ def print_response_stream(
     show_message: bool = True,
     show_reasoning: bool = True,
     show_full_reasoning: bool = False,
+    show_member_responses: Optional[bool] = None,
     tags_to_include_in_markdown: Optional[Set[str]] = None,
     session_id: Optional[str] = None,
     session_state: Optional[Dict[str, Any]] = None,
@@ -313,10 +335,13 @@ def print_response_stream(
     videos: Optional[Sequence[Video]] = None,
     files: Optional[Sequence[File]] = None,
     markdown: bool = False,
+    stream_events: bool = False,
     stream_intermediate_steps: bool = False,  # type: ignore
-    knowledge_filters: Optional[Dict[str, Any]] = None,
+    knowledge_filters: Optional[Union[Dict[str, Any], List[FilterExpr]]] = None,
     add_history_to_context: Optional[bool] = None,
     dependencies: Optional[Dict[str, Any]] = None,
+    add_dependencies_to_context: Optional[bool] = None,
+    add_session_state_to_context: Optional[bool] = None,
     metadata: Optional[Dict[str, Any]] = None,
     debug_mode: Optional[bool] = None,
     **kwargs: Any,
@@ -335,7 +360,7 @@ def print_response_stream(
     if not tags_to_include_in_markdown:
         tags_to_include_in_markdown = {"think", "thinking"}
 
-    stream_intermediate_steps = True  # With streaming print response, we need to stream intermediate steps
+    stream_events = True  # With streaming print response, we need to stream intermediate steps
 
     _response_content: str = ""
     _response_reasoning_content: str = ""
@@ -379,19 +404,23 @@ def print_response_stream(
             videos=videos,
             files=files,
             stream=True,
-            stream_intermediate_steps=stream_intermediate_steps,
+            stream_events=stream_events,
             session_id=session_id,
             session_state=session_state,
             user_id=user_id,
             knowledge_filters=knowledge_filters,
             add_history_to_context=add_history_to_context,
             dependencies=dependencies,
+            add_dependencies_to_context=add_dependencies_to_context,
+            add_session_state_to_context=add_session_state_to_context,
             metadata=metadata,
             debug_mode=debug_mode,
-            yield_run_response=True,
+            yield_run_output=True,
             **kwargs,
         )
 
+        input_content = get_text_from_message(input)
+
         team_markdown = None
         member_markdown = {}
 
@@ -427,6 +456,10 @@ def print_response_stream(
                 if hasattr(resp, "reasoning_steps") and resp.reasoning_steps is not None:  # type: ignore
                     reasoning_steps = resp.reasoning_steps  # type: ignore
 
+                if resp.event == TeamRunEvent.pre_hook_completed:  # type: ignore
+                    if resp.run_input is not None:  # type: ignore
+                        input_content = get_text_from_message(resp.run_input.input_content)  # type: ignore
+
                 # Collect team tool calls, avoiding duplicates
                 if resp.event == TeamRunEvent.tool_call_completed and resp.tool:  # type: ignore
                     tool = resp.tool  # type: ignore
@@ -440,7 +473,7 @@ def print_response_stream(
                         team_tool_calls.append(tool)
 
             # Collect member tool calls, avoiding duplicates
-            if hasattr(resp, "member_responses") and resp.member_responses:
+            if show_member_responses and hasattr(resp, "member_responses") and resp.member_responses:
                 for member_response in resp.member_responses:
                     member_id = None
                     if isinstance(member_response, RunOutput) and member_response.agent_id is not None:
@@ -471,12 +504,11 @@ def print_response_stream(
             # Create new panels for each chunk
             panels = []
 
-            if input and show_message:
+            if input_content and show_message:
                 render = True
                 # Convert message to a panel
-                message_content = get_text_from_message(input)
                 message_panel = create_panel(
-                    content=Text(message_content, style="green"),
+                    content=Text(input_content, style="green"),
                     title="Message",
                     border_style="cyan",
                 )
@@ -489,7 +521,7 @@ def print_response_stream(
                     reasoning_panel = build_reasoning_step_panel(i, step, show_full_reasoning)
                     panels.append(reasoning_panel)
 
-            if len(_response_reasoning_content) > 0:
+            if len(_response_reasoning_content) > 0 and show_reasoning:
                 render = True
                 # Create panel for thinking
                 thinking_panel = create_panel(
@@ -503,7 +535,9 @@ def print_response_stream(
                 panels.append(status)
 
             # Process member responses and their tool calls
-            for member_response in resp.member_responses if hasattr(resp, "member_responses") else []:
+            for member_response in (
+                resp.member_responses if show_member_responses and hasattr(resp, "member_responses") else []
+            ):
                 member_id = None
                 member_name = "Team Member"
                 if isinstance(member_response, RunOutput) and member_response.agent_id is not None:
@@ -536,7 +570,7 @@ def print_response_stream(
                         panels.append(member_tool_calls_panel)
 
                 # Process member response content
-                if team.show_members_responses and member_id is not None:
+                if show_member_responses and member_id is not None:
                     show_markdown = False
                     if markdown:
                         show_markdown = True
@@ -655,10 +689,9 @@ def print_response_stream(
         final_panels = []
 
         # Start with the message
-        if input and show_message:
-            message_content = get_text_from_message(input)
+        if input_content and show_message:
             message_panel = create_panel(
-                content=Text(message_content, style="green"),
+                content=Text(input_content, style="green"),
                 title="Message",
                 border_style="cyan",
             )
@@ -671,7 +704,7 @@ def print_response_stream(
                 final_panels.append(reasoning_panel)
 
         # Add thinking panel if available
-        if _response_reasoning_content:
+        if _response_reasoning_content and show_reasoning:
             thinking_panel = create_panel(
                 content=Text(_response_reasoning_content),
                 title=f"Thinking ({response_timer.elapsed:.1f}s)",
@@ -680,7 +713,7 @@ def print_response_stream(
             final_panels.append(thinking_panel)
 
         # Add member tool calls and responses in correct order
-        if run_response is not None and hasattr(run_response, "member_responses"):
+        if show_member_responses and run_response is not None and hasattr(run_response, "member_responses"):
             for i, member_response in enumerate(run_response.member_responses):  # type: ignore
                 member_id = None
                 if isinstance(member_response, RunOutput) and member_response.agent_id is not None:
@@ -829,6 +862,7 @@ async def aprint_response(
     show_message: bool = True,
     show_reasoning: bool = True,
     show_full_reasoning: bool = False,
+    show_member_responses: Optional[bool] = None,
     tags_to_include_in_markdown: Optional[Set[str]] = None,
     session_id: Optional[str] = None,
     session_state: Optional[Dict[str, Any]] = None,
@@ -838,9 +872,11 @@ async def aprint_response(
     videos: Optional[Sequence[Video]] = None,
     files: Optional[Sequence[File]] = None,
     markdown: bool = False,
-    knowledge_filters: Optional[Dict[str, Any]] = None,
+    knowledge_filters: Optional[Union[Dict[str, Any], List[FilterExpr]]] = None,
     add_history_to_context: Optional[bool] = None,
     dependencies: Optional[Dict[str, Any]] = None,
+    add_dependencies_to_context: Optional[bool] = None,
+    add_session_state_to_context: Optional[bool] = None,
     metadata: Optional[Dict[str, Any]] = None,
     debug_mode: Optional[bool] = None,
     **kwargs: Any,
@@ -887,18 +923,35 @@ async def aprint_response(
             videos=videos,
             files=files,
             stream=False,
+            stream_events=True,
             session_id=session_id,
             session_state=session_state,
             user_id=user_id,
             knowledge_filters=knowledge_filters,
             add_history_to_context=add_history_to_context,
             dependencies=dependencies,
+            add_dependencies_to_context=add_dependencies_to_context,
+            add_session_state_to_context=add_session_state_to_context,
             metadata=metadata,
             debug_mode=debug_mode,
             **kwargs,
         )
         response_timer.stop()
 
+        if run_response.input is not None and run_response.input.input_content != input:
+            # Input was modified during the run
+            panels = [status]
+            if show_message:
+                # Convert message to a panel
+                message_content = get_text_from_message(run_response.input.input_content)
+                message_panel = create_panel(
+                    content=Text(message_content, style="green"),
+                    title="Message",
+                    border_style="cyan",
+                )
+                panels.append(message_panel)  # type: ignore
+                live_console.update(Group(*panels))
+
         team_markdown = False
         member_markdown = {}
         if markdown:
@@ -926,7 +979,7 @@ async def aprint_response(
                 panels.append(reasoning_panel)
             live_console.update(Group(*panels))
 
-        if isinstance(run_response, TeamRunOutput) and run_response.reasoning_content is not None:
+        if isinstance(run_response, TeamRunOutput) and run_response.reasoning_content is not None and show_reasoning:
             # Create panel for thinking
             thinking_panel = create_panel(
                 content=Text(run_response.reasoning_content),
@@ -938,7 +991,7 @@ async def aprint_response(
 
         if isinstance(run_response, TeamRunOutput):
             # Handle member responses
-            if team.show_members_responses:
+            if show_member_responses:
                 for member_response in run_response.member_responses:
                     # Handle member reasoning
                     reasoning_steps = []
@@ -1107,6 +1160,7 @@ async def aprint_response_stream(
     show_message: bool = True,
     show_reasoning: bool = True,
     show_full_reasoning: bool = False,
+    show_member_responses: Optional[bool] = None,
     tags_to_include_in_markdown: Optional[Set[str]] = None,
     session_id: Optional[str] = None,
     session_state: Optional[Dict[str, Any]] = None,
@@ -1116,10 +1170,13 @@ async def aprint_response_stream(
     videos: Optional[Sequence[Video]] = None,
     files: Optional[Sequence[File]] = None,
     markdown: bool = False,
+    stream_events: bool = False,
     stream_intermediate_steps: bool = False,  # type: ignore
-    knowledge_filters: Optional[Dict[str, Any]] = None,
+    knowledge_filters: Optional[Union[Dict[str, Any], List[FilterExpr]]] = None,
     add_history_to_context: Optional[bool] = None,
     dependencies: Optional[Dict[str, Any]] = None,
+    add_dependencies_to_context: Optional[bool] = None,
+    add_session_state_to_context: Optional[bool] = None,
     metadata: Optional[Dict[str, Any]] = None,
     debug_mode: Optional[bool] = None,
     **kwargs: Any,
@@ -1136,7 +1193,7 @@ async def aprint_response_stream(
     if not tags_to_include_in_markdown:
         tags_to_include_in_markdown = {"think", "thinking"}
 
-    stream_intermediate_steps = True  # With streaming print response, we need to stream intermediate steps
+    stream_events = True  # With streaming print response, we need to stream intermediate steps
 
     _response_content: str = ""
     _response_reasoning_content: str = ""
@@ -1182,6 +1239,8 @@ async def aprint_response_stream(
         # Dict to track member response panels by member_id
         member_response_panels = {}
 
+        input_content = get_text_from_message(input)
+
         final_run_response = None
         async for resp in team.arun(  # type: ignore
             input=input,
@@ -1190,16 +1249,18 @@ async def aprint_response_stream(
             videos=videos,
             files=files,
             stream=True,
-            stream_intermediate_steps=stream_intermediate_steps,
+            stream_events=stream_events,
             session_id=session_id,
             session_state=session_state,
             user_id=user_id,
             knowledge_filters=knowledge_filters,
             add_history_to_context=add_history_to_context,
+            add_dependencies_to_context=add_dependencies_to_context,
+            add_session_state_to_context=add_session_state_to_context,
             dependencies=dependencies,
             metadata=metadata,
             debug_mode=debug_mode,
-            yield_run_response=True,
+            yield_run_output=True,
             **kwargs,
         ):
             if team_markdown is None:
@@ -1229,6 +1290,10 @@ async def aprint_response_stream(
                 if hasattr(resp, "reasoning_steps") and resp.reasoning_steps is not None:  # type: ignore
                     reasoning_steps = resp.reasoning_steps  # type: ignore
 
+                if resp.event == TeamRunEvent.pre_hook_completed:  # type: ignore
+                    if resp.run_input is not None:  # type: ignore
+                        input_content = get_text_from_message(resp.run_input.input_content)  # type: ignore
+
                 # Collect team tool calls, avoiding duplicates
                 if resp.event == TeamRunEvent.tool_call_completed and resp.tool:  # type: ignore
                     tool = resp.tool  # type: ignore
@@ -1242,7 +1307,7 @@ async def aprint_response_stream(
                         team_tool_calls.append(tool)
 
             # Collect member tool calls, avoiding duplicates
-            if hasattr(resp, "member_responses") and resp.member_responses:
+            if show_member_responses and hasattr(resp, "member_responses") and resp.member_responses:
                 for member_response in resp.member_responses:
                     member_id = None
                     if isinstance(member_response, RunOutput) and member_response.agent_id is not None:
@@ -1272,12 +1337,11 @@ async def aprint_response_stream(
             # Create new panels for each chunk
             panels = []
 
-            if input and show_message:
+            if input_content and show_message:
                 render = True
                 # Convert message to a panel
-                message_content = get_text_from_message(input)
                 message_panel = create_panel(
-                    content=Text(message_content, style="green"),
+                    content=Text(input_content, style="green"),
                     title="Message",
                     border_style="cyan",
                 )
@@ -1290,7 +1354,7 @@ async def aprint_response_stream(
                     reasoning_panel = build_reasoning_step_panel(i, step, show_full_reasoning)
                     panels.append(reasoning_panel)
 
-            if len(_response_reasoning_content) > 0:
+            if len(_response_reasoning_content) > 0 and show_reasoning:
                 render = True
                 # Create panel for thinking
                 thinking_panel = create_panel(
@@ -1304,7 +1368,9 @@ async def aprint_response_stream(
                 panels.append(status)
 
             # Process member responses and their tool calls
-            for member_response in resp.member_responses if hasattr(resp, "member_responses") else []:
+            for member_response in (
+                resp.member_responses if show_member_responses and hasattr(resp, "member_responses") else []
+            ):
                 member_id = None
                 member_name = "Team Member"
                 if isinstance(member_response, RunOutput) and member_response.agent_id is not None:
@@ -1337,7 +1403,7 @@ async def aprint_response_stream(
                         panels.append(member_tool_calls_panel)
 
                 # Process member response content
-                if team.show_members_responses and member_id is not None:
+                if show_member_responses and member_id is not None:
                     show_markdown = False
                     if markdown:
                         show_markdown = True
@@ -1457,10 +1523,9 @@ async def aprint_response_stream(
         final_panels = []
 
         # Start with the message
-        if input and show_message:
-            message_content = get_text_from_message(input)
+        if input_content and show_message:
             message_panel = create_panel(
-                content=Text(message_content, style="green"),
+                content=Text(input_content, style="green"),
                 title="Message",
                 border_style="cyan",
             )
@@ -1473,7 +1538,7 @@ async def aprint_response_stream(
                 final_panels.append(reasoning_panel)
 
         # Add thinking panel if available
-        if _response_reasoning_content:
+        if _response_reasoning_content and show_reasoning:
             thinking_panel = create_panel(
                 content=Text(_response_reasoning_content),
                 title=f"Thinking ({response_timer.elapsed:.1f}s)",
@@ -1482,7 +1547,7 @@ async def aprint_response_stream(
             final_panels.append(thinking_panel)
 
         # Add member tool calls and responses in correct order
-        if run_response is not None and hasattr(run_response, "member_responses"):
+        if show_member_responses and run_response is not None and hasattr(run_response, "member_responses"):
             for i, member_response in enumerate(run_response.member_responses):
                 member_id = None
                 if isinstance(member_response, RunOutput) and member_response.agent_id is not None: