4. langgraph中的react agent使用 (在react agent添加人机交互)

1. 初始化模型

首先,我们需要初始化要使用的模型。

python 复制代码
# First we initialize the model we want to use.
from langchain_openai import ChatOpenAI

model = ChatOpenAI(
    temperature=0,
    model="glm-4-plus",
    openai_api_key="your_api_key",
    openai_api_base="https://open.bigmodel.cn/api/paas/v4/"
)

2. 定义自定义工具

我们将使用一个自定义工具来返回预定义的天气信息。

python 复制代码
from typing import Literal
from langchain_core.tools import tool

@tool
def get_weather(location: str):
    """Use this to get weather information from a given location."""
    if location.lower() in ["nyc", "new york"]:
        return "It might be cloudy in nyc"
    elif location.lower() in ["sf", "san francisco"]:
        return "It's always sunny in sf"
    else:
        raise AssertionError("Unknown Location, please try again use different expression")

tools = [get_weather]

3. 设置检查点

为了支持人机交互模式,我们需要一个检查点。

python 复制代码
from langgraph.checkpoint.memory import MemorySaver

memory = MemorySaver()

4. 定义图形

使用LangGraph的预构建功能来创建一个REACT代理。

python 复制代码
from langgraph.prebuilt import create_react_agent

graph = create_react_agent(
    model, tools=tools, interrupt_before=["tools"], checkpointer=memory
)

5. 打印工具流

定义一个工具来美化打印流。

python 复制代码
def print_stream(stream):
    """A utility to pretty print the stream."""
    for s in stream:
        message = s["messages"][-1]
        if isinstance(message, tuple):
            print(message)
        else:
            message.pretty_print()

6. 发送查询并打印结果

发送一个查询并打印结果。

python 复制代码
from langchain_core.messages import HumanMessage

config = {"configurable": {"thread_id": "42"}}
inputs = {"messages": [("user", "what is the weather in SF, CA?")]}

print_stream(graph.stream(inputs, config, stream_mode="values"))

输出示例:

复制代码
================================[1m Human Message [0m=================================
what is the weather in SF, CA?
================================[1m Ai Message [0m==================================
Tool Calls:
  get_weather (call_9208192248525421766)
 Call ID: call_9208192248525421766
  Args:
    location: SF, CA

7. 获取当前状态

获取当前状态以了解下一步操作。

python 复制代码
snapshot = graph.get_state(config)
print("Next step: ", snapshot.next)

输出示例:

复制代码
Next step:  ('tools',)

8. 继续处理流

继续处理流以获取更多结果。

python 复制代码
print_stream(graph.stream(None, config, stream_mode="values"))

输出示例:

复制代码
================================[1m Ai Message [0m==================================
Tool Calls:
  get_weather (call_9208192248525421766)
 Call ID: call_9208192248525421766
  Args:
    location: SF, CA
================================[1m Tool Message [0m=================================
Name: get_weather

Error: AssertionError('Unknown Location, please try again use different expression')
 Please fix your mistakes.
================================[1m Ai Message [0m==================================
Tool Calls:
  get_weather (call_9208192214165585200)
 Call ID: call_9208192214165585200
  Args:
    location: San Francisco, CA

9. 更新状态

根据工具的错误信息更新状态。

python 复制代码
state = graph.get_state(config)

last_message = state.values["messages"][-1]
last_message.tool_calls[0]["args"] = {"location": "San Francisco"}

graph.update_state(config, {"messages": [last_message]})

输出示例:

复制代码
{'configurable': {'thread_id': '42',
  'checkpoint_ns': '',
  'checkpoint_id': '1efa3728-5fb2-6e2f-8004-c28044c43c8b'}}

10. 再次打印流

再次打印流以获取最终结果。

python 复制代码
print_stream(graph.stream(None, config, stream_mode="values"))

输出示例:

复制代码
================================[1m Ai Message [0m==================================
Tool Calls:
  get_weather (call_9208192214165585200)
 Call ID: call_9208192214165585200
  Args:
    location: San Francisco
================================[1m Tool Message [0m=================================
Name: get_weather

It's always sunny in sf
================================[1m Ai Message [0m==================================

The weather in San Francisco is always sunny!

参考链接:https://langchain-ai.github.io/langgraph/how-tos/create-react-agent-hitl/

相关推荐
AndrewHZ13 分钟前
【图像处理基石】如何入门色彩评估?
图像处理·人工智能·深度学习·色彩科学·hvs·色彩评估·颜色工程
TomatoSCI13 分钟前
聚类的可视化选择:PCA / t-SNE丨TomatoSCI分析日记
人工智能·机器学习
大咖分享课15 分钟前
深度剖析:最新发布的ChatGPT Agent 技术架构与应用场景
人工智能·openai·智能助手·ai代理·chatgpt agent·自主任务执行
lucky_lyovo26 分钟前
卷积神经网络--网络性能提升
人工智能·神经网络·cnn
liliangcsdn30 分钟前
smolagents - 如何在mac用agents做简单算术题
人工智能·macos·prompt
nju_spy34 分钟前
周志华《机器学习导论》第8章 集成学习 Ensemble Learning
人工智能·随机森林·机器学习·集成学习·boosting·bagging·南京大学
都叫我大帅哥1 小时前
LangChain加载HTML内容全攻略:从入门到精通
python·langchain
静心问道1 小时前
TrOCR: 基于Transformer的光学字符识别方法,使用预训练模型
人工智能·深度学习·transformer·多模态
说私域1 小时前
基于开源AI大模型、AI智能名片与S2B2C商城小程序源码的用户价值引导与核心用户沉淀策略研究
人工智能·开源
亲持红叶1 小时前
GLU 变种:ReGLU 、 GEGLU 、 SwiGLU
人工智能·深度学习·神经网络·激活函数