add graphs to langgraph studio

Author: ks6088ts

langgraph.json

@@ -3,13 +3,16 @@
     "."
   ],
   "graphs": {
-    "supervisor_agent": "template_langgraph.agents.supervisor_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.agent:graph",
+    "demo_agents_parallel_processor_agent": "template_langgraph.agents.demo_agents.parallel_processor_agent.agent:graph",
+    "demo_agents_multi_agent": "template_langgraph.agents.demo_agents.multi_agent:graph",
+    "demo_agents_weather_agent": "template_langgraph.agents.demo_agents.weather_agent:graph",
+    "image_classifier_agent": "template_langgraph.agents.image_classifier_agent.agent:graph",
     "issue_formatter_agent": "template_langgraph.agents.issue_formatter_agent.agent:graph",
-    "task_decomposer_agent": "template_langgraph.agents.task_decomposer_agent.agent:graph",
+    "kabuto_helpdesk_agent": "template_langgraph.agents.kabuto_helpdesk_agent.agent:graph",
     "news_summarizer_agent": "template_langgraph.agents.news_summarizer_agent.agent:graph",
-    "image_classifier_agent": "template_langgraph.agents.image_classifier_agent.agent:graph"
+    "supervisor_agent": "template_langgraph.agents.supervisor_agent.agent:graph",
+    "task_decomposer_agent": "template_langgraph.agents.task_decomposer_agent.agent:graph"
   },
   "env": ".env"
 }

scripts/demo_agents_operator.py

@@ -3,9 +3,9 @@
 import typer
 from dotenv import load_dotenv
 
-from template_langgraph.agents.demo_agents.multi_agent import app as multi_agent_app
-from template_langgraph.agents.demo_agents.parallel_processor_agent.agent import app as parallel_processor_agent_app
-from template_langgraph.agents.demo_agents.weather_agent import app as weather_agent_app
+from template_langgraph.agents.demo_agents.multi_agent import graph as multi_agent_graph
+from template_langgraph.agents.demo_agents.parallel_processor_agent.agent import graph as parallel_processor_agent_graph
+from template_langgraph.agents.demo_agents.weather_agent import graph as weather_agent_graph
 from template_langgraph.loggers import get_logger
 
 app = typer.Typer(
@@ -33,7 +33,7 @@ def weather_agent(
     if verbose:
         logger.setLevel(logging.DEBUG)
 
-    response = weather_agent_app.invoke(
+    response = weather_agent_graph.invoke(
         {
             "messages": [
                 {"role": "user", "content": query},
@@ -62,7 +62,7 @@ def multi_agent(
     if verbose:
         logger.setLevel(logging.DEBUG)
 
-    response = multi_agent_app.invoke(
+    response = multi_agent_graph.invoke(
         {
             "messages": [
                 {"role": "user", "content": query},
@@ -91,7 +91,7 @@ def parallel_processor_agent(
     if verbose:
         logger.setLevel(logging.DEBUG)
 
-    for event in parallel_processor_agent_app.stream(
+    for event in parallel_processor_agent_graph.stream(
         input={
             "goal": goal,
         },

template_langgraph/agents/demo_agents/multi_agent.py

@@ -3,7 +3,7 @@
 from langgraph.graph import END, START, MessagesState, StateGraph
 from langgraph.types import Command
 
-from template_langgraph.agents.demo_agents.weather_agent import app
+from template_langgraph.agents.demo_agents.weather_agent import graph
 from template_langgraph.llms.azure_openais import AzureOpenAiWrapper
 from template_langgraph.loggers import get_logger
 
@@ -37,6 +37,6 @@ def call_model(state: MessagesState) -> Command[Literal["weather_agent", END]]:
 workflow = StateGraph(MessagesState)
 
 workflow.add_node("agent", call_model)
-workflow.add_node("weather_agent", app)
+workflow.add_node("weather_agent", graph)
 workflow.add_edge(START, "agent")
-app = workflow.compile()
+graph = workflow.compile()

template_langgraph/agents/demo_agents/parallel_processor_agent/agent.py

@@ -36,4 +36,4 @@ def create_graph(self) -> CompiledStateGraph:
         return workflow.compile()
 
 
-app = ParallelProcessorAgent(AzureOpenAiWrapper().chat_model).create_graph()
+graph = ParallelProcessorAgent(AzureOpenAiWrapper().chat_model).create_graph()

template_langgraph/agents/demo_agents/weather_agent.py

@@ -42,4 +42,4 @@ def call_model(state: MessagesState) -> Command[Literal["tools", END]]:
 workflow.add_node("tools", tool_node)
 workflow.add_edge(START, "agent")
 workflow.add_edge("tools", "agent")
-app = workflow.compile()
+graph = workflow.compile()