Source code for langchain_community.agent_toolkits.ainetwork.toolkit
from __future__ import annotations
from typing import TYPE_CHECKING, List, Literal, Optional
from langchain_core.pydantic_v1 import root_validator
from langchain_core.tools import BaseToolkit
from langchain_community.tools import BaseTool
from langchain_community.tools.ainetwork.app import AINAppOps
from langchain_community.tools.ainetwork.owner import AINOwnerOps
from langchain_community.tools.ainetwork.rule import AINRuleOps
from langchain_community.tools.ainetwork.transfer import AINTransfer
from langchain_community.tools.ainetwork.utils import authenticate
from langchain_community.tools.ainetwork.value import AINValueOps
if TYPE_CHECKING:
from ain.ain import Ain
[docs]class AINetworkToolkit(BaseToolkit):
"""与AINetwork区块链进行交互的工具包。
*安全提示*: 该工具包包含可以读取和修改服务状态的工具;
例如,通过读取、创建、更新、删除与该服务相关联的数据。
有关更多信息,请参阅 https://python.langchain.com/docs/security。"""
network: Optional[Literal["mainnet", "testnet"]] = "testnet"
interface: Optional[Ain] = None
@root_validator(pre=True)
def set_interface(cls, values: dict) -> dict:
if not values.get("interface"):
values["interface"] = authenticate(network=values.get("network", "testnet"))
return values
class Config:
"""Pydantic配置。"""
validate_all = True
arbitrary_types_allowed = True
[docs] def get_tools(self) -> List[BaseTool]:
"""获取工具包中的工具。"""
return [
AINAppOps(),
AINOwnerOps(),
AINRuleOps(),
AINTransfer(),
AINValueOps(),
]