agno 2.3.26__py3-none-any.whl → 2.4.1__py3-none-any.whl

agno/os/routers/workflows/router.py

@@ -14,6 +14,7 @@ from fastapi import (
 from fastapi.responses import JSONResponse, StreamingResponse
 from pydantic import BaseModel
 
+from agno.db.base import BaseDb
 from agno.exceptions import InputCheckError, OutputCheckError
 from agno.os.auth import (
     get_auth_token_from_request,
@@ -61,7 +62,9 @@ async def handle_workflow_via_websocket(websocket: WebSocket, message: dict, os:
             return
 
         # Get workflow from OS
-        workflow = get_workflow_by_id(workflow_id, os.workflows, create_fresh=True)
+        workflow = get_workflow_by_id(
+            workflow_id=workflow_id, workflows=os.workflows, db=os.db, registry=os.registry, create_fresh=True
+        )
         if not workflow:
             await websocket.send_text(json.dumps({"event": "error", "error": f"Workflow {workflow_id} not found"}))
             return
@@ -141,7 +144,9 @@ async def handle_workflow_subscription(websocket: WebSocket, message: dict, os:
         if buffer_status is None:
             # Run not in buffer - check database
             if workflow_id and session_id:
-                workflow = get_workflow_by_id(workflow_id, os.workflows, create_fresh=True)
+                workflow = get_workflow_by_id(
+                    workflow_id=workflow_id, workflows=os.workflows, db=os.db, registry=os.registry, create_fresh=True
+                )
                 if workflow and isinstance(workflow, Workflow):
                     workflow_run = await workflow.aget_run_output(run_id, session_id)
 
@@ -526,9 +531,6 @@ def get_workflow_router(
         },
     )
     async def get_workflows(request: Request) -> List[WorkflowSummaryResponse]:
-        if os.workflows is None:
-            return []
-
         # Filter workflows based on user's scopes (only if authorization is enabled)
         if getattr(request.state, "authorization_enabled", False):
             from agno.os.auth import filter_resources_by_access, get_accessible_resources
@@ -538,11 +540,24 @@ def get_workflow_router(
             if not accessible_ids:
                 raise HTTPException(status_code=403, detail="Insufficient permissions")
 
-            accessible_workflows = filter_resources_by_access(request, os.workflows, "workflows")
+            accessible_workflows = filter_resources_by_access(request, os.workflows or [], "workflows")
         else:
-            accessible_workflows = os.workflows
+            accessible_workflows = os.workflows or []
+
+        workflows: List[WorkflowSummaryResponse] = []
+        if accessible_workflows:
+            for workflow in accessible_workflows:
+                workflows.append(WorkflowSummaryResponse.from_workflow(workflow=workflow))
+
+        if os.db and isinstance(os.db, BaseDb):
+            from agno.workflow.workflow import get_workflows
 
-        return [WorkflowSummaryResponse.from_workflow(workflow) for workflow in accessible_workflows]
+            db_workflows = get_workflows(db=os.db, registry=os.registry)
+            if db_workflows:
+                for db_workflow in db_workflows:
+                    workflows.append(WorkflowSummaryResponse.from_workflow(workflow=db_workflow))
+
+        return workflows
 
     @router.get(
         "/workflows/{workflow_id}",
@@ -571,7 +586,9 @@ def get_workflow_router(
         dependencies=[Depends(require_resource_access("workflows", "read", "workflow_id"))],
     )
     async def get_workflow(workflow_id: str, request: Request) -> WorkflowResponse:
-        workflow = get_workflow_by_id(workflow_id, os.workflows, create_fresh=True)
+        workflow = get_workflow_by_id(
+            workflow_id=workflow_id, workflows=os.workflows, db=os.db, registry=os.registry, create_fresh=True
+        )
         if workflow is None:
             raise HTTPException(status_code=404, detail="Workflow not found")
         if isinstance(workflow, RemoteWorkflow):
@@ -622,6 +639,7 @@ def get_workflow_router(
         stream: bool = Form(True),
         session_id: Optional[str] = Form(None),
         user_id: Optional[str] = Form(None),
+        version: Optional[int] = Form(None),
     ):
         kwargs = await get_request_kwargs(request, create_workflow_run)
 
@@ -650,7 +668,14 @@ def get_workflow_router(
             kwargs["metadata"] = metadata
 
         # Retrieve the workflow by ID
-        workflow = get_workflow_by_id(workflow_id, os.workflows, create_fresh=True)
+        workflow = get_workflow_by_id(
+            workflow_id=workflow_id,
+            workflows=os.workflows,
+            db=os.db,
+            version=version,
+            registry=os.registry,
+            create_fresh=True,
+        )
         if workflow is None:
             raise HTTPException(status_code=404, detail="Workflow not found")
 
@@ -716,7 +741,9 @@ def get_workflow_router(
         dependencies=[Depends(require_resource_access("workflows", "run", "workflow_id"))],
     )
     async def cancel_workflow_run(workflow_id: str, run_id: str):
-        workflow = get_workflow_by_id(workflow_id, os.workflows, create_fresh=True)
+        workflow = get_workflow_by_id(
+            workflow_id=workflow_id, workflows=os.workflows, db=os.db, registry=os.registry, create_fresh=True
+        )
 
         if workflow is None:
             raise HTTPException(status_code=404, detail="Workflow not found")

agno/os/routers/workflows/schema.py

@@ -121,7 +121,7 @@ class WorkflowResponse(BaseModel):
 
     @classmethod
     async def from_workflow(cls, workflow: Workflow) -> "WorkflowResponse":
-        workflow_dict = workflow.to_dict()
+        workflow_dict = workflow.to_dict_for_steps()
         steps = workflow_dict.get("steps")
 
         if steps:

agno/os/schema.py

@@ -16,11 +16,7 @@ from agno.os.config import (
     SessionConfig,
     TracesConfig,
 )
-from agno.os.utils import (
-    extract_input_media,
-    get_run_input,
-    get_session_name,
-)
+from agno.os.utils import extract_input_media, get_run_input, get_session_name, to_utc_datetime
 from agno.session import AgentSession, TeamSession, WorkflowSession
 from agno.team.remote import RemoteTeam
 from agno.team.team import Team
@@ -79,10 +75,12 @@ class InternalServerErrorResponse(BaseModel):
 
 
 class HealthResponse(BaseModel):
-    model_config = ConfigDict(json_schema_extra={"example": {"status": "ok", "instantiated_at": "1760169236.778903"}})
+    model_config = ConfigDict(
+        json_schema_extra={"example": {"status": "ok", "instantiated_at": "2025-06-10T12:00:00Z"}}
+    )
 
     status: str = Field(..., description="Health status of the service")
-    instantiated_at: str = Field(..., description="Unix timestamp when service was instantiated")
+    instantiated_at: datetime = Field(..., description="Timestamp when service was instantiated")
 
 
 class InterfaceResponse(BaseModel):
@@ -145,7 +143,8 @@ class ConfigResponse(BaseModel):
     name: Optional[str] = Field(None, description="Name of the OS instance")
     description: Optional[str] = Field(None, description="Description of the OS instance")
     available_models: Optional[List[str]] = Field(None, description="List of available models")
-    databases: List[str] = Field(..., description="List of database IDs")
+    os_database: Optional[str] = Field(None, description="ID of the database used for the OS instance")
+    databases: List[str] = Field(..., description="List of database IDs used by the components of the OS instance")
     chat: Optional[ChatConfig] = Field(None, description="Chat configuration")
 
     session: Optional[SessionConfig] = Field(None, description="Session configuration")
@@ -212,8 +211,8 @@ class SessionSchema(BaseModel):
                     return None
             return None
 
-        created_at = parse_datetime(session.get("created_at", 0))
-        updated_at = parse_datetime(session.get("updated_at", created_at))
+        created_at = to_utc_datetime(session.get("created_at", 0))
+        updated_at = to_utc_datetime(session.get("updated_at", created_at))
         return cls(
             session_id=session.get("session_id", ""),
             session_name=session_name,
@@ -282,8 +281,8 @@ class AgentSessionDetailSchema(BaseModel):
             metrics=session.session_data.get("session_metrics", {}) if session.session_data else None,  # type: ignore
             metadata=session.metadata,
             chat_history=[message.to_dict() for message in session.get_chat_history()],
-            created_at=created_at,
-            updated_at=updated_at,
+            created_at=to_utc_datetime(created_at),
+            updated_at=to_utc_datetime(updated_at),
         )
 
 
@@ -322,8 +321,8 @@ class TeamSessionDetailSchema(BaseModel):
             metrics=session.session_data.get("session_metrics", {}) if session.session_data else None,
             metadata=session.metadata,
             chat_history=[message.to_dict() for message in session.get_chat_history()],
-            created_at=created_at,
-            updated_at=updated_at,
+            created_at=to_utc_datetime(created_at),
+            updated_at=to_utc_datetime(updated_at),
         )
 
 
@@ -339,13 +338,15 @@ class WorkflowSessionDetailSchema(BaseModel):
     workflow_data: Optional[dict] = Field(None, description="Workflow-specific data")
     metadata: Optional[dict] = Field(None, description="Additional metadata")
 
-    created_at: Optional[int] = Field(None, description="Unix timestamp of session creation")
-    updated_at: Optional[int] = Field(None, description="Unix timestamp of last update")
+    created_at: Optional[datetime] = Field(None, description="Session creation timestamp")
+    updated_at: Optional[datetime] = Field(None, description="Last update timestamp")
 
     @classmethod
     def from_session(cls, session: WorkflowSession) -> "WorkflowSessionDetailSchema":
         session_dict = session.to_dict()
         session_name = get_session_name({**session_dict, "session_type": "workflow"})
+        created_at = datetime.fromtimestamp(session.created_at, tz=timezone.utc) if session.created_at else None
+        updated_at = datetime.fromtimestamp(session.updated_at, tz=timezone.utc) if session.updated_at else created_at
         return cls(
             session_id=session.session_id,
             user_id=session.user_id,
@@ -356,8 +357,8 @@ class WorkflowSessionDetailSchema(BaseModel):
             session_state=session.session_data.get("session_state", None) if session.session_data else None,
             workflow_data=session.workflow_data,
             metadata=session.metadata,
-            created_at=session.created_at,
-            updated_at=session.updated_at or session.created_at,
+            created_at=to_utc_datetime(created_at),
+            updated_at=to_utc_datetime(updated_at),
         )
 
 
@@ -418,9 +419,7 @@ class RunSchema(BaseModel):
             files=run_dict.get("files", []),
             response_audio=run_dict.get("response_audio", None),
             input_media=extract_input_media(run_dict),
-            created_at=datetime.fromtimestamp(run_dict.get("created_at", 0), tz=timezone.utc)
-            if run_dict.get("created_at") is not None
-            else None,
+            created_at=to_utc_datetime(run_dict.get("created_at")),
         )
 
 
@@ -468,9 +467,7 @@ class TeamRunSchema(BaseModel):
             messages=[message for message in run_dict.get("messages", [])] if run_dict.get("messages") else None,
             tools=[tool for tool in run_dict.get("tools", [])] if run_dict.get("tools") else None,
             events=[event for event in run_dict["events"]] if run_dict.get("events") else None,
-            created_at=datetime.fromtimestamp(run_dict.get("created_at", 0), tz=timezone.utc)
-            if run_dict.get("created_at") is not None
-            else None,
+            created_at=to_utc_datetime(run_dict.get("created_at")),
             references=run_dict.get("references", []),
             citations=run_dict.get("citations", None),
             reasoning_messages=run_dict.get("reasoning_messages", []),
@@ -496,7 +493,7 @@ class WorkflowRunSchema(BaseModel):
     step_results: Optional[list[dict]] = Field(None, description="Results from each workflow step")
     step_executor_runs: Optional[list[dict]] = Field(None, description="Executor runs for each step")
     metrics: Optional[dict] = Field(None, description="Performance and usage metrics")
-    created_at: Optional[int] = Field(None, description="Unix timestamp of run creation")
+    created_at: Optional[datetime] = Field(None, description="Run creation timestamp")
     reasoning_content: Optional[str] = Field(None, description="Reasoning content if reasoning was enabled")
     reasoning_steps: Optional[List[dict]] = Field(None, description="List of reasoning steps")
     references: Optional[List[dict]] = Field(None, description="References cited in the workflow")
@@ -525,7 +522,7 @@ class WorkflowRunSchema(BaseModel):
             metrics=run_response.get("metrics", {}),
             step_results=run_response.get("step_results", []),
             step_executor_runs=run_response.get("step_executor_runs", []),
-            created_at=run_response["created_at"],
+            created_at=to_utc_datetime(run_response.get("created_at")),
             reasoning_content=run_response.get("reasoning_content", ""),
             reasoning_steps=run_response.get("reasoning_steps", []),
             references=run_response.get("references", []),
@@ -560,3 +557,72 @@ class PaginatedResponse(BaseModel, Generic[T]):
 
     data: List[T] = Field(..., description="List of items for the current page")
     meta: PaginationInfo = Field(..., description="Pagination metadata")
+
+
+class ComponentType(str, Enum):
+    AGENT = "agent"
+    TEAM = "team"
+    WORKFLOW = "workflow"
+
+
+class ComponentCreate(BaseModel):
+    name: str = Field(..., description="Display name")
+    component_id: Optional[str] = Field(
+        None, description="Unique identifier for the entity. Auto-generated from name if not provided."
+    )
+    component_type: ComponentType = Field(..., description="Type of entity: agent, team, or workflow")
+    description: Optional[str] = Field(None, description="Optional description")
+    metadata: Optional[Dict[str, Any]] = Field(None, description="Optional metadata")
+    # Config parameters are optional, but if provided, they will be used to create the initial config
+    config: Optional[Dict[str, Any]] = Field(None, description="Optional configuration")
+    label: Optional[str] = Field(None, description="Optional label (e.g., 'stable')")
+    stage: str = Field("draft", description="Stage: 'draft' or 'published'")
+    notes: Optional[str] = Field(None, description="Optional notes")
+    set_current: bool = Field(True, description="Set as current version")
+
+
+class ComponentResponse(BaseModel):
+    component_id: str
+    component_type: ComponentType
+    name: Optional[str] = None
+    description: Optional[str] = None
+    current_version: Optional[int] = None
+    metadata: Optional[Dict[str, Any]] = None
+    created_at: int
+    updated_at: Optional[int] = None
+
+
+class ConfigCreate(BaseModel):
+    config: Dict[str, Any] = Field(..., description="The configuration data")
+    version: Optional[int] = Field(None, description="Optional version number")
+    label: Optional[str] = Field(None, description="Optional label (e.g., 'stable')")
+    stage: str = Field("draft", description="Stage: 'draft' or 'published'")
+    notes: Optional[str] = Field(None, description="Optional notes")
+    links: Optional[List[Dict[str, Any]]] = Field(None, description="Optional links to child components")
+    set_current: bool = Field(True, description="Set as current version")
+
+
+class ComponentConfigResponse(BaseModel):
+    component_id: str
+    version: int
+    label: Optional[str] = None
+    stage: str
+    config: Dict[str, Any]
+    notes: Optional[str] = None
+    created_at: int
+    updated_at: Optional[int] = None
+
+
+class ComponentUpdate(BaseModel):
+    name: Optional[str] = None
+    description: Optional[str] = None
+    component_type: Optional[str] = None
+    metadata: Optional[Dict[str, Any]] = None
+
+
+class ConfigUpdate(BaseModel):
+    config: Optional[Dict[str, Any]] = None
+    label: Optional[str] = None
+    stage: Optional[str] = None
+    notes: Optional[str] = None
+    links: Optional[List[Dict[str, Any]]] = None

agno/os/utils.py

@@ -14,6 +14,7 @@ from agno.media import Audio, Image, Video
 from agno.media import File as FileMedia
 from agno.models.message import Message
 from agno.os.config import AgentOSConfig
+from agno.registry import Registry
 from agno.remote.base import RemoteDb, RemoteKnowledge
 from agno.run.agent import RunOutputEvent
 from agno.run.team import TeamRunOutputEvent
@@ -24,6 +25,20 @@ from agno.utils.log import log_warning, logger
 from agno.workflow import RemoteWorkflow, Workflow
 
 
+def to_utc_datetime(value: Optional[Union[int, float, datetime]]) -> Optional[datetime]:
+    """Convert a timestamp to a UTC datetime."""
+    if value is None:
+        return None
+
+    if isinstance(value, datetime):
+        # If already a datetime, make sure the timezone is UTC
+        if value.tzinfo is None:
+            return value.replace(tzinfo=timezone.utc)
+        return value
+
+    return datetime.fromtimestamp(value, tz=timezone.utc)
+
+
 async def get_request_kwargs(request: Request, endpoint_func: Callable) -> Dict[str, Any]:
     """Given a Request and an endpoint function, return a dictionary with all extra form data fields.
 
@@ -416,6 +431,9 @@ def extract_format(file: UploadFile) -> Optional[str]:
 def get_agent_by_id(
     agent_id: str,
     agents: Optional[List[Union[Agent, RemoteAgent]]] = None,
+    db: Optional[Union[BaseDb, AsyncBaseDb]] = None,
+    registry: Optional[Registry] = None,
+    version: Optional[int] = None,
     create_fresh: bool = False,
 ) -> Optional[Union[Agent, RemoteAgent]]:
     """Get an agent by ID, optionally creating a fresh instance for request isolation.
@@ -432,14 +450,28 @@ def get_agent_by_id(
     Returns:
         The agent instance (shared or fresh copy based on create_fresh)
     """
-    if agent_id is None or agents is None:
+    if agent_id is None:
         return None
 
-    for agent in agents:
-        if agent.id == agent_id:
-            if create_fresh and isinstance(agent, Agent):
-                return agent.deep_copy()
-            return agent
+    # Try to get the agent from the list of agents
+    if agents:
+        for agent in agents:
+            if agent.id == agent_id:
+                if create_fresh and isinstance(agent, Agent):
+                    return agent.deep_copy()
+                return agent
+
+    # Try to get the agent from the database
+    if db and isinstance(db, BaseDb):
+        from agno.agent.agent import get_agent_by_id as get_agent_by_id_db
+
+        try:
+            db_agent = get_agent_by_id_db(db=db, id=agent_id, version=version, registry=registry)
+            return db_agent
+        except Exception as e:
+            logger.error(f"Error getting agent {agent_id} from database: {e}")
+            return None
+
     return None
 
 
@@ -447,6 +479,9 @@ def get_team_by_id(
     team_id: str,
     teams: Optional[List[Union[Team, RemoteTeam]]] = None,
     create_fresh: bool = False,
+    db: Optional[Union[BaseDb, AsyncBaseDb]] = None,
+    version: Optional[int] = None,
+    registry: Optional[Registry] = None,
 ) -> Optional[Union[Team, RemoteTeam]]:
     """Get a team by ID, optionally creating a fresh instance for request isolation.
 
@@ -461,14 +496,26 @@ def get_team_by_id(
     Returns:
         The team instance (shared or fresh copy based on create_fresh)
     """
-    if team_id is None or teams is None:
+    if team_id is None:
         return None
 
-    for team in teams:
-        if team.id == team_id:
-            if create_fresh and isinstance(team, Team):
-                return team.deep_copy()
-            return team
+    if teams:
+        for team in teams:
+            if team.id == team_id:
+                if create_fresh and isinstance(team, Team):
+                    return team.deep_copy()
+                return team
+
+    if db and isinstance(db, BaseDb):
+        from agno.team.team import get_team_by_id as get_team_by_id_db
+
+        try:
+            db_team = get_team_by_id_db(db=db, id=team_id, version=version, registry=registry)
+            return db_team
+        except Exception as e:
+            logger.error(f"Error getting team {team_id} from database: {e}")
+            return None
+
     return None
 
 
@@ -476,6 +523,9 @@ def get_workflow_by_id(
     workflow_id: str,
     workflows: Optional[List[Union[Workflow, RemoteWorkflow]]] = None,
     create_fresh: bool = False,
+    db: Optional[Union[BaseDb, AsyncBaseDb]] = None,
+    version: Optional[int] = None,
+    registry: Optional[Registry] = None,
 ) -> Optional[Union[Workflow, RemoteWorkflow]]:
     """Get a workflow by ID, optionally creating a fresh instance for request isolation.
 
@@ -486,18 +536,33 @@ def get_workflow_by_id(
         workflow_id: The workflow ID to look up
         workflows: List of workflows to search
         create_fresh: If True, creates a new instance using deep_copy()
+        db: Optional database interface
+        version: Workflow version, if needed
+        registry: Optional Registry instance
 
     Returns:
         The workflow instance (shared or fresh copy based on create_fresh)
     """
-    if workflow_id is None or workflows is None:
+    if workflow_id is None:
         return None
 
-    for workflow in workflows:
-        if workflow.id == workflow_id:
-            if create_fresh and isinstance(workflow, Workflow):
-                return workflow.deep_copy()
-            return workflow
+    if workflows:
+        for workflow in workflows:
+            if workflow.id == workflow_id:
+                if create_fresh and isinstance(workflow, Workflow):
+                    return workflow.deep_copy()
+                return workflow
+
+    if db and isinstance(db, BaseDb):
+        from agno.workflow.workflow import get_workflow_by_id as get_workflow_by_id_db
+
+        try:
+            db_workflow = get_workflow_by_id_db(db=db, id=workflow_id, version=version, registry=registry)
+            return db_workflow
+        except Exception as e:
+            logger.error(f"Error getting workflow {workflow_id} from database: {e}")
+            return None
+
     return None
 
 
@@ -871,7 +936,7 @@ def format_duration_ms(duration_ms: Optional[int]) -> str:
     return f"{duration_ms / 1000:.2f}s"
 
 
-def parse_datetime_to_utc(datetime_str: str, param_name: str = "datetime") -> "datetime":
+def timestamp_to_datetime(datetime_str: str, param_name: str = "datetime") -> "datetime":
     """Parse an ISO 8601 datetime string and convert to UTC.
 
     Args:

agno/reasoning/anthropic.py

@@ -69,7 +69,7 @@ def get_anthropic_reasoning_stream(
     redacted_reasoning_content: Optional[str] = None
 
     try:
-        for event in reasoning_agent.run(input=messages, stream=True, stream_intermediate_steps=True):
+        for event in reasoning_agent.run(input=messages, stream=True, stream_events=True):
             if hasattr(event, "event"):
                 if event.event == RunEvent.run_content:
                     # Stream reasoning content as it arrives
@@ -140,7 +140,7 @@ async def aget_anthropic_reasoning_stream(
     redacted_reasoning_content: Optional[str] = None
 
     try:
-        async for event in reasoning_agent.arun(input=messages, stream=True, stream_intermediate_steps=True):
+        async for event in reasoning_agent.arun(input=messages, stream=True, stream_events=True):
             if hasattr(event, "event"):
                 if event.event == RunEvent.run_content:
                     # Stream reasoning content as it arrives

agno/reasoning/azure_ai_foundry.py

@@ -86,7 +86,7 @@ def get_ai_foundry_reasoning_stream(
     reasoning_content: str = ""
 
     try:
-        for event in reasoning_agent.run(input=messages, stream=True, stream_intermediate_steps=True):
+        for event in reasoning_agent.run(input=messages, stream=True, stream_events=True):
             if hasattr(event, "event"):
                 if event.event == RunEvent.run_content:
                     # Check for reasoning_content attribute first (native reasoning)
@@ -132,7 +132,7 @@ async def aget_ai_foundry_reasoning_stream(
     reasoning_content: str = ""
 
     try:
-        async for event in reasoning_agent.arun(input=messages, stream=True, stream_intermediate_steps=True):
+        async for event in reasoning_agent.arun(input=messages, stream=True, stream_events=True):
             if hasattr(event, "event"):
                 if event.event == RunEvent.run_content:
                     # Check for reasoning_content attribute first (native reasoning)

agno/reasoning/deepseek.py

@@ -69,7 +69,7 @@ def get_deepseek_reasoning_stream(
     reasoning_content: str = ""
 
     try:
-        for event in reasoning_agent.run(input=messages, stream=True, stream_intermediate_steps=True):
+        for event in reasoning_agent.run(input=messages, stream=True, stream_events=True):
             if hasattr(event, "event"):
                 if event.event == RunEvent.run_content:
                     # Stream reasoning content as it arrives
@@ -140,7 +140,7 @@ async def aget_deepseek_reasoning_stream(
     reasoning_content: str = ""
 
     try:
-        async for event in reasoning_agent.arun(input=messages, stream=True, stream_intermediate_steps=True):
+        async for event in reasoning_agent.arun(input=messages, stream=True, stream_events=True):
             if hasattr(event, "event"):
                 if event.event == RunEvent.run_content:
                     # Stream reasoning content as it arrives

agno/reasoning/default.py

@@ -1,10 +1,11 @@
 from __future__ import annotations
 
 from textwrap import dedent
-from typing import Any, Callable, Dict, List, Literal, Optional, Union
+from typing import Callable, Dict, List, Literal, Optional, Union
 
 from agno.models.base import Model
 from agno.reasoning.step import ReasoningSteps
+from agno.run.base import RunContext
 from agno.tools import Toolkit
 from agno.tools.function import Function
 
@@ -19,9 +20,7 @@ def get_default_reasoning_agent(
     telemetry: bool = True,
     debug_mode: bool = False,
     debug_level: Literal[1, 2] = 1,
-    session_state: Optional[Dict[str, Any]] = None,
-    dependencies: Optional[Dict[str, Any]] = None,
-    metadata: Optional[Dict[str, Any]] = None,
+    run_context: Optional[RunContext] = None,
 ) -> Optional["Agent"]:  # type: ignore  # noqa: F821
     from agno.agent import Agent
 
@@ -89,9 +88,9 @@ def get_default_reasoning_agent(
         telemetry=telemetry,
         debug_mode=debug_mode,
         debug_level=debug_level,
-        session_state=session_state,
-        dependencies=dependencies,
-        metadata=metadata,
+        session_state=run_context.session_state if run_context else None,
+        dependencies=run_context.dependencies if run_context else None,
+        metadata=run_context.metadata if run_context else None,
     )
 
     return agent

agno/reasoning/gemini.py

@@ -94,7 +94,7 @@ def get_gemini_reasoning_stream(
     reasoning_content: str = ""
 
     try:
-        for event in reasoning_agent.run(input=messages, stream=True, stream_intermediate_steps=True):
+        for event in reasoning_agent.run(input=messages, stream=True, stream_events=True):
             if hasattr(event, "event"):
                 if event.event == RunEvent.run_content:
                     # Stream reasoning content as it arrives
@@ -134,7 +134,7 @@ async def aget_gemini_reasoning_stream(
     reasoning_content: str = ""
 
     try:
-        async for event in reasoning_agent.arun(input=messages, stream=True, stream_intermediate_steps=True):
+        async for event in reasoning_agent.arun(input=messages, stream=True, stream_events=True):
             if hasattr(event, "event"):
                 if event.event == RunEvent.run_content:
                     # Stream reasoning content as it arrives

agno/reasoning/helpers.py

@@ -1,8 +1,9 @@
-from typing import Any, Dict, List, Literal, Optional
+from typing import List, Literal, Optional
 
 from agno.models.base import Model
 from agno.models.message import Message
 from agno.reasoning.step import NextAction, ReasoningStep
+from agno.run.base import RunContext
 from agno.run.messages import RunMessages
 from agno.utils.log import logger
 
@@ -12,9 +13,7 @@ def get_reasoning_agent(
     telemetry: bool = False,
     debug_mode: bool = False,
     debug_level: Literal[1, 2] = 1,
-    session_state: Optional[Dict[str, Any]] = None,
-    dependencies: Optional[Dict[str, Any]] = None,
-    metadata: Optional[Dict[str, Any]] = None,
+    run_context: Optional[RunContext] = None,
 ) -> "Agent":  # type: ignore  # noqa: F821
     from agno.agent import Agent
 
@@ -23,9 +22,9 @@ def get_reasoning_agent(
         telemetry=telemetry,
         debug_mode=debug_mode,
         debug_level=debug_level,
-        session_state=session_state,
-        dependencies=dependencies,
-        metadata=metadata,
+        session_state=run_context.session_state if run_context else None,
+        dependencies=run_context.dependencies if run_context else None,
+        metadata=run_context.metadata if run_context else None,
     )