add graph png generator script

Author: ks6088ts

docs/index.md

@@ -26,7 +26,6 @@ uv run python -m template_langgraph.tasks.run_kabuto_helpdesk_agent "KABUTOの
 uv run python -m template_langgraph.tasks.run_kabuto_helpdesk_agent "KABUTOのマニュアルから禅モードに関する情報を教えて下さい"
 
 # ChatWithToolsAgent
-uv run python -m template_langgraph.tasks.draw_chat_with_tools_agent_mermaid_png "data/chat_with_tools_agent.png"
 uv run python -m template_langgraph.tasks.run_chat_with_tools_agent
 # KABUTOの起動時に、画面全体が紫色に点滅し、システムがフリーズします。KABUTO のマニュアルから、関連する情報を取得したり過去のシステムのトラブルシュート事例が蓄積されたデータベースから、関連する情報を取得して質問に答えてください
 # 天狗のいたずら という現象について KABUTO のマニュアルから、関連する情報を取得したり過去のシステムのトラブルシュート事例が蓄積されたデータベースから、関連する情報を取得して質問に答えてください
@@ -35,6 +34,11 @@ uv run python -m template_langgraph.tasks.run_chat_with_tools_agent
 uv run python -m template_langgraph.tasks.run_issue_formatter_agent
 # KABUTOにログインできない！パスワードは合ってるはずなのに…若手社員である山田太郎は、Windows 11 を立ち上げ、日課のように自社の業務システムKABUTOのログイン画面を開きます。しかし、そこには、意味をなさない「虚無」という文字だけがただひっそりと表示されていたのです。これは質問でもあり不具合の報告でもあります。岡本太郎さんに本件調査依頼します。
 
+# Draw mermaid diagram for Agents
+AGENT_NAME=chat_with_tools_agent
+uv run python scripts/draw_mermaid_png.py \
+  --name $AGENT_NAME \
+  --output data/$AGENT_NAME.png
 ```
 
 ## References

scripts/draw_mermaid_png.py

@@ -0,0 +1,65 @@
+import logging
+
+import typer
+from dotenv import load_dotenv
+
+from template_langgraph.loggers import get_logger
+
+# Initialize the Typer application
+app = typer.Typer(
+    add_completion=False,
+    help="template-langgraph CLI",
+)
+
+# Set up logging
+logger = get_logger(__name__)
+
+
+@app.command()
+def png(
+    name: str = typer.Option(
+        "chat_with_tools_agent",
+        "--name",
+        "-n",
+        help="Name of the agent to draw",
+    ),
+    output_file_path: str = typer.Option(
+        "output.png",
+        "--output",
+        "-o",
+        help="Path to the output PNG file",
+    ),
+    verbose: bool = typer.Option(
+        False,
+        "--verbose",
+        "-v",
+        help="Enable verbose output",
+    ),
+):
+    # Set up logging
+    if verbose:
+        logger.setLevel(logging.DEBUG)
+
+    logger.debug(f"This is a debug message with name: {name}")
+    if name == "chat_with_tools_agent":
+        from template_langgraph.agents.chat_with_tools_agent.agent import graph
+    if name == "issue_formatter_agent":
+        from template_langgraph.agents.issue_formatter_agent.agent import graph
+    if name == "task_decomposer_agent":
+        from template_langgraph.agents.task_decomposer_agent.agent import graph
+    if name == "kabuto_helpdesk_agent":
+        from template_langgraph.agents.kabuto_helpdesk_agent import graph
+
+    typer.echo(f"Drawing agent: {name}")
+    graph.get_graph().draw_mermaid_png(
+        output_file_path=output_file_path,
+    )
+    typer.echo(f"Graph saved to {output_file_path}")
+
+
+if __name__ == "__main__":
+    load_dotenv(
+        override=True,
+        verbose=True,
+    )
+    app()

template_langgraph/agents/issue_formatter_agent/agent.py

@@ -37,9 +37,5 @@ def analyze(self, state: AgentState) -> AgentState:
         state["issue"] = issue
         return state
 
-    def draw_mermaid_png(self) -> bytes:
-        """Draw the graph in Mermaid format."""
-        return self.create_graph().get_graph().draw_mermaid_png()
-
 
 graph = IssueFormatterAgent().create_graph()

template_langgraph/agents/task_decomposer_agent/agent.py

@@ -74,9 +74,5 @@ def route_human_feedback(
         logger.info("Looping back to chat for further processing.")
         return "loopback"
 
-    def draw_mermaid_png(self) -> bytes:
-        """Draw the graph in Mermaid format."""
-        return self.create_graph().get_graph().draw_mermaid_png()
-
 
 graph = TaskDecomposerAgent().create_graph()