Universal AI Workflow
Infrastructure

# Define an AI-powered component with just a decorator
from flowmason_core import node, NodeInput, NodeOutput, Field

@node(name="summarizer", category="reasoning")
class SummarizerNode:
    class Input(NodeInput):
        text: str = Field(description="Text to summarize")
        max_length: int = 200

    class Output(NodeOutput):
        summary: str
        word_count: int

    async def execute(self, input: Input, context) -> Output:
        response = await context.llm.generate(
            prompt=f"Summarize in {input.max_length} chars: {input.text}"
        )
        return self.Output(
            summary=response.text,
            word_count=len(response.text.split())
        )