pydantic-graph 1.2.1__py3-none-any.whl → 1.22.0__py3-none-any.whl

pydantic_graph/beta/util.py

@@ -0,0 +1,90 @@
+"""Utility types and functions for type manipulation and introspection.
+
+This module provides helper classes and functions for working with Python's type system,
+including workarounds for type checker limitations and utilities for runtime type inspection.
+"""
+
+from dataclasses import dataclass
+from typing import Any, Generic, cast, get_args, get_origin
+
+from typing_extensions import TypeAliasType, TypeVar
+
+T = TypeVar('T', infer_variance=True)
+"""Generic type variable with inferred variance."""
+
+
+class TypeExpression(Generic[T]):
+    """A workaround for type checker limitations when using complex type expressions.
+
+        This class serves as a wrapper for types that cannot normally be used in positions
+    requiring `type[T]`, such as `Any`, `Union[...]`, or `Literal[...]`. It provides a
+        way to pass these complex type expressions to functions expecting concrete types.
+
+    Example:
+            Instead of `output_type=Union[str, int]` (which may cause type errors),
+            use `output_type=TypeExpression[Union[str, int]]`.
+
+    Note:
+            This is a workaround for the lack of TypeForm in the Python type system.
+    """
+
+    pass
+
+
+TypeOrTypeExpression = TypeAliasType('TypeOrTypeExpression', type[TypeExpression[T]] | type[T], type_params=(T,))
+"""Type alias allowing both direct types and TypeExpression wrappers.
+
+This alias enables functions to accept either regular types (when compatible with type checkers)
+or TypeExpression wrappers for complex type expressions. The correct type should be inferred
+automatically in either case.
+"""
+
+
+def unpack_type_expression(type_: TypeOrTypeExpression[T]) -> type[T]:
+    """Extract the actual type from a TypeExpression wrapper or return the type directly.
+
+    Args:
+        type_: Either a direct type or a TypeExpression wrapper.
+
+    Returns:
+        The unwrapped type, ready for use in runtime type operations.
+    """
+    if get_origin(type_) is TypeExpression:
+        return get_args(type_)[0]
+    return cast(type[T], type_)
+
+
+@dataclass
+class Some(Generic[T]):
+    """Container for explicitly present values in Maybe type pattern.
+
+    This class represents a value that is definitely present, as opposed to None.
+    It's part of the Maybe pattern, similar to Option/Maybe in functional programming,
+    allowing distinction between "no value" (None) and "value is None" (Some(None)).
+    """
+
+    value: T
+    """The wrapped value."""
+
+
+Maybe = TypeAliasType('Maybe', Some[T] | None, type_params=(T,))
+"""Optional-like type that distinguishes between absence and None values.
+
+Unlike Optional[T], Maybe[T] can differentiate between:
+- No value present: represented as None
+- Value is None: represented as Some(None)
+
+This is particularly useful when None is a valid value in your domain.
+"""
+
+
+def get_callable_name(callable_: Any) -> str:
+    """Extract a human-readable name from a callable object.
+
+    Args:
+        callable_: Any callable object (function, method, class, etc.).
+
+    Returns:
+        The callable's __name__ attribute if available, otherwise its string representation.
+    """
+    return getattr(callable_, '__name__', str(callable_))

pydantic_graph/exceptions.py

@@ -15,6 +15,28 @@ class GraphSetupError(TypeError):
         super().__init__(message)
 
 
+class GraphBuildingError(ValueError):
+    """An error raised during graph-building."""
+
+    message: str
+    """The error message."""
+
+    def __init__(self, message: str):
+        self.message = message
+        super().__init__(message)
+
+
+class GraphValidationError(ValueError):
+    """An error raised during graph validation."""
+
+    message: str
+    """The error message."""
+
+    def __init__(self, message: str):
+        self.message = message
+        super().__init__(message)
+
+
 class GraphRuntimeError(RuntimeError):
     """Error caused by an issue during graph execution."""
 

pydantic_graph/graph.py

@@ -238,8 +238,12 @@ class Graph(Generic[StateT, DepsT, RunEndT]):
         with ExitStack() as stack:
             entered_span: AbstractSpan | None = None
             if span is None:
-                if self.auto_instrument:
-                    entered_span = stack.enter_context(logfire_span('run graph {graph.name}', graph=self))
+                if self.auto_instrument:  # pragma: no branch
+                    # Separate variable because we actually don't want logfire's f-string magic here,
+                    # we want the span_name to be preformatted for other backends
+                    # as requested in https://github.com/pydantic/pydantic-ai/issues/3173.
+                    span_name = f'run graph {self.name}'
+                    entered_span = stack.enter_context(logfire_span(span_name, graph=self))
             else:
                 entered_span = stack.enter_context(span)
             traceparent = None if entered_span is None else get_traceparent(entered_span)
