How to use Hierarchical Agent Teams in langgraph studio?

[cris-m]

from langchain_core.messages import HumanMessage
from langgraph.graph import START, StateGraph
from teams.compute_agent.graph import compute_graph
from teams.configuration import Configuration
from teams.explore_agent.graph import explore_graph
from teams.state import AgentState
from teams.utils import create_supervisor_node, load_chat_model

llm = load_chat_model(Configuration.llm_model)
teams_supervisor = create_supervisor_node(llm, ["compute_team", "explore_team"])


def compute_team(state: AgentState) -> AgentState:
    response = compute_graph.invoke({"messages": state["messages"][-1]})
    return {
        "messages": [
            HumanMessage(content=response["messages"][-1].content, name="compute_team")
        ]
    }


def explore_team(state: AgentState) -> AgentState:
    response = explore_graph.invoke({"messages": state["messages"][-1]})
    return {
        "messages": [
            HumanMessage(content=response["messages"][-1].content, name="explore_team")
        ]
    }


builder = StateGraph(AgentState)
builder.add_node("supervisor", teams_supervisor)
builder.add_node("compute_team", compute_team)
builder.add_node("explore_team", explore_team)

builder.add_edge(START, "supervisor")
builder.add_edge("compute_team", "supervisor")
builder.add_edge("explore_team", "supervisor")
builder.add_conditional_edges("supervisor", lambda state: state["next"])

graph = builder.compile()
graph.name = "Agent"

Here is the langgraph.json

{
    "dependencies": ["."],
    "graphs": {
        "explore_team": "./teams/explore_agent/graph.py:explore_graph",
        "compute_team": "./teams/compute_agent/graph.py:compute_graph",
        "agent": "./teams/graph.py:graph"
    },

    "env": ".env"
}

I don't see the graph

Based on log it seems it cannot found the subgraph?

File "/api/langgraph_api/shared/route.py", line 37, in app
  File "/usr/local/lib/python3.11/site-packages/starlette/_exception_handler.py", line 53, in wrapped_app
    raise exc
  File "/usr/local/lib/python3.11/site-packages/starlette/_exception_handler.py", line 42, in wrapped_app
    await app(scope, receive, sender)
  File "/api/langgraph_api/shared/route.py", line 32, in app
  File "/usr/local/lib/python3.11/site-packages/langgraph_storage/retry.py", line 34, in wrapper
    return await func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/api/langgraph_api/api/assistants.py", line 145, in get_assistant_graph
  File "/usr/local/lib/python3.11/site-packages/langgraph/graph/graph.py", line 603, in get_graph
    add_edge(start, end)
  File "/usr/local/lib/python3.11/site-packages/langgraph/graph/graph.py", line 561, in add_edge
    start_nodes[start],
    ~~~~~~~~~~~^^^^^^^
KeyError: 'compute_team'

[cris-m]

I can see the two subgraph, but I cannot see the main graph.

compute_team graph

explore_team graph

I imported them as follow in the main graph

from teams.compute_agent.graph import compute_graph
from teams.explore_agent.graph import explore_graph

They are add to the main builder graph as node

def compute_team(state: AgentState) -> AgentState:
    response = compute_graph.invoke({"messages": state["messages"][-1]})
    return {
        "messages": [
            HumanMessage(content=response["messages"][-1].content, name="compute_team")
        ]
    }


def explore_team(state: AgentState) -> AgentState:
    response = explore_graph.invoke({"messages": state["messages"][-1]})
    return {
        "messages": [
            HumanMessage(content=response["messages"][-1].content, name="explore_team")
        ]
    }
    
 
builder = StateGraph(AgentState)
builder.add_node("supervisor", teams_supervisor)
builder.add_node("compute_team", compute_team)
builder.add_node("explore_team", explore_team)