Pydantic Graph
Graph and finite state machine library.
This library is developed as part of Pydantic AI, however it has no dependency
on pydantic-ai or related packages and can be considered as a pure graph-based state machine library. You may find it useful whether or not you're using Pydantic AI or even building with GenAI.
As with Pydantic AI, this library prioritizes type safety and use of common Python syntax over esoteric, domain-specific use of Python syntax.
pydantic-graph allows you to define graphs using standard Python syntax. In particular, edges are defined using the return type hint of nodes.
Full documentation is available at ai.pydantic.dev/graph.
Here's a basic example:
from __future__ import annotations
from dataclasses import dataclass
from pydantic_graph import BaseNode, End, GraphBuilder, GraphRunContext, StepContext
@dataclass
class DivisibleBy5(BaseNode[None, None, int]):
foo: int
async def run(
self,
ctx: GraphRunContext,
) -> Increment | End[int]:
if self.foo % 5 == 0:
return End(self.foo)
else:
return Increment(self.foo)
@dataclass
class Increment(BaseNode):
foo: int
async def run(self, ctx: GraphRunContext) -> DivisibleBy5:
return DivisibleBy5(self.foo + 1)
g = GraphBuilder(input_type=int, output_type=int)
@g.step
async def start(ctx: StepContext[None, None, int]) -> DivisibleBy5:
return DivisibleBy5(ctx.inputs)
g.add(
g.node(DivisibleBy5),
g.node(Increment),
g.edge_from(g.start_node).to(start),
)
fives_graph = g.build()
async def main():
result = await fives_graph.run(inputs=4)
print(result)
#> 5
1 maintainer
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file pydantic_graph-2.24.0.tar.gz.
File metadata
- Download URL: pydantic_graph-2.24.0.tar.gz
- Upload date:
- Size: 45.2 kB
- Tags: Source
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/6.1.0 CPython/3.13.13
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
04546807cdc5c36793088a3c42dcffd74825e192b06cafd7d43f0726dc8a302e
|
|
| MD5 |
7d0fef2cf13824335572c7ca7cf3afad
|
|
| BLAKE2b-256 |
c6371ace6e245823b2f0387e9f28390e5f1309686b2dc17a799e3a530f986c53
|
Provenance
The following attestation bundles were made for pydantic_graph-2.24.0.tar.gz:
Publisher:
ci.yml on pydantic/pydantic-ai
-
Statement:
-
Statement type:
https://in-toto.io/Statement/v1 -
Predicate type:
https://docs.pypi.org/attestations/publish/v1 -
Subject name:
pydantic_graph-2.24.0.tar.gz -
Subject digest:
04546807cdc5c36793088a3c42dcffd74825e192b06cafd7d43f0726dc8a302e - Sigstore transparency entry: 2341953722
- Sigstore integration time:
-
Permalink:
pydantic/pydantic-ai@7a64d049c3f5271a975cd1d64b2fa876d83ede1d -
Branch / Tag:
refs/tags/v2.24.0 - Owner: https://github.com/pydantic
-
Access:
public
-
Token Issuer:
https://token.actions.githubusercontent.com -
Runner Environment:
github-hosted -
Publication workflow:
ci.yml@7a64d049c3f5271a975cd1d64b2fa876d83ede1d -
Trigger Event:
push
-
Statement type:
File details
Details for the file pydantic_graph-2.24.0-py3-none-any.whl.
File metadata
- Download URL: pydantic_graph-2.24.0-py3-none-any.whl
- Upload date:
- Size: 52.7 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/6.1.0 CPython/3.13.13
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
be32705d3e92fad0c3149f9b4b8708fd2583e1aa02c36c26556538f8ecc1c8de
|
|
| MD5 |
7afb97125718a89ebe944d5b6c5ec896
|
|
| BLAKE2b-256 |
54a30d6eadc5caeb536198f998fd21bd5ade05459edd58057ff4c0322a8ef773
|
Provenance
The following attestation bundles were made for pydantic_graph-2.24.0-py3-none-any.whl:
Publisher:
ci.yml on pydantic/pydantic-ai
-
Statement:
-
Statement type:
https://in-toto.io/Statement/v1 -
Predicate type:
https://docs.pypi.org/attestations/publish/v1 -
Subject name:
pydantic_graph-2.24.0-py3-none-any.whl -
Subject digest:
be32705d3e92fad0c3149f9b4b8708fd2583e1aa02c36c26556538f8ecc1c8de - Sigstore transparency entry: 2341954278
- Sigstore integration time:
-
Permalink:
pydantic/pydantic-ai@7a64d049c3f5271a975cd1d64b2fa876d83ede1d -
Branch / Tag:
refs/tags/v2.24.0 - Owner: https://github.com/pydantic
-
Access:
public
-
Token Issuer:
https://token.actions.githubusercontent.com -
Runner Environment:
github-hosted -
Publication workflow:
ci.yml@7a64d049c3f5271a975cd1d64b2fa876d83ede1d -
Trigger Event:
push
-
Statement type:
