agno 2.2.0__py3-none-any.whl → 2.2.2__py3-none-any.whl

agno/utils/message.py

@@ -1,8 +1,68 @@
+from copy import deepcopy
 from typing import Dict, List, Union
 
 from pydantic import BaseModel
 
 from agno.models.message import Message
+from agno.utils.log import log_debug
+
+
+def filter_tool_calls(messages: List[Message], max_tool_calls: int) -> None:
+    """
+    Filter messages (in-place) to keep only the most recent N tool calls.
+
+    Args:
+        messages: List of messages to filter (modified in-place)
+        max_tool_calls: Number of recent tool calls to keep
+    """
+    # Count total tool calls
+    tool_call_count = sum(1 for m in messages if m.role == "tool")
+
+    # No filtering needed
+    if tool_call_count <= max_tool_calls:
+        return
+
+    # Collect tool_call_ids to keep (most recent N)
+    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)
+
+    tool_call_ids_to_keep: set[str] = set(tool_call_ids_list)
+
+    # Filter messages in-place
+    filtered_messages = []
+    for msg in messages:
+        if msg.role == "tool":
+            # Keep only tool results in our window
+            if msg.tool_call_id in tool_call_ids_to_keep:
+                filtered_messages.append(msg)
+        elif msg.role == "assistant" and msg.tool_calls:
+            # Filter tool_calls within the assistant message
+            # Use deepcopy to ensure complete isolation of the filtered message
+            filtered_msg = deepcopy(msg)
+            # 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
+                ]
+
+            if filtered_msg.tool_calls:
+                # Has tool_calls remaining, keep it
+                filtered_messages.append(filtered_msg)
+            # skip empty messages
+            elif filtered_msg.content:
+                filtered_msg.tool_calls = None
+                filtered_messages.append(filtered_msg)
+        else:
+            filtered_messages.append(msg)
+
+    messages[:] = filtered_messages
+
+    # Log filtering information
+    num_filtered = tool_call_count - len(tool_call_ids_to_keep)
+    log_debug(f"Filtered {num_filtered} tool calls, kept {len(tool_call_ids_to_keep)}")
 
 
 def get_text_from_message(message: Union[List, Dict, str, Message, BaseModel]) -> str:

agno/utils/print_response/workflow.py

@@ -7,7 +7,7 @@ from rich.markdown import Markdown
 from rich.status import Status
 from rich.text import Text
 
