AI Agent开发学习系列 - LangGraph(2): 用LangGraph创建我们的第一个Agent(练习解答)
在AI Agent开发学习系列 - LangGraph(1): 用LangGraph创建我们的第一个Agent中,我们学习了如何用LangGraph来创建一个简单的Agent。为了巩固学习,我们来做一个练习。
用LangGraph创建如下图的一个Agent:
要求:
- 输入你的名字
- 在greeter这个节点里,把你输入名字拼接到下面这句话前:you’re doing an amazing job learning LangGraph!
解答:
from typing import TypedDict
from langgraph.graph import StateGraph# Create an agentstate
class AgentState(TypedDict):message: strdef greeting_node(state: AgentState) -> AgentState:"""Simple note that adds a greeting mesage to the state"""state["message"] = state["message"] + ", you're doing an amazing job learning LangGraph!"return stategraph = StateGraph(AgentState)graph.add_node("greeter", greeting_node)
graph.set_entry_point("greeter")
graph.set_finish_point("greeter")app = graph.compile()from IPython.display import Image, display
display(Image(app.get_graph().draw_mermaid_png()))result = app.invoke({"message": "Bob"})
result["message"]
运行结果:
"Bob, you're doing an amazing job learning LangGraph!"