Doc Pipeline

from typing import Callable, Anyfrom dataclasses import dataclass
@dataclassclass Stage:    name: str    operation: Callable    class Pipeline:    def __init__(self, name: str):        self.name = name        self.stages: list[Stage] = []        def add_stage(self, name: str, operation: Callable):        self.stages.append(Stage(name, operation))        return self  # Fluent API        def run(self, input_data: Any) -> Any:        data = input_data        for stage in self.stages:            print(f"Running stage: {stage.name}")            data = stage.operation(data)        return data
# Example usagepipeline = Pipeline("contract-review")pipeline.add_stage("extract", extract_pdf_text)pipeline.add_stage("analyze", analyze_with_ai)pipeline.add_stage("generate", create_docx_report)
result = pipeline.run("/path/to/contract.pdf")

class ConditionalPipeline(Pipeline):    def add_conditional_stage(self, name: str, condition: Callable,                                if_true: Callable, if_false: Callable):        def conditional_op(data):            if condition(data):                return if_true(data)            return if_false(data)        return self.add_stage(name, conditional_op)
# Usagepipeline.add_conditional_stage(    "ocr_if_needed",    condition=lambda d: d.get("has_images"),    if_true=run_ocr,    if_false=lambda d: d)