Merge branch 'refactor/until-callbacks' into v3

Author: monoxgas

pyproject.toml

@@ -114,7 +114,6 @@ target-version = "py310"
 line-length = 100
 extend-exclude = [
     "*.ipynb", # jupyter notebooks
-    "tests/*"
 ]
 
 [tool.ruff.lint]

rigging/__init__.py

@@ -1,5 +1,13 @@
 from rigging import data, error, generator, logging, model, parsing, watchers
-from rigging.chat import Chat, ChatPipeline, MapChatCallback, ThenChatCallback
+from rigging.chat import (
+    Chat,
+    ChatPipeline,
+    MapChatCallback,
+    PipelineStep,
+    PipelineStepContextManager,
+    PipelineStepGenerator,
+    ThenChatCallback,
+)
 from rigging.completion import (
     Completion,
     CompletionPipeline,
@@ -65,6 +73,9 @@
     "MapChatCallback",
     "ThenCompletionCallback",
     "MapCompletionCallback",
+    "PipelineStep",
+    "PipelineStepGenerator",
+    "PipelineStepContextManager",
     "generator",
     "mcp",
     "robopages",