rename from basic workflow to chat with tools

Author: ks6088ts

docs/index.md

@@ -25,9 +25,9 @@ uv run python -m template_langgraph.tasks.search_documents_on_elasticsearch
 uv run python -m template_langgraph.tasks.run_kabuto_helpdesk_agent "KABUTOの起動時に、画面全体が紫色に点滅し、システムがフリーズします。"
 uv run python -m template_langgraph.tasks.run_kabuto_helpdesk_agent "KABUTOのマニュアルから禅モードに関する情報を教えて下さい"
 
-# BasicWorkflowAgent
-uv run python -m template_langgraph.tasks.draw_basic_workflow_agent_mermaid_png "data/basic_workflow_agent.png"
-uv run python -m template_langgraph.tasks.run_basic_workflow_agent
+# 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 のマニュアルから、関連する情報を取得したり過去のシステムのトラブルシュート事例が蓄積されたデータベースから、関連する情報を取得して質問に答えてください
 

langgraph.json

@@ -3,7 +3,7 @@
     "."
   ],
   "graphs": {
-    "basic_workflow_agent": "template_langgraph.agents.basic_workflow_agent.agent:graph",
+    "chat_with_tools_agent": "template_langgraph.agents.chat_with_tools_agent.agent:graph",
     "kabuto_helpdesk_agent": "template_langgraph.agents.kabuto_helpdesk_agent:graph",
     "issue_formatter_agent": "template_langgraph.agents.issue_formatter_agent.agent:graph"
   },

template_langgraph/agents/basic_workflow_agent/__init__.py renamed to template_langgraph/agents/chat_with_tools_agent/__init__.py

@@ -3,7 +3,7 @@
 from langchain_core.messages import ToolMessage
 from langgraph.graph import END, StateGraph
 
-from template_langgraph.agents.basic_workflow_agent.models import AgentState
+from template_langgraph.agents.chat_with_tools_agent.models import AgentState
 from template_langgraph.llms.azure_openais import AzureOpenAiWrapper
 from template_langgraph.loggers import get_logger
 from template_langgraph.tools.elasticsearch_tool import search_elasticsearch
@@ -36,7 +36,7 @@ def __call__(self, inputs: dict):
         return {"messages": outputs}
 
 
-class BasicWorkflowAgent:
+class ChatWithToolsAgent:
     def __init__(self):
         self.llm = AzureOpenAiWrapper().chat_model
 
@@ -80,7 +80,7 @@ def create_graph(self):
 
     def initialize(self, state: AgentState) -> AgentState:
         """Initialize the agent with the given state."""
-        logger.info(f"Initializing BasicWorkflowAgent with state: {state}")
+        logger.info(f"Initializing ChatWithToolsAgent with state: {state}")
         # Here you can add any initialization logic if needed
         return state
 
@@ -119,7 +119,7 @@ def route_tools(
 
     def finalize(self, state: AgentState) -> AgentState:
         """Finalize the agent's work and prepare the output."""
-        logger.info(f"Finalizing BasicWorkflowAgent with state: {state}")
+        logger.info(f"Finalizing ChatWithToolsAgent with state: {state}")
         # Here you can add any finalization logic if needed
         return state
 
@@ -128,4 +128,4 @@ def draw_mermaid_png(self) -> bytes:
         return self.create_graph().get_graph().draw_mermaid_png()
 
 
-graph = BasicWorkflowAgent().create_graph()
+graph = ChatWithToolsAgent().create_graph()

template_langgraph/agents/basic_workflow_agent/agent.py renamed to template_langgraph/agents/chat_with_tools_agent/agent.py

@@ -0,0 +1,11 @@
+import sys
+
+from template_langgraph.agents.chat_with_tools_agent.agent import ChatWithToolsAgent
+
+if __name__ == "__main__":
+    png_path = "data/chat_with_tools_agent.png"
+    if len(sys.argv) > 1:
+        png_path = sys.argv[1]
+
+    with open(png_path, "wb") as f:
+        f.write(ChatWithToolsAgent().draw_mermaid_png())

template_langgraph/agents/basic_workflow_agent/models.py renamed to template_langgraph/agents/chat_with_tools_agent/models.py

@@ -1,7 +1,7 @@
 import logging
 
-from template_langgraph.agents.basic_workflow_agent.agent import AgentState
-from template_langgraph.agents.basic_workflow_agent.agent import graph as basic_workflow_agent_graph
+from template_langgraph.agents.chat_with_tools_agent.agent import AgentState
+from template_langgraph.agents.chat_with_tools_agent.agent import graph as chat_with_tools_agent_graph
 from template_langgraph.loggers import get_logger
 
 logger = get_logger(__name__)
@@ -11,7 +11,7 @@
 def stream_graph_updates(
     state: AgentState,
 ) -> dict:
-    for event in basic_workflow_agent_graph.stream(input=state):
+    for event in chat_with_tools_agent_graph.stream(input=state):
         logger.info("-" * 20)
         logger.info(f"Event: {event}")
     return event