Subgraphs

from langgraph.graph import StateGraph

# Import your compiled subgraph
from my_agents.search_agent import graph as search_graph

builder = StateGraph(State)
builder.add_node("coordinator", coordinator_node)
builder.add_node("search_agent", search_graph)  # Add graph as a node
builder.add_node("summarizer", summarize_node)

builder.set_entry_point("coordinator")
builder.add_edge("coordinator", "search_agent")
builder.add_edge("search_agent", "summarizer")

graph = builder.compile()

{
  "graphs": {
    "orchestrator": "./src/orchestrator/graph.py:graph"
  }
}

async for chunk in client.runs.stream(
    thread_id=thread_id,
    assistant_id="orchestrator",
    input={"messages": [{"type": "human", "content": "Research this topic"}]},
    stream_subgraphs=True,
):
    print(chunk)

from langgraph.types import interrupt

# Subgraph with an interrupt
def approval_node(state):
    response = interrupt({"action": "approve_purchase", "amount": state["amount"]})
    return {"approved": response[0]["type"] == "accept"}

# List subgraphs
subgraphs = await client.assistants.get_subgraphs(assistant_id)

# Recursive listing
subgraphs = await client.assistants.get_subgraphs(assistant_id, recurse=True)

# Xray graph visualization (expands subgraph nodes)
graph = await client.assistants.get_graph(assistant_id, xray=True)

# Control xray depth
graph = await client.assistants.get_graph(assistant_id, xray=2)

# Get state of a specific subgraph namespace
state = await client.threads.get_state(
    thread_id,
    subgraphs=True,
)

from langgraph.graph import StateGraph

from agents.search_agent import graph as search_graph
from agents.code_agent import graph as code_graph


def router(state):
    """Route to the appropriate specialist agent."""
    intent = state["intent"]
    if intent == "search":
        return "search_agent"
    elif intent == "code":
        return "code_agent"
    return "__end__"


builder = StateGraph(State)
builder.add_node("classify", classify_intent)
builder.add_node("search_agent", search_graph)
builder.add_node("code_agent", code_graph)

builder.set_entry_point("classify")
builder.add_conditional_edges("classify", router)

graph = builder.compile()