agno 1.8.1__py3-none-any.whl → 2.0.0a1__py3-none-any.whl

agno/run/workflow.py

@@ -1,28 +1,66 @@
 from dataclasses import asdict, dataclass, field
 from enum import Enum
 from time import time
-from typing import Any, Dict, Optional, Union
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
 
 from pydantic import BaseModel
 
+from agno.media import AudioArtifact, AudioResponse, ImageArtifact, VideoArtifact
+from agno.run.agent import RunOutput
+from agno.run.base import RunStatus
+from agno.run.team import TeamRunOutput
 from agno.utils.log import log_error
 
+if TYPE_CHECKING:
+    from agno.workflow.types import StepOutput, WorkflowMetrics
 
-class RunEvent(str, Enum):
-    """Events that can be sent by the run() functions"""
+
+class WorkflowRunEvent(str, Enum):
+    """Events that can be sent by workflow execution"""
 
     workflow_started = "WorkflowStarted"
     workflow_completed = "WorkflowCompleted"
+    workflow_cancelled = "WorkflowCancelled"
+    workflow_error = "WorkflowError"
+
+    step_started = "StepStarted"
+    step_completed = "StepCompleted"
+    step_error = "StepError"
+
+    loop_execution_started = "LoopExecutionStarted"
+    loop_iteration_started = "LoopIterationStarted"
+    loop_iteration_completed = "LoopIterationCompleted"
+    loop_execution_completed = "LoopExecutionCompleted"
+
+    parallel_execution_started = "ParallelExecutionStarted"
+    parallel_execution_completed = "ParallelExecutionCompleted"
+
+    condition_execution_started = "ConditionExecutionStarted"
+    condition_execution_completed = "ConditionExecutionCompleted"
+
+    router_execution_started = "RouterExecutionStarted"
+    router_execution_completed = "RouterExecutionCompleted"
+
+    steps_execution_started = "StepsExecutionStarted"
+    steps_execution_completed = "StepsExecutionCompleted"
+
+    step_output = "StepOutput"
 
 
 @dataclass
-class BaseWorkflowRunResponseEvent:
-    run_id: str
+class BaseWorkflowRunOutputEvent:
+    """Base class for all workflow run response events"""
+
     created_at: int = field(default_factory=lambda: int(time()))
     event: str = ""
 
-    # For backwards compatibility
-    content: Optional[Any] = None
+    # Workflow-specific fields
+    workflow_id: Optional[str] = None
+    workflow_name: Optional[str] = None
+    session_id: Optional[str] = None
+    run_id: Optional[str] = None
+    step_id: Optional[str] = None
+    parent_step_id: Optional[str] = None
 
     def to_dict(self) -> Dict[str, Any]:
         _dict = {k: v for k, v in asdict(self).items() if v is not None}
@@ -30,6 +68,22 @@ class BaseWorkflowRunResponseEvent:
         if hasattr(self, "content") and self.content and isinstance(self.content, BaseModel):
             _dict["content"] = self.content.model_dump(exclude_none=True)
 
+        # Handle StepOutput fields that contain Message objects
+        if hasattr(self, "step_results") and self.step_results is not None:
+            _dict["step_results"] = [step.to_dict() for step in self.step_results]
+
+        if hasattr(self, "step_response") and self.step_response is not None:
+            _dict["step_response"] = self.step_response.to_dict()
+
+        if hasattr(self, "iteration_results") and self.iteration_results is not None:
+            _dict["iteration_results"] = [step.to_dict() for step in self.iteration_results]
+
+        if hasattr(self, "all_results") and self.all_results is not None:
+            _dict["all_results"] = [[step.to_dict() for step in iteration] for iteration in self.all_results]
+
+        if hasattr(self, "step_results") and self.step_results is not None:
+            _dict["step_results"] = [step.to_dict() for step in self.step_results]
+
         return _dict
 
     def to_json(self) -> str:
@@ -43,17 +97,488 @@ class BaseWorkflowRunResponseEvent:
 
         return json.dumps(_dict, indent=2)
 
+    @property
+    def is_cancelled(self):
+        return False
+
+    @property
+    def is_error(self):
+        return False
+
+    @property
+    def status(self):
+        status = "Completed"
+        if self.is_error:
+            status = "Error"
+        if self.is_cancelled:
+            status = "Cancelled"
+
+        return status
+
 
 @dataclass
