Rename graph v2 to beta

Author: DouweM

examples/pydantic_ai_examples/dr2/nodes.py

@@ -8,9 +8,9 @@
 from typing_extensions import TypeVar
 
 from pydantic_ai import Agent, models
-from pydantic_graph.v2.id_types import NodeId
-from pydantic_graph.v2.step import StepContext
-from pydantic_graph.v2.util import TypeOrTypeExpression, unpack_type_expression
+from pydantic_graph.beta.id_types import NodeId
+from pydantic_graph.beta.step import StepContext
+from pydantic_graph.beta.util import TypeOrTypeExpression, unpack_type_expression
 
 InputT = TypeVar('InputT', infer_variance=True)
 OutputT = TypeVar('OutputT', infer_variance=True)

examples/pydantic_ai_examples/dr2/plan_outline_graph.py

@@ -25,9 +25,9 @@
 
 from pydantic import BaseModel
 
-from pydantic_graph.v2.graph_builder import GraphBuilder
-from pydantic_graph.v2.step import StepContext
-from pydantic_graph.v2.util import TypeExpression
+from pydantic_graph.beta.graph_builder import GraphBuilder
+from pydantic_graph.beta.step import StepContext
+from pydantic_graph.beta.util import TypeExpression
 
 from .nodes import Interruption, Prompt
 from .shared_types import MessageHistory, Outline

examples/pydantic_ai_examples/temporal_graph.py

@@ -17,14 +17,14 @@
 from typing_extensions import TypeVar
 
 with workflow.unsafe.imports_passed_through():
