如何使用内置工具和工具包
Prerequisites
本指南假设您熟悉以下概念:
工具
LangChain 拥有大量的第三方工具集合。请访问 工具集成 查看可用工具的列表。
important
使用第三方工具时,请确保您了解该工具的工作原理及其拥有的权限。仔细阅读其文档,并从安全角度检查是否有任何要求。请参阅我们的安全指南以获取更多信息。
让我们试试Wikipedia 集成。
!pip install -qU langchain-community wikipedia
from langchain_community.tools import WikipediaQueryRun
from langchain_community.utilities import WikipediaAPIWrapper
api_wrapper = WikipediaAPIWrapper(top_k_results=1, doc_content_chars_max=100)
tool = WikipediaQueryRun(api_wrapper=api_wrapper)
print(tool.invoke({"query": "langchain"}))
API Reference:WikipediaQueryRun | WikipediaAPIWrapper
Page: LangChain
Summary: LangChain is a framework designed to simplify the creation of applications
该工具有以下默认设置:
print(f"Name: {tool.name}")
print(f"Description: {tool.description}")
print(f"args schema: {tool.args}")
print(f"returns directly?: {tool.return_direct}")
Name: wikipedia
Description: A wrapper around Wikipedia. Useful for when you need to answer general questions about people, places, companies, facts, historical events, or other subjects. Input should be a search query.
args schema: {'query': {'description': 'query to look up on wikipedia', 'title': 'Query', 'type': 'string'}}
returns directly?: False
自定义默认工具
我们还可以修改内置的名称、描述和参数的JSON模式。
在定义参数的JSON模式时,重要的是输入保持与函数相同,因此您不应更改它。但您可以轻松地为每个输入定义自定义描述。
from langchain_community.tools import WikipediaQueryRun
from langchain_community.utilities import WikipediaAPIWrapper
from pydantic import BaseModel, Field
class WikiInputs(BaseModel):
"""Inputs to the wikipedia tool."""
query: str = Field(
description="query to look up in Wikipedia, should be 3 or less words"
)
tool = WikipediaQueryRun(
name="wiki-tool",
description="look up things in wikipedia",
args_schema=WikiInputs,
api_wrapper=api_wrapper,
return_direct=True,
)
print(tool.run("langchain"))
API Reference:WikipediaQueryRun | WikipediaAPIWrapper
Page: LangChain
Summary: LangChain is a framework designed to simplify the creation of applications
print(f"Name: {tool.name}")
print(f"Description: {tool.description}")
print(f"args schema: {tool.args}")
print(f"returns directly?: {tool.return_direct}")
Name: wiki-tool
Description: look up things in wikipedia
args schema: {'query': {'description': 'query to look up in Wikipedia, should be 3 or less words', 'title': 'Query', 'type': 'string'}}
returns directly?: True
如何使用内置工具包
工具包是为特定任务设计的一系列工具的集合。它们具有方便的加载方法。
所有工具包都暴露了一个get_tools
方法,该方法返回一个工具列表。
通常你应该这样使用它们:
# Initialize a toolkit
toolkit = ExampleTookit(...)
# Get list of tools
tools = toolkit.get_tools()