wip

Author: ks6088ts

docs/index.md

@@ -24,6 +24,10 @@ uv run python -m template_langgraph.tasks.search_documents_on_elasticsearch
 # 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のマニュアルから禅モードに関する情報を教えて下さい"
+
+# 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 "KABUTOの起動時に、画面全体が紫色に点滅し、システムがフリーズします。"
 ```
 
 ## References

template_langgraph/agents/basic_workflow_agent/__init__.py

@@ -0,0 +1,69 @@
+from langgraph.graph import END, START, StateGraph
+
+from template_langgraph.agents.basic_workflow_agent.models import AgentInput, AgentOutput, 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
+from template_langgraph.tools.qdrants import search_qdrant
+
+logger = get_logger(__name__)
+
+
+class BasicWorkflowAgent:
+    def __init__(self, tools=None):
+        if tools is None:
+            # Default tool for searching Qdrant and Elasticsearch
+            tools = [
+                search_qdrant,
+                search_elasticsearch,
+                # Add other tools as needed
+            ]
+        self.llm = AzureOpenAiWrapper().chat_model.bind_tools(
+            tools=tools,
+        )
+
+    def create_graph(self):
+        """Create the main graph for the agent."""
+        # Create the workflow state graph
+        workflow = StateGraph(AgentState)
+
+        # Create nodes
+        workflow.add_node("initialize", self.initialize)
+        workflow.add_node("do_something", self.do_something)
+
+        # Create edges
+        workflow.add_edge(START, "initialize")
+        workflow.add_edge("initialize", "do_something")
+        workflow.add_edge("do_something", END)
+
+        # Compile the graph
+        return workflow.compile()
+
+    def initialize(self, state: AgentState) -> AgentState:
+        """Initialize the agent with the given state."""
+        logger.info(f"Initializing BasicWorkflowAgent with state: {state}")
+        # Here you can add any initialization logic if needed
+        return state
+
+    def do_something(self, state: AgentState) -> AgentState:
+        """Perform some action with the given state."""
+        logger.info(f"Doing something with state: {state}")
+        # Here you can add the logic for the action
+        return state
+
+    def run_agent(self, input: AgentInput) -> AgentOutput:
+        """Run the agent with the given input."""
+        logger.info(f"Running BasicWorkflowAgent with question: {input.model_dump_json(indent=2)}")
+        app = self.create_graph()
+        initial_state: AgentState = {
+            "request": input.request,
+        }
+        final_state = app.invoke(initial_state)
+        logger.info(f"Final state after running agent: {final_state}")
+        return AgentOutput(
+            response=final_state.get("response", "No response"),
+        )
+
+    def draw_mermaid_png(self) -> bytes:
+        """Draw the graph in Mermaid format."""
+        return self.create_graph().get_graph().draw_mermaid_png()

template_langgraph/agents/basic_workflow_agent/agent.py

@@ -0,0 +1,15 @@
+from typing import TypedDict
+
+from pydantic import BaseModel, Field
+
+
+class AgentInput(BaseModel):
+    request: str = Field(..., description="ユーザーからのリクエスト")
+
+
+class AgentOutput(BaseModel):
+    response: str = Field(..., description="エージェントの応答")
+
+
+class AgentState(TypedDict):
+    request: str

template_langgraph/agents/basic_workflow_agent/models.py

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

template_langgraph/tasks/draw_basic_workflow_agent_mermaid_png.py

@@ -0,0 +1,28 @@
+import logging
+import sys
+
+from template_langgraph.agents.basic_workflow_agent.agent import AgentInput, BasicWorkflowAgent
+from template_langgraph.loggers import get_logger
+
+logger = get_logger(__name__)
+logger.setLevel(logging.INFO)
+
+if __name__ == "__main__":
+    question = "「鬼灯」を実行すると、KABUTOが急に停止します。原因と対策を教えてください。"
+    if len(sys.argv) > 1:
+        # sys.argv[1] が最初の引数
+        question = sys.argv[1]
+
+    # Agentのインスタンス化
+    agent = BasicWorkflowAgent(
+        tools=None,  # ツールはカスタムせず、デフォルトのツールを使用
+    )
+    # AgentInputの作成
+    agent_input = AgentInput(
+        request=question,
+    )
+
+    # エージェントの実行
+    logger.info(f"Running BasicWorkflowAgent with input: {agent_input.model_dump_json(indent=2)}")
+    agent_output = agent.run_agent(input=agent_input)
+    logger.info(f"Agent output: {agent_output.model_dump_json(indent=2)}")