agno 2.2.6__py3-none-any.whl → 2.2.7__py3-none-any.whl

agno/workflow/agent.py

@@ -1,9 +1,10 @@
 """WorkflowAgent - A restricted Agent for workflow orchestration"""
 
-from typing import TYPE_CHECKING, Any, Callable, Dict, Optional
+from typing import TYPE_CHECKING, Any, Callable, Optional
 
 from agno.agent import Agent
 from agno.models.base import Model
+from agno.run import RunContext
 from agno.workflow.types import WebSocketHandler
 
 if TYPE_CHECKING:
@@ -81,7 +82,7 @@ Guidelines:
         workflow: "Any",  # Workflow type
         session: "WorkflowSession",
         execution_input: "WorkflowExecutionInput",
-        session_state: Optional[Dict[str, Any]],
+        run_context: RunContext,
         stream: bool = False,
     ) -> Callable:
         """
@@ -91,7 +92,7 @@ Guidelines:
             workflow: The workflow instance
             session: The workflow session
             execution_input: The execution input
-            session_state: The session state
+            run_context: The run context
             stream: Whether to stream the workflow execution
         Returns:
             Callable tool function
@@ -150,9 +151,9 @@ Guidelines:
                 final_content = ""
                 for event in workflow._execute_stream(
                     session=session_from_db,
+                    run_context=run_context,
                     execution_input=workflow_execution_input,
                     workflow_run_response=workflow_run_response,
-                    session_state=session_state,
                     stream_events=True,
                 ):
                     yield event
@@ -170,7 +171,7 @@ Guidelines:
                     session=session_from_db,
                     execution_input=workflow_execution_input,
                     workflow_run_response=workflow_run_response,
-                    session_state=session_state,
+                    run_context=run_context,
                 )
 
                 if isinstance(result.content, str):
@@ -187,7 +188,7 @@ Guidelines:
         workflow: "Any",  # Workflow type
         session: "WorkflowSession",
         execution_input: "WorkflowExecutionInput",
-        session_state: Optional[Dict[str, Any]],
+        run_context: RunContext,
         stream: bool = False,
         websocket_handler: Optional[WebSocketHandler] = None,
     ) -> Callable:
@@ -199,7 +200,7 @@ Guidelines:
             workflow: The workflow instance
             session: The workflow session
             execution_input: The execution input
-            session_state: The session state
+            run_context: The run context
             stream: Whether to stream the workflow execution
 
         Returns:
@@ -267,7 +268,7 @@ Guidelines:
                     user_id=session_from_db.user_id,
                     execution_input=workflow_execution_input,
                     workflow_run_response=workflow_run_response,
-                    session_state=session_state,
+                    session_state=run_context.session_state,
                     stream_events=True,
                     websocket_handler=websocket_handler,
                 ):
@@ -285,7 +286,7 @@ Guidelines:
                     user_id=session_from_db.user_id,
                     execution_input=workflow_execution_input,
                     workflow_run_response=workflow_run_response,
-                    session_state=session_state,
+                    session_state=run_context.session_state,
                 )
 
                 if isinstance(result.content, str):

agno/workflow/step.py

@@ -10,6 +10,7 @@ from typing_extensions import TypeGuard
 from agno.agent import Agent
 from agno.media import Audio, Image, Video
 from agno.models.metrics import Metrics
+from agno.run import RunContext
 from agno.run.agent import RunOutput
 from agno.run.team import TeamRunOutput
 from agno.run.workflow import (
@@ -210,6 +211,7 @@ class Step:
         session_id: Optional[str] = None,
         user_id: Optional[str] = None,
         workflow_run_response: Optional["WorkflowRunOutput"] = None,
+        run_context: Optional[RunContext] = None,
         session_state: Optional[Dict[str, Any]] = None,
         store_executor_outputs: bool = True,
         workflow_session: Optional[WorkflowSession] = None,
@@ -224,7 +226,11 @@ class Step:
 
         if workflow_session:
             step_input.workflow_session = workflow_session
-        session_state_copy = copy(session_state) if session_state is not None else {}
+
+        if run_context is not None and run_context.session_state is not None:
+            session_state_copy = copy(run_context.session_state)
+        else:
+            session_state_copy = copy(session_state) if session_state is not None else {}
 
         # Execute with retries
         for attempt in range(self.max_retries + 1):
@@ -407,6 +413,7 @@ class Step:
         stream_intermediate_steps: bool = False,
         stream_executor_events: bool = True,
         workflow_run_response: Optional["WorkflowRunOutput"] = None,
+        run_context: Optional[RunContext] = None,
         session_state: Optional[Dict[str, Any]] = None,
         step_index: Optional[Union[int, tuple]] = None,
         store_executor_outputs: bool = True,
@@ -422,8 +429,12 @@ class Step:
 
         if workflow_session:
             step_input.workflow_session = workflow_session
+
         # Create session_state copy once to avoid duplication
-        session_state_copy = copy(session_state) if session_state is not None else {}
+        if run_context is not None and run_context.session_state is not None:
+            session_state_copy = copy(run_context.session_state)
+        else:
+            session_state_copy = copy(session_state) if session_state is not None else {}
 
         # Considering both stream_events and stream_intermediate_steps (deprecated)
         stream_events = stream_events or stream_intermediate_steps

agno/workflow/workflow.py

@@ -29,8 +29,8 @@ 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 import RunContext, RunStatus
 from agno.run.agent import RunContentEvent, RunEvent, RunOutput
-from agno.run.base import RunStatus
 from agno.run.cancel import (
     cancel_run as cancel_run_global,
 )
@@ -1216,7 +1216,7 @@ class Workflow:
         session: WorkflowSession,
         execution_input: WorkflowExecutionInput,
         workflow_run_response: WorkflowRunOutput,
-        session_state: Optional[Dict[str, Any]] = None,
+        run_context: RunContext,
         **kwargs: Any,
     ) -> WorkflowRunOutput:
         """Execute a specific pipeline by name synchronously"""
@@ -1283,7 +1283,7 @@ class Workflow:
                         session_id=session.session_id,
                         user_id=self.user_id,
                         workflow_run_response=workflow_run_response,
-                        session_state=session_state,
+                        session_state=run_context.session_state,
                         store_executor_outputs=self.store_executor_outputs,
                         workflow_session=session,
                         add_workflow_history_to_steps=self.add_workflow_history_to_steps
@@ -1377,7 +1377,7 @@ class Workflow:
         session: WorkflowSession,
         execution_input: WorkflowExecutionInput,
         workflow_run_response: WorkflowRunOutput,
-        session_state: Optional[Dict[str, Any]] = None,
+        run_context: RunContext,
         stream_events: bool = False,
         **kwargs: Any,
     ) -> Iterator[WorkflowRunOutputEvent]:
@@ -1468,7 +1468,7 @@ class Workflow:
                         stream_events=stream_events,
                         stream_executor_events=self.stream_executor_events,
                         workflow_run_response=workflow_run_response,
-                        session_state=session_state,
+                        session_state=run_context.session_state,
                         step_index=i,
                         store_executor_outputs=self.store_executor_outputs,
                         workflow_session=session,
@@ -2267,6 +2267,13 @@ class Workflow:
             session_id=session_id, user_id=user_id, session_state=session_state
         )
 
+        run_context = RunContext(
+            run_id=run_id,
+            session_id=session_id,
+            user_id=user_id,
+            session_state=session_state,
+        )
+
         self._prepare_steps()
 
         # Create workflow run response with PENDING status
@@ -2312,10 +2319,8 @@ class Workflow:
                 if self.agent is not None:
                     self._aexecute_workflow_agent(
                         user_input=input,  # type: ignore
-                        session_id=session_id,
-                        user_id=user_id,
                         execution_input=inputs,
-                        session_state=session_state,
+                        run_context=run_context,
                         stream=False,
                         **kwargs,
                     )
@@ -2375,6 +2380,13 @@ class Workflow:
             session_id=session_id, user_id=user_id, session_state=session_state
         )
 
+        run_context = RunContext(
+            run_id=run_id,
+            session_id=session_id,
+            user_id=user_id,
+            session_state=session_state,
+        )
+
         self._prepare_steps()
 
         # Create workflow run response with PENDING status
@@ -2406,10 +2418,8 @@ class Workflow:
                 if self.agent is not None:
                     result = self._aexecute_workflow_agent(
                         user_input=input,  # type: ignore
-                        session_id=session_id,
-                        user_id=user_id,
+                        run_context=run_context,
                         execution_input=inputs,
-                        session_state=session_state,
                         stream=True,
                         websocket_handler=websocket_handler,
                         **kwargs,
@@ -2437,7 +2447,7 @@ class Workflow:
                         execution_input=inputs,
                         workflow_run_response=workflow_run_response,
                         stream_events=stream_events,
-                        session_state=session_state,
+                        run_context=run_context,
                         websocket_handler=websocket_handler,
                         **kwargs,
                     ):
@@ -2491,7 +2501,7 @@ class Workflow:
         self,
         session: WorkflowSession,
         execution_input: WorkflowExecutionInput,
-        session_state: Optional[Dict[str, Any]],
+        run_context: RunContext,
         stream: bool = False,
     ) -> None:
         """Initialize the workflow agent with tools (but NOT context - that's passed per-run)"""
@@ -2501,7 +2511,7 @@ class Workflow:
             workflow=self,
             session=session,
             execution_input=execution_input,
-            session_state=session_state,
+            run_context=run_context,
             stream=stream,
         )
         workflow_tool = Function.from_callable(workflow_tool_func)
