class Bridge: def __init__(self, transformers): self.transformers = transformers def process_prompt(self, prompt, generate_fn): # Split the prompt into tokens. tokens = prompt.split() # Determine the workflow. workflow = {} for i, transformer in enumerate(self.transformers): workflow[chr(ord('A') + i)] = generate_fn(transformer, tokens) return workflow def generate(self, workflow): # Generate text for each transformer. for transformer, text in workflow.items(): yield text