Skip to main content
Open In ColabOpen on GitHub

Compass LangChain 工具包

与各种DeFi协议进行交互。非托管工具返回未签名的交易。该工具包基于通用DeFi API(Compass API)构建,允许代理执行以下金融操作:

  • 交换代币 在 Uniswap 和 Aerodrome 上
  • 借出借入 资产,使用 Aave 上的协议
  • 提供流动性 在Aerodrome和Uniswap上
  • 在钱包之间转账
  • 查询余额、投资组合以及监控持仓

概览

集成详情

可序列化的JS 支持最新包裹
LangchainCompassToolkitlangchain-compassPyPI - Version

工具特性

以下是该工具包提供的工具示例(每天可能会有变动):

  • aave_supply: 向Aave提供资产以赚取利息。
  • aave_borrow: 使用抵押品从Aave借入资产。
  • uniswap_swap_sell_exactly: 在 Uniswap 上交换特定数量的一种代币。
  • generic_portfolio_get: 检索钱包的投资组合(以美元和代币余额表示)。
  • generic_transfer_erc20: 在地址之间转移ERC20代币。

设置

在这里,我们将:

  1. 安装langchain包
  2. 导入并实例化工具包
  3. 通过 toolkit.get_tools() 将工具传递给您的代理

安装

该工具包位于 langchain-compass 包中:

%pip install -qU langchain-compass

环境设置

要运行这些示例,请确保 LangChain 能够访问 LLM 服务。例如,如果你使用的是 GPT-4o,请创建一个包含以下内容的 .env 文件:

# .env file
OPENAI_API_KEY=<your_openai_api_key_here>

实例化

现在我们可以实例化我们的工具包:

from langchain_compass.toolkits import LangchainCompassToolkit

toolkit = LangchainCompassToolkit(compass_api_key=None)

工具

查看可用工具

tools = toolkit.get_tools()
for tool in tools:
print(tool.name)
# Expected output:

aave_supply
aave_borrow
aave_repay
aave_withdraw
aave_asset_price_get
aave_liquidity_change_get
aave_user_position_summary_get
...

调用

以编程方式调用单个工具:

tool_name = "generic_ens_get"
tool = next(tool for tool in tools if tool.name == tool_name)
tool.invoke({"ens_name": "vitalik.eth", "chain": "ethereum:mainnet"})
EnsNameInfoResponse(wallet_address='0xd8dA6BF26964aF9D7eEd9e03E53415D37aA96045', registrant='0xd8dA6BF26964aF9D7eEd9e03E53415D37aA96045')

在代理中使用

我们需要一个LLM或聊天模型:

from dotenv import load_dotenv
from langchain_openai import ChatOpenAI

load_dotenv()

llm = ChatOpenAI(model="gpt-4o")
API 参考:ChatOpenAI

用工具初始化代理:

from langgraph.prebuilt import create_react_agent

tools = toolkit.get_tools()
agent_executor = create_react_agent(llm, tools)
API 参考:create_react_agent

使用示例:

example_query = "please set an allowance on Uniswap of 10 WETH for vitalic.eth."  # spelt wrong intentionally

events = agent_executor.stream(
{"messages": [("user", example_query)]},
stream_mode="values",
)
for event in events:
event["messages"][-1].pretty_print()

期望输出:

================================ Human Message =================================

please set an allowance on Uniswap of 10 WETH for vitalic.eth.
================================== Ai Message ==================================
Tool Calls:
generic_ens_get (call_MHIXRXxWH0L7iUEYHwvDUdU1)
Call ID: call_MHIXRXxWH0L7iUEYHwvDUdU1
Args:
chain: ethereum:mainnet
ens_name: vitalic.eth
================================= Tool Message =================================
Name: generic_ens_get

wallet_address='0x44761Ef63FaD902D8f8dC77e559Ab116929881Db' registrant='0x44761Ef63FaD902D8f8dC77e559Ab116929881Db'
================================== Ai Message ==================================
Tool Calls:
generic_allowance_set (call_IEBftbtBfKCkI1zFXXtEY8tq)
Call ID: call_IEBftbtBfKCkI1zFXXtEY8tq
Args:
amount: 10
chain: ethereum:mainnet
contract_name: UniswapV3Router
sender: 0x44761Ef63FaD902D8f8dC77e559Ab116929881Db
token: WETH
================================= Tool Message =================================
Name: generic_allowance_set

{"type": "unsigned_transaction", "content": {"chainId": 1, "data": "0x095ea7b300000000000000000000000068b3465833fb72a70ecdf485e0e4c7bd8665fc450000000000000000000000000000000000000000000000008ac7230489e80000", "from": "0x44761Ef63FaD902D8f8dC77e559Ab116929881Db", "gas": 46434, "to": "0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2", "value": 0, "nonce": 79, "maxFeePerGas": 2265376912, "maxPriorityFeePerGas": 6400594}}

API 参考

langchain-compass 基于Compass API构建。每个工具对应一个API端点。 请在此处查阅文档