add graph runner cli

Author: ks6088ts

docs/index.md

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

scripts/draw_mermaid_png.py

@@ -0,0 +1,149 @@
+import logging
+
+import typer
+from dotenv import load_dotenv
+
+from template_langgraph.agents.chat_with_tools_agent.agent import graph as chat_with_tools_agent_graph
+from template_langgraph.agents.issue_formatter_agent.agent import graph as issue_formatter_agent_graph
+from template_langgraph.agents.kabuto_helpdesk_agent import graph as kabuto_helpdesk_agent_graph
+from template_langgraph.agents.task_decomposer_agent.agent import graph as task_decomposer_agent_graph
+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__)
+
+
+def get_agent_graph(name: str):
+    if name == "chat_with_tools_agent":
+        return chat_with_tools_agent_graph
+    elif name == "issue_formatter_agent":
+        return issue_formatter_agent_graph
+    elif name == "task_decomposer_agent":
+        return task_decomposer_agent_graph
+    elif name == "kabuto_helpdesk_agent":
+        return kabuto_helpdesk_agent_graph
+    else:
+        raise ValueError(f"Unknown agent name: {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}")
+    typer.echo(f"Drawing agent: {name}")
+    get_agent_graph(name).get_graph().draw_mermaid_png(
+        output_file_path=output_file_path,
+    )
+    typer.echo(f"Graph saved to {output_file_path}")
+
+
+@app.command()
+def run(
+    name: str = typer.Option(
+        "chat_with_tools_agent",
+        "--name",
+        "-n",
+        help="Name of the agent to draw",
+    ),
+    question: str = typer.Option(
+        "「鬼灯」を実行すると、KABUTOが急に停止します。原因と対策を教えてください。",
+        "--question",
+        "-q",
+        help="Question to ask the agent",
+    ),
+    verbose: bool = typer.Option(
+        False,
+        "--verbose",
+        "-v",
+        help="Enable verbose output",
+    ),
+):
+    # Set up logging
+    if verbose:
+        logger.setLevel(logging.DEBUG)
+
+    if name == "chat_with_tools_agent":
+        from template_langgraph.agents.chat_with_tools_agent.agent import (
+            AgentState,
+        )
+
+        for event in chat_with_tools_agent_graph.stream(
+            input=AgentState(
+                messages=[
+                    {
+                        "role": "user",
+                        "content": question,
+                    }
+                ],
+            )
+        ):
+            logger.info("-" * 20)
+            logger.info(f"Event: {event}")
+
+    if name == "issue_formatter_agent":
+        from template_langgraph.agents.issue_formatter_agent.agent import (
+            AgentState,
+        )
+
+        for event in issue_formatter_agent_graph.stream(
+            input=AgentState(
+                messages=[
+                    {
+                        "role": "user",
+                        "content": question,
+                    }
+                ],
+            )
+        ):
+            logger.info("-" * 20)
+            logger.info(f"Event: {event}")
+
+    if name == "kabuto_helpdesk_agent":
+        from template_langgraph.agents.kabuto_helpdesk_agent import KabutoHelpdeskAgent
+
+        agent = KabutoHelpdeskAgent(
+            tools=None,  # ツールはカスタムせず、デフォルトのツールを使用
+        )
+        response = agent.run(
+            question=question,
+        )
+        logger.info(f"Agent result: {response}")
+
+        # エージェントの応答を表示
+        logger.info(f"Answer: {response['messages'][-1].content}")
+
+
+if __name__ == "__main__":
+    load_dotenv(
+        override=True,
+        verbose=True,
+    )
+    app()