agentworkflow-viz

Agent workflow graph visualization plugin for LangGraph and other frameworks.

Render live graphs of your agent workflows with a built-in web UI, trace logging, and full customization via Python callbacks or raw HTML/CSS/JS injection.

Installation

pip install agentworkflow-viz

For LangGraph support:

pip install "agentworkflow-viz[langgraph]"

Quick Start

import agentworkflow_viz

viz = agentworkflow_viz.VizPlugin()
viz.start()

# Wrap a LangGraph graph
graph = viz.wrap(my_langgraph_graph)

# Or use the manual tracer
tracer = agentworkflow_viz.VizTracer()
with tracer.span("my_step"):
    do_work()

viz.stop()

Then open http://localhost:8100/graph in your browser.

Customization

Full node customization with Python renderers, HTML templates, and CSS/JS injection:

import agentworkflow_viz

class MyRenderer(agentworkflow_viz.NodeRenderer):
    def render(self, node_id, node_data):
        if node_id == "agent":
            return {
                "icon": "\U0001f916",
                "css_class": "custom-agent",
                "width": 160,
                "height": 60,
            }
        return {}

customization = agentworkflow_viz.VizCustomization(
    node_renderer=MyRenderer(),
    custom_css=".custom-agent { fill: rgba(108,155,255,0.15); stroke: #6c9bff; }",
    title="My Agent Graph",
)

viz = agentworkflow_viz.VizPlugin(config=agentworkflow_viz.VizConfig(customization=customization))
viz.start()

See examples/custom_nodes.py for a full working example.

Configuration

Environment variables:

The default host is loopback (127.0.0.1) so the UI — which can execute injected JS — is not exposed to your network unless you opt in with 0.0.0.0.

Building from source

To set up a dev environment, run the examples, or build the wheel/sdist yourself, see BUILDING.md. Quick version:

python -m venv .venv
# activate it (see BUILDING.md), then:
python -m pip install -e ".[langgraph]"   # editable dev install
python -m pip install --upgrade build twine
python -m build                           # -> dist/*.whl and dist/*.tar.gz
python -m twine check dist/*

License

MIT