Conditional edges:
ветвление на основе результата

from langgraph.graph import END

def route(state: AgentState) -> str:
    """Решает, куда идти после узла agent. Возвращает имя узла (строку)."""
    last = state["messages"][-1]
    if last.tool_calls:        # LLM попросила инструмент
        return "tools"
    return END                 # ответ готов — завершаем граф

builder.add_conditional_edges(
    "agent",                        # 1. из какого узла ветвимся
    route,                          # 2. функция-маршрутизатор
    {                               # 3. карта: результат route → узел
        "tools": "tools",
        END: END,
    },
)

from langgraph.errors import GraphRecursionError

try:
    result = graph.invoke(
        {"messages": [...]},
        {"recursion_limit": 25},     # не больше 25 шагов за запуск
    )
except GraphRecursionError:
    print("Агент не сошёлся за 25 шагов — обрываем")

from typing import Annotated, TypedDict
from langgraph.graph import StateGraph, START, END
from langgraph.graph.message import add_messages
from langchain_openai import ChatOpenAI

# ---- 1. Инструмент ----
def get_weather(city: str) -> str:
    return f"В городе {city} сейчас +18°C, ясно."

# ---- 2. Состояние (reducer из прошлого урока) ----
class State(TypedDict):
    messages: Annotated[list, add_messages]

# ---- 3. LLM с привязанным инструментом ----
llm = ChatOpenAI(model="gpt-4o-mini")
llm_with_tools = llm.bind_tools([get_weather])

# ---- 4. Узлы ----
def agent(state: State) -> dict:
    """Спрашивает LLM по текущей истории."""
    return {"messages": [llm_with_tools.invoke(state["messages"])]}

def tools(state: State) -> dict:
    """Выполняет инструменты, которые попросила LLM."""
    last = state["messages"][-1]
    results = []
    for call in last.tool_calls:
        output = get_weather(**call["args"])
        results.append({"role": "tool", "content": output,
                        "tool_call_id": call["id"]})
    return {"messages": results}

# ---- 5. Маршрутизатор ----
def route(state: State) -> str:
    return "tools" if state["messages"][-1].tool_calls else END

# ---- 6. Сборка графа с ветвлением ----
builder = StateGraph(State)
builder.add_node("agent", agent)
builder.add_node("tools", tools)

builder.add_edge(START, "agent")
builder.add_conditional_edges("agent", route, {"tools": "tools", END: END})
builder.add_edge("tools", "agent")        # ← цикл: после инструментов снова к LLM

graph = builder.compile()

# ---- 7. Запуск ----
result = graph.invoke(
    {"messages": [{"role": "user", "content": "Какая погода в Москве?"}]},
    {"recursion_limit": 10},
)
print(result["messages"][-1].content)
# → "Сейчас в Москве +18°C и ясно."

def classify(state: State) -> str:
    intent = state["intent"]          # его проставил предыдущий узел
    if intent == "code":
        return "code_branch"
    if intent == "search":
        return "search_branch"
    return "chitchat_branch"

builder.add_conditional_edges(
    "router_node",
    classify,
    {
        "code_branch":     "write_code",
        "search_branch":   "web_search",
        "chitchat_branch": "small_talk",
    },
)

from langgraph.graph import StateGraph, START, END

# 1. Маршрутизатор: state -> имя узла (НИЧЕГО не меняет в состоянии!)
def route(state) -> str:
    return "tools" if state["messages"][-1].tool_calls else END

# 2. Подключение: из узла, router, карта исходов
builder.add_conditional_edges("agent", route, {"tools": "tools", END: END})

# 3. Цикл — ребро назад создаёт петлю
builder.add_edge("tools", "agent")          # agent <-> tools

# 4. Защита от зацикливания
graph.invoke(state, {"recursion_limit": 25})   # иначе GraphRecursionError

# 5. Несколько исходов — router возвращает разные метки
def classify(state) -> str:
    return state["intent"]                  # "code" | "search" | "chat"
builder.add_conditional_edges("node", classify,
    {"code": "a", "search": "b", "chat": "c"})

# Правила:
#  • router без сайд-эффектов; работа — в узлах
#  • в карте перечислены ВСЕ возвраты router (вкл. END)
#  • в цикле обязателен путь к END