-class WorkflowRunResponseStartedEvent(BaseWorkflowRunResponseEvent):
-    event: str = RunEvent.workflow_started.value
+class WorkflowStartedEvent(BaseWorkflowRunOutputEvent):
+    """Event sent when workflow execution starts"""
+
+    event: str = WorkflowRunEvent.workflow_started.value
 
 
 @dataclass
-class WorkflowCompletedEvent(BaseWorkflowRunResponseEvent):
-    event: str = RunEvent.workflow_completed.value
+class WorkflowCompletedEvent(BaseWorkflowRunOutputEvent):
+    """Event sent when workflow execution completes"""
+
+    event: str = WorkflowRunEvent.workflow_completed.value
     content: Optional[Any] = None
     content_type: str = "str"
 
+    # Store actual step execution results as StepOutput objects
+    step_results: List["StepOutput"] = field(default_factory=list)  # noqa: F821
+    metadata: Optional[Dict[str, Any]] = None
+
+
+@dataclass
+class WorkflowErrorEvent(BaseWorkflowRunOutputEvent):
+    """Event sent when workflow execution fails"""
+
+    event: str = WorkflowRunEvent.workflow_error.value
+    error: Optional[str] = None
+
+
+@dataclass
+class WorkflowCancelledEvent(BaseWorkflowRunOutputEvent):
+    """Event sent when workflow execution is cancelled"""
+
+    event: str = WorkflowRunEvent.workflow_cancelled.value
+    reason: Optional[str] = None
+
+    @property
+    def is_cancelled(self):
+        return True
+
+
+@dataclass
+class StepStartedEvent(BaseWorkflowRunOutputEvent):
+    """Event sent when step execution starts"""
+
+    event: str = WorkflowRunEvent.step_started.value
+    step_name: Optional[str] = None
+    step_index: Optional[Union[int, tuple]] = None
+
+
+@dataclass
+class StepCompletedEvent(BaseWorkflowRunOutputEvent):
+    """Event sent when step execution completes"""
+
+    event: str = WorkflowRunEvent.step_completed.value
+    step_name: Optional[str] = None
+    step_index: Optional[Union[int, tuple]] = None
+
+    content: Optional[Any] = None
+    content_type: str = "str"
+
+    # Media content fields
+    images: Optional[List[ImageArtifact]] = None
+    videos: Optional[List[VideoArtifact]] = None
+    audio: Optional[List[AudioArtifact]] = None
+    response_audio: Optional[AudioResponse] = None
+
+    # Store actual step execution results as StepOutput objects
+    step_response: Optional["StepOutput"] = None  # noqa: F821
+
+
+@dataclass
+class StepErrorEvent(BaseWorkflowRunOutputEvent):
+    """Event sent when step execution fails"""
+
+    event: str = WorkflowRunEvent.step_error.value
+    step_name: Optional[str] = None
+    step_index: Optional[Union[int, tuple]] = None
+    error: Optional[str] = None
+
+
+@dataclass
+class LoopExecutionStartedEvent(BaseWorkflowRunOutputEvent):
+    """Event sent when loop execution starts"""
+
+    event: str = WorkflowRunEvent.loop_execution_started.value
+    step_name: Optional[str] = None
+    step_index: Optional[Union[int, tuple]] = None
+    max_iterations: Optional[int] = None
+
+
+@dataclass
+class LoopIterationStartedEvent(BaseWorkflowRunOutputEvent):
+    """Event sent when loop iteration starts"""
+
+    event: str = WorkflowRunEvent.loop_iteration_started.value
+    step_name: Optional[str] = None
+    step_index: Optional[Union[int, tuple]] = None
+    iteration: int = 0
+    max_iterations: Optional[int] = None
+
+
+@dataclass
+class LoopIterationCompletedEvent(BaseWorkflowRunOutputEvent):
+    """Event sent when loop iteration completes"""
+
+    event: str = WorkflowRunEvent.loop_iteration_completed.value
+    step_name: Optional[str] = None
+    step_index: Optional[Union[int, tuple]] = None
+    iteration: int = 0
+    max_iterations: Optional[int] = None
+    iteration_results: List["StepOutput"] = field(default_factory=list)  # noqa: F821
+    should_continue: bool = True
+
+
+@dataclass
+class LoopExecutionCompletedEvent(BaseWorkflowRunOutputEvent):
+    """Event sent when loop execution completes"""
+
+    event: str = WorkflowRunEvent.loop_execution_completed.value
+    step_name: Optional[str] = None
+    step_index: Optional[Union[int, tuple]] = None
+    total_iterations: int = 0
+    max_iterations: Optional[int] = None
+    all_results: List[List["StepOutput"]] = field(default_factory=list)  # noqa: F821
+
+
+@dataclass
+class ParallelExecutionStartedEvent(BaseWorkflowRunOutputEvent):
+    """Event sent when parallel step execution starts"""
+
+    event: str = WorkflowRunEvent.parallel_execution_started.value
+    step_name: Optional[str] = None
+    step_index: Optional[Union[int, tuple]] = None
+    parallel_step_count: Optional[int] = None
+
+
+@dataclass
+class ParallelExecutionCompletedEvent(BaseWorkflowRunOutputEvent):
+    """Event sent when parallel step execution completes"""
+
+    event: str = WorkflowRunEvent.parallel_execution_completed.value
+    step_name: Optional[str] = None
+    step_index: Optional[Union[int, tuple]] = None
+    parallel_step_count: Optional[int] = None
+
+    # Results from all parallel steps
+    step_results: List["StepOutput"] = field(default_factory=list)  # noqa: F821
+
+
+@dataclass
+class ConditionExecutionStartedEvent(BaseWorkflowRunOutputEvent):
+    """Event sent when condition step execution starts"""
+
+    event: str = WorkflowRunEvent.condition_execution_started.value
+    step_name: Optional[str] = None
+    step_index: Optional[Union[int, tuple]] = None
+    condition_result: Optional[bool] = None
+
+
+@dataclass
+class ConditionExecutionCompletedEvent(BaseWorkflowRunOutputEvent):
+    """Event sent when condition step execution completes"""
+
+    event: str = WorkflowRunEvent.condition_execution_completed.value
+    step_name: Optional[str] = None
+    step_index: Optional[Union[int, tuple]] = None
+    condition_result: Optional[bool] = None
+    executed_steps: Optional[int] = None
+
+    # Results from executed steps
+    step_results: List["StepOutput"] = field(default_factory=list)  # noqa: F821
+
+
+@dataclass
+class RouterExecutionStartedEvent(BaseWorkflowRunOutputEvent):
+    """Event sent when router step execution starts"""
+
+    event: str = WorkflowRunEvent.router_execution_started.value
+    step_name: Optional[str] = None
+    step_index: Optional[Union[int, tuple]] = None
+    # Names of steps selected by router
+    selected_steps: List[str] = field(default_factory=list)
+
+
+@dataclass
+class RouterExecutionCompletedEvent(BaseWorkflowRunOutputEvent):
+    """Event sent when router step execution completes"""
+
+    event: str = WorkflowRunEvent.router_execution_completed.value
+    step_name: Optional[str] = None
+    step_index: Optional[Union[int, tuple]] = None
+    # Names of steps that were selected
+    selected_steps: List[str] = field(default_factory=list)
+    executed_steps: Optional[int] = None
+
+    # Results from executed steps
+    step_results: List["StepOutput"] = field(default_factory=list)  # noqa: F821
+
+
+@dataclass
+class StepsExecutionStartedEvent(BaseWorkflowRunOutputEvent):
+    """Event sent when steps execution starts"""
+
+    event: str = WorkflowRunEvent.steps_execution_started.value
+    step_name: Optional[str] = None
+    step_index: Optional[Union[int, tuple]] = None
+    steps_count: Optional[int] = None
+
+
+@dataclass
+class StepsExecutionCompletedEvent(BaseWorkflowRunOutputEvent):
+    """Event sent when steps execution completes"""
+
+    event: str = WorkflowRunEvent.steps_execution_completed.value
+    step_name: Optional[str] = None
+    step_index: Optional[Union[int, tuple]] = None
+    steps_count: Optional[int] = None
+    executed_steps: Optional[int] = None
+
+    # Results from executed steps
+    step_results: List["StepOutput"] = field(default_factory=list)  # noqa: F821
+
+
+@dataclass
+class StepOutputEvent(BaseWorkflowRunOutputEvent):
+    """Event sent when a step produces output - replaces direct StepOutput yielding"""
+
+    event: str = "StepOutput"
+    step_name: Optional[str] = None
+    step_index: Optional[Union[int, tuple]] = None
+
+    # Store actual step execution result as StepOutput object
+    step_output: Optional["StepOutput"] = None  # noqa: F821
+
+    # Properties for backward compatibility
+    @property
+    def content(self) -> Optional[Union[str, Dict[str, Any], List[Any], BaseModel, Any]]:
+        return self.step_output.content if self.step_output else None
+
+    @property
+    def images(self) -> Optional[List[ImageArtifact]]:
+        return self.step_output.images if self.step_output else None
+
+    @property
+    def videos(self) -> Optional[List[VideoArtifact]]:
+        return self.step_output.videos if self.step_output else None
+
+    @property
+    def audio(self) -> Optional[List[AudioArtifact]]:
+        return self.step_output.audio if self.step_output else None
+
+    @property
+    def success(self) -> bool:
+        return self.step_output.success if self.step_output else True
+
+    @property
+    def error(self) -> Optional[str]:
+        return self.step_output.error if self.step_output else None
+
+    @property
+    def stop(self) -> bool:
+        return self.step_output.stop if self.step_output else False
+
+
+# Union type for all workflow run response events
+WorkflowRunOutputEvent = Union[
+    WorkflowStartedEvent,
+    WorkflowCompletedEvent,
+    WorkflowErrorEvent,
+    WorkflowCancelledEvent,
+    StepStartedEvent,
+    StepCompletedEvent,
+    StepErrorEvent,
+    LoopExecutionStartedEvent,
+    LoopIterationStartedEvent,
+    LoopIterationCompletedEvent,
+    LoopExecutionCompletedEvent,
+    ParallelExecutionStartedEvent,
+    ParallelExecutionCompletedEvent,
+    ConditionExecutionStartedEvent,
+    ConditionExecutionCompletedEvent,
+    RouterExecutionStartedEvent,
+    RouterExecutionCompletedEvent,
+    StepsExecutionStartedEvent,
+    StepsExecutionCompletedEvent,
+    StepOutputEvent,
+]
+
+
+@dataclass
+class WorkflowRunOutput:
+    """Response returned by Workflow.run() functions - kept for backwards compatibility"""
+
+    content: Optional[Union[str, Dict[str, Any], List[Any], BaseModel, Any]] = None
+    content_type: str = "str"
+
+    # Workflow-specific fields
+    workflow_id: Optional[str] = None
+    workflow_name: Optional[str] = None
+
+    run_id: Optional[str] = None
+    session_id: Optional[str] = None
+
+    # Media content fields
+    images: Optional[List[ImageArtifact]] = None
+    videos: Optional[List[VideoArtifact]] = None
+    audio: Optional[List[AudioArtifact]] = None
+    response_audio: Optional[AudioResponse] = None
+
+    # Store actual step execution results as StepOutput objects
+    step_results: List[Union["StepOutput", List["StepOutput"]]] = field(default_factory=list)  # noqa: F821
+
+    # Store agent/team responses separately with parent_run_id references
+    step_executor_runs: Optional[List[Union[RunOutput, TeamRunOutput]]] = None
+
+    # Store events from workflow execution
+    events: Optional[List[WorkflowRunOutputEvent]] = None
+
+    # Workflow metrics aggregated from all steps
+    metrics: Optional["WorkflowMetrics"] = None
+
+    metadata: Optional[Dict[str, Any]] = None
+    created_at: int = field(default_factory=lambda: int(time()))
+
+    status: RunStatus = RunStatus.pending
+
+    @property
+    def is_cancelled(self):
+        return self.status == RunStatus.cancelled
+
+    def to_dict(self) -> Dict[str, Any]:
+        _dict = {
+            k: v
+            for k, v in asdict(self).items()
+            if v is not None
+            and k
+            not in [
+                "metadata",
+                "images",
+                "videos",
+                "audio",
+                "response_audio",
+                "step_results",
+                "step_executor_runs",
+                "events",
+                "metrics",
+            ]
+        }
+
+        if self.status is not None:
+            _dict["status"] = self.status.value if isinstance(self.status, RunStatus) else self.status
+
+        if self.metadata is not None:
+            _dict["metadata"] = self.metadata
+
+        if self.images is not None:
+            _dict["images"] = [img.to_dict() for img in self.images]
+
+        if self.videos is not None:
+            _dict["videos"] = [vid.to_dict() for vid in self.videos]
+
+        if self.audio is not None:
+            _dict["audio"] = [aud.to_dict() for aud in self.audio]
+
+        if self.response_audio is not None:
+            _dict["response_audio"] = self.response_audio.to_dict()
+
+        if self.step_results:
+            flattened_responses = []
+            for step_response in self.step_results:
+                if isinstance(step_response, list):
+                    # Handle List[StepOutput] from workflow components like Steps
+                    flattened_responses.extend([s.to_dict() for s in step_response])
+                else:
+                    # Handle single StepOutput
+                    flattened_responses.append(step_response.to_dict())
+            _dict["step_results"] = flattened_responses
+
+        if self.step_executor_runs:
+            _dict["step_executor_runs"] = [run.to_dict() for run in self.step_executor_runs]
+
+        if self.metrics is not None:
+            _dict["metrics"] = self.metrics.to_dict()
+
+        if self.content and isinstance(self.content, BaseModel):
+            _dict["content"] = self.content.model_dump(exclude_none=True)
+
+        if self.events is not None:
+            _dict["events"] = [e.to_dict() for e in self.events]
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> "WorkflowRunOutput":
+        # Import here to avoid circular import
+        from agno.workflow.step import StepOutput
+
+        workflow_metrics_dict = data.pop("metrics", {})
+        workflow_metrics = None
+        if workflow_metrics_dict:
+            from agno.workflow.workflow import WorkflowMetrics
+
+            workflow_metrics = WorkflowMetrics.from_dict(workflow_metrics_dict)
+
+        step_results = data.pop("step_results", [])
+        parsed_step_results: List[Union["StepOutput", List["StepOutput"]]] = []
+        if step_results:
+            for step_output_dict in step_results:
+                # Reconstruct StepOutput from dict
+                parsed_step_results.append(StepOutput.from_dict(step_output_dict))
+
+        # Parse step_executor_runs
+        step_executor_runs_data = data.pop("step_executor_runs", [])
+        step_executor_runs: List[Union[RunOutput, TeamRunOutput]] = []
+        if step_executor_runs_data:
+            step_executor_runs = []
+            for run_data in step_executor_runs_data:
+                if "team_id" in run_data or "team_name" in run_data:
+                    step_executor_runs.append(TeamRunOutput.from_dict(run_data))
+                else:
+                    step_executor_runs.append(RunOutput.from_dict(run_data))
+
+        metadata = data.pop("metadata", None)
+
+        images = data.pop("images", [])
+        images = [ImageArtifact.model_validate(image) for image in images] if images else None
+
+        videos = data.pop("videos", [])
+        videos = [VideoArtifact.model_validate(video) for video in videos] if videos else None
+
+        audio = data.pop("audio", [])
+        audio = [AudioArtifact.model_validate(audio) for audio in audio] if audio else None
+
+        response_audio = data.pop("response_audio", None)
+        response_audio = AudioResponse.model_validate(response_audio) if response_audio else None
+
+        events = data.pop("events", [])
+
+        return cls(
+            step_results=parsed_step_results,
+            metadata=metadata,
+            images=images,
+            videos=videos,
+            audio=audio,
+            response_audio=response_audio,
+            events=events,
+            metrics=workflow_metrics,
+            step_executor_runs=step_executor_runs,
+            **data,
+        )
+
+    def get_content_as_string(self, **kwargs) -> str:
+        import json
+
+        from pydantic import BaseModel
+
+        if isinstance(self.content, str):
+            return self.content
+        elif isinstance(self.content, BaseModel):
+            return self.content.model_dump_json(exclude_none=True, **kwargs)
+        else:
+            return json.dumps(self.content, **kwargs)
 
-WorkflowRunResponseEvent = Union[WorkflowRunResponseStartedEvent, WorkflowCompletedEvent]
+    def has_completed(self) -> bool:
+        """Check if the workflow run is completed (either successfully or with error)"""
+        return self.status in [RunStatus.completed, RunStatus.error]

agno/session/__init__.py

@@ -0,0 +1,10 @@
+from typing import Union
+
+from agno.session.agent import AgentSession
+from agno.session.summary import SessionSummaryManager
+from agno.session.team import TeamSession
+from agno.session.workflow import WorkflowSession
+
+Session = Union[AgentSession, TeamSession, WorkflowSession]
+
+__all__ = ["AgentSession", "TeamSession", "WorkflowSession", "Session", "SessionSummaryManager"]