From bab4790c1087d1e5501948d65f6a2593503d633e Mon Sep 17 00:00:00 2001 From: Shay Date: Thu, 14 May 2026 13:24:19 -0700 Subject: [PATCH] feat(generate): export SentenceAssembler, SentencePlan, assemble_surface from __init__ --- generate/__init__.py | 4 ++++ 1 file changed, 4 insertions(+) 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", ]