-from agno.media import Audio, Image, Video
+from agno.media import Audio, File, Image, Video
 from agno.models.message import Message
 from agno.run.workflow import (
     ConditionExecutionCompletedEvent,
@@ -48,6 +48,7 @@ def print_response(
     audio: Optional[List[Audio]] = None,
     images: Optional[List[Image]] = None,
     videos: Optional[List[Video]] = None,
+    files: Optional[List[File]] = None,
     markdown: bool = True,
     show_time: bool = True,
     show_step_details: bool = True,
@@ -76,6 +77,8 @@ def print_response(
         media_info.append(f"Images: {len(images)}")
     if videos:
         media_info.append(f"Videos: {len(videos)}")
+    if files:
+        media_info.append(f"Files: {len(files)}")
 
     workflow_info = f"""**Workflow:** {workflow.name}"""
     if workflow.description:
@@ -126,6 +129,7 @@ def print_response(
                 audio=audio,
                 images=images,
                 videos=videos,
+                files=files,
                 **kwargs,
             )  # type: ignore
 
@@ -186,6 +190,7 @@ def print_response_stream(
     audio: Optional[List[Audio]] = None,
     images: Optional[List[Image]] = None,
     videos: Optional[List[Video]] = None,
+    files: Optional[List[File]] = None,
     stream_events: bool = False,
     stream_intermediate_steps: bool = False,
     markdown: bool = True,
@@ -210,6 +215,8 @@ def print_response_stream(
         media_info.append(f"Images: {len(images)}")
     if videos:
         media_info.append(f"Videos: {len(videos)}")
+    if files:
+        media_info.append(f"Files: {len(files)}")
 
     workflow_info = f"""**Workflow:** {workflow.name}"""
     if workflow.description:
@@ -314,6 +321,7 @@ def print_response_stream(
                 audio=audio,
                 images=images,
                 videos=videos,
+                files=files,
                 stream=True,
                 stream_events=stream_events,
                 **kwargs,
@@ -832,6 +840,7 @@ async def aprint_response(
     audio: Optional[List[Audio]] = None,
     images: Optional[List[Image]] = None,
     videos: Optional[List[Video]] = None,
+    files: Optional[List[File]] = None,
     markdown: bool = True,
     show_time: bool = True,
     show_step_details: bool = True,
@@ -860,6 +869,8 @@ async def aprint_response(
         media_info.append(f"Images: {len(images)}")
     if videos:
         media_info.append(f"Videos: {len(videos)}")
+    if files:
+        media_info.append(f"Files: {len(files)}")
 
     workflow_info = f"""**Workflow:** {workflow.name}"""
     if workflow.description:
@@ -910,6 +921,7 @@ async def aprint_response(
                 audio=audio,
                 images=images,
                 videos=videos,
+                files=files,
                 **kwargs,
             )  # type: ignore
 
@@ -970,6 +982,7 @@ async def aprint_response_stream(
     audio: Optional[List[Audio]] = None,
     images: Optional[List[Image]] = None,
     videos: Optional[List[Video]] = None,
+    files: Optional[List[File]] = None,
     stream_events: bool = False,
     stream_intermediate_steps: bool = False,
     markdown: bool = True,
@@ -994,6 +1007,8 @@ async def aprint_response_stream(
         media_info.append(f"Images: {len(images)}")
     if videos:
         media_info.append(f"Videos: {len(videos)}")
+    if files:
+        media_info.append(f"Files: {len(files)}")
 
     workflow_info = f"""**Workflow:** {workflow.name}"""
     if workflow.description:
@@ -1098,6 +1113,7 @@ async def aprint_response_stream(
                 audio=audio,
                 images=images,
                 videos=videos,
+                files=files,
                 stream=True,
                 stream_events=stream_events,
                 **kwargs,

agno/utils/team.py

@@ -1,6 +1,10 @@
-from typing import TYPE_CHECKING, Optional, Union
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
 
 from agno.agent import Agent
+from agno.media import Audio, File, Image, Video
+from agno.run.agent import RunOutput
+from agno.run.team import TeamRunOutput
+from agno.utils.log import log_debug
 from agno.utils.string import is_valid_uuid, url_safe_string
 
 if TYPE_CHECKING:
@@ -49,3 +53,87 @@ def get_member_id(member: Union[Agent, "Team"]) -> str:
     else:
         url_safe_member_id = None
     return url_safe_member_id
+
+
+def add_interaction_to_team_run_context(
+    team_run_context: Dict[str, Any],
+    member_name: str,
+    task: str,
+    run_response: Union[RunOutput, TeamRunOutput],
+) -> None:
+    if "member_responses" not in team_run_context:
+        team_run_context["member_responses"] = []
+    team_run_context["member_responses"].append(
+        {
+            "member_name": member_name,
+            "task": task,
+            "run_response": run_response,
+        }
+    )
+    log_debug(f"Updated team run context with member name: {member_name}")
+
+
+def get_team_member_interactions_str(team_run_context: Dict[str, Any]) -> str:
+    if not team_run_context:
+        return ""
+    team_member_interactions_str = ""
+    if "member_responses" in team_run_context:
+        team_member_interactions_str += "<member_interaction_context>\nSee below interactions wit other team members.\n"
+
+        for interaction in team_run_context["member_responses"]:
+            response_dict = interaction["run_response"].to_dict()
+            response_content = (
+                response_dict.get("content")
+                or ",".join([tool.get("content", "") for tool in response_dict.get("tools", [])])
+                or ""
+            )
+            team_member_interactions_str += f"Member: {interaction['member_name']}\n"
+            team_member_interactions_str += f"Task: {interaction['task']}\n"
+            team_member_interactions_str += f"Response: {response_content}\n"
+            team_member_interactions_str += "\n"
+        team_member_interactions_str += "</member_interaction_context>\n"
+    return team_member_interactions_str
+
+
+def get_team_run_context_images(team_run_context: Dict[str, Any]) -> List[Image]:
+    if not team_run_context:
+        return []
+    images = []
+    if "member_responses" in team_run_context:
+        for interaction in team_run_context["member_responses"]:
+            if interaction["run_response"].images:
+                images.extend(interaction["run_response"].images)
+    return images
+
+
+def get_team_run_context_videos(team_run_context: Dict[str, Any]) -> List[Video]:
+    if not team_run_context:
+        return []
+    videos = []
+    if "member_responses" in team_run_context:
+        for interaction in team_run_context["member_responses"]:
+            if interaction["run_response"].videos:
+                videos.extend(interaction["run_response"].videos)
+    return videos
+
+
+def get_team_run_context_audio(team_run_context: Dict[str, Any]) -> List[Audio]:
+    if not team_run_context:
+        return []
+    audio = []
+    if "member_responses" in team_run_context:
+        for interaction in team_run_context["member_responses"]:
+            if interaction["run_response"].audio:
+                audio.extend(interaction["run_response"].audio)
+    return audio
+
+
+def get_team_run_context_files(team_run_context: Dict[str, Any]) -> List[File]:
+    if not team_run_context:
+        return []
+    files = []
+    if "member_responses" in team_run_context:
+        for interaction in team_run_context["member_responses"]:
+            if interaction["run_response"].files:
+                files.extend(interaction["run_response"].files)
+    return files

agno/workflow/step.py

@@ -382,7 +382,6 @@ class Step:
         """Enrich event with step and workflow context information"""
         if workflow_run_response is None:
             return event
-
         if hasattr(event, "workflow_id"):
             event.workflow_id = workflow_run_response.workflow_id
         if hasattr(event, "workflow_run_id"):

agno/workflow/types.py

@@ -10,6 +10,12 @@ from agno.media import Audio, File, Image, Video
 from agno.models.metrics import Metrics
 from agno.session.workflow import WorkflowSession
 from agno.utils.log import log_warning
+from agno.utils.media import (
+    reconstruct_audio_list,
+    reconstruct_files,
+    reconstruct_images,
+    reconstruct_videos,
+)
 from agno.utils.serialize import json_serializer
 
 
@@ -308,21 +314,10 @@ class StepOutput:
     def from_dict(cls, data: Dict[str, Any]) -> "StepOutput":
         """Create StepOutput from dictionary"""
         # Reconstruct media artifacts
-        images = data.get("images")
-        if images:
-            images = [Image.model_validate(img) for img in images]
-
-        videos = data.get("videos")
-        if videos:
-            videos = [Video.model_validate(vid) for vid in videos]
-
-        audio = data.get("audio")
-        if audio:
-            audio = [Audio.model_validate(aud) for aud in audio]
-
-        files = data.get("files")
-        if files:
-            files = [File.model_validate(file) for file in files]
+        images = reconstruct_images(data.get("images"))
+        videos = reconstruct_videos(data.get("videos"))
+        audio = reconstruct_audio_list(data.get("audio"))
+        files = reconstruct_files(data.get("files"))
 
         metrics_data = data.get("metrics")
         metrics = None

agno/workflow/workflow.py

@@ -29,7 +29,7 @@ from agno.exceptions import InputCheckError, OutputCheckError, RunCancelledExcep
 from agno.media import Audio, File, Image, Video
 from agno.models.message import Message
 from agno.models.metrics import Metrics
-from agno.run.agent import RunEvent
+from agno.run.agent import RunContentEvent, RunEvent
 from agno.run.base import RunStatus
 from agno.run.cancel import (
     cancel_run as cancel_run_global,
@@ -39,6 +39,7 @@ from agno.run.cancel import (
     raise_if_cancelled,
     register_run,
 )
+from agno.run.team import RunContentEvent as TeamRunContentEvent
 from agno.run.team import TeamRunEvent
 from agno.run.workflow import (
     StepOutputEvent,
@@ -1177,6 +1178,15 @@ class Workflow:
             logger.error(f"Function signature inspection failed: {e}. Falling back to original calling convention.")
             return func(**kwargs)
 
+    def _accumulate_partial_step_data(
+        self, event: Union[RunContentEvent, TeamRunContentEvent], partial_step_content: str
+    ) -> str:
+        """Accumulate partial step data from streaming events"""
+        if isinstance(event, (RunContentEvent, TeamRunContentEvent)) and event.content:
+            if isinstance(event.content, str):
+                partial_step_content += event.content
+        return partial_step_content
+
     def _execute(
         self,
         session: WorkflowSession,
@@ -1400,11 +1410,22 @@ class Workflow:
 
                 early_termination = False
 
+                # Track partial step data in case of cancellation
+                current_step_name = ""
+                current_step = None
+                partial_step_content = ""
+
                 for i, step in enumerate(self.steps):  # type: ignore[arg-type]
                     raise_if_cancelled(workflow_run_response.run_id)  # type: ignore
                     step_name = getattr(step, "name", f"step_{i + 1}")
                     log_debug(f"Streaming step {i + 1}/{self._get_step_count()}: {step_name}")
 
+                    # Track current step for cancellation handler
+                    current_step_name = step_name
+                    current_step = step
+                    # Reset partial data for this step
+                    partial_step_content = ""
+
                     # Create enhanced StepInput
                     step_input = self._create_step_input(
                         execution_input=execution_input,
@@ -1433,6 +1454,10 @@ class Workflow:
                         num_history_runs=self.num_history_runs,
                     ):
                         raise_if_cancelled(workflow_run_response.run_id)  # type: ignore
+
+                        # Accumulate partial data from streaming events
+                        partial_step_content = self._accumulate_partial_step_data(event, partial_step_content)  # type: ignore
+
                         # Handle events
                         if isinstance(event, StepOutput):
                             step_output = event
@@ -1547,6 +1572,29 @@ class Workflow:
                 logger.info(f"Workflow run {workflow_run_response.run_id} was cancelled during streaming")
                 workflow_run_response.status = RunStatus.cancelled
                 workflow_run_response.content = str(e)
+
+                # Capture partial progress from the step that was cancelled mid-stream
+                if partial_step_content:
+                    logger.info(
+                        f"Step with name  '{current_step_name}' was cancelled. Setting its partial progress as step output."
+                    )
+                    partial_step_output = StepOutput(
+                        step_name=current_step_name,
+                        step_id=getattr(current_step, "step_id", None) if current_step else None,
+                        step_type=StepType.STEP,
+                        executor_type=getattr(current_step, "executor_type", None) if current_step else None,
+                        executor_name=getattr(current_step, "executor_name", None) if current_step else None,
+                        content=partial_step_content,
+                        success=False,
+                        error="Cancelled during execution",
+                    )
+                    collected_step_outputs.append(partial_step_output)
+
+                # Preserve all progress (completed steps + partial step) before cancellation
+                if collected_step_outputs:
+                    workflow_run_response.step_results = collected_step_outputs
+                    workflow_run_response.metrics = self._aggregate_workflow_metrics(collected_step_outputs)
+
                 cancelled_event = WorkflowCancelledEvent(
                     run_id=workflow_run_response.run_id or "",
                     workflow_id=self.id,
@@ -1922,12 +1970,22 @@ class Workflow:
 
                 early_termination = False
 
+                # Track partial step data in case of cancellation
+                current_step_name = ""
+                current_step = None
+                partial_step_content = ""
+
                 for i, step in enumerate(self.steps):  # type: ignore[arg-type]
                     if workflow_run_response.run_id:
                         raise_if_cancelled(workflow_run_response.run_id)
                     step_name = getattr(step, "name", f"step_{i + 1}")
                     log_debug(f"Async streaming step {i + 1}/{self._get_step_count()}: {step_name}")
 
+                    current_step_name = step_name
+                    current_step = step
+                    # Reset partial data for this step
+                    partial_step_content = ""
+
                     # Create enhanced StepInput
                     step_input = self._create_step_input(
                         execution_input=execution_input,
@@ -1957,6 +2015,10 @@ class Workflow:
                     ):
                         if workflow_run_response.run_id:
                             raise_if_cancelled(workflow_run_response.run_id)
+
+                        # Accumulate partial data from streaming events
+                        partial_step_content = self._accumulate_partial_step_data(event, partial_step_content)  # type: ignore
+
                         if isinstance(event, StepOutput):
                             step_output = event
                             collected_step_outputs.append(step_output)
@@ -2073,6 +2135,29 @@ class Workflow:
                 logger.info(f"Workflow run {workflow_run_response.run_id} was cancelled during streaming")
                 workflow_run_response.status = RunStatus.cancelled
                 workflow_run_response.content = str(e)
+
+                # Capture partial progress from the step that was cancelled mid-stream
+                if partial_step_content:
+                    logger.info(
+                        f"Step with name  '{current_step_name}' was cancelled. Setting its partial progress as step output."
+                    )
+                    partial_step_output = StepOutput(
+                        step_name=current_step_name,
+                        step_id=getattr(current_step, "step_id", None) if current_step else None,
+                        step_type=StepType.STEP,
+                        executor_type=getattr(current_step, "executor_type", None) if current_step else None,
+                        executor_name=getattr(current_step, "executor_name", None) if current_step else None,
+                        content=partial_step_content,
+                        success=False,
+                        error="Cancelled during execution",
+                    )
+                    collected_step_outputs.append(partial_step_output)
+
+                # Preserve all progress (completed steps + partial step) before cancellation
+                if collected_step_outputs:
+                    workflow_run_response.step_results = collected_step_outputs
+                    workflow_run_response.metrics = self._aggregate_workflow_metrics(collected_step_outputs)
+
                 cancelled_event = WorkflowCancelledEvent(
                     run_id=workflow_run_response.run_id or "",
                     workflow_id=self.id,

{agno-2.2.0.dist-info → agno-2.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agno
-Version: 2.2.0
+Version: 2.2.2
 Summary: Agno: a lightweight library for building Multi-Agent Systems
 Author-email: Ashpreet Bedi <ashpreet@agno.com>
 Project-URL: homepage, https://agno.com
@@ -135,6 +135,8 @@ Provides-Extra: fal
 Requires-Dist: fal_client; extra == "fal"
 Provides-Extra: firecrawl
 Requires-Dist: firecrawl-py; extra == "firecrawl"
+Provides-Extra: tavily
+Requires-Dist: tavily-python; extra == "tavily"
 Provides-Extra: crawl4ai
 Requires-Dist: crawl4ai; extra == "crawl4ai"
 Provides-Extra: github
@@ -238,6 +240,8 @@ Requires-Dist: pypdf; extra == "pdf"
 Requires-Dist: rapidocr_onnxruntime; extra == "pdf"
 Provides-Extra: docx
 Requires-Dist: python-docx; extra == "docx"
+Provides-Extra: pptx
+Requires-Dist: python-pptx; extra == "pptx"
 Provides-Extra: text
 Requires-Dist: aiofiles; extra == "text"
 Provides-Extra: csv
@@ -294,6 +298,7 @@ Requires-Dist: agno[duckdb]; extra == "tools"
 Requires-Dist: agno[newspaper]; extra == "tools"
 Requires-Dist: agno[youtube]; extra == "tools"
 Requires-Dist: agno[firecrawl]; extra == "tools"
+Requires-Dist: agno[tavily]; extra == "tools"
 Requires-Dist: agno[crawl4ai]; extra == "tools"
 Requires-Dist: agno[github]; extra == "tools"
 Requires-Dist: agno[gmail]; extra == "tools"
@@ -346,6 +351,7 @@ Requires-Dist: agno[upstash]; extra == "vectordbs"
 Provides-Extra: knowledge
 Requires-Dist: agno[pdf]; extra == "knowledge"
 Requires-Dist: agno[docx]; extra == "knowledge"
+Requires-Dist: agno[pptx]; extra == "knowledge"
 Requires-Dist: agno[text]; extra == "knowledge"
 Requires-Dist: agno[csv]; extra == "knowledge"
 Requires-Dist: agno[markdown]; extra == "knowledge"
@@ -446,6 +452,30 @@ if __name__ == "__main__":
     agent_os.serve(app="agno_agent:app", reload=True)
 ```
 
+## AgentOS - Production Runtime for Multi-Agent Systems
+
+AgentOS is Agno's high-performance runtime for serving multi-agent systems in production. Key features include:
+
+1. **Pre-built FastAPI Runtime**: AgentOS ships with a ready-to-use FastAPI app for orchestrating your agents, teams, and workflows. This provides a major head start when building an AI product.
+
+2. **Integrated Control Plane**: The [AgentOS UI](https://os.agno.com) connects directly to your runtime, letting you test, monitor, and manage your system in real time. This gives you unmatched visibility and control over your system.
+
+3. **Private by Design**: AgentOS runs entirely in your cloud, ensuring complete data privacy. No data ever leaves your system. This is ideal for security-conscious enterprises.
+
+Here's what the [AgentOS UI](https://os.agno.com) looks like in action:
+
+https://github.com/user-attachments/assets/feb23db8-15cc-4e88-be7c-01a21a03ebf6
+
+## The Complete Agentic Solution
+
+For companies building agents, Agno provides the complete agentic solution:
+
+- The fastest framework for building agents, multi-agent teams and agentic workflows.
+- A ready-to-use FastAPI app that gets you building AI products on day one.
+- A control plane for testing, monitoring and managing your system.
+
+Agno brings a novel architecture that no other framework provides, your AgentOS runs securely in your cloud, and the control plane connects directly to it from your browser. You don't need to send data to any external services or pay retention costs, you get complete privacy and control.
+
 ## Designed for Agent Engineering
 
 Agno is a remarkably feature-rich framework, purpose-built for large-scale multi-agent deployments.
@@ -474,30 +504,6 @@ Agno is a remarkably feature-rich framework, purpose-built for large-scale multi
 
 Every part of Agno is built for real-world deployment — where developer experience meets production performance.
 
-## AgentOS - Production Runtime for Multi-Agent Systems
-
-AgentOS is Agno's high-performance runtime for serving multi-agent systems in production. Key features include:
-
-1. **Pre-built FastAPI Runtime**: AgentOS ships with a ready-to-use FastAPI app for orchestrating your agents, teams, and workflows. This provides a major head start when building an AI product.
-
-2. **Integrated Control Plane**: The [AgentOS UI](https://os.agno.com) connects directly to your runtime, letting you test, monitor, and manage your system in real time. This gives you unmatched visibility and control over your system.
-
-3. **Private by Design**: AgentOS runs entirely in your cloud, ensuring complete data privacy. No data ever leaves your system. This is ideal for security-conscious enterprises.
-
-Here's what the [AgentOS UI](https://os.agno.com) looks like in action:
-
-https://github.com/user-attachments/assets/feb23db8-15cc-4e88-be7c-01a21a03ebf6
-
-## The Complete Agentic Solution
-
-For companies building agents, Agno provides the complete agentic solution:
-
-- The fastest framework for building agents, multi-agent teams and agentic workflows.
-- A ready-to-use FastAPI app that gets you building AI products on day one.
-- A control plane for testing, monitoring and managing your system.
-
-Agno brings a novel architecture that no other framework provides, your AgentOS runs securely in your cloud, and the control plane connects directly to it from your browser. You don't need to send data to any external services or pay retention costs, you get complete privacy and control.
-
 ## Setup Your Coding Agent to Use Agno
 
 For LLMs and AI assistants to understand and navigate Agno's documentation, we provide an [llms.txt](https://docs.agno.com/llms.txt) or [llms-full.txt](https://docs.agno.com/llms-full.txt) file. This file is built for AI systems to efficiently parse and reference our documentation.