【LangChain学习笔记】创建智能体

基础创建：简易智能体

核心代码示例

python 复制代码

from langchain.agents import create_agent
from langchain_openai import ChatOpenAI

# 大模型初始化
model = ChatOpenAI(
    model_name="qwen-plus",
    base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",
    api_key="sk-94207ea3eb4b44eaae46683ed8ac55e5",
)

# 使用create_agent函数创建智能体，将前面初始化的模型作为参数传入
agent = create_agent(model=model)

# 通过智能体的invoke方法发起调用，传入构造好的输入数据
response = agent.invoke({"messages": [{"role": "human", "content": "1+1=?"}]})

# 打印智能体响应消息列表中最后一条消息的内容
print(response["messages"][-1].content)

废弃函数提醒

最新版本 LangChain 中，无法通过 langchain.agents.initialize_agent 函数创建智能体；
langgraph.prebuilt.create_react_agent 函数同样已被废弃，不建议继续使用。

实战实例：智能体创建网页

功能说明

创建绑定文件管理工具的智能体，实现根据用户指定主题，自动编写并生成 HTML 网页文件，同时返回结构化 JSON 格式响应。

完整代码示例

python 复制代码

from langchain_openai import ChatOpenAI
from langchain.agents import create_agent
from langchain_core.prompts import (
    ChatPromptTemplate,
    SystemMessagePromptTemplate,
    HumanMessagePromptTemplate,
)

# 从langchain_community.agent_toolkits模块导入FileManagementToolkit类，用于获取文件管理工具集
from langchain_community.agent_toolkits import FileManagementToolkit
from pydantic import BaseModel, Field
from langchain_core.output_parsers import JsonOutputParser
import os

# 大模型初始化
model = ChatOpenAI(
    model_name="qwen-plus",
    base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",
    api_key="[你的API Key]",
)

# 定义系统消息提示模板内容
system_message_prompt_template = SystemMessagePromptTemplate.from_template(
    "你是一个专业的前端开发，你可以根据用户的需求，编写前端代码。"
)

# 定义用户消息提示模板内容
human_message_prompt_template = HumanMessagePromptTemplate.from_template(
    "帮我编写一个主题为{topic}的html文件，最后返回json格式的结果，格式为{format_instructions}。"
)

# 使用BaseModel定义一个名为ResponseFormat的数据模型
class ResponseFormat(BaseModel):
    code: int = Field(description="状态码，0表示成功，1表示失败")
    message: str = Field(description="返回的消息")

# 使用JsonOutputParser类初始化一个JSON输出解析器，并关联ResponseFormat数据模型
json_parser = JsonOutputParser(
    pydantic_object=ResponseFormat,
)

# 使用ChatPromptTemplate类的from_messages方法创建聊天提示模板，由系统消息提示模板和用户消息提示模板组成
chat_prompt_template = ChatPromptTemplate.from_messages(
    [system_message_prompt_template, human_message_prompt_template]
)

# 使用partial方法为聊天提示模板预填充变量
chat_prompt_template = chat_prompt_template.partial(
    format_instructions=json_parser.get_format_instructions()
)

# 创建智能体，传入初始化后的模型以及FileManagementToolkit获取的工具集
agent = create_agent(
    model=model, tools=FileManagementToolkit(root_dir=os.getcwd()).get_tools()
)

# 通过智能体的invoke方法发起调用，传入构造好的输入数据
response = agent.invoke(
    {"messages": chat_prompt_template.format_messages(topic="静夜思")},
)

print(response["messages"][-1].content)