-    from pydantic_graph.nodes import BaseNode, End, GraphRunContext
-    from pydantic_graph.v2 import (
+    from pydantic_graph.beta import (
         GraphBuilder,
         NullReducer,
         StepContext,
         StepNode,
         TypeExpression,
     )
+    from pydantic_graph.nodes import BaseNode, End, GraphRunContext
 
 T = TypeVar('T', infer_variance=True)
 

pydantic_ai_slim/pydantic_ai/_agent_graph.py

@@ -19,8 +19,8 @@
 from pydantic_ai._utils import is_async_callable, run_in_executor
 from pydantic_ai.builtin_tools import AbstractBuiltinTool
 from pydantic_graph import BaseNode, GraphRunContext
+from pydantic_graph.beta import Graph, GraphBuilder
 from pydantic_graph.nodes import End, NodeRunEndT
-from pydantic_graph.v2 import Graph, GraphBuilder
 
 from . import _output, _system_prompt, exceptions, messages as _messages, models, result, usage as _usage
 from .exceptions import ToolRetryError

pydantic_ai_slim/pydantic_ai/run.py

@@ -7,8 +7,8 @@
 from typing import TYPE_CHECKING, Any, Generic, Literal, overload
 
 from pydantic_graph import BaseNode, End, GraphRunContext
-from pydantic_graph.v2.graph import EndMarker, GraphRun, GraphTask, JoinItem
-from pydantic_graph.v2.step import NodeStep
+from pydantic_graph.beta.graph import EndMarker, GraphRun, GraphTask, JoinItem
+from pydantic_graph.beta.step import NodeStep
 
 from . import (
     _agent_graph,

pydantic_graph/pydantic_graph/v2/__init__.py renamed to pydantic_graph/pydantic_graph/beta/__init__.py

@@ -13,13 +13,13 @@
 
 from typing_extensions import Never, Self, TypeVar
 
-from pydantic_graph.v2.id_types import ForkId, NodeId
-from pydantic_graph.v2.paths import Path, PathBuilder
-from pydantic_graph.v2.step import StepFunction
-from pydantic_graph.v2.util import TypeOrTypeExpression
+from pydantic_graph.beta.id_types import ForkId, NodeId
+from pydantic_graph.beta.paths import Path, PathBuilder
+from pydantic_graph.beta.step import StepFunction
+from pydantic_graph.beta.util import TypeOrTypeExpression
 
 if TYPE_CHECKING:
-    from pydantic_graph.v2.node_types import DestinationNode
+    from pydantic_graph.beta.node_types import DestinationNode
 
 StateT = TypeVar('StateT', infer_variance=True)
 """Type variable for graph state."""

pydantic_graph/pydantic_graph/v2/decision.py renamed to pydantic_graph/pydantic_graph/beta/decision.py

@@ -20,23 +20,30 @@
 
 from pydantic_graph import exceptions
 from pydantic_graph._utils import AbstractSpan, get_traceparent, logfire_span
-from pydantic_graph.nodes import BaseNode, End
-from pydantic_graph.v2.decision import Decision
-from pydantic_graph.v2.id_types import ForkStack, ForkStackItem, GraphRunId, JoinId, NodeId, NodeRunId, TaskId
-from pydantic_graph.v2.join import Join, Reducer
-from pydantic_graph.v2.node import (
+from pydantic_graph.beta.decision import Decision
+from pydantic_graph.beta.id_types import ForkStack, ForkStackItem, GraphRunId, JoinId, NodeId, NodeRunId, TaskId
+from pydantic_graph.beta.join import Join, Reducer
+from pydantic_graph.beta.node import (
     EndNode,
     Fork,
     StartNode,
 )
-from pydantic_graph.v2.node_types import AnyNode
-from pydantic_graph.v2.parent_forks import ParentFork
-from pydantic_graph.v2.paths import BroadcastMarker, DestinationMarker, LabelMarker, Path, SpreadMarker, TransformMarker
-from pydantic_graph.v2.step import NodeStep, Step, StepContext, StepNode
-from pydantic_graph.v2.util import unpack_type_expression
+from pydantic_graph.beta.node_types import AnyNode
+from pydantic_graph.beta.parent_forks import ParentFork
+from pydantic_graph.beta.paths import (
+    BroadcastMarker,
+    DestinationMarker,
+    LabelMarker,
+    Path,
+    SpreadMarker,
+    TransformMarker,
+)
+from pydantic_graph.beta.step import NodeStep, Step, StepContext, StepNode
+from pydantic_graph.beta.util import unpack_type_expression
+from pydantic_graph.nodes import BaseNode, End
 
 if TYPE_CHECKING:
-    from pydantic_graph.v2.mermaid import StateDiagramDirection
+    from pydantic_graph.beta.mermaid import StateDiagramDirection
 
 
 StateT = TypeVar('StateT', infer_variance=True)
@@ -239,7 +246,7 @@ def render(self, *, title: str | None = None, direction: StateDiagramDirection |
         Returns:
             A string containing the Mermaid diagram representation
         """
-        from pydantic_graph.v2.mermaid import build_mermaid_graph
+        from pydantic_graph.beta.mermaid import build_mermaid_graph
 
         return build_mermaid_graph(self).render(title=title, direction=direction)
 

pydantic_graph/pydantic_graph/v2/graph.py renamed to pydantic_graph/pydantic_graph/beta/graph.py

@@ -17,24 +17,23 @@
 from typing_extensions import Never, TypeAliasType, TypeVar
 
 from pydantic_graph import _utils, exceptions
-from pydantic_graph.nodes import BaseNode, End
-from pydantic_graph.v2.decision import Decision, DecisionBranch, DecisionBranchBuilder
-from pydantic_graph.v2.graph import Graph
-from pydantic_graph.v2.id_types import ForkId, JoinId, NodeId
-from pydantic_graph.v2.join import Join, Reducer
-from pydantic_graph.v2.node import (
+from pydantic_graph.beta.decision import Decision, DecisionBranch, DecisionBranchBuilder
+from pydantic_graph.beta.graph import Graph
+from pydantic_graph.beta.id_types import ForkId, JoinId, NodeId
+from pydantic_graph.beta.join import Join, Reducer
+from pydantic_graph.beta.node import (
     EndNode,
     Fork,
     StartNode,
 )
-from pydantic_graph.v2.node_types import (
+from pydantic_graph.beta.node_types import (
     AnyDestinationNode,
     AnyNode,
     DestinationNode,
     SourceNode,
 )
-from pydantic_graph.v2.parent_forks import ParentFork, ParentForkFinder
-from pydantic_graph.v2.paths import (
+from pydantic_graph.beta.parent_forks import ParentFork, ParentForkFinder
+from pydantic_graph.beta.paths import (
     BroadcastMarker,
     DestinationMarker,
     EdgePath,
@@ -43,8 +42,9 @@
     PathBuilder,
     SpreadMarker,
 )
-from pydantic_graph.v2.step import NodeStep, Step, StepFunction, StepNode
-from pydantic_graph.v2.util import TypeOrTypeExpression, get_callable_name, unpack_type_expression
+from pydantic_graph.beta.step import NodeStep, Step, StepFunction, StepNode
+from pydantic_graph.beta.util import TypeOrTypeExpression, get_callable_name, unpack_type_expression
+from pydantic_graph.nodes import BaseNode, End
 
 StateT = TypeVar('StateT', infer_variance=True)
 DepsT = TypeVar('DepsT', infer_variance=True)