@@ -2544,7 +2554,7 @@ class Workflow:
         user_input: Union[str, Dict[str, Any], List[Any], BaseModel],
         session: WorkflowSession,
         execution_input: WorkflowExecutionInput,
-        session_state: Optional[Dict[str, Any]],
+        run_context: RunContext,
         stream: bool = False,
         **kwargs: Any,
     ) -> Union[WorkflowRunOutput, Iterator[WorkflowRunOutputEvent]]:
@@ -2557,7 +2567,7 @@ class Workflow:
             user_input: The user's input
             session: The workflow session
             execution_input: The execution input
-            session_state: The session state
+            run_context: The run context
             stream: Whether to stream the response
             stream_intermediate_steps: Whether to stream intermediate steps
 
@@ -2569,7 +2579,7 @@ class Workflow:
                 agent_input=user_input,
                 session=session,
                 execution_input=execution_input,
-                session_state=session_state,
+                run_context=run_context,
                 stream=stream,
                 **kwargs,
             )
@@ -2578,7 +2588,7 @@ class Workflow:
                 agent_input=user_input,
                 session=session,
                 execution_input=execution_input,
-                session_state=session_state,
+                run_context=run_context,
                 stream=stream,
             )
 
@@ -2587,7 +2597,7 @@ class Workflow:
         agent_input: Union[str, Dict[str, Any], List[Any], BaseModel],
         session: WorkflowSession,
         execution_input: WorkflowExecutionInput,
