AI创想
标题:
LangGraph极简入门
[打印本页]
作者:
创想小编
时间:
昨天 23:02
标题:
LangGraph极简入门
作者:CSDN博客
什么是 LangGraph?
把一句话拆成多步、带循环、带分支、还能随时把中间结果拿出来——这就是 LangGraph。
它把“一条直线”的 Chain 变成了“一张图”:节点(Node)干活,边(Edge)告诉数据下一步去哪。
(, 下载次数: 0)
上传
点击文件名下载附件
运行示例
from langchain_openai import ChatOpenAI
from langgraph.constants import START
from langgraph.graph import StateGraph, END
from typing import TypedDict
# 1. 状态结构(图里所有节点共享)
class State(TypedDict):
topic: str # 用户话题
reply: str # LLM 回复
# 2. 节点只是普通函数,参数+返回值都叫 State
def ask_topic(state: State) -> State:
state["topic"] = input("你想聊啥? ")
return state
def answer(state: State) -> State:
# llm = ChatOpenAI(model="gpt-3.5-turbo")
llm = ChatOpenAI(base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",api_key="sk-3ad75cfac6b443159a***f8d3d473b2a",model="qwen-plus" )
state["reply"] = llm.invoke(f"用一句话介绍{state['topic']}").content
print(state["reply"])
return state
# 3. 建图、连边
builder = StateGraph(State)
builder.add_node("ask", ask_topic)
builder.add_node("answer", answer)
builder.add_edge(START,"ask")
builder.add_edge("ask", "answer")
builder.add_edge("answer", END)
graph = builder.compile()
# 4. 跑起来
graph.invoke({"topic": "", "reply": ""})
复制代码
StateGraph
<
原文地址:https://blog.csdn.net/jianlee1991/article/details/154989578
欢迎光临 AI创想 (https://www.llms-ai.com/)
Powered by Discuz! X3.4