Source code for langchain_community.agent_toolkits.connery.toolkit

from typing import List

from langchain_core.pydantic_v1 import root_validator
from langchain_core.tools import BaseTool, BaseToolkit

from langchain_community.tools.connery import ConneryService


class ConneryToolkit(BaseToolkit):
    """工具包，其中包含康纳利动作列表作为工具。"""

    tools: List[BaseTool]

    def get_tools(self) -> List[BaseTool]:
        """
        返回康纳动作列表。
        """
        return self.tools

    @root_validator()
    def validate_attributes(cls, values: dict) -> dict:
        """验证ConneryToolkit类的属性。
参数：
    values（字典）：要验证的参数。
返回：
    字典：经过验证的参数。
"""

        if not values.get("tools"):
            raise ValueError("The attribute 'tools' must be set.")

        return values

    @classmethod
    def create_instance(cls, connery_service: ConneryService) -> "ConneryToolkit":
        """使用Connery服务创建一个Connery工具包。
参数：
    connery_service (ConneryService): Connery服务，用于获取Connery操作列表。
返回：
    ConneryToolkit: Connery工具包。
"""

        instance = cls(tools=connery_service.list_actions())  # type: ignore[arg-type]

        return instance