agentworkflow-viz 0.2.0

Plug-and-play agent graph visualization plugin for LangGraph and other frameworks

agent-viz

Plug-and-play agent 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 agent-viz

For LangGraph support:

pip install "agent-viz[langgraph]"

Quick Start

import agentviz

viz = agentviz.VizPlugin()
viz.start()

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

# Or use the manual tracer
tracer = agentviz.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 agentviz

class MyRenderer(agentviz.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 = agentviz.VizCustomization(
    node_renderer=MyRenderer(),
    custom_css=".custom-agent { fill: rgba(108,155,255,0.15); stroke: #6c9bff; }",
    title="My Agent Graph",
)

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

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

Configuration

Environment variables:

Variable	Default	Description
AGENT_VIZ_ENABLED	true	Enable/disable the plugin
AGENT_VIZ_PORT	8100	Server port
AGENT_VIZ_HOST	0.0.0.0	Server host
AGENT_VIZ_MOUNT	false	Mount mode (no standalone server)

License

MIT