-        session_state: Optional[Dict[str, Any]],
+        run_context: RunContext,
         stream: bool = False,
         **kwargs: Any,
     ) -> Iterator[WorkflowRunOutputEvent]:
@@ -2606,10 +2616,10 @@ class Workflow:
         from agno.run.workflow import WorkflowCompletedEvent, WorkflowRunOutputEvent
 
         # Initialize agent with stream_intermediate_steps=True so tool yields events
-        self._initialize_workflow_agent(session, execution_input, session_state, stream=stream)
+        self._initialize_workflow_agent(session, execution_input, run_context=run_context, stream=stream)
 
         # Build dependencies with workflow context
-        dependencies = self._get_workflow_agent_dependencies(session)
+        run_context.dependencies = self._get_workflow_agent_dependencies(session)
 
         # Run agent with streaming - workflow events will bubble up from the tool
         agent_response: Optional[RunOutput] = None
@@ -2647,7 +2657,8 @@ class Workflow:
             stream_intermediate_steps=True,
             yield_run_response=True,
             session_id=session.session_id,
-            dependencies=dependencies,  # Pass context dynamically per-run
+            dependencies=run_context.dependencies,  # Pass context dynamically per-run
+            session_state=run_context.session_state,  # Pass session state dynamically per-run
         ):  # type: ignore
             if isinstance(event, tuple(get_args(WorkflowRunOutputEvent))):
                 yield event  # type: ignore[misc]
