diff --git a/generate/__init__.py b/generate/__init__.py
index c941619f..03207c4b 100644
--- a/generate/__init__.py
+++ b/generate/__init__.py
@@ -14,6 +14,7 @@ from .dialogue import (
     trajectory_blade,
 )
 from .stream import generate, agenerate
+from .surface import SentenceAssembler, SentencePlan, assemble as assemble_surface
 
 __all__ = [
     "DialogueRole",
@@ -29,4 +30,7 @@ __all__ = [
     "propose",
     "propose_dialogue",
     "trajectory_blade",
+    "SentenceAssembler",
+    "SentencePlan",
+    "assemble_surface",
 ]