diff --git a/src/graph/__init__.py b/src/graph/__init__.py
new file mode 100644
index 0000000..0c44772
--- /dev/null
+++ b/src/graph/__init__.py
@@ -0,0 +1,3 @@
+from src.graph.builder import GraphBuilder, GraphType, NodeType
+
+__all__ = ["GraphBuilder", "GraphType", "NodeType"]