@@ -2751,7 +2762,7 @@ class Workflow:
         agent_input: Union[str, Dict[str, Any], List[Any], BaseModel],
         session: WorkflowSession,
         execution_input: WorkflowExecutionInput,
-        session_state: Optional[Dict[str, Any]],
+        run_context: RunContext,
         stream: bool = False,
     ) -> WorkflowRunOutput:
         """
@@ -2764,16 +2775,17 @@ class Workflow:
         """
 
         # Initialize the agent
-        self._initialize_workflow_agent(session, execution_input, session_state, stream=stream)
+        self._initialize_workflow_agent(session, execution_input, run_context=run_context, stream=stream)
 
         # Build dependencies with workflow context
-        dependencies = self._get_workflow_agent_dependencies(session)
+        run_context.dependencies = self._get_workflow_agent_dependencies(session)
 
         # Run the agent
         agent_response: RunOutput = self.agent.run(  # type: ignore[union-attr]
             input=agent_input,
             session_id=session.session_id,
-            dependencies=dependencies,
+            dependencies=run_context.dependencies,
+            session_state=run_context.session_state,
             stream=stream,
         )  # type: ignore
 
@@ -2865,7 +2877,7 @@ class Workflow:
         self,
         session: WorkflowSession,
         execution_input: WorkflowExecutionInput,
-        session_state: Optional[Dict[str, Any]],
+        run_context: RunContext,
         websocket_handler: Optional[WebSocketHandler] = None,
         stream: bool = False,
     ) -> None:
@@ -2876,7 +2888,7 @@ class Workflow:
             workflow=self,
             session=session,
             execution_input=execution_input,
-            session_state=session_state,
+            run_context=run_context,
             stream=stream,
             websocket_handler=websocket_handler,
         )
