Skip to content

WorkflowVisualizer

llmfy.flow_engine.visualizer.visualizer.WorkflowVisualizer

Source code in llmfy/flow_engine/visualizer/visualizer.py 
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
class WorkflowVisualizer:
    @staticmethod
    def create_mermaid_diagram(workflow) -> str:
        """Generate Mermaid diagram markdown from workflow."""
        nodes = workflow.nodes
        edges = workflow.edges

        # Start building the diagram
        mermaid = [
            "graph TD",
            "    %% Node Styles",
            "    %% Nodes",
        ]

        # Add all nodes
        for name, node in nodes.items():
            # Style nodes based on their type
            if name == START:
                mermaid.append(f"    {name}([{name}])")
                # mermaid.append(f"    class {name} start")
            elif name == END:
                mermaid.append(f"    {name}([{name}])")
                # mermaid.append(f"    class {name} end")
            else:
                mermaid.append(f"    {name}({name})")

        mermaid.append("    %% Edges")

        # Add all edges
        for edge in edges:
            source = edge.source

            if edge.condition is None:
                # Simple edges
                for target in edge.targets:
                    mermaid.append(f"    {source} --> {target}")
            else:
                # Conditional edges
                for target in edge.targets:
                    mermaid.append(f"    {source} -.->|condition| {target}")

        mermaid.append("style __start__ fill:#189C82,stroke:#066F62,color:white")
        mermaid.append("style __end__ fill:#189C82,stroke:#066F62,color:white")

        return "\n".join(mermaid)

    @staticmethod
    def generate_diagram_url(mermaid_code: str) -> str:
        """Generate URL for Mermaid diagram image."""
        encoded = base64.b64encode(mermaid_code.encode()).decode()
        return f"https://mermaid.ink/img/{encoded}"

create_mermaid_diagram(workflow) staticmethod

Generate Mermaid diagram markdown from workflow.

Source code in llmfy/flow_engine/visualizer/visualizer.py 
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
@staticmethod
def create_mermaid_diagram(workflow) -> str:
    """Generate Mermaid diagram markdown from workflow."""
    nodes = workflow.nodes
    edges = workflow.edges

    # Start building the diagram
    mermaid = [
        "graph TD",
        "    %% Node Styles",
        "    %% Nodes",
    ]

    # Add all nodes
    for name, node in nodes.items():
        # Style nodes based on their type
        if name == START:
            mermaid.append(f"    {name}([{name}])")
            # mermaid.append(f"    class {name} start")
        elif name == END:
            mermaid.append(f"    {name}([{name}])")
            # mermaid.append(f"    class {name} end")
        else:
            mermaid.append(f"    {name}({name})")

    mermaid.append("    %% Edges")

    # Add all edges
    for edge in edges:
        source = edge.source

        if edge.condition is None:
            # Simple edges
            for target in edge.targets:
                mermaid.append(f"    {source} --> {target}")
        else:
            # Conditional edges
            for target in edge.targets:
                mermaid.append(f"    {source} -.->|condition| {target}")

    mermaid.append("style __start__ fill:#189C82,stroke:#066F62,color:white")
    mermaid.append("style __end__ fill:#189C82,stroke:#066F62,color:white")

    return "\n".join(mermaid)

generate_diagram_url(mermaid_code) staticmethod

Generate URL for Mermaid diagram image.

Source code in llmfy/flow_engine/visualizer/visualizer.py 
52
53
54
55
56
@staticmethod
def generate_diagram_url(mermaid_code: str) -> str:
    """Generate URL for Mermaid diagram image."""
    encoded = base64.b64encode(mermaid_code.encode()).decode()
    return f"https://mermaid.ink/img/{encoded}"