o i @sdZddlmZddlmZddlmZddlmZddl m Z m Z m Z eddd Z ed dd Z ed dd Z Gd d d eeZGdddeeZeGdddeeefZdS)zCore node types for graph construction and execution. This module defines the fundamental node types used to build execution graphs, including start/end nodes and fork nodes for parallel execution. ) annotations) dataclass)Generic)TypeVar)ForkIDJoinIDNodeIDStateTT)infer_varianceOutputTInputTc@seZdZdZedZdS) StartNodezqEntry point node for graph execution. The StartNode represents the beginning of a graph execution flow. __start__N)__name__ __module__ __qualname____doc__ridrr^/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/pydantic_graph/beta/node.pyr sr c@s$eZdZdZedZ d ddZd S) EndNodezTerminal node representing the completion of graph execution. The EndNode marks the successful completion of a graph execution flow and can collect the final output data. __end__inputsr returnNonecCtd)a+Force type variance for proper generic typing. This method exists solely for type checking purposes and should never be called. Args: inputs: Input data of type InputT. Raises: RuntimeError: Always, as this method should never be executed. KThis method should never be called, it is just defined for typing purposes. RuntimeErrorselfrrrr_force_variance.s zEndNode._force_varianceN)rr rr)rrrrrrr!rrrrr$s rc@s:eZdZUdZded< ded< ded< dd d ZdS)ForkaFork node that creates parallel execution branches. A Fork node splits the execution flow into multiple parallel branches, enabling concurrent execution of downstream nodes. It can either map a sequence across multiple branches or duplicate data to each branch. rrboolis_mapz JoinID | Nonedownstream_join_idrr rr cCr)asForce type variance for proper generic typing. This method exists solely for type checking purposes and should never be called. Args: inputs: Input data to be forked. Returns: Output data type (never actually returned). Raises: RuntimeError: Always, as this method should never be executed. rrrrrrr!QszFork._force_varianceN)rr rr )rrrr__annotations__r!rrrrr"<s r"N)r __future__r dataclassesrtypingrtyping_extensionsrpydantic_graph.beta.id_typesrrrr r r r rr"rrrrs