@@ -2899,10 +2911,8 @@ class Workflow:
     def _aexecute_workflow_agent(
         self,
         user_input: Union[str, Dict[str, Any], List[Any], BaseModel],
-        session_id: str,
-        user_id: Optional[str],
+        run_context: RunContext,
         execution_input: WorkflowExecutionInput,
-        session_state: Optional[Dict[str, Any]],
         stream: bool = False,
         websocket_handler: Optional[WebSocketHandler] = None,
         **kwargs: Any,
@@ -2914,27 +2924,27 @@ class Workflow:
 
         Args:
             user_input: The user's input
-            session_id: The workflow session ID
-            user_id: The user ID
+            session: The workflow session
+            run_context: The run context
             execution_input: The execution input
-            session_state: The session state
             stream: Whether to stream the response
             websocket_handler: The WebSocket handler
 
         Returns:
             Coroutine[WorkflowRunOutput] if stream=False, AsyncIterator[WorkflowRunOutputEvent] if stream=True
         """
+
         if stream:
 
             async def _stream():
                 session, session_state_loaded = await self._aload_session_for_workflow_agent(
-                    session_id, user_id, session_state
+                    run_context.session_id, run_context.user_id, run_context.session_state
                 )
                 async for event in self._arun_workflow_agent_stream(
                     agent_input=user_input,
                     session=session,
                     execution_input=execution_input,
-                    session_state=session_state_loaded,
+                    run_context=run_context,
                     stream=stream,
                     websocket_handler=websocket_handler,
                     **kwargs,
@@ -2946,13 +2956,13 @@ class Workflow:
 
             async def _execute():
                 session, session_state_loaded = await self._aload_session_for_workflow_agent(
-                    session_id, user_id, session_state
+                    run_context.session_id, run_context.user_id, run_context.session_state
                 )
                 return await self._arun_workflow_agent(
                     agent_input=user_input,
                     session=session,
                     execution_input=execution_input,
-                    session_state=session_state_loaded,
+                    run_context=run_context,
                     stream=stream,
                 )
 
@@ -2963,7 +2973,7 @@ class Workflow:
         agent_input: Union[str, Dict[str, Any], List[Any], BaseModel],
         session: WorkflowSession,
         execution_input: WorkflowExecutionInput,
-        session_state: Optional[Dict[str, Any]],
+        run_context: RunContext,
         stream: bool = False,
         websocket_handler: Optional[WebSocketHandler] = None,
         **kwargs: Any,
@@ -2986,10 +2996,14 @@ class Workflow:
         log_debug(f"User input: {agent_input}")
 
         self._async_initialize_workflow_agent(
-            session, execution_input, session_state, stream=stream, websocket_handler=websocket_handler
+            session,
+            execution_input,
+            run_context=run_context,
+            stream=stream,
+            websocket_handler=websocket_handler,
         )
 
-        dependencies = self._get_workflow_agent_dependencies(session)
+        run_context.dependencies = self._get_workflow_agent_dependencies(session)
 
         agent_response: Optional[RunOutput] = None
         workflow_executed = False
@@ -3027,7 +3041,8 @@ class Workflow:
             stream_intermediate_steps=True,
             yield_run_response=True,
             session_id=session.session_id,
-            dependencies=dependencies,  # Pass context dynamically per-run
+            dependencies=run_context.dependencies,  # Pass context dynamically per-run
+            session_state=run_context.session_state,  # Pass session state dynamically per-run
         ):  # type: ignore
             if isinstance(event, tuple(get_args(WorkflowRunOutputEvent))):
                 yield event  # type: ignore[misc]
@@ -3150,7 +3165,7 @@ class Workflow:
         agent_input: Union[str, Dict[str, Any], List[Any], BaseModel],
         session: WorkflowSession,
         execution_input: WorkflowExecutionInput,
-        session_state: Optional[Dict[str, Any]],
+        run_context: RunContext,
         stream: bool = False,
     ) -> WorkflowRunOutput:
         """
@@ -3162,16 +3177,17 @@ class Workflow:
             WorkflowRunOutput: The workflow run output with agent response
         """
         # Initialize the agent
-        self._async_initialize_workflow_agent(session, execution_input, session_state, stream=stream)
+        self._async_initialize_workflow_agent(session, execution_input, run_context=run_context, stream=stream)
 
         # Build dependencies with workflow context
-        dependencies = self._get_workflow_agent_dependencies(session)
+        run_context.dependencies = self._get_workflow_agent_dependencies(session)
 
         # Run the agent
         agent_response: RunOutput = await self.agent.arun(  # type: ignore[union-attr]
             input=agent_input,
             session_id=session.session_id,
-            dependencies=dependencies,
+            dependencies=run_context.dependencies,
+            session_state=run_context.session_state,
             stream=stream,
         )  # type: ignore
 
@@ -3366,6 +3382,14 @@ class Workflow:
         # Update session state from DB
         session_state = self._load_session_state(session=workflow_session, session_state=session_state)
 
+        # Initialize run context
+        run_context = RunContext(
+            run_id=run_id,
+            session_id=session_id,
+            user_id=user_id,
+            session_state=session_state,
+        )
+
         log_debug(f"Workflow Run Start: {self.name}", center=True)
 
         # Use simple defaults
@@ -3404,7 +3428,7 @@ class Workflow:
                 user_input=input,  # type: ignore
                 session=workflow_session,
                 execution_input=inputs,
-                session_state=session_state,
+                run_context=run_context,
                 stream=stream,
                 **kwargs,
             )
@@ -3425,7 +3449,7 @@ class Workflow:
                 execution_input=inputs,  # type: ignore[arg-type]
                 workflow_run_response=workflow_run_response,
                 stream_events=stream_events,
-                session_state=session_state,
+                run_context=run_context,
                 **kwargs,
             )
         else:
@@ -3433,7 +3457,7 @@ class Workflow:
                 session=workflow_session,
                 execution_input=inputs,  # type: ignore[arg-type]
                 workflow_run_response=workflow_run_response,
-                session_state=session_state,
+                run_context=run_context,
                 **kwargs,
             )
 
@@ -3548,6 +3572,14 @@ class Workflow:
         self.initialize_workflow()
         session_id, user_id = self._initialize_session(session_id=session_id, user_id=user_id)
 
+        # Initialize run context
+        run_context = RunContext(
+            run_id=run_id,
+            session_id=session_id,
+            user_id=user_id,
+            session_state=session_state,
+        )
+
         log_debug(f"Async Workflow Run Start: {self.name}", center=True)
 
         # Use simple defaults
@@ -3582,10 +3614,8 @@ class Workflow:
         if self.agent is not None:
             return self._aexecute_workflow_agent(  # type: ignore
                 user_input=input,  # type: ignore
-                session_id=session_id,
-                user_id=user_id,
                 execution_input=inputs,
-                session_state=session_state,
+                run_context=run_context,
                 stream=stream,
                 **kwargs,
             )
@@ -3687,6 +3717,7 @@ class Workflow:
             audio: Audio input
             images: Image input
             videos: Video input
+            files: File input
             stream: Whether to stream the response content
             stream_events: Whether to stream intermediate steps
             markdown: Whether to render content as markdown
@@ -3780,6 +3811,7 @@ class Workflow:
             audio: Audio input
             images: Image input
             videos: Video input
+            files: Files input
             stream: Whether to stream the response content
             stream_events: Whether to stream intermediate steps
             markdown: Whether to render content as markdown
@@ -3886,7 +3918,7 @@ class Workflow:
                 step_dict["team"] = step.team if hasattr(step, "team") else None  # type: ignore
 
             # Handle nested steps for Router/Loop
-            if isinstance(step, (Router)):
+            if isinstance(step, Router):
                 step_dict["steps"] = (
                     [serialize_step(step) for step in step.choices] if hasattr(step, "choices") else None
                 )

{agno-2.2.6.dist-info → agno-2.2.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agno
-Version: 2.2.6
+Version: 2.2.7
 Summary: Agno: a lightweight library for building Multi-Agent Systems
 Author-email: Ashpreet Bedi <ashpreet@agno.com>
 Project-URL: homepage, https://agno.com
@@ -35,12 +35,12 @@ Requires-Dist: rich
 Requires-Dist: typer
 Requires-Dist: typing-extensions
 Provides-Extra: dev
-Requires-Dist: mypy; extra == "dev"
+Requires-Dist: mypy==1.18.2; extra == "dev"
+Requires-Dist: ruff==0.14.3; extra == "dev"
 Requires-Dist: pytest; extra == "dev"
 Requires-Dist: pytest-asyncio; extra == "dev"
 Requires-Dist: pytest-cov; extra == "dev"
 Requires-Dist: pytest-mock; extra == "dev"
-Requires-Dist: ruff; extra == "dev"
 Requires-Dist: timeout-decorator; extra == "dev"
 Requires-Dist: types-pyyaml; extra == "dev"
 Requires-Dist: types-aiofiles; extra == "dev"
@@ -206,6 +206,7 @@ Provides-Extra: firestore
 Requires-Dist: google-cloud-firestore; extra == "firestore"
 Provides-Extra: redis
 Requires-Dist: redis; extra == "redis"
+Requires-Dist: redisvl; extra == "redis"
 Provides-Extra: pgvector
 Requires-Dist: pgvector; extra == "pgvector"
 Provides-Extra: chromadb
@@ -261,6 +262,8 @@ Provides-Extra: a2a
 Requires-Dist: a2a-sdk; extra == "a2a"
 Provides-Extra: huggingface
 Requires-Dist: huggingface-hub; extra == "huggingface"
+Provides-Extra: vllm
+Requires-Dist: vllm; extra == "vllm"
 Provides-Extra: performance
 Requires-Dist: memory_profiler; extra == "performance"
 Provides-Extra: cookbooks
@@ -361,6 +364,7 @@ Requires-Dist: agno[markdown]; extra == "knowledge"
 Requires-Dist: agno[chonkie]; extra == "knowledge"
 Provides-Extra: embedders
 Requires-Dist: agno[huggingface]; extra == "embedders"
+Requires-Dist: agno[vllm]; extra == "embedders"
 Provides-Extra: tests
 Requires-Dist: agno[dev]; extra == "tests"
 Requires-Dist: agno[models]; extra == "tests"