@@ -723,8 +727,12 @@ class GraphRun(Generic[StateT, DepsT, RunEndT]):
             raise exceptions.GraphRuntimeError(f'Node `{node}` is not in the graph.')
 
         with ExitStack() as stack:
-            if self.graph.auto_instrument:
-                stack.enter_context(logfire_span('run node {node_id}', node_id=node_id, node=node))
+            if self.graph.auto_instrument:  # pragma: no branch
+                # Separate variable because we actually don't want logfire's f-string magic here,
+                # we want the span_name to be preformatted for other backends
+                # as requested in https://github.com/pydantic/pydantic-ai/issues/3173.
+                span_name = f'run node {node_id}'
+                stack.enter_context(logfire_span(span_name, node_id=node_id, node=node))
 
             async with self.persistence.record_run(node_snapshot_id):
                 ctx = GraphRunContext(state=self.state, deps=self.deps)

pydantic_graph/nodes.py

@@ -28,8 +28,6 @@ DepsT = TypeVar('DepsT', default=None, contravariant=True)
 class GraphRunContext(Generic[StateT, DepsT]):
     """Context for a graph."""
 
-    # TODO: Can we get rid of this struct and just pass both these things around..?
-
     state: StateT
     """The state of the graph."""
     deps: DepsT

pydantic_graph/persistence/in_mem.py

@@ -65,7 +65,7 @@ class SimpleStatePersistence(BaseStatePersistence[StateT, RunEndT]):
         start = perf_counter()
         try:
             yield
-        except Exception:
+        except Exception:  # pragma: no cover
             self.last_snapshot.duration = perf_counter() - start
             self.last_snapshot.status = 'error'
             raise

