modified types to make BasicWorkflowAgent available on LangGraph Studio

Author: ks6088ts

template_langgraph/agents/basic_workflow_agent/agent.py

@@ -62,7 +62,8 @@ def create_graph(self):
         workflow.add_node("finalize", self.finalize)
 
         # Create edges
-        workflow.add_edge(START, "initialize")
+        # workflow.add_edge(START, "initialize")
+        workflow.add_edge(START, "chat_with_tools")
         workflow.add_edge("initialize", "do_something")
         workflow.add_edge("do_something", "extract_profile")
         workflow.add_edge("extract_profile", "chat_with_tools")

template_langgraph/agents/basic_workflow_agent/models.py

@@ -1,5 +1,12 @@
-from typing import Annotated, TypedDict
-
+from collections.abc import Sequence
+from typing import (
+    Annotated,
+    TypedDict,
+)
+
+from langchain_core.messages import (
+    BaseMessage,
+)
 from langgraph.graph.message import add_messages
 from pydantic import BaseModel, Field
 
@@ -21,5 +28,5 @@ class AgentOutput(BaseModel):
 
 
 class AgentState(TypedDict):
-    messages: Annotated[list, add_messages]
+    messages: Annotated[Sequence[BaseMessage], add_messages]
     profile: Profile | None = Field(None, description="抽出されたプロファイル情報")