🦜🕸️LangGraph
⚡ Building language agents as graphs ⚡
Overview
LangGraph is a library for building stateful, multi-actor applications with LLMs, and is used to build agent and multi-agent workflows. Compared to other LLM frameworks it has these core benefits: cycles, controllability and persistence. LangGraph is a very low-level framework - this allows you to have fine-grained control over both the flow and the state of your application, crucial for creating reliable agents. LangGraph comes with built-in persistence, allowing for advanced human-in-the-loop and memory features.
LangGraph is inspired by Pregel and Apache Beam. The public interface draws inspiration from NetworkX. LangGraph is built by LangChain Inc, the creators of LangChain, but can be used with LangChain.
Key Features
- Cycles and Branching: Implement loops and conditionals in your apps.
- Persistence: Automatically save state after each step in the graph. Pause and resume the graph execution at any point to support error recovery, human-in-the-loop workflows, time travel and more.
- Human-in-the-Loop: Interrupt graph execution to approve or edit next action planned by the agent.
- Streaming Support: Stream outputs as they are produced by each node (including token streaming).
- Integration with LangChain: LangGraph integrates seamlessly with LangChain and LangSmith.
Installation
pip install -U langgraph
Example
One of the central concepts of LangGraph is state. Each graph execution creates a state that is passed between nodes in the graph as they execute, and each node updates this internal state with its return value after it executes. The way that the graph updates its internal state is defined by either the type of graph chosen or a custom function.
Let's take a look at a simple example of an agent that can search the web using Tavily Search API.
pip install langchain_openai langchain_community
export OPENAI_API_KEY=sk-...
export TAVILY_API_KEY=tvly-...
Optionally, we can set up LangSmith for best-in-class observability.
export LANGCHAIN_TRACING_V2="true"
export LANGCHAIN_API_KEY=ls__...
from typing import Annotated, Literal, TypedDict
from langchain_core.messages import HumanMessage
from langchain_community.tools.tavily_search import TavilySearchResults
from langchain_openai import ChatOpenAI
from langgraph.graph import END, StateGraph, MessagesState
from langgraph.prebuilt import ToolNode
# Define the tools for the agent to use
tools = [TavilySearchResults(max_results=1)]
tool_node = ToolNode(tools)
model = ChatOpenAI(temperature=0).bind_tools(tools)
# Define the function that determines whether to continue or not
def should_continue(state: AgentState) -> Literal["tools", "__end__"]:
messages = state['messages']
last_message = messages[-1]
# If the LLM makes a tool call, then we route to the "tools" node
if last_message.tool_calls:
return "tools"
# Otherwise, we stop (reply to the user)
return "__end__"
# Define the function that calls the model
def call_model(state: AgentState):
messages = state['messages']
response = model.invoke(messages)
# We return a list, because this will get added to the existing list
return {"messages": [response]}
# Define a new graph
workflow = StateGraph(MessagesState)
# Define the two nodes we will cycle between
workflow.add_node("agent", call_model)
workflow.add_node("tools", tool_node)
# Set the entrypoint as `agent`
# This means that this node is the first one called
workflow.set_entry_point("agent")
# We now add a conditional edge
workflow.add_conditional_edges(
# First, we define the start node. We use `agent`.
# This means these are the edges taken after the `agent` node is called.
"agent",
# Next, we pass in the function that will determine which node is called next.
should_continue,
)
# We now add a normal edge from `tools` to `agent`.
# This means that after `tools` is called, `agent` node is called next.
workflow.add_edge("tools", 'agent')
# Finally, we compile it!
# This compiles it into a LangChain Runnable,
# meaning you can use it as you would any other runnable
app = workflow.compile()
# Use the Runnable
final_state = app.invoke({"messages": [HumanMessage(content="what is the weather in sf")]})
final_state["messages"][-1].content
'The current weather in San Francisco is as follows:\n- Temperature: 60.1°F (15.6°C)\n- Condition: Partly cloudy\n- Wind: 5.6 mph (9.0 kph) from SSW\n- Humidity: 83%\n- Visibility: 9.0 miles (16.0 km)\n- UV Index: 4.0\n\nFor more details, you can visit [Weather API](https://www.weatherapi.com/).'
Step-by-step Breakdown:
-
Initialize the model, tools and the graph.
- we use
ChatOpenAIas our LLM. NOTE: we need make sure the model knows that it has these tools available to call. We can do this by converting the LangChain tools into the format for OpenAI tool calling using the.bind_tools()method. - we define the tools we want to use -- a web search tool in our case. It is really easy to create your own tools - see documentation here on how to do that here.
- we use
-
Define graph nodes.
There are two main nodes we need for this:
- The
agentnode: responsible for deciding what (if any) actions to take. - The
toolsnode that invokes tools: if the agent decides to take an action, this node will then execute that action.
- The
-
Define graph edges We define both normal and conditional edges. Conditional edge means that the destination depends on the contents of the graph's State. In our case, the destination is not known until the agent (LLM) decides.
Our graph has one of each type of edge:
- Conditional edge: after the agent is called, we should either:
- a. Run tools if the agent said to take an action, OR
- b. Finish (respond to the user) if the agent did not ask to run tools
- Normal edge: after the tools are invoked, the graph should always return to the agent to decide what to do next
- Conditional edge: after the agent is called, we should either:
-
Set the entry point for graph execution (
agent). -
Compile the graph.
When we compile the graph, we are translating it to low-level Pregel operations
-
Execute the graph
- LangGraph adds the input message to the internal state, then passes the state to the entrypoint node,
"agent". - The
"agent"node executes, invoking the chat model. - The chat model returns an
AIMessage. LangGraph adds this to the state. - Graph cycles the following steps until there are no more
tool_callsonAIMessage:
- If
AIMessagehastool_calls,"tools"node executes - The
"agent"node executes again and returnsAIMessage
- Execution progresses to the special
ENDvalue and outputs the final state. And as a result, we get a list of all our chat messages as output.
- LangGraph adds the input message to the internal state, then passes the state to the entrypoint node,
Documentation
- Tutorials: Learn to build with LangGraph through guided examples.
- How-to Guides: Accomplish specific things within LangGraph, from streaming, to adding memory & persistence, to common design patterns (branching, subgraphs, etc.), these are the place to go if you want to copy and run a specific code snippet.
- Conceptual Guides: In-depth explanations of the key concepts and principles behind LangGraph, such as nodes, edges, state and more.
- API Reference: Review important classes and methods, simple examples of how to use the graph and checkpointing APIs, higher-level prebuilt components and more.
Contributing
As an open-source project in a rapidly developing field, we are extremely open to contributions, whether it be in the form of a new feature, improved infrastructure, or better documentation.
For detailed information on how to contribute, see here.