{pydantic_graph-1.2.1.dist-info → pydantic_graph-1.22.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydantic-graph
-Version: 1.2.1
+Version: 1.22.0
 Summary: Graph and state machine library
 Project-URL: Homepage, https://ai.pydantic.dev/graph/tree/main/pydantic_graph
 Project-URL: Source, https://github.com/pydantic/pydantic-ai

pydantic_graph-1.22.0.dist-info/RECORD

@@ -0,0 +1,28 @@
+pydantic_graph/__init__.py,sha256=qkrSmWLpnNhD7mLBWV70iS46vy2vFiU2zUModG31wPQ,593
+pydantic_graph/_utils.py,sha256=f0B1VIRxfAslj5UIfBVRzXhIJkDXyF8P3B9V22Q7o7U,6782
+pydantic_graph/exceptions.py,sha256=aeaBf2H18dV7YCNZKZmiXiI6Fyys2qQdunZwd7TSCPk,1648
+pydantic_graph/graph.py,sha256=rEm_5PzRs-5k6Y0mmaF5SGhF0wPA2JSclNAEZtBUZpA,33942
+pydantic_graph/mermaid.py,sha256=u8xM8eEAOWV0TkqEAPJJ9jL2XEZnJ_H7yNGhulg7SL4,10045
+pydantic_graph/nodes.py,sha256=CkY3lrC6jqZtzwhSRjFzmM69TdFFFrr58XSDU4THKHA,7450
+pydantic_graph/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pydantic_graph/beta/__init__.py,sha256=VVmbEFaCSXYHwXqS4pANg4B3cn_c86tT62tW_EXcuyw,751
+pydantic_graph/beta/decision.py,sha256=x-Ta549b-j5hyBPUWFdwRQDRaJqnBHF1pfBP9L8I3vI,11239
+pydantic_graph/beta/graph.py,sha256=CBCqWTJ1u7o0k2Rud193IJq2NffydEOJ5UMl0A_tYJs,42790
+pydantic_graph/beta/graph_builder.py,sha256=dCw4LePreagujGNtTdCVZfRVWkCs35MpoPEtAncLo5U,43326
+pydantic_graph/beta/id_types.py,sha256=FZ3rYSubF6g_Ocv0faL3yJsy1lNN9AGZl9f_izvORUg,2814
+pydantic_graph/beta/join.py,sha256=rzCumDX_YgaU_a5bisfbjbbOuI3IwSZsCZs9TC0T9E4,8002
+pydantic_graph/beta/mermaid.py,sha256=Bj8a3CODPcojwT7BnrYqLBKTp0AbA1T3XsTmK2St3v4,7127
+pydantic_graph/beta/node.py,sha256=cTEGKiT3Lutg-PWxBbZDihpnBTVoPMSyCbfB50fjKeY,3071
+pydantic_graph/beta/node_types.py,sha256=Ha1QPbAHqmJ1ARb359b8LOJK-jZDMO_ZyHkYv9Zbglw,3399
+pydantic_graph/beta/parent_forks.py,sha256=lMCT3slwDuZtiLZImqXuW-i0ZftONCWGr7RTpCAe9dY,9691
+pydantic_graph/beta/paths.py,sha256=LkFvgnyNa1lHdFkN83F7Dgsdg9Q2y0zYrLyqprQiQiY,16068
+pydantic_graph/beta/step.py,sha256=n0JstmxM6Z2rCc2EPUrSAp4MS4IjM2mZsE0ymeekzxU,8683
+pydantic_graph/beta/util.py,sha256=F7IkSC0U-tU1yOxncslyOrZ5HlrZIdafBJARsPetIHQ,3153
+pydantic_graph/persistence/__init__.py,sha256=NLBGvUWhem23EdMHHxtX0XgTS2vyixmuWtWmZKj_U58,8968
+pydantic_graph/persistence/_utils.py,sha256=6ySxCc1lFz7bbLUwDLkoZWNqi8VNLBVU4xxJbKI23fQ,2264
+pydantic_graph/persistence/file.py,sha256=XZy295cGc86HfUl_KuB-e7cECZW3bubiEdyJMVQ1OD0,6906
+pydantic_graph/persistence/in_mem.py,sha256=MmahaVpdzmDB30Dm3ZfSCZBqgmx6vH4HXdBaWwVF0K0,6799
+pydantic_graph-1.22.0.dist-info/METADATA,sha256=p67ZgI9FdZRQbiHwV6u0A8K3bbOiB2I2NVru1IH10rA,3895
+pydantic_graph-1.22.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+pydantic_graph-1.22.0.dist-info/licenses/LICENSE,sha256=vA6Jc482lEyBBuGUfD1pYx-cM7jxvLYOxPidZ30t_PQ,1100
+pydantic_graph-1.22.0.dist-info/RECORD,,

pydantic_graph-1.2.1.dist-info/RECORD

@@ -1,15 +0,0 @@
-pydantic_graph/__init__.py,sha256=qkrSmWLpnNhD7mLBWV70iS46vy2vFiU2zUModG31wPQ,593
-pydantic_graph/_utils.py,sha256=Z-CCyFQV47Otpup8G43oWho4W0LJrUSE4dRDCg-jjNo,5453
-pydantic_graph/exceptions.py,sha256=OUz0YaUHG9CqHYvV4uU4mwF-E2GQCLWKjsWZBu3sMe8,1164
-pydantic_graph/graph.py,sha256=nJNUcMQHSj-Yg2d7pgpAxD-3FdVDhtxJ91x_z3UK7xY,33281
-pydantic_graph/mermaid.py,sha256=u8xM8eEAOWV0TkqEAPJJ9jL2XEZnJ_H7yNGhulg7SL4,10045
-pydantic_graph/nodes.py,sha256=FVMc8YtspydzqUNrYeu1xfosjorafZT2OoNJe16ZvbU,7535
-pydantic_graph/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pydantic_graph/persistence/__init__.py,sha256=NLBGvUWhem23EdMHHxtX0XgTS2vyixmuWtWmZKj_U58,8968
-pydantic_graph/persistence/_utils.py,sha256=6ySxCc1lFz7bbLUwDLkoZWNqi8VNLBVU4xxJbKI23fQ,2264
-pydantic_graph/persistence/file.py,sha256=XZy295cGc86HfUl_KuB-e7cECZW3bubiEdyJMVQ1OD0,6906
-pydantic_graph/persistence/in_mem.py,sha256=TmXB9V-gVMz7qIWov-cWzEa1Cw0Tf9TKGti0CB5U_dQ,6779
-pydantic_graph-1.2.1.dist-info/METADATA,sha256=h7wPmdIuGpdhsSCnyCi2-7-toQ2q9TsEi-Awn1FyBVc,3894
-pydantic_graph-1.2.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-pydantic_graph-1.2.1.dist-info/licenses/LICENSE,sha256=vA6Jc482lEyBBuGUfD1pYx-cM7jxvLYOxPidZ30t_PQ,1100
-pydantic_graph-1.2.1.dist-info/RECORD,,