"j" SrSSKJr SSKJrJr SSKJr SSKJ r J r J r J r J r Jr SSKJr SSKJrJrJr SSKJr \"S S S 9r\"S S S 9r\"S S S 9r\"SS S 9r\"SS9"SS\ \\\455r"SS\ \\\\45r"SS\ \\\\45r\\ \ \ \ 4r\"SS9"SS\ \\\\455r\"SS\\\\ 455r "SS\\\\ \\\\ 4\\ -45r!g)zStep-based graph execution components. This module provides the core abstractions for step-based graph execution, including step contexts, step functions, and step nodes that bridge between the declarative `BaseNode` API and the builder graph. ) annotations) AsyncIterator Awaitable) dataclass)AnyGenericProtocolcast get_originoverload)TypeVar)BaseNodeEndGraphRunContext)NodeIDStateTT)infer_varianceDepsTInputTOutputTF)initc\rSrSr%SrS\S'S\S'S\S'SS jr\SS j5r\SS j5r \SS j5r S r g) StepContexta5Context information passed to step functions during graph execution. The step context provides access to the current graph state, dependencies, and input data for a step. Type Parameters: StateT: The type of the graph state DepsT: The type of the dependencies InputT: The type of the input data r_stater_depsr_inputsc(XlX lX0lgN)rrr)selfstatedepsinputss P/home/mande/repo/quber/.venv/lib/python3.13/site-packages/pydantic_graph/step.py__init__StepContext.__init__,s   cUR$r)rr s r$r!StepContext.state1s {{r'cUR$r)rr)s r$r"StepContext.deps5s zzr'cUR$)z[The input data for this step. This must be a property to ensure correct variance behavior )rr)s r$r#StepContext.inputs9s ||r')rrrN)r!rr"rr#r)returnr)r/r)r/r) __name__ __module__ __qualname____firstlineno____doc____annotations__r%propertyr!r"r#__static_attributes__r'r$rrsb N" L% O' r'rc"\rSrSrSrSSjrSrg) StepFunctionBaFProtocol for step functions that can be executed in the graph. Step functions are async callables that receive a step context and return a result. Type Parameters: StateT: The type of the graph state DepsT: The type of the dependencies InputT: The type of the input data OutputT: The type of the output data c[e)zExecute the step function with the given context. Args: ctx: The step context containing state, dependencies, and inputs Returns: An awaitable that resolves to the step's output NotImplementedErrorr ctxs r$__call__StepFunction.__call__Ns "!r'r8N)r@"StepContext[StateT, DepsT, InputT]r/zAwaitable[OutputT]r0r1r2r3r4rAr7r8r'r$r:r:Bs   "r'r:c"\rSrSrSrSSjrSrg)StreamFunctionZaSProtocol for stream functions that can be executed in the graph. Stream functions are async callables that receive a step context and return an async iterator. Type Parameters: StateT: The type of the graph state DepsT: The type of the dependencies InputT: The type of the input data OutputT: The type of the output data c## [e7f)zExecute the stream function with the given context. Args: ctx: The step context containing state, dependencies, and inputs Returns: An async iterator yielding the streamed output r=r?s r$rAStreamFunction.__call__fs "!s r8N)r@rCr/zAsyncIterator[OutputT]rDr8r'r$rFrFZs   r'rFc\rSrSr%SrS\S'S\S'S\S'S S .SS jjr\SS j5r\ SSS jj5r \ SSj5r SSSjjr Sr g )StepwanA step in the graph execution that wraps a step function. Steps represent individual units of execution in the graph, encapsulating a step function along with metadata like ID and label. Type Parameters: StateT: The type of the graph state DepsT: The type of the dependencies InputT: The type of the input data OutputT: The type of the output data rid,StepFunction[StateT, DepsT, InputT, OutputT]_call str | NonelabelN)rQc(XlX lX0lgr)rMrOrQ)r rMcallrQs r$r% Step.__init__s  r'cUR$)zXThe step function to execute. This needs to be a property for proper variance inference.)rOr)s r$rS Step.callszzr'cgrr8r r#s r$as_node Step.as_nodesGJr'cgrr8rXs r$rYrZsBEr'c[X5$)zCreate a step node with bound inputs. Args: inputs: The input data to bind to this step, or None Returns: A [`StepNode`][pydantic_graph.step.StepNode] with this step and the bound inputs )StepNoderXs r$rYrZs%%r')rOrMrQ)rMrrSrNrQrP)r/rNr)r#Noner/StepNode[StateT, DepsT])r#rr/r_)r#z InputT | Noner/r_) r0r1r2r3r4r5r%r6rSr rYr7r8r'r$rKrKwsg  J* 77' 6nr JJ EE & &r'rKc<\rSrSr%SrS\S'S\S'S SjrSrg ) r]aA `BaseNode` that represents a builder step with bound inputs. `StepNode` lets a [`BaseNode`][pydantic_graph.BaseNode] subclass hand off to a builder [`Step`][pydantic_graph.step.Step] by wrapping the step together with the value it should receive as `inputs`. It is not meant to be run directly; returning a `StepNode` from a `BaseNode.run` method tells the graph builder which step to invoke next. zStep[StateT, DepsT, Any, Any]steprr#c # [S5e7f)zAttempt to run the step node. Args: ctx: The graph execution context Returns: The result of step execution Raises: NotImplementedError: Always raised as StepNode is not meant to be run directly z`StepNode` is not meant to be run directly, it is meant to be returned from a `BaseNode` subclass to indicate a transition to a builder step.r=r?s r$run StepNode.runs" \  s r8N)r@zGraphRunContext[StateT, DepsT]r/'BaseNode[StateT, DepsT, Any] | End[Any])r0r1r2r3r4r5rdr7r8r'r$r]r]s (' K( r'r]c\^\rSrSr%SrS\S'SSS.S U4SjjjrS SjrS rU=r $) NodeStepa+A step that wraps a `BaseNode` type for execution by the builder graph. `NodeStep` lets a [`BaseNode`][pydantic_graph.BaseNode] subclass participate as a step in the builder graph. It validates that the input is an instance of the expected node type and runs it with the appropriate graph context. "type[BaseNode[StateT, DepsT, Any]] node_typeN)rMrQc>[TU]U=(d [UR55URUS9 [ U5=(d UUlg)zInitialize a node step. Args: node_type: The BaseNode class this step will execute id: Optional unique identifier, defaults to the node's get_node_id() label: Optional human-readable label for this step )rMrSrQN)superr%r get_node_id _call_noder rk)r rkrMrQ __class__s r$r%NodeStep.__init__sJ 4VI1134  $I.;)r'c># URn[X R5(d[SUSUR35e[ [ [ [[4U5nUR[URURS95IShvN $N7f)a Execute the wrapped node with the step context. Args: ctx: The step context containing the node instance to run Returns: The result of running the node, either another BaseNode or End Raises: ValueError: If the input node is not of the expected type zNode z is not of type )r!r"N) r# isinstancerk ValueErrorr rrrrrdrr!r")r r@nodes r$roNodeStep._call_nodesyzz$//uTF*:4>>:JKL LHVUC/0$7XXoCIICHHMNNNNsBBBB)rk)rkrjrMz NodeID | NonerQrP)r@zStepContext[StateT, DepsT, Any]r/rf) r0r1r2r3r4r5r%ror7 __classcell__)rps@r$rhrhsM21/ ! <5<  <  <<.OOr'rhN)"r4 __future__rcollections.abcrr dataclassesrtypingrrr r r r typing_extensionsr pydantic_graph.basenoderrrpydantic_graph.id_typesrrrrrrr:rFAnyStepFunctionrKr]rhr8r'r$rsX#4!EE%BB* $ /- $ / )D 1 %'&%/0%%P"8FE67:;"0XfeVW<=2sCc12> -&765&'1 2-&-&`  xs*+   D2OtFE31C(Ds3x(OOP2Or'