《状态机驱动的 LLM 应用架构:基于流水线范式解析 LangGraph 的 Graph 核心》

前言

LangGraph其实就是把llm和函数、业务逻辑杂糅在一起做了一个流程编排的应用框架。它的核心是Graph图。那么这篇文章就是帮助你理解这个Graph图的核心应用。文中引入了应用案例:一个聊天机器人。

首先我们需呀安装依赖

python 复制代码
pip install -U langgraph langsmith

关于langgraph的图Graph

我们可以把图Graph想象成一个流程图或者工厂的流水线。

流水线的三要素

  1. 产品-The State-状态
    1. 工厂流水线上的产品比如汽车的底盘就是一个产品。我们可以把langgraph创建的一次会话当成一个产品,也就是State对象,它里面装着message列表。
  2. 工作站-The-Node-节点
    1. 节点就相当于流水线上的工作站,在工厂里,产品到了工作站,工人就开始对产品做一些操作,比如说安装轮胎。我们可以把langgraph里的节点想象成一个函数,这个函数做指定的操作。
    2. 我们的代码里, chatbot 函数就是一个工作站。当"对话"这个产品流到这里时, chatbot 函数就对它进行一步操作: 调用大语言模型(LLM)获取回复 。
  3. 传送带-The-Edge-边
    1. 在工厂里,传送带决定了流水线上的产品从哪个工作站传送到哪个工作站。
    2. 在我们的代码里, add_edge 定义的就是传送带的走向。它告诉程序,产品应该从哪里送到哪里。
定义State-一次会话-一个产品
python 复制代码
class State(TypedDict):
    messages: Annotated[list, add_messages]
定义节点-函数
python 复制代码
def chatbot(state: State):
    return {"messages": [llm.invoke(state["messages"])]}
定义边
py 复制代码
graph_builder.add_edge(START, "chatbot")

这个边我们从开始节点指向了函数节点chatbot,相当于这个流程走完到这个函数就终止了。

markdown 复制代码
[ START (开始) ] ---------> [ "chatbot" 工作站 ]

langgraph的核心价值

它提供了一个清晰的框架,让我们能够像画流程图一样,去设计和组织复杂的 AI 应用逻辑,而不是把所有 if-else 判断都写在一个巨大的函数里。

当前代码里的"图",是一个只有一个步骤的、最简单的流程图。它之所以存在,是为了给我们提供一个强大的框架,以便未来可以轻松地在其中添加更多的"工作站"(功能)和"传送带"(逻辑)。

状态机画流程图-StateGraph()

我们可以通过创建StateGraph()对象来创建一个新的、空白的流程图蓝图,并规定在这张蓝图上流动的所有数据都必须符合 State 这个格式。

python 复制代码
graph_builder = StateGraph(State)
(...) 里的 State - "规定产品的规格"

在这张流程图(Graph)的任何传送带(Edge)上流动的数据,都必须是 State 这个样子的。

完整代码

python 复制代码
import os
from typing import Annotated

from dotenv import load_dotenv
from langchain.chat_models import init_chat_model
from langchain_openai import ChatOpenAI
from typing_extensions import TypedDict

from langgraph.graph import StateGraph, START
from langgraph.graph.message import add_messages
##构建一个基本聊天机器人¶

load_dotenv()
class State(TypedDict):
    messages: Annotated[list, add_messages]


graph_builder = StateGraph(State)

llm = ChatOpenAI(
    api_key=os.getenv("DASHSCOPE_API_KEY"),
    base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",
    model="qwen-plus",
    temperature=0
)


def chatbot(state: State):
    return {"messages": [llm.invoke(state["messages"])]}


# The first argument is the unique node name
# The second argument is the function or object that will be called whenever
# the node is used.cc g
graph_builder.add_node("chatbot", chatbot)
graph_builder.add_edge(START, "chatbot")
graph = graph_builder.compile()


while True:
    user_input = input("User: ")
    if user_input.lower() in ["quit", "exit", "q"]:
        print("Goodbye!")
        break
    for event in graph.stream({"messages": [("user", user_input)]}):
        if "chatbot" in event:
            print("AI:", event["chatbot"]["messages"][-1].content)
相关推荐
minstbe3 小时前
IC设计私有化AI助手实战:基于Docker+OpenCode+Ollama的数字前端综合增强方案(进阶版)
人工智能·python·语言模型·llama
GinoInterpreter4 小时前
什么是翻译的去中心化?
人工智能·自然语言处理·去中心化·区块链·机器翻译·机器翻译模型·机器翻译引擎
码农小白AI4 小时前
IACheck AI报告文档审核:高端制造合规新助力,保障标准引用报告质量
大数据·人工智能·制造
_YiFei5 小时前
哪个降论文AI率工具最好用?
人工智能·深度学习·神经网络
放下华子我只抽RuiKe55 小时前
机器学习全景指南-直觉篇——基于距离的 K-近邻 (KNN) 算法
人工智能·gpt·算法·机器学习·语言模型·chatgpt·ai编程
kisshuan123965 小时前
[特殊字符]【深度学习】DA3METRIC-LARGE单目深度估计算法详解
人工智能·深度学习·算法
sali-tec5 小时前
C# 基于OpenCv的视觉工作流-章33-Blod分析
图像处理·人工智能·opencv·算法·计算机视觉
老星*5 小时前
Trae-cn一句话安装OpenClaw:AI智能体框架快速部署指南
人工智能·编辑器
昨夜见军贴06166 小时前
IACheck结合AI报告审核:轨道扣件横向阻力检测报告确保无误差
人工智能
Qt学视觉6 小时前
AI2-Paddle环境搭建
c++·人工智能·python·opencv·paddle